reverting tools.py and app.py

Signed-off-by: apple muncy <[email protected]>

app.py

@@ -1,7 +1,6 @@
 from typing import TypedDict, Annotated
 from langgraph.graph.message import add_messages
 from langchain_core.messages import AnyMessage, HumanMessage, AIMessage, SystemMessage
-from langchain_core.tools import tool
 from langgraph.prebuilt import ToolNode
 from langgraph.graph import START, StateGraph
 from langgraph.prebuilt import tools_condition
@@ -33,16 +32,15 @@ chat = ChatOllama(model="qwen2:7b",
 
 # Define available tools
 tools = [
-#        load_guest_dataset, 
-#        search_tool, 
+        load_guest_dataset, 
+        search_tool, 
         weather_info_tool, 
-#        hub_stats_tool
+        hub_stats_tool
         ]
 
 # Bind tools to the chat model
 chat_with_tools = chat.bind_tools(tools)
 
-toolnode =ToolNode(Tools)
 # Generate the AgentState and Agent graph
 class AgentState(TypedDict):
     messages: Annotated[list[AnyMessage], add_messages]
@@ -64,7 +62,7 @@ builder = StateGraph(AgentState)
 
 # Define nodes: these do the work
 builder.add_node("assistant", assistant)
-builder.add_node("tools", toolnode)
+builder.add_node("tools", ToolNode(tools))
 
 # Define edges: these determine how the control flow moves
 builder.add_edge(START, "assistant")
@@ -104,4 +102,4 @@ with gr.Blocks() as demo:
     send_btn.click(fn=chat_fn, inputs=[msg, chatbot], outputs=[chatbot, msg])
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()

tools.py

@@ -1,13 +1,16 @@
+#from smolagents import DuckDuckGoSearchTool
+#from smolagents import Tool
 from langchain_community.tools import DuckDuckGoSearchRun
 import random
+from huggingface_hub import list_models
+from langchain.tools import Tool
 
-from langchain_core.tools import tool
 
 # Initialize the DuckDuckGo search tool
 
 
 search_tool = DuckDuckGoSearchRun()
-@tool
+
 def get_weather_info(location: str) -> str:
     """Fetches dummy weather information for a given location."""
     # Dummy weather data
@@ -21,7 +24,7 @@ def get_weather_info(location: str) -> str:
     return f"Weather in {location}: {data['condition']}, {data['temp_c']}°C"
 
 # Initialize the tool
-weather_info_tool = tool(
+weather_info_tool = Tool(
     name="get_weather_info",
     func=get_weather_info,
     description="Fetches dummy weather information for a given location."
@@ -47,6 +50,4 @@ hub_stats_tool = Tool(
     name="get_hub_stats",
     func=get_hub_stats,
     description="Fetches the most downloaded model from a specific author on the Hugging Face Hub."
-)
-
-
+)