Upload agent.py

agent.py

@@ -0,0 +1,34 @@
+from typing import TypedDict, Annotated
+from langgraph.graph.message import add_messages
+from langchain_core.messages import AnyMessage
+from langgraph.graph import START, StateGraph
+from langgraph.prebuilt import ToolNode, tools_condition
+from langchain_ollama import ChatOllama  
+from tools import search_tool, weather_info_tool, hub_stats_tool
+from dotenv import load_dotenv
+load_dotenv()
+
+llm = ChatOllama(model="mistral") 
+
+
+tools = [search_tool, weather_info_tool, hub_stats_tool]
+chat_with_tools = llm.bind_tools(tools)
+
+
+class AgentState(TypedDict):
+    messages: Annotated[list[AnyMessage], add_messages]
+
+
+def assistant(state: AgentState):
+    return {
+        "messages": [chat_with_tools.invoke(state["messages"])]
+    }
+
+
+builder = StateGraph(AgentState)
+builder.add_node("assistant", assistant)
+builder.add_node("tools", ToolNode(tools))
+builder.add_edge(START, "assistant")
+builder.add_conditional_edges("assistant", tools_condition)
+builder.add_edge("tools", "assistant")
+alfred = builder.compile()