test out langfuse

app.py

@@ -5,8 +5,31 @@ import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
 
+import os
+import base64
+
 from Gradio_UI import GradioUI
 
+LANGFUSE_PUBLIC_KEY = os.getenv("LANGFUSE_PUBLIC_KEY")
+LANGFUSE_SECRET_KEY = os.getenv("LANGFUSE_SECRET_KEY")
+
+LANGFUSE_AUTH=base64.b64encode(f"{LANGFUSE_PUBLIC_KEY}:{LANGFUSE_SECRET_KEY}".encode()).decode()
+
+#os.environ["OTEL_EXPORTER_OTLP_ENDPOINT"] = "https://cloud.langfuse.com/api/public/otel" # EU data region
+os.environ["OTEL_EXPORTER_OTLP_ENDPOINT"] = "https://us.cloud.langfuse.com/api/public/otel" # US data region
+os.environ["OTEL_EXPORTER_OTLP_HEADERS"] = f"Authorization=Basic {LANGFUSE_AUTH}"
+
+from opentelemetry.sdk.trace import TracerProvider
+
+from openinference.instrumentation.smolagents import SmolagentsInstrumentor
+from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
+from opentelemetry.sdk.trace.export import SimpleSpanProcessor
+
+trace_provider = TracerProvider()
+trace_provider.add_span_processor(SimpleSpanProcessor(OTLPSpanExporter()))
+
+SmolagentsInstrumentor().instrument(tracer_provider=trace_provider)
+
 @tool
 def multiplication_tool(arg1:int, arg2:int)-> int:
     #Keep this format for the description / args / args description but feel free to modify the tool
@@ -56,7 +79,7 @@ agent = CodeAgent(
     model=model,
     tools=[final_answer,multiplication_tool,image_generation_tool],
     max_steps=6,
-    verbosity_level=1,
+#    verbosity_level=1,
     grammar=None,
     planning_interval=None,
     name=None,

requirements.txt

@@ -3,3 +3,7 @@ smolagents
 requests
 duckduckgo_search
 pandas
+
+opentelemetry-sdk
+opentelemetry-exporter-otlp
+openinference-instrumentation-smolagents