Scott Cogan commited on
Commit
10894eb
·
1 Parent(s): 94720ac
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -175,7 +175,7 @@ class BasicAgent:
175
  , google_search
176
  ]
177
 
178
- self.agent = self.llm.bind_tools(self.tools)
179
 
180
  self.sys_msg = SystemMessage('''You are a general AI assistant. I will ask you a question. Only provide YOUR FINAL ANSWER and nothing else.
181
  YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
@@ -210,7 +210,7 @@ class BasicAgent:
210
  print("BasicAgent initialized.")
211
 
212
  def assistant(self, state):
213
- return {"messages": [self.agent.invoke([self.sys_msg] + state["messages"])]}
214
 
215
  def tools_node(self, state):
216
  # Execute the tool and return the result
 
175
  , google_search
176
  ]
177
 
178
+ self.agent = self.llm # Use the LLM directly
179
 
180
  self.sys_msg = SystemMessage('''You are a general AI assistant. I will ask you a question. Only provide YOUR FINAL ANSWER and nothing else.
181
  YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
 
210
  print("BasicAgent initialized.")
211
 
212
  def assistant(self, state):
213
+ return {"messages": [self.llm.invoke([self.sys_msg] + state["messages"])]}
214
 
215
  def tools_node(self, state):
216
  # Execute the tool and return the result