scriptoriumadmin/ReMe
1
0
Fork 0
mirror of https://github.com/agentscope-ai/ReMe.git synced 2026-10-08 03:10:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[dev] update cli, delete try catch in while loop

Browse source
This commit is contained in:
hs 2024-06-27 14:53:02 +08:00
parent 093946e999
commit f16034d6a1
1 changed files with 21 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

41
memory_scope/chat/cli_memory_chat.py
View file

@ -64,7 +64,7 @@ class CliMemoryChat(BaseMemoryChat):
for result in self.generation_model.call(messages=[system_message, new_message], stream=self.stream):
yield result
self.memory_service.add_messages(result.text)
self.memory_service.add_messages(result.message)
def run(self):
self.USER_COMMANDS.update({f"/{k}": v for k, v in self.memory_service.op_description_dict.items()})
@ -118,22 +118,23 @@ class CliMemoryChat(BaseMemoryChat):
continue
while True:
try:
if self.stream:
for msg in self.chat_with_memory(query=query):
print(msg.text, flush=True)
print()
else:
msg = self.chat_with_memory(query=query)
print(msg.text)
break
except KeyboardInterrupt:
questionary.print("User interrupt occurred.")
retry = questionary.confirm("Retry chat_with_memory()?").ask()
if not retry:
break
except Exception as e:
questionary.print(f"An exception occurred when running chat_with_memory(): {e}")
retry = questionary.confirm("Retry chat_with_memory()?").ask()
if not retry:
break
# try:
if self.stream:
for msg in self.chat_with_memory(query=query):
print(msg.delta, end="")
print()
else:
msg = self.chat_with_memory(query=query)
print(msg.message.content)
break
# except KeyboardInterrupt:
# questionary.print("User interrupt occurred.")
# retry = questionary.confirm("Retry chat_with_memory()?").ask()
# if not retry:
# break
# except Exception as e:
# questionary.print(f"An exception occurred when running chat_with_memory(): {e}")
# # retry = questionary.confirm("Retry chat_with_memory()?").ask()
# # if not retry:
# # break
# raise e