mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-10-05 02:41:43 +00:00
28 lines
767 B
Python
28 lines
767 B
Python
from typing import List
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
from experiencemaker.schema.experience import BaseExperienceNode
|
|
from experiencemaker.schema.message import Message
|
|
|
|
|
|
class BaseResponse(BaseModel):
|
|
success: bool = Field(default=True)
|
|
metadata: dict = Field(default_factory=dict)
|
|
|
|
|
|
class RetrieverResponse(BaseResponse):
|
|
experience_nodes: List[BaseExperienceNode] = Field(default_factory=list)
|
|
experience_merged: str = Field(default="")
|
|
|
|
|
|
class SummarizerResponse(BaseResponse):
|
|
experience_nodes: List[BaseExperienceNode] = Field(default_factory=list)
|
|
|
|
|
|
class VectorStoreResponse(BaseResponse):
|
|
...
|
|
|
|
class AgentResponse(BaseResponse):
|
|
answer: str = Field(default="")
|
|
messages: List[Message] = Field(default_factory=list)
|