ReMe/experiencemaker/schema/response.py
2025-07-10 20:39:58 +08:00

28 lines
753 B
Python

from typing import List
from pydantic import BaseModel, Field
from experiencemaker.schema.experience import BaseExperience
from experiencemaker.schema.message import Message
class BaseResponse(BaseModel):
success: bool = Field(default=True)
metadata: dict = Field(default_factory=dict)
class RetrieverResponse(BaseResponse):
experience_list: List[BaseExperience] = Field(default_factory=list)
experience_merged: str = Field(default="")
class SummarizerResponse(BaseResponse):
experience_list: List[BaseExperience] = Field(default_factory=list)
class VectorStoreResponse(BaseResponse):
...
class AgentResponse(BaseResponse):
answer: str = Field(default="")
messages: List[Message] = Field(default_factory=list)