Update agent.py
Browse files
agent.py
CHANGED
|
@@ -21,7 +21,7 @@ class State(TypedDict):
|
|
| 21 |
tools = [websearch]
|
| 22 |
|
| 23 |
llm = ChatGroq(
|
| 24 |
-
model="
|
| 25 |
temperature=0.1,
|
| 26 |
max_tokens= None,
|
| 27 |
reasoning_format="parsed",
|
|
@@ -35,7 +35,7 @@ def ask_agent(agent_state: State):
|
|
| 35 |
content = text_prompt
|
| 36 |
)
|
| 37 |
query = agent_state["messages"][-1] # HumanMessage
|
| 38 |
-
response = llm_with_tools.invoke(query.content)
|
| 39 |
return {"messages": [response]}
|
| 40 |
|
| 41 |
graph_builder = StateGraph(State)
|
|
|
|
| 21 |
tools = [websearch]
|
| 22 |
|
| 23 |
llm = ChatGroq(
|
| 24 |
+
model="deepseek-r1-distill-llama-70b",
|
| 25 |
temperature=0.1,
|
| 26 |
max_tokens= None,
|
| 27 |
reasoning_format="parsed",
|
|
|
|
| 35 |
content = text_prompt
|
| 36 |
)
|
| 37 |
query = agent_state["messages"][-1] # HumanMessage
|
| 38 |
+
response = llm_with_tools.invoke(text_prompt + query.content)
|
| 39 |
return {"messages": [response]}
|
| 40 |
|
| 41 |
graph_builder = StateGraph(State)
|