Create data_structures.py

data_structures.py

@@ -0,0 +1,71 @@
+from pydantic import BaseModel, Field
+from typing import List, Dict, Optional, Union, Literal
+
+# --- 1. THE MATTER FORMAT (For 3D/World Agents) ---
+class Vector3(BaseModel):
+    x: float
+    y: float
+    z: float
+
+class MatterObject(BaseModel):
+    id: str = Field(description="Unique ID for this specific part")
+    group_id: Optional[str] = Field(None, description="Logical grouping (e.g., 'Lobby_Wall_North')")
+    class_name: str = Field("Part", description="Roblox ClassName: Part, MeshPart, WedgePart")
+    
+    # Transform
+    position: Vector3
+    rotation: Vector3
+    size: Vector3
+    
+    # Appearance
+    color_hex: str = "#A3A2A5"
+    material: str = "Plastic"
+    transparency: float = 0.0
+    reflectance: float = 0.0
+    
+    # External Assets
+    mesh_id: Optional[str] = None
+    texture_id: Optional[str] = None
+
+class MatterManifest(BaseModel):
+    """The output of the 3D Artist or World Builder."""
+    assets: List[MatterObject]
+
+# --- 2. THE PROJECT FLOW FORMAT (For PM/Supervisors) ---
+class TaskNode(BaseModel):
+    task_id: str
+    title: str
+    role_required: Literal["PROJECT_MANAGER", "SUPERVISOR_WORLD", "SUPERVISOR_SCRIPT", "3D_ARTIST", "SCRIPTING_ENGINEER", "ERRANDS"]
+    instruction: str
+    assigned_instance_id: Optional[str] = None # Which specific agent instance handles this
+    
+    # Recursive dependency structure
+    subtasks: List['TaskNode'] = [] 
+    dependencies: List[str] = [] # IDs of tasks that must finish first
+
+class ProjectFlow(BaseModel):
+    """The Master GDD (Game Design Document) and Execution Plan."""
+    project_id: str
+    game_concept_summary: str
+    root_tasks: List[TaskNode]
+
+# --- 3. COMMUNICATION PROTOCOLS ---
+class AgentMessage(BaseModel):
+    id: str
+    from_agent: str
+    to_agents: List[str] # Visibility Constraint
+    role: str # 'user' or 'model' (Gemini format)
+    content: str # Text or JSON string
+    image_base64: Optional[str] = None # For reference injection
+
+class GenerationRequest(BaseModel):
+    agent_type: str
+    agent_instance_id: str
+    prompt: str
+    
+    # The Visibility Layer
+    # The agent responding needs to know who they are talking to
+    # to filter history correctly.
+    conversation_partner_id: str 
+    
+    image_base64: Optional[str] = None # If the prompt includes an image