Update agent.py

agent.py

@@ -8,25 +8,27 @@ 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_google_genai import ChatGoogleGenerativeAI
+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"],
+# HF Chat interface
+llm = HuggingFaceEndpoint(
+    repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
+    huggingfacehub_api_token=os.environ["HUGGINGFACEHUB_API_TOKEN"],
+)
+chat = ChatHuggingFace(llm=llm, verbose=True)
+
+# Google's chat interface
+# chat = ChatGoogleGenerativeAI(
+#     model="gemini-2.0-flash",
+#     temperature=0.7,
+#     max_retries=2,
 # )
-# chat = ChatHuggingFace(llm=llm, verbose=True)
 
-chat = ChatGoogleGenerativeAI(
-    model="gemini-2.0-flash",
-    temperature=0.7,
-    max_retries=2,
-)
 tools = [search_tool]
 chat_with_tools = chat.bind_tools(tools)