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

updated CLI interface to properly print searches on archival memory + recall memory (#731)

Browse Source
This commit is contained in:
Charles Packer
2023-12-28 12:06:20 -08:00
committed by GitHub
parent b5fd0d5a17
commit e2e2a5c78c
1 changed files with 18 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
memgpt/interface.py
View File

@@ -159,8 +159,11 @@ class CLIInterface(AgentInterface):
if match:
function_name = match.group(1)
function_args = match.group(2)
if "memory" in function_name:
print_function_message("🧠", f"updating memory with {function_name}")
if function_name in ["archival_memory_insert", "archival_memory_search"]:
if function_name == "archival_memory_insert":
print_function_message("🧠", f"updating memory with {function_name}")
elif function_name == "archival_memory_search":
print_function_message("🧠", f"searching memory with {function_name}")
try:
msg_dict = eval(function_args)
if function_name == "archival_memory_search":
@@ -186,6 +189,19 @@ class CLIInterface(AgentInterface):
printd(str(e))
printd(msg_dict)
pass
elif function_name in ["conversation_search", "conversation_search_date"]:
print_function_message("🧠", f"searching memory with {function_name}")
try:
msg_dict = eval(function_args)
output = f'\tquery: {msg_dict["query"]}, page: {msg_dict["page"]}'
if STRIP_UI:
print(output)
else:
print(f"{Fore.RED}{output}{Style.RESET_ALL}")
except Exception as e:
printd(str(e))
printd(msg_dict)
pass
else:
printd(f"{CLI_WARNING_PREFIX}did not recognize function message")
printd_function_message("", msg)