Add 89999999999999

enhanced_websocket_handler.py

@@ -240,20 +240,18 @@ async def handle_text_message(websocket: WebSocket, data: dict, session_data: di
     try:
         if use_hybrid:
             # Stream hybrid LLM service response
-            provider_used = None
-            full_response = ""
-            async for chunk, provider in get_hybrid_response(
+            response_chunks = []
+            async for chunk in get_hybrid_response(
                 user_message, session_data["context"], config, knowledge_base
             ):
-                if provider_used is None:
-                    provider_used = provider
-                full_response += chunk
+                response_chunks.append(chunk)
                 await websocket.send_json({
                     "type": "streaming_response",
-                    "message": chunk,
-                    "provider_used": provider_used
+                    "message": chunk
                 })
-            response_text = full_response
+            response_text = "".join(response_chunks)
+            provider_used = hybrid_llm_service.choose_llm_provider(user_message)
+            provider_used = provider_used.value if provider_used else "unknown"
         else:
             # Use traditional graph approach
             session_data["messages"].append(HumanMessage(content=user_message))
@@ -361,20 +359,18 @@ async def handle_voice_message(websocket: WebSocket, data: dict, session_data: d
 
         # Process as text message with language context
         if use_hybrid:
-            provider_used = None
-            full_response = ""
-            async for chunk, provider in get_hybrid_response(
+            response_chunks = []
+            async for chunk in get_hybrid_response(
                 enhanced_message, session_data["context"], config, knowledge_base
             ):
-                if provider_used is None:
-                    provider_used = provider
-                full_response += chunk
+                response_chunks.append(chunk)
                 await websocket.send_json({
                     "type": "streaming_response",
-                    "message": chunk,
-                    "provider_used": provider_used
+                    "message": chunk
                 })
-            response_text = full_response
+            response_text = "".join(response_chunks)
+            provider_used = hybrid_llm_service.choose_llm_provider(enhanced_message)
+            provider_used = provider_used.value if provider_used else "unknown"
         else:
             session_data["messages"].append(HumanMessage(content=enhanced_message))
             result = await graph.ainvoke({"messages": session_data["messages"]}, config)