haiku.rag/examples/a2a-server/haiku_rag_a2a/a2a/models.py
2026-01-05 11:53:30 +02:00

18 lines
467 B
Python

from pydantic import BaseModel, Field
from haiku.rag.client import HaikuRAG
class A2AConfig(BaseModel):
"""Configuration for A2A (Agent-to-Agent) protocol server."""
max_contexts: int = Field(
default=1000, description="Maximum number of conversations to keep in memory"
)
class AgentDependencies(BaseModel):
"""Dependencies for the A2A conversational agent."""
model_config = {"arbitrary_types_allowed": True}
client: HaikuRAG