Update data_structures.py

data_structures.py

@@ -6,7 +6,7 @@ class UserGoal(BaseModel):
     user_id: str
     project_id: str
     goal_text: str
-    roblox_user_id: Optional[str] = None 
+    roblox_user_id: Optional[str] = "0"
     image_base64: Optional[str] = Field(None, description="Base64 encoded context image.")
 
 class RecallRequest(BaseModel):
@@ -21,9 +21,10 @@ class ResetRequest(BaseModel):
 # --- FEEDBACK LOOP ---
 class SceneObjectState(BaseModel):
     """Represents a snapshot of an object currently in Roblox."""
-    name: str
+    id: str # This maps to the Instance.Name in Roblox
     class_name: str
     position: Dict[str, float] # {x, y, z}
+    rotation: Dict[str, float] # {x, y, z}
     size: Dict[str, float]
     parent_name: str
 
@@ -42,7 +43,7 @@ class Vector3(BaseModel):
     z: float = 0.0
 
 class MatterObject(BaseModel):
-    id: str 
+    id: str # CRITICAL: The AI MUST assign this (e.g., "Floor_Base", "Wall_North")
     class_name: str = "Part"
     position: Vector3
     rotation: Vector3
@@ -51,21 +52,14 @@ class MatterObject(BaseModel):
     anchored: bool = True
     transparency: float = 0.0
     material: str = "Plastic"
+    shape: str = "Block" # Block, Ball, Cylinder
     mesh_id: Optional[str] = None
     texture_id: Optional[str] = None
 
-class ToolboxAsset(BaseModel):
-    id: str
-    asset_id: str 
-    position: Vector3
-    rotation: Vector3
-    scale: float = 1.0
-
-class RiggingCommand(BaseModel):
-    part0_id: str
-    part1_id: str
-    joint_type: Literal["Weld", "Motor6D", "HingeConstraint", "BallSocketConstraint"]
-    position: Vector3
+class EditorCommand(BaseModel):
+    type: Literal["SELECT", "DELETE", "RENAME", "MOVE", "RESIZE"]
+    target_id: str # The Name of the object
+    payload: Dict[str, Any] # e.g., new position
 
 class ScriptPayload(BaseModel):
     name: str
@@ -94,7 +88,8 @@ class TaskNode(BaseModel):
 class ProjectFlow(BaseModel):
     project_id: str
     game_concept_summary: str
-    root_tasks: List[TaskNode]
+    is_finished: bool = False # If True, PM stops the loop
+    next_task: Optional[TaskNode] = None # The immediate next step
 
 # --- COMMUNICATION ---
 class AgentMessage(BaseModel):