Spaces:
Running
Running
File size: 558 Bytes
a55ea57 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | from typing import Any
from pydantic import BaseModel, ConfigDict, Field
class AgentRequest(BaseModel):
model_config = ConfigDict(extra="forbid")
class AgentSearchRequest(AgentRequest):
query: str = Field(min_length=1)
limit: int = Field(default=10, ge=1, le=100)
class AgentContextRequest(AgentRequest):
query: str = Field(min_length=1)
max_tokens: int = Field(default=4000, ge=100, le=32000)
class AgentMemoryRequest(AgentRequest):
text: str = Field(min_length=1)
metadata: dict[str, Any] = Field(default_factory=dict)
|