ai-ebash 0.1.5a0__tar.gz → 0.1.7__tar.gz

{ai_ebash-0.1.5a0/src/ai_ebash.egg-info → ai_ebash-0.1.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ai-ebash
-Version: 0.1.5a0
+Version: 0.1.7
 Summary: Console utility for integrating artificial intelligence into a Linux terminal.
 Author: Andrey Bochkarev
 Author-email: andrey.bch.1976@gmail.com

{ai_ebash-0.1.5a0 → ai_ebash-0.1.7}/setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = ai-ebash
-version = 0.1.5a
+version = 0.1.7
 author = Andrey Bochkarev
 author_email = andrey.bch.1976@gmail.com
 description = Console utility for integrating artificial intelligence into a Linux terminal.

{ai_ebash-0.1.5a0 → ai_ebash-0.1.7/src/ai_ebash.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ai-ebash
-Version: 0.1.5a0
+Version: 0.1.7
 Summary: Console utility for integrating artificial intelligence into a Linux terminal.
 Author: Andrey Bochkarev
 Author-email: andrey.bch.1976@gmail.com

{ai_ebash-0.1.5a0 → ai_ebash-0.1.7}/src/ai_ebash.egg-info/SOURCES.txt

@@ -13,4 +13,7 @@ src/aiebash/__main__.py
 src/aiebash/api_client.py
 src/aiebash/block_runner.py
 src/aiebash/formatter_text.py
+src/aiebash/llm_factory.py
+src/aiebash/llm_interface.py
+src/aiebash/openai_client.py
 src/aiebash/settings.py

ai_ebash-0.1.7/src/aiebash/__main__.py

@@ -0,0 +1,119 @@
+#!/usr/bin/env python3
+import sys
+import threading
+import time
+from pathlib import Path
+from rich.console import Console
+from rich.markdown import Markdown
+from rich.rule import Rule
+
+# sys.path, как раньше
+sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
+
+from aiebash.llm_factory import create_llm_client
+from aiebash.formatter_text import annotate_bash_blocks
+from aiebash.block_runner import run_code_selection
+from aiebash.settings import settings
+
+# Получаем настройки для конкретного бэкенда, например 'openai_over_proxy'
+BACKEND = "openai_over_proxy"
+
+MODEL = settings.get(BACKEND, "MODEL")
+API_URL = settings.get(BACKEND, "API_URL")
+API_KEY = settings.get(BACKEND, "API_KEY")
+
+# Глобальные настройки
+DEBUG = settings.get_bool("global", "DEBUG")
+CONTEXT = settings.get("global", "CONTEXT")
+
+# Клиент
+llm_client = create_llm_client(
+    backend=BACKEND,
+    model=MODEL,
+    api_url=API_URL,
+    api_key=API_KEY,
+)
+
+# Управление прогрессом
+stop_event = threading.Event()
+def run_progress():
+    console = Console()
+    with console.status("[bold green]Ai печатает...[/bold green]", spinner="dots"):
+        while not stop_event.is_set():
+            time.sleep(0.1)
+
+
+def main():
+    if len(sys.argv) < 2:
+        print("Использование: ai [-run] [-chat] ваш запрос к ИИ без кавычек")
+        sys.exit(0)
+
+    console = Console()
+
+    # Режимы
+    run_mode = "-run" in sys.argv
+    chat_mode = "-chat" in sys.argv
+    args = [a for a in sys.argv[1:] if a not in ("-run", "-chat")]
+
+    prompt = " ".join(args)
+
+    try:
+        if chat_mode:
+            # Диалоговый режим
+            console.print(Rule(" Вход в чатовый режим ", style="cyan"))
+            messages = []
+            if CONTEXT:
+                messages.append({"role": "system", "content": CONTEXT})
+
+            while True:
+                user_input = console.input("[bold green]Вы:[/bold green] ")
+                if user_input.strip().lower() in ("exit", "quit", "выход"):
+                    break
+
+                messages.append({"role": "user", "content": user_input})
+
+                stop_event.clear()
+                progress_thread = threading.Thread(target=run_progress)
+                progress_thread.start()
+
+                answer = llm_client.send_chat(messages)
+
+                stop_event.set()
+                progress_thread.join()
+
+                messages.append({"role": "assistant", "content": answer})
+                console.print(Markdown(answer))
+                console.print(Rule("", style="green"))
+
+        else:
+            # Обычный режим (один вопрос → один ответ)
+            stop_event.clear()
+            progress_thread = threading.Thread(target=run_progress)
+            progress_thread.start()
+
+            answer = llm_client.send_prompt(prompt, system_context=CONTEXT)
+
+            stop_event.set()
+            progress_thread.join()
+
+            if DEBUG:
+                print("=== RAW RESPONSE ===")
+                print(answer)
+                print("=== /RAW RESPONSE ===")
+
+            annotated_answer, code_blocks = annotate_bash_blocks(answer)
+
+            if run_mode and code_blocks:
+                console.print(Markdown(annotated_answer))
+                run_code_selection(console, code_blocks)
+            else:
+                console.print(Markdown(answer))
+
+            console.print(Rule("", style="green"))
+
+    except Exception as e:
+        print("Ошибка:", e)
+
+
+if __name__ == "__main__":
+    main()

ai_ebash-0.1.7/src/aiebash/llm_factory.py

@@ -0,0 +1,15 @@
+# llm_factory.py
+from .llm_interface import LLMClient
+from .openai_client import OpenAIClient
+# сюда в будущем можно добавить HuggingFaceClient, OllamaClient и др.
+
+def create_llm_client(backend: str, model: str, api_url: str, api_key: str = None, timeout: int = 60) -> LLMClient:
+    backend = backend.lower().strip()
+    if backend == "openai_over_proxy":
+        return OpenAIClient(model=model, api_url=api_url, api_key=api_key, timeout=timeout)
+    # elif backend == "huggingface":
+    #     return HuggingFaceClient(...)
+    # elif backend == "ollama":
+    #     return OllamaClient(...)
+    else:
+        raise ValueError(f"Неизвестный backend: {backend}")

ai_ebash-0.1.7/src/aiebash/llm_interface.py

@@ -0,0 +1,6 @@
+# llm_interface.py
+from typing import List, Protocol
+
+class LLMClient(Protocol):
+    def send_chat(self, messages: List[dict]) -> str: ...
+    def send_prompt(self, prompt: str, system_context: str = "") -> str: ...

ai_ebash-0.1.7/src/aiebash/openai_client.py

@@ -0,0 +1,32 @@
+# openai_client.py
+import json, requests
+from typing import List
+from .llm_interface import LLMClient
+
+class OpenAIClient(LLMClient):
+    def __init__(self, model: str, api_url: str, api_key: str = None, timeout: int = 60):
+        self.model = model
+        self.api_url = api_url
+        self.api_key = api_key
+        self.timeout = timeout
+
+    def send_chat(self, messages: List[dict]) -> str:
+        payload = {"model": self.model, "messages": messages}
+        headers = {"Content-Type": "application/json"}
+        if self.api_key:
+            headers["Authorization"] = f"Bearer {self.api_key}"
+
+        resp = requests.post(self.api_url, headers=headers, json=payload, timeout=self.timeout)
+        resp.raise_for_status()
+        data = resp.json()
+
+        if "choices" in data and data["choices"]:
+            return data["choices"][0]["message"]["content"]
+        raise RuntimeError("Unexpected API response format")
+
+    def send_prompt(self, prompt: str, system_context: str = "") -> str:
+        messages = []
+        if system_context:
+            messages.append({"role": "system", "content": system_context})
+        messages.append({"role": "user", "content": prompt})
+        return self.send_chat(messages)

ai_ebash-0.1.7/src/aiebash/settings.py

@@ -0,0 +1,39 @@
+import configparser
+import shutil
+from pathlib import Path
+from platformdirs import user_config_dir
+from typing import Optional  # <--- ДОБАВЛЕНО
+
+# --- Пути к конфигурации ---
+APP_NAME = "ai-ebash"
+# Путь к конфигу пользователя (e.g., %APPDATA%\ai-ebash\config.ini)
+USER_CONFIG_PATH = Path(user_config_dir(APP_NAME)) / "config.ini"
+# Путь к дефолтному конфигу, который лежит рядом с этим файлом
+DEFAULT_CONFIG_PATH = Path(__file__).parent / "default_config.ini"
+
+# --- Логика инициализации ---
+# Если у пользователя нет config.ini, копируем ему дефолтный
+if not USER_CONFIG_PATH.exists():
+    USER_CONFIG_PATH.parent.mkdir(parents=True, exist_ok=True)
+    shutil.copy(DEFAULT_CONFIG_PATH, USER_CONFIG_PATH)
+
+# --- Чтение и предоставление доступа к настройкам ---
+_config = configparser.ConfigParser()
+_config.read(USER_CONFIG_PATH, encoding="utf-8")
+
+
+class Settings:
+    """Простая обертка для доступа к настройкам из config.ini."""
+    def get(self, section: str, key: str, fallback: Optional[str] = None) -> Optional[str]: # <--- ИЗМЕНЕНО
+        """Получить значение из [section] по ключу key."""
+        return _config.get(section, key, fallback=fallback)
+
+    def get_bool(self, section: str, key: str, fallback: bool = False) -> bool:
+        """Получить булево значение."""
+        return _config.getboolean(section, key, fallback=fallback)
+
+# --- Экземпляр для всего проекта ---
+# В других файлах импортируйте его: from aiebash.settings import settings
+settings = Settings()
+
+

ai_ebash-0.1.5a0/src/aiebash/__main__.py

@@ -1,100 +0,0 @@
-#!/usr/bin/env python3
-import sys
-from pathlib import Path
-
-# Убедимся, что parent (src) в sys.path, чтобы можно было import aibash.*
-# Это нужно, если вы запускаете файл напрямую: python src/aibash/__main__.py
-sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
-
-
-from rich.markdown import Markdown
-from rich.console import Console
-from rich.rule import Rule
-import threading
-import time
-
-
-from aiebash.api_client import send_prompt
-from aiebash.formatter_text import annotate_bash_blocks
-from aiebash.block_runner import run_code_selection  # добавлен импорт
-
-from aiebash.settings import settings as user_settings
-
-# нормализуем DEBUG (возможные варианты: True/False, "true"/"False", "1"/"0")
-_raw_debug = user_settings.get("DEBUG")
-if isinstance(_raw_debug, bool):
-    DEBUG = _raw_debug
-else:
-    DEBUG = str(_raw_debug).strip().lower() in ("1", "true", "yes", "on")
-CONTEXT = user_settings.get("CONTEXT")
-MODEL = user_settings.get("MODEL")
-API_URL = user_settings.get("API_URL")
-API_KEY = user_settings.get("API_KEY")
-
-# Флаг для остановки потока (заменён на Event)
-import threading
-stop_event = threading.Event()
-
-def run_progress():
-    console = Console()
-    with console.status("[bold green]Ai печатает...[/bold green]", spinner="dots") as status:
-         while not stop_event.is_set():
-            time.sleep(0.1)
-
-def main():
-    # Если нет аргументов, выводим подсказку по использованию
-    if len(sys.argv) < 2:
-        print("Использование: ai [-run] ваш запрос к ИИ без кавычек")
-        sys.exit(0)
-
-    console = Console()
-
-    # Проверяем ключ -run
-    run_mode = False
-    args = sys.argv[1:]
-    if "-run" in args:
-        run_mode = True
-        args.remove("-run")
-
-    # Собираем текст запроса из оставшихся аргументов
-    prompt = " ".join(args)
-
-    try:
-        # Запуск прогресс-бара в отдельном потоке
-        progress_thread = threading.Thread(target=run_progress)
-        progress_thread.start()
-
-        # Получаем ответ от API через новый интерфейс
-        answer = send_prompt(prompt, MODEL, API_URL, API_KEY, CONTEXT)
-
-        # Сигнализируем потоку прогресса остановиться
-        stop_event.set()
-        progress_thread.join()  # Ждём завершения потока
-
-        # В режиме DEBUG выводим исходную (неформатированную) версию ответа
-        if not DEBUG:
-            print("=== RAW RESPONSE (from send_prompt) ===")
-            print(answer)
-            print("=== /RAW RESPONSE === \n")
-
-        # Размечаем bash-блоки и получаем список кодов
-        annotated_answer, code_blocks = annotate_bash_blocks(answer)
-        
-
-        # Если включён режим выполнения и есть блоки кода — предлагаем выбрать
-        if run_mode and code_blocks:
-            console.print(Markdown(annotated_answer))
-            run_code_selection(console, code_blocks)
-        else:
-            console.print(Markdown(answer))
-        
-        console.print(Rule("", style="green"))
-
-    except Exception as e:
-        # Прочие ошибки (сеть, JSON, и т.д.)
-        print("Ошибка:", e)
-
-
-
-if __name__ == "__main__":
-    main()

ai_ebash-0.1.5a0/src/aiebash/settings.py

@@ -1,65 +0,0 @@
-
-from pathlib import Path
-import configparser
-from platformdirs import user_config_dir
-
-APP_NAME = "ai-bash"
-
-# Дефолтные значения хранятся в словаре
-DEFAULTS = {
-    "API_URL": "https://openai-proxy.andrey-bch-1976.workers.dev/v1/chat/completions", # Прокси для OpenAI API
-    "API_KEY": "",  # Ключ API (если нужен) 
-    "MODEL": "gpt-4o-mini", # Модель по умолчанию
-    "CONTEXT": ( # Контекст по умолчанию
-        "Ты профессиональный системный администратор, помощник по Linux. "
-        "Пользователь - новичок в Linux, помоги ему. "
-        "Система Ubuntu, оболочка $SHELL. "
-        "При ответе учитывай, что пользователь работает в терминале Bash."
-        "Отвечай всегда на русском языке. "
-        "Ты разговариваешь с пользователем в терминале. "
-    ), 
-    "DEBUG": False  # Включить отладочную информацию
-}
-
-# Где хранится INI-файл
-# Ubuntu → ~/.config/ai-bash/config.ini
-# Windows → C:\Users\<user>\AppData\Local\ai-bash\ai-bash\config.ini
-
-class Settings:
-    def __init__(self):
-        # Кроссплатформенная директория конфигов
-        self.config_dir = Path(user_config_dir(APP_NAME))
-        self.config_dir.mkdir(parents=True, exist_ok=True)
-
-        self.config_file = self.config_dir / "config.ini"
-        self.config = configparser.ConfigParser()
-
-        # Если файл есть — читаем, иначе создаём с дефолтами
-        if self.config_file.exists():
-            self.config.read(self.config_file)
-            if "settings" not in self.config:
-                self.config["settings"] = DEFAULTS
-                self._save()
-        else:
-            self.config["settings"] = DEFAULTS
-            self._save()
-            print(f"[INFO] Создан новый конфиг: {self.config_file}")
-
-    def _save(self):
-        with open(self.config_file, "w") as f:
-            self.config.write(f)
-
-    # Получение значений
-    def get(self, key):
-        return self.config["settings"].get(key, DEFAULTS.get(key))
-
-    # Изменение значений
-    def set(self, key, value):
-        self.config["settings"][key] = str(value)
-        self._save()
-
-
-# Глобальный объект настроек
-settings = Settings()
-
-