Update agents/agents_nodes.py

agents/agents_nodes.py

@@ -22,23 +22,17 @@ from tools.financial_tools import time_value_tool
 #     return {"messages": [response]}
 
 import os
-from langchain_community.chat_models import ChatHuggingFace
-from langchain_community.llms import HuggingFaceEndpoint
-from langchain_core.messages import AIMessage
+from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint
+
 
-# Initialize Hugging Face endpoint (replace with your model)
-HF_MODEL = "google/gemma-2b-it"
 llm_endpoint = HuggingFaceEndpoint(
-    endpoint_url=f"https://huggingface.co/models/{HF_MODEL}",
-    huggingfacehub_api_token=os.environ["HF_TOKEN"],
+    endpoint_url="https://api-inference.huggingface.co/models/google/gemma-2b-it",
+    huggingfacehub_api_token=os.environ["HF_TOKEN"],  # Set in Space secrets
     max_new_tokens=500,
     temperature=0
 )
 
-# Wrap in chat model
 llm = ChatHuggingFace(llm=llm_endpoint)
-
-# Bind tools to model
 llm_instantiated = llm.bind_tools(
     [time_value_tool],
     tool_choice={"type": "function", "function": {"name": "time_value_tool"}}