Scott Cogan commited on
Commit ·
10894eb
1
Parent(s): 94720ac
gradio
Browse files
app.py
CHANGED
|
@@ -175,7 +175,7 @@ class BasicAgent:
|
|
| 175 |
, google_search
|
| 176 |
]
|
| 177 |
|
| 178 |
-
self.agent = self.llm
|
| 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.
|
| 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
|