mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-05-06 14:22:47 +02:00
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
"""Contracts for multi_agent_v1 orchestrator and subagent communication."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Literal
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class WorkerBudget(BaseModel):
|
|
max_steps: int = Field(default=1, ge=1)
|
|
max_duration_ms: int = Field(default=15_000, ge=100)
|
|
|
|
|
|
class SubagentTaskPlan(BaseModel):
|
|
domain: str = Field(..., min_length=1)
|
|
goal: str = Field(..., min_length=1)
|
|
constraints: list[str] = Field(default_factory=list)
|
|
budget: WorkerBudget = Field(default_factory=WorkerBudget)
|
|
|
|
|
|
class GroundingEvidence(BaseModel):
|
|
claim: str = Field(..., min_length=1)
|
|
source_type: str = Field(..., min_length=1)
|
|
source_ref: str = Field(..., min_length=1)
|
|
confidence: float = Field(default=0.0, ge=0.0, le=1.0)
|
|
snippet: str = ""
|
|
|
|
|
|
class SubagentResult(BaseModel):
|
|
status: Literal["success", "partial", "error"]
|
|
summary: str = ""
|
|
evidence: list[GroundingEvidence] = Field(default_factory=list)
|
|
artifacts: list[str] = Field(default_factory=list)
|
|
needs_human: bool = False
|
|
error_class: str | None = None
|