Update agent.py

agent.py

@@ -8,20 +8,26 @@ from langgraph.prebuilt import ToolNode
 from langgraph.graph import START, StateGraph
 from langgraph.checkpoint.memory import MemorySaver
 from langgraph.prebuilt import tools_condition
-from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
+# from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
+from langchain_google_genai import ChatGoogleGenerativeAI
 from smolagents import CodeAgent, HfApiModel
 
 # Import our custom tools from their modules
 from tools import hub_stats_tool, search_tool
 
 # Generate the chat interface, including the tools
-llm = HuggingFaceEndpoint(
-    repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
-    huggingfacehub_api_token=os.environ["HUGGINGFACEHUB_API_TOKEN"],
+# llm = HuggingFaceEndpoint(
+#     repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
+#     huggingfacehub_api_token=os.environ["HUGGINGFACEHUB_API_TOKEN"],
+# )
+# chat = ChatHuggingFace(llm=llm, verbose=True)
+
+chat = ChatGoogleGenerativeAI(
+    model="gemini-2.0-flash",
+    temperature=0.7,
+    max_retries=2,
 )
-
-chat = ChatHuggingFace(llm=llm, verbose=True)
-tools = [search_tool, hub_stats_tool]
+tools = [search_tool]
 chat_with_tools = chat.bind_tools(tools)
 
 # Generate the AgentState and Agent graph