fastchat.conversation

Conversation prompt templates. We kindly request that you import fastchat instead of copying this file if you want to use it. You can contribute back the changes you want to make.

Module Contents

Classes

SeparatorStyle	Separator styles.
Conversation	A class that manages prompt templates and keeps all conversation history.

Functions

register_conv_template(template[, override])	Register a new conversation template.
get_conv_template(→ Conversation)	Get a conversation template.

class fastchat.conversation.SeparatorStyle

Separator styles.

class fastchat.conversation.Conversation

A class that manages prompt templates and keeps all conversation history.

get_prompt() → str

Get the prompt for generation.

set_system_message(system_message: str)

Set the system message.

append_message(role: str, message: str)

Append a new message.

update_last_message(message: str)

Update the last output. The last message is typically set to be None when constructing the prompt, so we need to update it in-place after getting the response from a model.

to_gradio_chatbot()

Convert the conversation to gradio chatbot format.

to_openai_api_messages()

Convert the conversation to OpenAI chat completion format.

fastchat.conversation.register_conv_template(template: Conversation, override: bool = False)

Register a new conversation template.

fastchat.conversation.get_conv_template(name: str) → Conversation

Get a conversation template.