sdsdd

modules/analyzer.py

@@ -20,7 +20,7 @@ class Analyzer:
         
         # Connect to Redis cache
         self.cache = RedisServerStatusCache()
-    
+
     def is_server_ready(self):
         # Check cache first
         cached_status = self.cache.get(self.cache_key)
@@ -61,7 +61,7 @@ class Analyzer:
         self.cache.set(self.cache_key, is_ready, ttl=60)  # Cache for 1 minute
         log_server_status(self.cache_key, is_ready)
         return is_ready
-    
+
     def wait_for_server(self, timeout=300, interval=15):  # Increased timeout to 5 minutes
         """Wait for server to be ready with user feedback"""
         if self.is_server_ready():
@@ -81,7 +81,7 @@ class Analyzer:
         
         logging.warning("⏰ Server initialization timeout reached")
         return False
-    
+
     def analyze_stream(self, query, search_results):
         """ Analyze search results using the custom LLM with streaming output
         Yields chunks of the response as they arrive """
@@ -92,8 +92,7 @@ class Analyzer:
             for result in search_results[:5]  # Limit to top 5 for context
         ])
         
-        prompt = f"""
-You are an expert research analyst. Analyze the following query and information to provide a comprehensive summary.
+        prompt = f"""You are an expert research analyst. Analyze the following query and information to provide a comprehensive summary.
 
 Query: {query}
 Information: {context}
@@ -105,8 +104,7 @@ Please provide:
 4. Potential implications or future directions
 5. Any controversies or conflicting viewpoints
 
-Structure your response clearly with these sections. If there is insufficient information, state that clearly.
-"""
+Structure your response clearly with these sections. If there is insufficient information, state that clearly."""
         
         try:
             # First check if server is ready with extended timeout and user feedback
@@ -134,14 +132,12 @@ Structure your response clearly with these sections. If there is insufficient in
                 stream=True  # Enable streaming
             )
             
-            # Yield chunks as they arrive with proper spacing
-            accumulated_response = ""
+            # Yield chunks as they arrive
             for chunk in response:
                 if chunk.choices[0].delta.content:
                     content = chunk.choices[0].delta.content
-                    accumulated_response += content
-                    # Yield the accumulated response for better streaming display
-                    yield accumulated_response
+                    # Yield only the new content, not the accumulated response
+                    yield content
             
             logging.info("Analysis streaming completed successfully")
             
@@ -157,7 +153,7 @@ Structure your response clearly with these sections. If there is insufficient in
                 yield "⚠️ The AI model endpoint was not found. Please check the configuration."
             else:
                 yield f"Analysis failed: {str(e)}"
-    
+
     def analyze(self, query, search_results):
         """ Non-streaming version for compatibility """
         
@@ -167,8 +163,7 @@ Structure your response clearly with these sections. If there is insufficient in
             for result in search_results[:5]  # Limit to top 5 for context
         ])
         
-        prompt = f"""
-You are an expert research analyst. Analyze the following query and information to provide a comprehensive summary.
+        prompt = f"""You are an expert research analyst. Analyze the following query and information to provide a comprehensive summary.
 
 Query: {query}
 Information: {context}
@@ -180,8 +175,7 @@ Please provide:
 4. Potential implications or future directions
 5. Any controversies or conflicting viewpoints
 
-Structure your response clearly with these sections. If there is insufficient information, state that clearly.
-"""
+Structure your response clearly with these sections. If there is insufficient information, state that clearly."""
         
         try:
             # First check if server is ready with extended timeout