Update agent.py

agent.py

@@ -5,6 +5,7 @@ from langgraph.prebuilt import ToolNode, tools_condition
 from langchain_core.messages import HumanMessage, AIMessage
 from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain_core.runnables import Runnable
+from llama_index.core.agent.workflow import AgentWorkflow, ReActAgentfrom llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
 from tools import TOOLS
 import pandas as pd
 
@@ -18,7 +19,8 @@ def build_graph():
     
     # Initialize the LLM (e.g., Gemini Flash, zero temperature)
     # llm = ChatGoogleGenerativeAI(model="gemini-1.5-flash", temperature=0)
-    llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
+    # llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
+    llm = HuggingFaceInferenceAPI(model_name="Qwen/Qwen2.5-Coder-32B-Instruct")
     llm_with_tools = llm.bind_tools(TOOLS)
 
     # Step 1: Retriever node