src/cdp/__main__.py - develop - pythonpractice/consultant-data-provider

src/cdp/__main__.py

75 строк3 KB

Кузьма W600DEV Даждъбогин

TSKCNSLTNTDTP-1. base preparing. оживил проект. прописал базовые команды для мок-запроса и для реального.

15 май 2026, 16:09

15 май 2026, 16:09cab050c

"""Точка входа в приложение."""

import sys
from cdp.config import load_config
from cdp.clients.mock_client import MockLLMClient
from cdp.clients.gigachat_client import GigaChatClient


def main():
    print("=" * 50)
    print("Consultant-Data-Provider. Диалог с гарантированным результатом достижения поставленной цели.")
    print("=" * 50)
    print(f"Python: {sys.version}")
    print("Статус: вижу ок✅")
    print("=" * 50)

    # Команды, которые будем поддерживать
    if len(sys.argv) > 1:
        command = sys.argv[1]
        print(f"\nПолучена команда: {command}")

        if command == "hello":
            print("Hello from CDP!")

        elif command == "version":
            print(f"cdp version 0.1.0")

        elif command == "config":
            config = load_config()
            print(f"Client ID: {config.client_id}")
            print(f"Model: {config.model}")
            print(f"Timeout: {config.timeout} сек")

        elif command == "mock":
            # Создаём мок-клиент
            client = MockLLMClient(delay_seconds=0.5)

            # Если есть второй аргумент — используем как промпт
            if len(sys.argv) > 2:
                prompt = " ".join(sys.argv[2:])
            else:
                prompt = "Расскажи шутку про Python"

            print(f"\n📤 Отправляю промпт: {prompt}")
            print("Ожидаю ответ от LLM...")

            response = client.generate(prompt)

            print(f"\n Ответ:\n{response}")
            print("\n Мок-клиент отработал")
        elif command == "chat":
            if len(sys.argv) > 2:
                prompt = " ".join(sys.argv[2:])
            else:
                prompt = "Привет! Как дела?"

            print(f"\n📤 Отправляю в GigaChat: {prompt}")
            print("⏳ Ожидание ответа...\n")

            with GigaChatClient() as client:
                response = client.generate(prompt)
                print(f"🤖 GigaChat ответил:\n{response}\n")

            print("✅ Запрос завершён")
        else:
            print(f"⚠️ Неизвестная команда: {command}")
            print("\nДоступные команды: hello, version, config, mock")
    else:
        print("\nИспользование: python -m cdp [команда]")
        print("  Доступные команды: hello, version, config, mock, chat")
        print("  Пример: python -m cdp mock 'Привет, как дела?'")


if __name__ == "__main__":
    main()