Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -317,7 +317,7 @@ class BasicAgent:
|
|
| 317 |
return fixed_answer
|
| 318 |
|
| 319 |
def __graph_compile__(self):
|
| 320 |
-
def assistant(
|
| 321 |
"""Assistant Node"""
|
| 322 |
return {"message": [self.model_with_tools.invoke(state["messages"])]}
|
| 323 |
|
|
|
|
| 317 |
return fixed_answer
|
| 318 |
|
| 319 |
def __graph_compile__(self):
|
| 320 |
+
def assistant(state: MessagesState):
|
| 321 |
"""Assistant Node"""
|
| 322 |
return {"message": [self.model_with_tools.invoke(state["messages"])]}
|
| 323 |
|