antonioschiro commited on
Commit
61d8471
·
verified ·
1 Parent(s): 3210875

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -21,7 +21,7 @@ class State(TypedDict):
21
  tools = [websearch]
22
 
23
  llm = ChatGroq(
24
- model="meta-llama/llama-4-maverick-17b-128e-instruct",
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)