Fimeg/letta-server
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Fix crash from bad key access into response_message without function_call (#437)

Browse Source 
Signed-off-by: Claudio Cambra <developer@claudiocambra.com>
This commit is contained in:
Claudio Cambra
2023-12-05 05:58:36 +08:00
committed by GitHub
parent 408ccec317
commit 14fbfa1406
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
memgpt/agent.py
View File

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