+ observability of traces

app.py

@@ -7,20 +7,24 @@ import pandas as pd
 from langraph_agent import build_graph
 import asyncio
 import aiohttp
+from langfuse.langchain import CallbackHandler
+
+# Initialize Langfuse CallbackHandler for LangGraph/Langchain (tracing)
+langfuse_handler = CallbackHandler()
 
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
-# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
+# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------     
 class BasicAgent:
     def __init__(self):
         self.agent = build_graph()
         print("BasicAgent initialized.")
     async def aquery(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        response = await self.agent.run_query(question)
+        response = await self.agent.run_query(question, config={"callbacks": [langfuse_handler]})
         print(f"Agent returning fixed answer: {response}")
         return response
 

langraph_agent.py

@@ -16,6 +16,11 @@ from langchain_core.tools import tool
 from langchain.tools.retriever import create_retriever_tool
 from supabase.client import Client, create_client
 
+from langfuse.langchain import CallbackHandler
+
+# Initialize Langfuse CallbackHandler for LangGraph/Langchain (tracing)
+langfuse_handler = CallbackHandler()
+
 load_dotenv()
 
 @tool
@@ -209,6 +214,6 @@ if __name__ == "__main__":
     graph = build_graph(provider="groq")
     # Run the graph
     messages = [HumanMessage(content=question)]
-    messages = graph.invoke({"messages": messages})
+    messages = graph.invoke({"messages": messages}, config={"callbacks": [langfuse_handler]})
     for m in messages["messages"]:
         m.pretty_print()