Update models.py

models.py

@@ -40,3 +40,14 @@ class AnalysisResult(BaseModel):
     key_clauses: List[Dict[str, Any]]
     chunk_count: int
     processing_time: str
+
+class ChatInput(BaseModel):
+    message: str
+    document_id: str
+    conversation_history: List[Dict[str, str]] = []
+
+class ChatResponse(BaseModel):
+    response: str
+    sources: List[Dict[str, Any]]
+    document_id: str
+    confidence: Optional[float] = None