mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-09-06 08:16:00 +00:00
24 lines
603 B
Python
24 lines
603 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)
|
|
workspace_id: str = Field(default="")
|
|
|
|
|
|
class AgentWrapperRequest(BaseRequest):
|
|
query: str = Field(default="")
|
|
|
|
|
|
class ContextGeneratorRequest(BaseRequest):
|
|
trajectory: Trajectory = Field(default_factory=Trajectory)
|
|
retrieve_top_k: int = Field(default=1)
|
|
|
|
|
|
class SummarizerRequest(BaseRequest):
|
|
trajectories: List[Trajectory] = Field(default_factory=dict)
|