HCho commited on
Commit
fcb8fc2
·
verified ·
1 Parent(s): 94eedaa

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -108,7 +108,7 @@ def build_graph(provider: str = "google"):
108
  """ Use the tools to answer the query. you have add,subtract,multiply,divide,web_search,wikipedia_search,arxiv_search tools."""
109
  response = llm_with_tools.invoke([system_message]+state["messages"])
110
  time.sleep(5) # 무료 티어의 한계
111
- return {"messages": state["messages"] + [response)]}
112
 
113
 
114
  builder = StateGraph(MessagesState)
 
108
  """ Use the tools to answer the query. you have add,subtract,multiply,divide,web_search,wikipedia_search,arxiv_search tools."""
109
  response = llm_with_tools.invoke([system_message]+state["messages"])
110
  time.sleep(5) # 무료 티어의 한계
111
+ return {"messages": state["messages"] + [response]}
112
 
113
 
114
  builder = StateGraph(MessagesState)