mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-08-28 05:25:04 +00:00
24 lines
620 B
Python
24 lines
620 B
Python
from abc import ABC
|
|
from typing import List
|
|
|
|
from pydantic import Field
|
|
|
|
from experiencemaker.schema.module_loader import ModuleLoader
|
|
from experiencemaker.schema.trajectory import Trajectory
|
|
|
|
|
|
class BaseRequest(ModuleLoader, 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_samples: bool = Field(default=False)
|