Update agents.py

agents.py

@@ -0,0 +1,50 @@
+import os
+from typing import TypedDict, Annotated, List
+from langchain.tools import DuckDuckGoSearchResults
+from langchain_groq import ChatGroq
+from langgraph.graph import StateGraph, END, START
+from langgraph.graph.message import AnyMessage, add_messages
+
+# --- Setup API Key ---
+groq_api_key = os.getenv("GROQ_API_KEY")
+
+# --- Setup Search Tool ---
+search = DuckDuckGoSearchResults()
+
+# --- Setup Model ---
+model = ChatGroq(
+    model_name="llama3-8b-8192",
+    api_key=groq_api_key,
+    temperature=0
+)
+
+# --- Tool List ---
+tools = [search]
+
+# --- Bind Model with Tools ---
+llm_with_tools = model.bind_tools(
+    tools,
+    parallel_tool_calls=True
+)
+
+# --- Define State Class ---
+class AgentState(TypedDict):
+    messages: Annotated[List[AnyMessage], add_messages]
+
+# --- Define Search Node ---
+def search_node(state: AgentState):
+    response = llm_with_tools.invoke(state["messages"])
+    return {"messages": [response]}
+
+# --- Build Graph ---
+builder = StateGraph(AgentState)
+builder.add_node("search", search_node)
+builder.add_edge(START, "search")
+builder.add_edge("search", END)
+graph = builder.compile()
+
+# --- Chat Function ---
+def chat(question: str):
+    initial_messages = [{"role": "user", "content": question}]
+    result = graph.invoke({"messages": initial_messages})
+    return result