llama-index

1
"""Agent utils."""
2

3
from llama_index.legacy.agent.types import TaskStep
4
from llama_index.legacy.core.llms.types import MessageRole
5
from llama_index.legacy.llms.base import ChatMessage
6
from llama_index.legacy.memory import BaseMemory
7

8

9
def add_user_step_to_memory(
10
    step: TaskStep, memory: BaseMemory, verbose: bool = False
11
) -> None:
12
    """Add user step to memory."""
13
    user_message = ChatMessage(content=step.input, role=MessageRole.USER)
14
    memory.put(user_message)
15
    if verbose:
16
        print(f"Added user message to memory: {step.input}")
17