Create schemas.py

schemas.py

@@ -0,0 +1,42 @@
+from __future__ import annotations
+from pydantic import BaseModel, Field
+from typing import Optional
+from datetime import datetime
+
+
+class GenerateRequest(BaseModel):
+    prompt: str
+    app_type: Optional[str] = "saas"
+    systems: Optional[list[str]] = None
+
+
+class FixRequest(BaseModel):
+    error_message: str
+    file_path: Optional[str] = None
+
+
+class AgentMessage(BaseModel):
+    event_type: str
+    agent: str
+    content: str = ""
+    file_path: Optional[str] = None
+    timestamp: datetime = Field(default_factory=datetime.utcnow)
+    metadata: dict = Field(default_factory=dict)
+
+
+class ProjectState(BaseModel):
+    session_id: str
+    user_prompt: str
+    app_type: str = "saas"
+    status: str = "queued"
+    current_agent: str = ""
+    systems: list[str] = Field(default_factory=list)
+    messages: list[AgentMessage] = Field(default_factory=list)
+    generated_files: dict[str, str] = Field(default_factory=dict)
+    file_tree: list[str] = Field(default_factory=list)
+    errors: list[str] = Field(default_factory=list)
+    research_output: dict = Field(default_factory=dict)
+    blueprint: dict = Field(default_factory=dict)
+
+    class Config:
+        arbitrary_types_allowed = True