ai-ebash 0.1.5__py3-none-any.whl → 0.1.6__py3-none-any.whl

{ai_ebash-0.1.5.dist-info → ai_ebash-0.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ai-ebash
-Version: 0.1.5
+Version: 0.1.6
 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.6.dist-info/RECORD

@@ -0,0 +1,14 @@
+aiebash/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+aiebash/__main__.py,sha256=MNnFYCUxiS-z47d75k7o5c5_rzWuxwCTGeWdksluolc,3802
+aiebash/api_client.py,sha256=-hcgY3CP0lsV9Jct_-HQ6WpDosDChwalJjy4aFQ4CIM,2004
+aiebash/block_runner.py,sha256=JNcdBklJmrlO7ZX7xXiO0fUGqrcNJiRqdwM-yr3OP3k,1599
+aiebash/formatter_text.py,sha256=NRz_pCpakZ9k8477_4Gx1scDXrEbbOaOi-U7WRp6ZWo,847
+aiebash/llm_factory.py,sha256=SDG4mSStKtairW2FaExbX49vDXITt5LBvHIKJ72nJfk,718
+aiebash/llm_interface.py,sha256=LufTw-QvXkcQJMRON8RElmalKntk0YRUDHBEcjCDfaM,222
+aiebash/openai_client.py,sha256=94v86FCrAd1v9SgDRtrCu87iSCnq_o2Zz0iXQdnlWbQ,1264
+aiebash/settings.py,sha256=lD8qCEczEe5Fc4gFcqHqnEEmDko0lD6QEldOE68bOBE,1868
+ai_ebash-0.1.6.dist-info/METADATA,sha256=gwM4t6wafhhH9bB3hcqaVYk4SUJIUvl8tdBhEfrWTlk,2384
+ai_ebash-0.1.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ai_ebash-0.1.6.dist-info/entry_points.txt,sha256=bstvFHHX76KHqYpjM8FPoRVvNbQprST74CI30Mr5TIM,45
+ai_ebash-0.1.6.dist-info/top_level.txt,sha256=JJ-bWdhWlFCfCQGLhmqzN4_bNO0pfVPRsdbx8o9A9AA,8
+ai_ebash-0.1.6.dist-info/RECORD,,

aiebash/__main__.py

@@ -1,100 +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
 
-# Убедимся, что parent (src) в sys.path, чтобы можно было import aibash.*
-# Это нужно, если вы запускаете файл напрямую: python src/aibash/__main__.py
+# 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
 
-from rich.markdown import Markdown
-from rich.console import Console
-from rich.rule import Rule
-import threading
-import time
+# Получаем настройки для конкретного бэкенда, например '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")
 
-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()
+# Глобальные настройки
+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") as status:
-         while not stop_event.is_set():
+    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] ваш запрос к ИИ без кавычек")
+        print("Использование: ai [-run] [-chat] ваш запрос к ИИ без кавычек")
         sys.exit(0)
 
     console = Console()
 
-    # Проверяем ключ -run
-    run_mode = False
-    args = sys.argv[1:]
-    if "-run" in args:
-        run_mode = True
-        args.remove("-run")
+    # Режимы
+    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:
-        # Запуск прогресс-бара в отдельном потоке
-        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)
+        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:
-            console.print(Markdown(answer))
-        
-        console.print(Rule("", style="green"))
+            # Обычный режим (один вопрос → один ответ)
+            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:
-        # Прочие ошибки (сеть, JSON, и т.д.)
         print("Ошибка:", e)
 
 
-
 if __name__ == "__main__":
     main()

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}")

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: ...

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)

aiebash/settings.py

@@ -1,65 +1,39 @@
-
-from pathlib import Path
 import configparser
+import shutil
+from pathlib import Path
 from platformdirs import user_config_dir
+from typing import Optional  # <--- ДОБАВЛЕНО
 
-APP_NAME = "ai-bash"
+# --- Пути к конфигурации ---
+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"
 
-# Дефолтные значения хранятся в словаре
-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  # Включить отладочную информацию
-}
+# --- Логика инициализации ---
+# Если у пользователя нет 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)
 
-# Где хранится 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)
+# --- Чтение и предоставление доступа к настройкам ---
+_config = configparser.ConfigParser()
+_config.read(USER_CONFIG_PATH, encoding="utf-8")
 
-        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()
+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.5.dist-info/RECORD

@@ -1,11 +0,0 @@
-aiebash/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-aiebash/__main__.py,sha256=x7e3HI9_0UUyhohmEvTi8Yf2mTQ9YgyicJ6xSPgOHcE,3725
-aiebash/api_client.py,sha256=-hcgY3CP0lsV9Jct_-HQ6WpDosDChwalJjy4aFQ4CIM,2004
-aiebash/block_runner.py,sha256=JNcdBklJmrlO7ZX7xXiO0fUGqrcNJiRqdwM-yr3OP3k,1599
-aiebash/formatter_text.py,sha256=NRz_pCpakZ9k8477_4Gx1scDXrEbbOaOi-U7WRp6ZWo,847
-aiebash/settings.py,sha256=Ml9_ixXl-DVpFYfCygMAhGSH-1luTxN22YYz1mHlsfw,2671
-ai_ebash-0.1.5.dist-info/METADATA,sha256=ITMnYGPQdqAVbuVZATgQ9d-QCcGO7FSHUeiiG9kFW9s,2384
-ai_ebash-0.1.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ai_ebash-0.1.5.dist-info/entry_points.txt,sha256=bstvFHHX76KHqYpjM8FPoRVvNbQprST74CI30Mr5TIM,45
-ai_ebash-0.1.5.dist-info/top_level.txt,sha256=JJ-bWdhWlFCfCQGLhmqzN4_bNO0pfVPRsdbx8o9A9AA,8
-ai_ebash-0.1.5.dist-info/RECORD,,