Update agent.py

agent.py

@@ -183,6 +183,8 @@ def build_graph(provider: str = "huggingface", huggingface_model: str = "mistral
     else:
         raise ValueError("Invalid provider. Choose 'google' or 'huggingface'.")
 
+    return llm
+
     # ✅ Bind tools if defined
     llm_with_tools = llm.bind_tools(tools)  # Make sure `tools` is defined/imported
     return llm_with_tools