simple-chat-structure 0.1.2

Библиотека для управления структурой чатов и сообщений

Simple Chat Structure

Библиотека для управления структурой чатов и сообщений в диалоговых системах. Простой и удобный инструмент для работы с историей сообщений в чатах, ботах и других приложениях с диалоговым интерфейсом.

Возможности

• 📝 Управление сообщениями с различными ролями (пользователь, ассистент, система, функция)
• 📚 Работа с историей сообщений, ограничение по количеству
• 👥 Менеджер историй для нескольких пользователей
• 💾 Сериализация и десериализация в JSON
• ⏰ Автоматическая отметка времени для каждого сообщения
• ✨ Простой и понятный API

Установка

pip install simple-chat-structure

Требования

• Python 3.8 или выше

Использование

Основные примеры

Создание сообщений

from simple_chat_structure import Message, MessageRole

# Создание сообщения с ролью пользователя
user_message = Message(MessageRole.USER, "Привет, как дела?")

# Создание сообщения с ролью ассистента
assistant_message = Message(MessageRole.ASSISTANT, "Здравствуйте! У меня всё отлично, чем могу помочь?")

# Создание сообщения с ролью системы
system_message = Message(MessageRole.SYSTEM, "Вы общаетесь с AI-ассистентом.")

# Создание сообщения с указанием роли в виде строки
message = Message("user", "Какая сегодня погода?")

# Получение информации о сообщении
print(user_message.role)  # MessageRole.USER
print(user_message.content)  # "Привет, как дела?"
print(user_message.timestamp)  # datetime.datetime объект

Работа с историей сообщений

from simple_chat_structure import MessageHistory, MessageRole

# Создание истории сообщений
history = MessageHistory()

# Добавление сообщений
history.add_message(MessageRole.USER, "Привет!")
history.add_message(MessageRole.ASSISTANT, "Здравствуйте! Чем могу помочь?")
history.add_message(MessageRole.USER, "Какая сегодня погода?")

# Получение последних сообщений
last_messages = history.get_last_n_messages(2)
for msg in last_messages:
    print(f"{msg.role.value}: {msg.content}")

# Преобразование в формат для API
api_format = history.to_api_format()
print(api_format)
# [{"role": "user", "content": "Привет!"}, 
#  {"role": "assistant", "content": "Здравствуйте! Чем могу помочь?"}, 
#  {"role": "user", "content": "Какая сегодня погода?"}]

# Сохранение и загрузка из JSON
json_data = history.to_json()
new_history = MessageHistory().from_json(json_data)

# Очистка истории
history.clear()

# Создание истории с ограничением количества сообщений
limited_history = MessageHistory(max_messages=10)

Менеджер историй для нескольких пользователей

from simple_chat_structure import MessageHistoryManager, MessageRole

# Создание менеджера историй
manager = MessageHistoryManager(default_max_messages=50)

# Получение истории для конкретного пользователя
user_id = 123
history = manager[user_id]

# Добавление сообщений пользователю
history.add_message(MessageRole.USER, "Привет!")
history.add_message(MessageRole.ASSISTANT, "Здравствуйте!")

# Проверка наличия пользователя
if user_id in manager:
    print(f"У пользователя {user_id} есть история сообщений")

# Получение списка всех пользователей
users = manager.get_user_ids()
print(users)  # [123]

# Сохранение всех историй в JSON
json_data = manager.to_json()

# Загрузка историй из JSON
new_manager = MessageHistoryManager().from_json(json_data)

# Удаление истории пользователя
manager.remove_user(user_id)

# Очистка всех историй
manager.clear_all()

Документация API

Message

Класс для представления сообщений в чате.

Атрибуты

• role - Роль отправителя сообщения (MessageRole)
• content - Содержимое сообщения (str)
• timestamp - Время создания сообщения (datetime)

Методы

• to_dict() - Преобразование в словарь для API
• to_full_dict() - Преобразование в полный словарь с временем
• to_json() - Преобразование в JSON-строку

MessageRole

Перечисление для обозначения ролей сообщений.

• USER - Сообщение пользователя
• ASSISTANT - Сообщение ассистента
• SYSTEM - Системное сообщение
• FUNCTION - Функциональное сообщение

MessageHistory

Класс для управления историей сообщений.

Методы

• add_message(role, content) - Добавление сообщения в историю
• clear() - Очистка истории
• pop() - Удаление и возврат последнего сообщения
• get_last_n_messages(n) - Получение последних n сообщений
• to_api_format() - Преобразование в формат для API
• to_json() - Преобразование в JSON-строку
• from_json(json_data) - Загрузка из JSON-строки

MessageHistoryManager

Класс для управления историями сообщений нескольких пользователей.

Методы

• get_user_ids() - Получение списка ID пользователей
• remove_user(user_id) - Удаление истории пользователя
• clear_all() - Очистка всех историй
• to_json() - Преобразование в JSON-строку
• from_json(json_data) - Загрузка из JSON-строки