add langfuse for observability

app.py

@@ -1,19 +1,37 @@
 from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 import datetime
-import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
 from tools.visit_webpage import VisitWebpageTool
 from tools.web_search import DuckDuckGoSearchTool
-import requests
 from Gradio_UI import GradioUI
 import os
+from openinference.instrumentation.smolagents import SmolagentsInstrumentor
+from langfuse import get_client
 from dotenv import load_dotenv
 load_dotenv()
 
 SERPAPI_KEY = os.getenv("SERPAPI_KEY")
 
+langfuse = get_client()
+ 
+# Verify connection
+if langfuse.auth_check():
+    print("Langfuse client is authenticated and ready!")
+else:
+    print("Authentication failed. Please check your credentials and host.")
+
+# Try to instrument with SmolagentsInstrumentor, but handle compatibility issues gracefully
+try:
+    SmolagentsInstrumentor().instrument()
+    print("✅ SmolagentsInstrumentor successfully initialized!")
+except Exception as e:
+    print(f"⚠️ SmolagentsInstrumentor failed: {e}")
+    print("📊 Continuing with Langfuse-only tracking...")
+
+    
+
 @tool
 def youtube_search(query: str, max_results: int = 5) -> str:
     """A tool that searches YouTube and returns video titles and URLs.
@@ -76,6 +94,8 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 
+
+
 final_answer = FinalAnswerTool()
 
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
@@ -97,7 +117,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_in_timezone, youtube_search, DuckDuckGoSearchTool(), VisitWebpageTool()], ## add your tools here (don't remove final answer)
+    tools=[get_current_time_in_timezone, youtube_search, DuckDuckGoSearchTool(), VisitWebpageTool(), final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,

prompts.yaml

@@ -319,3 +319,12 @@
   "report": |-
     Here is the final answer from your managed agent '{{name}}':
     {{final_answer}}
+
+"final_answer":
+  "pre_messages": |-
+    You have reached the maximum number of steps. Please provide a final answer to the best of your ability.
+    Make sure to give as much context as possible in your answer.
+  "post_messages": |-
+    Thank you for providing your final answer.
+  "prompt": |-
+    Provide your final answer to the task.

requirements.txt

@@ -1,5 +1,11 @@
 markdownify
-smolagents==1.13.0
+smolagents
+smolagents[gradio]
 requests
 duckduckgo_search
 pandas
+transformers
+opentelemetry-sdk 
+opentelemetry-exporter-otlp 
+openinference-instrumentation-smolagents 
+langfuse

tools/superhero_party_theme.py

@@ -0,0 +1,25 @@
+from smolagents.tools import Tool
+
+class SuperheroPartyThemeTool(Tool):
+    name = "superhero_party_theme_generator"
+    description = """
+    This tool suggests creative superhero-themed party ideas based on a category.
+    It returns a unique party theme idea."""
+    
+    inputs = {
+        "category": {
+            "type": "string",
+            "description": "The type of superhero party (e.g., 'classic heroes', 'villain masquerade', 'futuristic Gotham').",
+        }
+    }
+    
+    output_type = "string"
+
+    def forward(self, category: str):
+        themes = {
+            "classic heroes": "Justice League Gala: Guests come dressed as their favorite DC heroes with themed cocktails like 'The Kryptonite Punch'.",
+            "villain masquerade": "Gotham Rogues' Ball: A mysterious masquerade where guests dress as classic Batman villains.",
+            "futuristic Gotham": "Neo-Gotham Night: A cyberpunk-style party inspired by Batman Beyond, with neon decorations and futuristic gadgets."
+        }
+        
+        return themes.get(category.lower(), "Themed party idea not found. Try 'classic heroes', 'villain masquerade', or 'futuristic Gotham'.")

tools/visit_webpage.py

@@ -1,8 +1,5 @@
 from typing import Any, Optional
 from smolagents.tools import Tool
-import requests
-import markdownify
-import smolagents
 import re
 
 class VisitWebpageTool(Tool):

tools/web_search.py

@@ -1,6 +1,5 @@
 from typing import Any, Optional
 from smolagents.tools import Tool
-import duckduckgo_search
 
 class DuckDuckGoSearchTool(Tool):
     name = "web_search"