Integrating with langfuse

Author: sa30nitk

plot_agent/langgraph_plot_agent/api.py

@@ -21,11 +21,13 @@
 logger = logging.getLogger(__name__)
 # Set logging level to DEBUG for more detailed logs
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
-
+from langfuse.callback import CallbackHandler
 from dotenv import load_dotenv
 
 load_dotenv('.env')
 
+langfuse_handler = CallbackHandler()
+
 # Create plots directory if it doesn't exist
 PLOTS_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "plots")
 os.makedirs(PLOTS_DIR, exist_ok=True)
@@ -165,7 +167,7 @@ async def process_query(job_id: str, query: str):
 
         try:
             # Invoke the agent to get final output
-            final_result = agent.invoke({"messages": messages})
+            final_result = agent.invoke({"messages": messages}, config={"callbacks": [langfuse_handler]})
 
             logger.debug(f"Agent invocation complete. Processing final result messages...")
 

plot_agent/langgraph_plot_agent/requirements.in

@@ -19,3 +19,4 @@ typing-extensions
 traceloop-sdk
 fastapi
 langchain_community
+langfuse

plot_agent/langgraph_plot_agent/requirements.txt

@@ -18,6 +18,7 @@ anyio==4.9.0
     # via
     #   anthropic
     #   httpx
+    #   langfuse
     #   openai
     #   starlette
 attrs==25.3.0
@@ -26,7 +27,9 @@ attrs==25.3.0
     #   jsonschema
     #   referencing
 backoff==2.2.1
-    # via posthog
+    # via
+    #   langfuse
+    #   posthog
 blinker==1.9.0
     # via streamlit
 cachetools==5.5.2
@@ -100,6 +103,7 @@ httpcore==1.0.9
 httpx==0.28.1
     # via
     #   anthropic
+    #   langfuse
     #   langgraph-sdk
     #   langsmith
     #   openai
@@ -111,6 +115,7 @@ idna==3.10
     # via
     #   anyio
     #   httpx
+    #   langfuse
     #   requests
     #   yarl
 importlib-metadata==8.6.1
@@ -156,6 +161,8 @@ langchain-openai==0.3.16
     # via -r requirements.in
 langchain-text-splitters==0.3.8
     # via langchain
+langfuse==2.60.4
+    # via -r requirements.in
 langgraph==0.4.3
     # via -r requirements.in
 langgraph-checkpoint==2.0.25
@@ -439,6 +446,7 @@ packaging==24.2
     #   altair
     #   huggingface-hub
     #   langchain-core
+    #   langfuse
     #   langsmith
     #   marshmallow
     #   matplotlib
@@ -475,6 +483,7 @@ pydantic==2.11.4
     #   fastapi
     #   langchain
     #   langchain-core
+    #   langfuse
     #   langgraph
     #   langsmith
     #   openai
@@ -521,6 +530,7 @@ requests==2.32.3
     #   huggingface-hub
     #   langchain
     #   langchain-community
+    #   langfuse
     #   langsmith
     #   opentelemetry-exporter-otlp-proto-http
     #   posthog
@@ -611,6 +621,7 @@ uvicorn==0.34.2
 wrapt==1.17.2
     # via
     #   deprecated
+    #   langfuse
     #   opentelemetry-instrumentation
     #   opentelemetry-instrumentation-sqlalchemy
     #   opentelemetry-instrumentation-threading