Spaces:

Leon4gr45
/

brain

Running

brain / src /services /api /constants /agent.py

Upload folder using huggingface_hub (part 3)

a55ea57 verified 20 days ago

558 Bytes

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)

	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)