Update models.py

models.py

@@ -1,9 +1,18 @@
+# models.py - Updated with new input models for Node.js integration
 from pydantic import BaseModel
 from typing import List, Dict, Any, Optional
 
 class AnalyzeDocumentInput(BaseModel):
     document_text: str
-    force_doc_id: Optional[str] = None
+    force_doc_id: Optional[str] = None  # From Node.js backend
+    max_tokens: int = 1600
+    stride: int = 50
+    max_length: int = 128
+    min_length: int = 24
+
+class AnalyzeDocumentURLInput(BaseModel):
+    document_url: str
+    force_doc_id: Optional[str] = None  # From Node.js backend
     max_tokens: int = 1600
     stride: int = 50
     max_length: int = 128
@@ -20,8 +29,9 @@ class SummarizeBatchInput(BaseModel):
     min_length: int = 24
 
 class ChatInput(BaseModel):
-    question: str
-    doc_id: str
+    message: str
+    document_id: str
+    conversation_history: List[Dict[str, str]] = []
 
 class RiskTerm(BaseModel):
     term: str
@@ -42,13 +52,9 @@ class AnalysisResult(BaseModel):
     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
+tional[float] = None