violinadoley25/multi-agent-task-alloc

1from __future__ import annotations2 3from typing import Any, Dict, List4 5from openenv.core.env_server.types import Action, Observation6from pydantic import Field7 8 9class TaskAllocationAction(Action):10    action_type: str = Field(default="allocate", description="allocate | plan | optimize")11    content: str = Field(..., description='JSON string: {"task_id": "t1", "agent_id": "alice"}')12 13 14class TaskAllocationObservation(Observation):15    task_name: str = Field(default="")16    game_state: Dict[str, Any] = Field(default_factory=dict)17    context: Dict[str, Any] = Field(default_factory=dict)18    step_count: int = Field(default=0)19    score: float = Field(default=0.0)20    feedback: str = Field(default="")21    valid_actions: List[str] = Field(default_factory=list)22