​
MemoryBlock

class MemoryBlock(ABC):

An abstract class serves as the fundamental component within the agent memory system. This class is equipped with “write” and “clear” functions. However, it intentionally does not define a retrieval interface, as the structure of the data to be retrieved may vary in different types of memory blocks.

​
write_records

def write_records(self, records: List[MemoryRecord]):

Writes records to the memory, appending them to existing ones.

Parameters:

  • records (List[MemoryRecord]): Records to be added to the memory.

​
write_record

def write_record(self, record: MemoryRecord):

Writes a record to the memory, appending it to existing ones.

Parameters:

  • record (MemoryRecord): Record to be added to the memory.

​
clear

def clear(self):

Clears all messages from the memory.

​
BaseContextCreator

class BaseContextCreator(ABC):

An abstract base class defining the interface for context creation strategies.

This class provides a foundational structure for different strategies to generate conversational context from a list of context records. The primary goal is to create a context that is aligned with a specified token count limit, allowing subclasses to define their specific approach.

Subclasses should implement the :obj:token_counter,:obj: token_limit, and :obj:create_context methods to provide specific context creation logic.

Attributes: token_counter (BaseTokenCounter): A token counter instance responsible for counting tokens in a message. token_limit (int): The maximum number of tokens allowed in the generated context.

​
token_counter

def token_counter(self):

​
token_limit

def token_limit(self):

​
create_context

def create_context(self, records: List[ContextRecord]):

An abstract method to create conversational context from the chat history.

Constructs the context from provided records. The specifics of how this is done and how the token count is managed should be provided by subclasses implementing this method. The output messages order should keep same as the input order.

Parameters:

  • records (List[ContextRecord]): A list of context records from which to generate the context.

Returns:

Tuple[List[OpenAIMessage], int]: A tuple containing the constructed context in OpenAIMessage format and the total token count.

​
AgentMemory

class AgentMemory(MemoryBlock, ABC):

Represents a specialized form of MemoryBlock, uniquely designed for direct integration with an agent. Two key abstract functions, “retrieve” and “get_context_creator”, are used for generating model context based on the memory records stored within the AgentMemory.

​
agent_id

def agent_id(self):

​
agent_id

def agent_id(self, val: Optional[str]):

​
retrieve

def retrieve(self):

Returns:

List[ContextRecord]: A record list for creating model context.

​
get_context_creator

def get_context_creator(self):

Returns:

BaseContextCreator: A model context creator.

​
get_context

def get_context(self):

Returns:

(List[OpenAIMessage], int): A tuple containing the constructed context in OpenAIMessage format and the total token count.

​
repr

def __repr__(self):

Returns:

str: A string in the format ‘ClassName(agent_id=<id>)’ if agent_id exists, otherwise just ‘ClassName()’.