ReMe/experiencemaker/schema/response.py
2025-06-10 10:51:37 +08:00

24 lines
674 B
Python

from abc import ABC
from typing import List
from pydantic import BaseModel, Field
from experiencemaker.schema.experience import Experience
from experiencemaker.schema.trajectory import Trajectory, ContextMessage
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):
context_msg: ContextMessage = Field(default_factory=ContextMessage)
class SummarizerResponse(BaseResponse):
experiences: List[Experience] = Field(default_factory=list)