include references

main.py

@@ -294,7 +294,7 @@ def analyze_data(query, data_type="news"):
     ]
 
     logger.info(f"{data_type.capitalize()} analysis completed")
-    return messages
+    return messages,search_data
 
 class QueryModel(BaseModel):
     query: str = Field(..., description="Search query")
@@ -312,7 +312,7 @@ class QueryModel(BaseModel):
 
 def search_assistant_api(query, data_type, model="openai/gpt-4o-mini"):
     logger.info(f"Received {data_type} assistant query: {query}")
-    messages = analyze_data(query, data_type)
+    messages, search_data = analyze_data(query, data_type)
     
     if not messages:
         logger.error(f"Failed to fetch {data_type} data")
@@ -326,7 +326,7 @@ def search_assistant_api(query, data_type, model="openai/gpt-4o-mini"):
             yield content
         logger.info(f"Completed {data_type} assistant response for query: {query}")
         logger.info(f"LLM Response: {full_response}")
-    
+        yield "<json>"+ json.dumps(search_data)+"</json>"
     return process_response
 
 def create_streaming_response(generator):