Update agent.py

agent.py

@@ -0,0 +1,51 @@
+from typing import TypedDict, Annotated
+from tools import add, substract, multiply, divide, DuckDuckGoSearchTool, WikipediaSearchTool, ArxivSearchTool, PubmedSearchTool
+
+from langgraph.graph.message import add_messages
+from langchain_core.messages import AnyMessage, HumanMessage, AIMessage
+from langgraph.graph import StateGraph, START, END, MessagesState
+from langgraph.prebuilt import ToolNode, tools_condition
+from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
+
+
+# load the system prompt from the file
+with open("system_prompt.txt", "r", encoding="utf-8") as f:
+    system_prompt = f.read()
+
+# System message
+sys_msg = SystemMessage(content=system_prompt)
+
+# Making the agent
+llm = HuggingFaceEndpoint(
+    repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
+    huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
+)
+
+chat = ChatHuggingFace(llm=llm, verbose=True)
+tools = [add, 
+         substract, 
+         multiply, 
+         divide, 
+         DuckDuckGoSearchTool, 
+         WikipediaSearchTool, 
+         ArxivSearchTool, 
+         PubmedSearchTool]
+
+chat_with_tools = chat.bind_tools(tools)
+
+## Defining our nodes
+def assistant(state: MessagesState):
+    """Assistant node"""
+    return {"messages": [sys_msg] + [chat_with_tools.invoke(state["messages"])]}
+
+# Build graph / nodes
+builder = StateGraph(State)
+builder.add_node("assistant", chat) # Assistant
+builder.add_node("tools", ToolNode(tools)) # Tools
+
+# Logic / edges
+builder.add_edge(START, "assistant")
+builder.add_conditional_edges("assistant", tools_condition)
+builder.add_edge("tools", "assistant")
+
+graph = builder.compile()