Hundao
589c5b06fe
- Auto-fixed 70 lint errors (import sorting, aliased errors, datetime.UTC)
- Fixed 85 remaining errors manually:
- E501: wrapped long lines in queen_profiles, catalog, routes_credentials
- F821: added missing TYPE_CHECKING imports for AgentHost, ToolRegistry,
HookContext, HookResult; added runtime imports where needed
- F811: removed duplicate method definitions in queen_lifecycle_tools
- F841/B007: removed unused variables in discovery.py
- W291: removed trailing whitespace in queen nodes
- E402: moved import to top of queen_memory_v2.py
- Fixed AgentRuntime -> AgentHost in example template type annotations
- Reformatted 343 files with ruff format
118 lines
3.8 KiB
Python
118 lines
3.8 KiB
Python
"""Goal Schema — Declarative definition of what a colony should achieve.
|
|
|
|
A Goal defines WHAT the colony should achieve, not HOW.
|
|
Workers are autonomous clones that figure out the how.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
from enum import StrEnum
|
|
from typing import Any
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class GoalStatus(StrEnum):
|
|
DRAFT = "draft"
|
|
READY = "ready"
|
|
ACTIVE = "active"
|
|
COMPLETED = "completed"
|
|
FAILED = "failed"
|
|
SUSPENDED = "suspended"
|
|
|
|
|
|
class SuccessCriterion(BaseModel):
|
|
id: str
|
|
description: str = Field(description="Human-readable description of what success looks like")
|
|
metric: str = Field(description="How to measure: 'output_contains', 'output_equals', 'llm_judge', 'custom'")
|
|
type: str = Field(default="success_rate", description="Runtime evaluation type")
|
|
target: Any = Field(description="The target value or condition")
|
|
weight: float = Field(default=1.0, ge=0.0, le=1.0, description="Relative importance (0-1)")
|
|
met: bool = False
|
|
|
|
model_config = {"extra": "allow"}
|
|
|
|
|
|
class Constraint(BaseModel):
|
|
id: str
|
|
description: str
|
|
constraint_type: str = Field(description="Type: 'hard' (must not violate) or 'soft' (prefer not to violate)")
|
|
category: str = Field(default="general", description="Category: 'time', 'cost', 'safety', 'scope', 'quality'")
|
|
check: str = Field(default="", description="How to check: expression, function name, or 'llm_judge'")
|
|
|
|
model_config = {"extra": "allow"}
|
|
|
|
|
|
class Goal(BaseModel):
|
|
"""Declarative definition of colony success.
|
|
|
|
Workers are autonomous — the goal provides direction, not step-by-step
|
|
instructions. The queen interprets the goal and spawns workers accordingly.
|
|
"""
|
|
|
|
id: str
|
|
name: str
|
|
description: str
|
|
status: GoalStatus = GoalStatus.DRAFT
|
|
|
|
success_criteria: list[SuccessCriterion] = Field(default_factory=list)
|
|
constraints: list[Constraint] = Field(default_factory=list)
|
|
|
|
context: dict[str, Any] = Field(
|
|
default_factory=dict,
|
|
description="Additional context: domain knowledge, user preferences, etc.",
|
|
)
|
|
|
|
required_capabilities: list[str] = Field(
|
|
default_factory=list,
|
|
description="What the colony needs: 'llm', 'web_search', 'code_execution', etc.",
|
|
)
|
|
|
|
input_schema: dict[str, Any] = Field(default_factory=dict)
|
|
output_schema: dict[str, Any] = Field(default_factory=dict)
|
|
|
|
version: str = "1.0.0"
|
|
parent_version: str | None = None
|
|
evolution_reason: str | None = None
|
|
|
|
created_at: datetime = Field(default_factory=datetime.now)
|
|
updated_at: datetime = Field(default_factory=datetime.now)
|
|
|
|
model_config = {"extra": "allow"}
|
|
|
|
def is_success(self) -> bool:
|
|
if not self.success_criteria:
|
|
return False
|
|
total_weight = sum(c.weight for c in self.success_criteria)
|
|
met_weight = sum(c.weight for c in self.success_criteria if c.met)
|
|
return met_weight >= total_weight * 0.9
|
|
|
|
def to_prompt_context(self) -> str:
|
|
if not self.success_criteria and not self.constraints and not self.context:
|
|
return ""
|
|
|
|
lines = [
|
|
f"# Goal: {self.name}",
|
|
f"{self.description}",
|
|
"",
|
|
"## Success Criteria:",
|
|
]
|
|
for sc in self.success_criteria:
|
|
lines.append(f"- {sc.description}")
|
|
|
|
if self.constraints:
|
|
lines.append("")
|
|
lines.append("## Constraints:")
|
|
for c in self.constraints:
|
|
severity = "MUST" if c.constraint_type == "hard" else "SHOULD"
|
|
lines.append(f"- [{severity}] {c.description}")
|
|
|
|
if self.context:
|
|
lines.append("")
|
|
lines.append("## Context:")
|
|
for key, value in self.context.items():
|
|
lines.append(f"- {key}: {value}")
|
|
|
|
return "\n".join(lines)
|