Update services/chat_service.py

services/chat_service.py

@@ -177,30 +177,34 @@ class ChatService:
     async def chat(
         self,
         user_input: str,
-        session_id: str,
+        session_id: Any,  # Allow any type to ensure flexibility, validate later
         max_length: int = 1000
     ) -> Tuple[str, List[Dict[str, Any]]]:
         """Main chat method that coordinates the entire conversation flow"""
         try:
+            # Ensure session_id is a string
+            if not isinstance(session_id, str):
+                session_id = str(session_id)
+    
             # Get chat history
             chat_history = self.conversation_manager.get_history(session_id)
-            
+    
             # Search all sources
             search_results = await self.search_all_sources(user_input)
-            
+    
             # Build context
             context = self.build_context(search_results, chat_history)
-            
+    
             # Create prompt
             prompt = (
                 f"Context:\n{context}\n\n"
                 f"User: {user_input}\n"
                 "Assistant:"
             )
-            
+    
             # Generate response
             response = await self.generate_response(prompt, max_length)
-            
+    
             # Store interaction
             self.conversation_manager.add_interaction(
                 session_id,
@@ -208,9 +212,9 @@ class ChatService:
                 response,
                 {'search_results': search_results}
             )
-            
+    
             return response, search_results
-            
+    
         except Exception as e:
             logger.error(f"Error in chat: {e}")
-            raise
+            raise