mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-09-06 08:16:00 +00:00
31 lines
1.3 KiB
Python
31 lines
1.3 KiB
Python
from pydantic import Field
|
|
|
|
from experiencemaker.schema.request import RetrieverRequest, SummarizerRequest, VectorStoreRequest, AgentRequest
|
|
from experiencemaker.schema.response import RetrieverResponse, SummarizerResponse, VectorStoreResponse, AgentResponse
|
|
from experiencemaker.utils.http_client import HttpClient
|
|
|
|
|
|
class ExperienceMakerClient(HttpClient):
|
|
base_url: str = Field(default=...)
|
|
|
|
def call_retriever(self, request: RetrieverRequest):
|
|
self.url = self.base_url + "/retriever"
|
|
return RetrieverResponse(**self.request(json_data=request.model_dump()))
|
|
|
|
def call_summarizer(self, request: SummarizerRequest):
|
|
self.url = self.base_url + "/summarizer"
|
|
return SummarizerResponse(**self.request(json_data=request.model_dump()))
|
|
|
|
def call_vector_store(self, request: VectorStoreRequest):
|
|
self.url = self.base_url + "/vector_store"
|
|
return VectorStoreResponse(**self.request(json_data=request.model_dump()))
|
|
|
|
def call_agent(self, request: AgentRequest):
|
|
self.url = self.base_url + "/agent"
|
|
return AgentResponse(**self.request(json_data=request.model_dump()))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
client = ExperienceMakerClient(base_url="http://0.0.0.0:8001")
|
|
response = client.call_retriever(RetrieverRequest(workspace_id="123", query="hello world"))
|
|
print(response.model_dump())
|