mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-08-28 05:25:04 +00:00
23 lines
568 B
Python
23 lines
568 B
Python
from abc import ABC
|
|
from typing import List
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
from experiencemaker.schema.trajectory import Trajectory
|
|
|
|
|
|
class BaseRequest(BaseModel, ABC):
|
|
metadata: dict = Field(default_factory=dict)
|
|
|
|
|
|
class AgentWrapperRequest(BaseRequest):
|
|
query: str = Field(default="")
|
|
|
|
|
|
class ContextGeneratorRequest(BaseRequest):
|
|
trajectory: Trajectory = Field(default_factory=Trajectory)
|
|
|
|
|
|
class SummarizerRequest(BaseRequest):
|
|
trajectories: List[Trajectory] = Field(default_factory=dict)
|
|
return_experience: bool = Field(default=True)
|