2026-03-13 01:01:54 +08:00
|
|
|
from __future__ import annotations
|
|
|
|
|
|
2026-03-13 14:10:13 +08:00
|
|
|
from enum import Enum
|
|
|
|
|
|
2026-03-13 01:01:54 +08:00
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
|
|
|
|
|
2026-03-13 14:10:13 +08:00
|
|
|
class AgentType(str, Enum):
|
|
|
|
|
ROUTER = "router"
|
|
|
|
|
WORKER = "worker"
|
|
|
|
|
|
|
|
|
|
|
2026-03-13 01:01:54 +08:00
|
|
|
class SystemAgentLLMConfig(BaseModel):
|
|
|
|
|
temperature: float | None = Field(default=None, ge=0.0, le=2.0)
|
|
|
|
|
max_tokens: int | None = Field(default=None, ge=1)
|
|
|
|
|
timeout_seconds: float | None = Field(default=30.0, gt=0.0, le=300.0)
|
