Update search_utils.py

search_utils.py

@@ -68,4 +68,23 @@ def analyze_data(query, data_type="news"):
     ]
 
     logger.info(f"{data_type.capitalize()} analysis completed")
-    return messages,data
+    return messages,data
+
+def search_assistant_api(query, data_type, model="openai/gpt-4o-mini"):
+    logger.info(f"Received {data_type} assistant query: {query}")
+    messages, search_data = analyze_data(query, data_type)
+    
+    if not messages:
+        logger.error(f"Failed to fetch {data_type} data")
+        raise HTTPException(status_code=500, detail=f"Failed to fetch {data_type} data")
+    
+    def process_response():
+        logger.info(f"Generating response using LLM: {messages}")
+        full_response = ""
+        for content in chat_with_llama_stream(messages, model=model):
+            full_response += content
+            yield content
+        logger.info(f"Completed {data_type} assistant response for query: {query}")
+        logger.info(f"LLM Response: {full_response}")
+        yield "<json><ref>"+ json.dumps(search_data)+"</ref></json>"
+    return process_response