scriptoriumadmin/ReMe
1
0
Fork 0
mirror of https://github.com/agentscope-ai/ReMe.git synced 2026-09-06 08:16:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

format response request

Browse source
This commit is contained in:
jinli.yl 2025-07-03 18:28:40 +08:00
parent 346cacdcf8
commit deb1e94fdb
2 changed files with 18 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
v1/schema/request.py
View file

@ -21,10 +21,10 @@ class SummarizerRequest(BaseRequest):
traj_list: List[Trajectory] = Field(default_factory=list)
class VectorStoreRequest(BaseModel):
class VectorStoreRequest(BaseRequest):
action: str = Field(default="")
params: dict = Field(default_factory=dict)
class AgentRequest(BaseModel):
class AgentRequest(BaseRequest):
query: str = Field(default="")

31
v1/schema/response.py
View file

@ -3,28 +3,29 @@ from typing import List
from pydantic import BaseModel, Field
from experiencemaker.schema.experience import Experience
from experiencemaker.schema.trajectory import Trajectory, ContextMessage
from v1.schema.experience import BaseExperienceNode
from v1.schema.message import Message
class BaseResponse(BaseModel, ABC):
success: bool = Field(default=True)
metadata: dict = Field(default_factory=dict)
#
# class AgentWrapperResponse(BaseResponse):
# trajectory: Trajectory = Field(default_factory=Trajectory)
class ContextGeneratorResponse(BaseResponse):
experience: list[dict] = Field(default_factory=list)
merge_experience: str = Field(default="")
# when to use, experience, response
class RetrieverResponse(BaseResponse):
experience_nodes: list[BaseExperienceNode] = Field(default_factory=list)
experience_merged: str = Field(default="")
class SummarizerResponse(BaseResponse):
experiences: List[dict] = Field(default_factory=list)
experience_nodes: list[BaseExperienceNode] = Field(default_factory=list)
class VectorStoreResponse(BaseResponse):
action: str = Field(default="")
params: dict = Field(default_factory=dict)
class AgentResponse(BaseResponse):
answer: str = Field(default="")
messages: List[Message] = Field(default_factory=list)