updates

agent.py

@@ -139,12 +139,12 @@ def build_graph():
 
     # --- Graph Definition ---
     builder = StateGraph(MessagesState)
-    builder.add_node("retriever", retriever_node)
+    # builder.add_node("retriever", retriever_node)
     builder.add_node("assistant", assistant)
     builder.add_node("tools", ToolNode(tools))
 
-    builder.add_edge(START, "retriever")
-    builder.add_edge("retriever", "assistant")
+    builder.add_edge(START, "assistant")
+    # builder.add_edge("retriever", "assistant")
     builder.add_conditional_edges("assistant", tools_condition)
     builder.add_edge("tools", "assistant")
 

tools.py

@@ -765,7 +765,7 @@ def reverse_sentence(text: str) -> str:
     """
     return text[::-1]
 
-
+@tool
 def get_max_bird_species_count_from_video(url: str) -> Dict:
     """
     Downloads a YouTube video and returns the maximum number of unique bird species