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 |
|