Agent Zero
dbeadd8ab1
Check Cross-Plugin Imports / check (push) Has been cancelled
- F-CTX: app/ai/context_builder.py (282 lines) — build_agent_context() + ReActSystemPromptBuilder - F-STR: app/ai/agent_stream.py (155 lines) — stream_react_loop() with SSE events (step, status, done, error) - F-DEF: AgentDefinition fields added (temperature, max_tokens, max_steps, trace_mode, skill_ids, trigger_config, ai_use_case_metadata) + migration 0122 - F-SKILL: app/ai/skill_registry.py (82 lines) — SkillDefinition + SkillRegistry singleton - F-TOOL: app/ai/agent_tools.py (117 lines) — get_agent_tools() with permission intersection - Skill CRUD routes: app/plugins/builtins/automation/skill_routes.py - Tests: test_skill_registry.py (97 lines), test_agent_tools.py (219 lines) - All Python compile checks pass, tests require PostgreSQL (infra issue, not code bug)
385 lines
11 KiB
Python
385 lines
11 KiB
Python
"""Pydantic v2 schemas for the Automation & Agents plugin API."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
# ─── Agent Definition Schemas ───
|
|
|
|
|
|
class AgentDefinitionCreate(BaseModel):
|
|
"""Create a new agent definition."""
|
|
|
|
name: str = Field(..., min_length=1, max_length=120)
|
|
description: str = Field(default="", max_length=500)
|
|
llm_model: str = Field(default="ollama/deepseek-v4-flash", max_length=100)
|
|
system_prompt: str = Field(default="")
|
|
tool_ids: list[str] = Field(default_factory=list)
|
|
heartbeat_interval_seconds: int = Field(default=0, ge=0)
|
|
mode: str = Field(default="reactive", pattern="^(proactive|reactive)$")
|
|
is_active: bool = Field(default=True)
|
|
max_executions_per_hour: int = Field(default=10, ge=1, le=1000)
|
|
max_duration_seconds: int = Field(default=300, ge=1, le=86400)
|
|
budget_limit_usd: float = Field(default=1.0, ge=0.0, le=10000.0)
|
|
temperature: float = Field(default=0.3, ge=0.0, le=2.0)
|
|
max_tokens: int = Field(default=1000, ge=1, le=100000)
|
|
max_steps: int = Field(default=20, ge=1, le=100)
|
|
trace_mode: str = Field(default="standard", pattern="^(standard|extended)$")
|
|
skill_ids: list[str] = Field(default_factory=list)
|
|
trigger_config: dict[str, Any] = Field(default_factory=dict)
|
|
ai_use_case_metadata: dict[str, Any] = Field(default_factory=dict)
|
|
|
|
|
|
class AgentDefinitionUpdate(BaseModel):
|
|
"""Update an existing agent definition (partial)."""
|
|
|
|
name: str | None = Field(None, max_length=120)
|
|
description: str | None = Field(None, max_length=500)
|
|
llm_model: str | None = Field(None, max_length=100)
|
|
system_prompt: str | None = None
|
|
tool_ids: list[str] | None = None
|
|
heartbeat_interval_seconds: int | None = Field(None, ge=0)
|
|
mode: str | None = Field(None, pattern="^(proactive|reactive)$")
|
|
is_active: bool | None = None
|
|
max_executions_per_hour: int | None = Field(None, ge=1, le=1000)
|
|
max_duration_seconds: int | None = Field(None, ge=1, le=86400)
|
|
budget_limit_usd: float | None = Field(None, ge=0.0, le=10000.0)
|
|
|
|
|
|
class AgentDefinitionResponse(BaseModel):
|
|
"""Agent definition response."""
|
|
|
|
id: str
|
|
name: str
|
|
description: str = ""
|
|
llm_model: str
|
|
system_prompt: str = ""
|
|
tool_ids: list[str] = []
|
|
heartbeat_interval_seconds: int = 0
|
|
mode: str
|
|
is_active: bool = True
|
|
max_executions_per_hour: int = 10
|
|
max_duration_seconds: int = 300
|
|
budget_limit_usd: float = 1.0
|
|
created_by: str | None = None
|
|
created_at: str | None = None
|
|
updated_at: str | None = None
|
|
|
|
|
|
# ─── Automation Definition Schemas ───
|
|
|
|
|
|
class AutomationDefinitionCreate(BaseModel):
|
|
"""Create a new automation definition."""
|
|
|
|
name: str = Field(..., min_length=1, max_length=120)
|
|
description: str = Field(default="", max_length=500)
|
|
trigger_type: str = Field(default="manual", pattern="^(event|schedule|manual|ui)$")
|
|
trigger_config: dict[str, Any] = Field(default_factory=dict)
|
|
conditions: list[dict[str, Any]] = Field(default_factory=list)
|
|
actions: list[dict[str, Any]] = Field(default_factory=list)
|
|
is_active: bool = Field(default=True)
|
|
dry_run: bool = Field(default=False)
|
|
|
|
|
|
class AutomationDefinitionUpdate(BaseModel):
|
|
"""Update an existing automation definition (partial)."""
|
|
|
|
name: str | None = Field(None, max_length=120)
|
|
description: str | None = Field(None, max_length=500)
|
|
trigger_type: str | None = Field(None, pattern="^(event|schedule|manual|ui)$")
|
|
trigger_config: dict[str, Any] | None = None
|
|
conditions: list[dict[str, Any]] | None = None
|
|
actions: list[dict[str, Any]] | None = None
|
|
is_active: bool | None = None
|
|
dry_run: bool | None = None
|
|
|
|
|
|
class AutomationDefinitionResponse(BaseModel):
|
|
"""Automation definition response."""
|
|
|
|
id: str
|
|
name: str
|
|
description: str = ""
|
|
trigger_type: str
|
|
trigger_config: dict[str, Any] = {}
|
|
conditions: list[dict[str, Any]] = []
|
|
actions: list[dict[str, Any]] = []
|
|
is_active: bool = True
|
|
dry_run: bool = False
|
|
created_by: str | None = None
|
|
created_at: str | None = None
|
|
updated_at: str | None = None
|
|
|
|
|
|
# ─── Cron Job Schemas ───
|
|
|
|
|
|
class CronJobResponse(BaseModel):
|
|
"""Cron job response."""
|
|
|
|
id: str
|
|
name: str
|
|
cron_expression: str
|
|
job_type: str
|
|
target_id: str | None = None
|
|
plugin_name: str = ""
|
|
is_active: bool = True
|
|
last_run_at: str | None = None
|
|
next_run_at: str | None = None
|
|
created_at: str | None = None
|
|
updated_at: str | None = None
|
|
|
|
|
|
# ─── Run Log Schemas ───
|
|
|
|
|
|
class AgentRunResponse(BaseModel):
|
|
"""Agent run log response."""
|
|
|
|
id: str
|
|
agent_id: str
|
|
status: str
|
|
started_at: str
|
|
completed_at: str | None = None
|
|
duration_seconds: float | None = None
|
|
result: str | None = None
|
|
error: str | None = None
|
|
cost_usd: float | None = None
|
|
trigger_type: str = "manual"
|
|
trigger_data: dict[str, Any] = {}
|
|
created_at: str | None = None
|
|
|
|
|
|
class AutomationRunResponse(BaseModel):
|
|
"""Automation run log response."""
|
|
|
|
id: str
|
|
automation_id: str
|
|
status: str
|
|
started_at: str
|
|
completed_at: str | None = None
|
|
duration_seconds: float | None = None
|
|
result: str | None = None
|
|
error: str | None = None
|
|
trigger_type: str = "manual"
|
|
trigger_data: dict[str, Any] = {}
|
|
dry_run: bool = False
|
|
created_at: str | None = None
|
|
|
|
|
|
# ─── Version Schemas ───
|
|
|
|
|
|
class AgentVersionResponse(BaseModel):
|
|
"""Agent version response."""
|
|
|
|
id: str
|
|
agent_id: str
|
|
version_number: int
|
|
snapshot: dict[str, Any]
|
|
changed_by: str | None = None
|
|
created_at: str | None = None
|
|
|
|
|
|
class AutomationVersionResponse(BaseModel):
|
|
"""Automation version response."""
|
|
|
|
id: str
|
|
automation_id: str
|
|
version_number: int
|
|
snapshot: dict[str, Any]
|
|
changed_by: str | None = None
|
|
created_at: str | None = None
|
|
|
|
|
|
# ─── Settings Schemas ───
|
|
|
|
|
|
class AgentMessageRequest(BaseModel):
|
|
"""Request to send a message from one agent to another."""
|
|
|
|
to_agent_name: str = Field(..., min_length=1, max_length=120)
|
|
message: str = Field(..., min_length=1)
|
|
|
|
|
|
class MiniAppCreate(BaseModel):
|
|
"""Create a custom MiniApp definition."""
|
|
|
|
app_id: str = Field(..., min_length=1, max_length=80)
|
|
name: str = Field(..., min_length=1, max_length=120)
|
|
icon: str = Field(default="AppWindow")
|
|
description: str = Field(default="")
|
|
render_schema: dict[str, Any] = Field(default_factory=dict)
|
|
|
|
|
|
class MiniAppResponse(BaseModel):
|
|
"""MiniApp definition response."""
|
|
|
|
app_id: str
|
|
name: str
|
|
icon: str = "AppWindow"
|
|
description: str = ""
|
|
plugin_name: str = ""
|
|
render_schema: dict[str, Any] = {}
|
|
|
|
|
|
class AutomationSettingsResponse(BaseModel):
|
|
"""Automation settings response."""
|
|
|
|
default_llm_model: str = "ollama/deepseek-v4-flash"
|
|
heartbeat_default_interval: int = 300
|
|
max_concurrent_agents: int = 5
|
|
log_level: str = "INFO"
|
|
|
|
|
|
class AutomationSettingsUpdate(BaseModel):
|
|
"""Automation settings update (partial)."""
|
|
|
|
default_llm_model: str | None = None
|
|
heartbeat_default_interval: int | None = None
|
|
max_concurrent_agents: int | None = None
|
|
log_level: str | None = None
|
|
|
|
|
|
# ─── List Response Schemas ───
|
|
|
|
|
|
class AgentDefinitionListResponse(BaseModel):
|
|
"""Paginated agent definition list."""
|
|
|
|
items: list[AgentDefinitionResponse]
|
|
total: int
|
|
|
|
|
|
class AutomationDefinitionListResponse(BaseModel):
|
|
"""Paginated automation definition list."""
|
|
|
|
items: list[AutomationDefinitionResponse]
|
|
total: int
|
|
|
|
|
|
class AgentRunListResponse(BaseModel):
|
|
"""Paginated agent run list."""
|
|
|
|
items: list[AgentRunResponse]
|
|
total: int
|
|
|
|
|
|
class AutomationRunListResponse(BaseModel):
|
|
"""Paginated automation run list."""
|
|
|
|
items: list[AutomationRunResponse]
|
|
total: int
|
|
|
|
|
|
class CronJobListResponse(BaseModel):
|
|
"""Paginated cron job list."""
|
|
|
|
items: list[CronJobResponse]
|
|
total: int
|
|
|
|
|
|
class AgentVersionListResponse(BaseModel):
|
|
"""Paginated agent version list."""
|
|
|
|
items: list[AgentVersionResponse]
|
|
total: int
|
|
|
|
|
|
class AutomationVersionListResponse(BaseModel):
|
|
"""Paginated automation version list."""
|
|
|
|
items: list[AutomationVersionResponse]
|
|
total: int
|
|
|
|
|
|
# ─── Subtask Schemas (Phase 5.8) ───
|
|
|
|
|
|
class SubtaskCreate(BaseModel):
|
|
"""Create a new subtask for multi-agent orchestration."""
|
|
|
|
parent_agent_id: str = Field(..., description="UUID of the parent agent")
|
|
child_agent_id: str = Field(..., description="UUID of the child agent that will execute")
|
|
task_description: str = Field(..., min_length=1, description="Description of the task")
|
|
|
|
|
|
class SubtaskUpdate(BaseModel):
|
|
"""Update a subtask (status, result)."""
|
|
|
|
status: str | None = Field(None, pattern="^(pending|running|completed|failed|cancelled)$")
|
|
result: dict[str, Any] | None = None
|
|
|
|
|
|
class SubtaskRead(BaseModel):
|
|
"""Subtask response."""
|
|
|
|
id: str
|
|
parent_agent_id: str
|
|
child_agent_id: str
|
|
task_description: str
|
|
status: str = "pending"
|
|
result: dict[str, Any] = {}
|
|
created_at: str | None = None
|
|
completed_at: str | None = None
|
|
updated_at: str | None = None
|
|
|
|
|
|
class SubtaskListResponse(BaseModel):
|
|
"""Paginated subtask list."""
|
|
|
|
items: list[SubtaskRead]
|
|
total: int
|
|
|
|
|
|
# ─── Skill Definition Schemas (Phase F-SKILL) ───
|
|
|
|
|
|
class SkillDefinitionCreate(BaseModel):
|
|
"""Create a new skill definition."""
|
|
|
|
name: str = Field(..., min_length=1, max_length=255)
|
|
description: str = Field(..., min_length=1)
|
|
instructions: str = Field(..., min_length=1)
|
|
allowed_tool_ids: list[str] = Field(default_factory=list)
|
|
context_policy: dict[str, Any] | None = None
|
|
category: str = Field(default="general", max_length=100)
|
|
is_active: bool = Field(default=True)
|
|
|
|
|
|
class SkillDefinitionUpdate(BaseModel):
|
|
"""Update an existing skill definition (partial)."""
|
|
|
|
name: str | None = Field(None, min_length=1, max_length=255)
|
|
description: str | None = Field(None, min_length=1)
|
|
instructions: str | None = Field(None, min_length=1)
|
|
allowed_tool_ids: list[str] | None = None
|
|
context_policy: dict[str, Any] | None = None
|
|
category: str | None = Field(None, max_length=100)
|
|
is_active: bool | None = None
|
|
|
|
|
|
class SkillDefinitionResponse(BaseModel):
|
|
"""Skill definition response."""
|
|
|
|
id: str
|
|
name: str
|
|
description: str
|
|
instructions: str
|
|
allowed_tool_ids: list[str] = []
|
|
context_policy: dict[str, Any] | None = None
|
|
category: str = "general"
|
|
is_active: bool = True
|
|
created_at: str | None = None
|
|
updated_at: str | None = None
|
|
|
|
|
|
class SkillDefinitionListResponse(BaseModel):
|
|
"""Paginated skill definition list."""
|
|
|
|
items: list[SkillDefinitionResponse]
|
|
total: int
|