fixed langfuse flush function

agent.py

@@ -3,6 +3,8 @@ from typing import TypedDict, List, Dict, Any, Optional, Annotated
 from functools import partial
 import os
 import re
+import time
+import uuid
 
 # from langchain_openai import ChatOpenAI
 # from langchain_huggingface.llms import HuggingFaceEndpoint
@@ -197,8 +199,6 @@ class BasicAgent:
         print(f"Agent received question (first 100 chars): {question[:100]}...")
         
         # Create a new Langfuse handler for this specific question to ensure separate traces
-        import time
-        import uuid
         handler = CallbackHandler()
         
         # Generate unique identifiers for this trace
@@ -229,8 +229,8 @@ class BasicAgent:
         response_text = response['messages'][-1].content
         answer = extract_answer(response_text)
         
-        # Flush the handler to ensure the trace is sent to Langfuse
-        handler.flush()
+        # Flush the langfuse client to ensure the trace is sent immediately
+        langfuse.flush()
         
         print(f"Trace logged for task_id: {task_id}")