Add Langfuse Code Logic

app.py

@@ -1,6 +1,7 @@
 import yaml
 import os
 from smolagents import GradioUI, CodeAgent, InferenceClientModel
+from smolagents import LangfuseCallback  # @mangoobee
 
 # Get current directory path
 CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
@@ -12,11 +13,16 @@ from tools.catering_service_tool import SimpleTool as CateringServiceTool
 from tools.superhero_party_theme_generator import SuperheroPartyThemeTool as SuperheroPartyThemeGenerator
 from tools.final_answer import FinalAnswerTool as FinalAnswer
 
-
+# Initialize Langfuse callback (reads from HF Spaces secrets) - @mangoobee
+langfuse_callback = LangfuseCallback(
+    public_key=os.environ["LANGFUSE_PUBLIC_KEY"],
+    secret_key=os.environ["LANGFUSE_SECRET_KEY"],
+    host=os.environ["LANGFUSE_HOST"]
+)
 
 model = InferenceClientModel(
-model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
-provider=None,
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+    provider=None,
 )
 
 web_search = WebSearch()
@@ -26,7 +32,6 @@ catering_service_tool = CateringServiceTool()
 superhero_party_theme_generator = SuperheroPartyThemeGenerator()
 final_answer = FinalAnswer()
 
-
 with open(os.path.join(CURRENT_DIR, "prompts.yaml"), 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
 
@@ -39,7 +44,9 @@ agent = CodeAgent(
     planning_interval=None,
     name=None,
     description=None,
-    prompt_templates=prompt_templates
+    prompt_templates=prompt_templates,
+    callbacks=[langfuse_callback]  # Add this parameter
 )
+
 if __name__ == "__main__":
     GradioUI(agent).launch()