Update agent.py

agent.py

@@ -14,6 +14,8 @@ from langchain_core.messages import SystemMessage, HumanMessage
 from langchain_core.tools import tool
 from langchain.tools.retriever import create_retriever_tool
 from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
+from smolagents import CodeAgent, InferenceClientModel
+
 
 load_dotenv()
 
@@ -136,7 +138,8 @@ def build_graph(provider: str = "huggingface"):
         # Groq https://console.groq.com/docs/models
         llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
     elif provider == "huggingface":
-         llm = HuggingFaceInferenceAPI(model_name="Qwen/Qwen2.5-Coder-32B-Instruct")
+         llm = InferenceClientModel()
+
     else:
         raise ValueError("Invalid provider. Choose 'google', 'groq' or 'huggingface'.")
     # Bind tools to LLM