added_ai_proxy

app/schemas.py

@@ -59,4 +59,21 @@ class AssetAssign(BaseModel):
     heir_name: str
 
 class DeclareGuale(BaseModel):
-    username: str
+    username: str
+
+# AI Proxy Schemas
+class AIMessage(BaseModel):
+    role: str
+    content: str
+
+class AIRequest(BaseModel):
+    messages: List[AIMessage]
+    model: Optional[str] = None
+
+class AIResponse(BaseModel):
+    id: str
+    object: str
+    created: int
+    model: str
+    choices: List[dict]
+    usage: dict