fix

count_max_bird_species_tool.py

@@ -4,7 +4,7 @@ from ultralytics import YOLO
 from langchain_core.tools import tool
 from utils import decode_base64_to_frame
 
-@tool
+@tool(return_direct=True)
 def count_max_bird_species_in_video(youtube_url: str) -> int:
     """
     Count the maximum number of bird species to be on camera simultaneously in a YouTube video.

langchain_agent.py

@@ -50,7 +50,6 @@ class LangChainAgent:
             agent=agent, 
             tools=tools, 
             verbose=True,
-            return_intermediate_steps=True,
             max_iterations=3)
 
     def __call__(self, question: str) -> str: