Chat

class Chat

Bases: object

Class for a chat session.

add_message(message: Message) → None

Add a message to the chat.

Parameters:

message (Message) – Message to be added to the chat

Returns:

None

Return type:

None

generate_prompt(prompt_format: PromptFormat) → str

Generate prompt from the chat.

Parameters:

prompt_format (PromptFormat) – Format of the prompt

Returns:

Generated prompt

Return type:

str

generate_prompt_anthropic() → str

Generate prompt from the chat in Anthropic’s format.

Returns:

Generated prompt

Return type:

str

property messages

List of messages in the chat.

Returns:

List of messages

Return type:

list

class Message(text: str, role: Role)

Bases: object

Class for a message.

Parameters:

• text (str) – Text of the message

• role (Role) – Role of the message

class PromptFormat(value, names=None, *, module=None, qualname=None, type=None, start=1, boundary=None)

List of prompt formats.

Anthropic = 1

class Role(value, names=None, *, module=None, qualname=None, type=None, start=1, boundary=None)

List of roles.

User = 1

AI = 2