init

vw-agentic-rag/service/schemas/messages.py

@@ -0,0 +1,34 @@
+from typing import Dict, Any, Optional
+from pydantic import BaseModel
+
+
+class UserMessage(BaseModel):
+    content: str
+    timestamp: Optional[str] = None
+
+
+class AssistantMessage(BaseModel):
+    content: str
+    citations_mapping_csv: Optional[str] = None
+    timestamp: Optional[str] = None
+
+
+class ToolMessage(BaseModel):
+    tool_name: str
+    tool_call_id: str
+    content: str  # Usually JSON string of results
+    timestamp: Optional[str] = None
+
+
+class ChatRequest(BaseModel):
+    session_id: str
+    messages: list[Dict[str, Any]]
+    client_hints: Optional[Dict[str, Any]] = None
+
+
+class ChatResponse(BaseModel):
+    """Base response for non-streaming endpoints"""
+    answer: str
+    citations_mapping_csv: str
+    tool_results: list[Dict[str, Any]]
+    session_id: str