Update agent.py
Browse files
agent.py
CHANGED
|
@@ -73,7 +73,7 @@ def build_agent():
|
|
| 73 |
|
| 74 |
def call_model(state: MessagesState):
|
| 75 |
"""Call the LLM with the given state."""
|
| 76 |
-
messages =
|
| 77 |
response = llm.invoke(messages)
|
| 78 |
return {"messages" : response}
|
| 79 |
|
|
|
|
| 73 |
|
| 74 |
def call_model(state: MessagesState):
|
| 75 |
"""Call the LLM with the given state."""
|
| 76 |
+
messages = state["messages"]
|
| 77 |
response = llm.invoke(messages)
|
| 78 |
return {"messages" : response}
|
| 79 |
|