tg-jarv-dialogue-manager 0.1.4

Jarv AI клиент и диалог-менеджер для Telegram с хранением истории в ClickHouse

tg-jarv-dialogue-manager

Асинхронный клиент Jarv AI и диалог-менеджер с хранением истории в ClickHouse.

Установка

pip install tg-jarv-dialogue-manager

Опционально для Telegram:

pip install tg-jarv-dialogue-manager[telegram]

Быстрый старт

import asyncio
from tg_jarv_dialogue_manager import JarvdialogueManager

async def main():
    dm = JarvdialogueManager(
        bot_name="DiscountBot",
        model="gpt-4o-mini",
        user_id="bot@synergetic",
        jarv_api_key="...",
        jarv_endpoint="https://api.jarv.tech/v1/chat",
        clickhouse_url="http://localhost:8123",
        clickhouse_database="jarv",
        clickhouse_user=None,
        clickhouse_password=None,
        table="dialogue_history",
        system_prompt="Ты — дружелюбный помощник.",
        pairs_limit=10,
        char_soft_limit=8000,
        payload_base={"temperature": 0.2},
    )
    answer = await dm.ask(tg_id=123456789, user_text="Привет!")
    print(answer)
    await dm.close()

asyncio.run(main())

Схема ClickHouse по умолчанию

CREATE TABLE IF NOT EXISTS dialogue_history (
  tg_id UInt64,
  bot_name String,
  seq_in_dialogue UInt32,
  role LowCardinality(String),
  content String,
  created_at DateTime DEFAULT now()
) ENGINE = MergeTree
ORDER BY (tg_id, bot_name, seq_in_dialogue);

Поддержка

Issues и PR приветствуются.