Update agent.py

agent.py

@@ -13,9 +13,14 @@ from langchain_experimental.tools.python.tool import PythonAstREPLTool  # for lo
 from tools import (calculator_basic, datetime_tools, transcribe_audio, transcribe_youtube, query_image, webpage_content, read_excel)
 from prompt import system_prompt
 
-# gpt-4.1-nano (cheaper for debugging) with temperature 0 for less randomness
-# for benchmarking use 04-mini (better reasoning) or gpt-4o-mini
+from langchain_core.runnables import RunnableConfig  # for LangSmith tracking
+
+# LangSmith to observe the agent
+langsmith_api_key = os.getenv("LANGSMITH_API_KEY")
+langsmith_tracing = os.getenv("LANGSMITH_TRACING")
 
+# gpt-4.1-nano (cheaper for debugging) with temperature 0 for less randomness
+# for benchmarking use 04-mini (better reasoning) or gpt-4o-mini (cheaper)
 llm = ChatOpenAI(
     model="gpt-4o-mini",
     api_key=os.getenv("OPENAI_API_KEY"),
@@ -104,7 +109,16 @@ class LangGraphAgent:
     def __call__(self, question: str) -> str:
         input_state = {"messages": [HumanMessage(content=question)]}  # prepare the initial user message
         print(f"Running LangGraphAgent with input: {question[:150]}...")
-        result = gaia_agent.invoke(input_state, {"recursion_limit": 30})  # prevents infinite looping when the LLM keeps calling tools over and over
+        
+        # tracing configuration for LangSmith
+        config = RunnableConfig(
+            config={
+                "run_name": "GAIA Agent",
+                "tags": ["gaia", "langgraph", "agent"],
+                "metadata": {"user_input": question}
+            }
+        )
+        result = gaia_agent.invoke(input_state, config, {"recursion_limit": 30})  # prevents infinite looping when the LLM keeps calling tools over and over
         final_response = result["messages"][-1].content
         
         try: