memoryscope.core.chat.cli_memory_chat

class memoryscope.core.chat.cli_memory_chat.CliMemoryChat(**kwargs)

基类：ApiMemoryChat

Command-line interface for chatting with an AI that integrates memory functionality. Allows users to interact, manage chat history, adjust streaming settings, and view commands' help.

USER_COMMANDS = {'clear': 'Clear the command history.', 'exit': 'Exit the CLI.', 'help': 'Display available CLI commands and their descriptions.', 'stream': 'Toggle between getting streamed responses from the model.'}

__init__(**kwargs)

print_logo()

Prints the logo of the CLI application to the console.

The logo is composed of multiple lines, which are iterated through and printed one by one to provide a visual identity for the chat interface.

chat_with_memory(query: str, role_name: str | None = None, system_prompt: str | None = None, memory_prompt: str | None = None, temporary_memories: str | None = None, history_message_strategy: Literal['auto', None] | int = 'auto', remember_response: bool = True, **kwargs)

The core function that carries out conversation with memory accepts user queries through query and returns the conversation results through model_response. The retrieved memories are stored in the memories within meta_data. :param query: User's query, includes the user's question. :type query: str :param role_name: User's role name. :type role_name: str, optional :param system_prompt: System prompt. Defaults to the system_prompt in "memory_chat_prompt.yaml". :type system_prompt: str, optional :param memory_prompt: Memory prompt, It takes effect when there is a memory and will be placed in

front of the retrieved memory. Defaults to the memory_prompt in "memory_chat_prompt.yaml".

参数:

• temporary_memories (str, optional) -- Manually added user memory in this function.

• history_message_strategy ("auto", None, int) --

  • If it is set to "auto"， the history messages in the conversation will retain those that have not

    yet been summarized. Default to "auto".

  • If it is set to None， no conversation history will be saved.

  • If it is set to an integer value "n", recent "n" message-pair[user, assistant] will be retained.

• remember_response (bool, optional) -- Flag indicating whether to save the AI's response to memory. Defaults to False.

返回:

In non-streaming mode, returns a complete AI response. - ModelResponseGen: In streaming mode, returns a generator yielding AI response parts. - Memories: To obtain the memory by invoking the method of model_response.meta_data[MEMORIES]

返回类型:

• ModelResponse

static parse_query_command(query: str)

Parses the user's input query command, separating it into the command and its associated keyword arguments.

参数:

query (str) -- The raw input string from the user which includes the command and its arguments.

返回:

A tuple containing the command (str) as the first element and a dictionary (kwargs) of keyword arguments as the second element.

返回类型:

tuple

process_commands(query: str) → bool

Parses and executes commands from user input in the CLI chat interface. Supports operations like exiting, clearing screen, showing help, toggling stream mode, executing predefined memory operations, and handling unknown commands.

参数:

query (str) -- The user's input command string.

返回:

Indicates whether to continue running the CLI after processing the command.

返回类型:

bool

run()

Runs the CLI chat loop, which handles user input, processes commands, communicates with the AI model, manages conversation memory, and controls the chat session including streaming responses, command execution, and error handling.

The loop continues until the user explicitly chooses to exit.

context: MemoryscopeContext

stream: bool

generation_model_kwargs: dict

kwargs: dict