From 14fbfa14060a5d0e76ca1aad31c6bcbb2c97551a Mon Sep 17 00:00:00 2001 From: Claudio Cambra Date: Tue, 5 Dec 2023 05:58:36 +0800 Subject: [PATCH] Fix crash from bad key access into response_message without function_call (#437) Signed-off-by: Claudio Cambra --- memgpt/agent.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/memgpt/agent.py b/memgpt/agent.py index 24bce1ef..0508b4df 100644 --- a/memgpt/agent.py +++ b/memgpt/agent.py @@ -424,7 +424,8 @@ class Agent(object): printd(f"First message didn't include function call: {response_message}") return False - function_name = response_message["function_call"]["name"] + function_call = response_message.get("function_call") + function_name = function_call.get("name") if function_call is not None else "" if require_send_message and function_name != "send_message" and function_name != "archival_memory_search": printd(f"First message function call wasn't send_message or archival_memory_search: {response_message}") return False