librecoding 0.3.0__py3-none-any.whl → 0.5.0__py3-none-any.whl

librecoding/__init__.py

@@ -1,24 +1,108 @@
-"""librecoding - модуль, который можно вызвать как функцию."""
+"""librecoding - вызываемый модуль: спрашивает промпт, шлёт в OpenRouter, сохраняет ответ в N.txt."""
 
+from __future__ import annotations
+
+import json
+import os
 import sys
+import urllib.error
+import urllib.request
+from pathlib import Path
 from types import ModuleType
 
 from .core import greet
 
-__version__ = "0.3.0"
+__version__ = "0.5.0"
+
+API_URL = "https://openrouter.ai/api/v1/chat/completions"
+DEFAULT_MODEL = "deepseek/deepseek-v4.1-flash"
+KEY_FILE = Path.home() / ".config" / "librecoding" / "api_key"
+PROMPT = "Введи промпт: "
+PAUSE_TEXT = "Нажми Enter, чтобы закрыть..."
+
+_state: dict = {"key": None, "model": DEFAULT_MODEL}
+
+
+# --- настройки ----------------------------------------------------------
+
+def set_api_key(key: str) -> None:
+    """Задать ключ OpenRouter на время работы процесса."""
+    _state["key"] = key.strip()
+
+
+def set_model(model: str) -> None:
+    """Сменить модель, например 'deepseek/deepseek-chat'."""
+    _state["model"] = model
+
+
+def get_model() -> str:
+    return _state["model"]
+
+
+def _api_key() -> str:
+    """Ключ: set_api_key() -> OPENROUTER_API_KEY -> ~/.config/librecoding/api_key."""
+    if _state["key"]:
+        return _state["key"]
+
+    env = os.environ.get("OPENROUTER_API_KEY", "").strip()
+    if env:
+        return env
+
+    if KEY_FILE.is_file():
+        for line in KEY_FILE.read_text(encoding="utf-8").splitlines():
+            line = line.strip()
+            if line and not line.startswith("#"):
+                return line
+
+    raise RuntimeError(
+        "API-ключ OpenRouter не найден. Задай его так:\n"
+        f"  echo 'sk-or-v1-...' > {KEY_FILE}\n"
+        f"  chmod 600 {KEY_FILE}\n"
+        "  либо: export OPENROUTER_API_KEY=sk-or-v1-...\n"
+        "  либо: librecoding.set_api_key('sk-or-v1-...')"
+    )
+
+
+# --- сеть ---------------------------------------------------------------
 
-BANNER = "Привет, мир! (из librecoding)"
-PROMPT = "Нажми Enter, чтобы закрыть..."
+def ask(prompt, model=None, temperature=0.7, timeout=180) -> str:
+    """Отправить промпт в OpenRouter, вернуть текст ответа."""
+    payload = {
+        "model": model or _state["model"],
+        "messages": [{"role": "user", "content": prompt}],
+        "temperature": temperature,
+    }
+    req = urllib.request.Request(
+        API_URL,
+        data=json.dumps(payload).encode("utf-8"),
+        headers={
+            "Authorization": f"Bearer {_api_key()}",
+            "Content-Type": "application/json",
+            "HTTP-Referer": "https://pypi.org/project/librecoding/",
+            "X-Title": "librecoding",
+        },
+        method="POST",
+    )
 
+    try:
+        with urllib.request.urlopen(req, timeout=timeout) as resp:
+            data = json.loads(resp.read().decode("utf-8"))
+    except urllib.error.HTTPError as exc:
+        body = exc.read().decode("utf-8", "replace")[:400]
+        raise RuntimeError(f"OpenRouter HTTP {exc.code}: {body}") from None
+    except urllib.error.URLError as exc:
+        raise RuntimeError(f"нет сети / таймаут: {exc.reason}") from None
+
+    try:
+        return data["choices"][0]["message"]["content"] or ""
+    except (KeyError, IndexError, TypeError):
+        raise RuntimeError(f"неожиданный ответ: {str(data)[:400]}") from None
 
-def hello() -> str:
-    """Явный способ попросить приветствие."""
-    print(BANNER)
-    return BANNER
 
+# --- ввод/вывод ---------------------------------------------------------
 
-def pause(prompt: str = PROMPT) -> None:
-    """Ждёт Enter, чтобы терминал не закрылся сразу."""
+def pause(prompt: str = PAUSE_TEXT) -> None:
+    """Ждёт Enter, чтобы терминал не закрылся. Молчит без tty."""
     if not sys.stdin.isatty():
         return
     try:
@@ -27,17 +111,77 @@ def pause(prompt: str = PROMPT) -> None:
         print()
 
 
-class _LibreCodingModule(ModuleType):
-    """Модуль, который ведёт себя как функция: librecoding()."""
+def _caller_dir() -> Path:
+    """Папка запущенного скрипта (test.py), а не библиотеки."""
+    path = getattr(sys.modules.get("__main__"), "__file__", None)
+    if path:
+        return Path(path).resolve().parent
+    if sys.argv and sys.argv[0]:
+        return Path(sys.argv[0]).resolve().parent
+    return Path.cwd()
+
+
+def next_filename(directory=None) -> Path:
+    """Следующий номер: 1.txt, 2.txt, 3.txt ... (максимальный + 1)."""
+    base = Path(directory) if directory else _caller_dir()
+    nums = [int(p.stem) for p in base.glob("*.txt") if p.stem.isdigit()]
+    return base / f"{max(nums, default=0) + 1}.txt"
+
+
+def interactive(prompt=None, filename=None, model=None, directory=None):
+    """Спросить промпт -> OpenRouter -> сохранить чистый ответ в N.txt рядом со скриптом."""
+    if not sys.stdin.isatty():
+        raise RuntimeError("нужен интерактивный терминал: запусти файл в консоли")
 
+    if prompt is None:
+        try:
+            prompt = input(PROMPT).strip()
+        except (EOFError, KeyboardInterrupt):
+            print()
+            return None
+
+    if not prompt:
+        print("Пустой промпт — выход.")
+        return None
+
+    print("думаю...")
+    answer = ask(prompt, model=model or _state["model"]).strip()
+    print(answer)
+
+    if filename:
+        path = Path(filename)
+        if not path.is_absolute():
+            path = (Path(directory) if directory else _caller_dir()) / path
+    else:
+        path = next_filename(directory)
+
+    path.parent.mkdir(parents=True, exist_ok=True)
+    path.write_text(answer + "\n", encoding="utf-8")
+    print(f"\nСохранено: {path}")
+    return path
+
+
+# --- магия: модуль, который можно вызвать -------------------------------
+
+class _LibreCodingModule(ModuleType):
     def __call__(self, obj=None, *args, **kwargs):
-        print(BANNER)
         if callable(obj) and not args and not kwargs:
-            return obj
+            return obj                      # @librecoding -> режим декоратора
+        try:
+            if isinstance(obj, str):        # librecoding("готовый промпт")
+                kwargs["prompt"] = obj
+            interactive(*args, **kwargs)
+        except Exception as exc:
+            print(f"Ошибка: {exc}", file=sys.stderr)
+            if os.environ.get("LIBRECODING_DEBUG"):
+                raise
         pause()
-        return BANNER
+        return None
 
 
 sys.modules[__name__].__class__ = _LibreCodingModule
 
-__all__ = ["greet", "hello", "pause", "BANNER"]
+__all__ = [
+    "greet", "ask", "interactive", "pause", "next_filename",
+    "set_api_key", "set_model", "get_model",
+]

{librecoding-0.3.0.dist-info → librecoding-0.5.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: librecoding
-Version: 0.3.0
+Version: 0.5.0
 Summary: Free and open-source coding library
 Author: ar
 License-Expression: MIT

librecoding-0.5.0.dist-info/RECORD

@@ -0,0 +1,5 @@
+librecoding/__init__.py,sha256=7DtzGQc78PhkoQyW9hjTgKDg70vfA8-9wnC98ap_r5Y,6433
+librecoding/core.py,sha256=UO5R_K8a-XwaJAhwl3cybnCSbk2tkpcnqrT0d-PIdTA,180
+librecoding-0.5.0.dist-info/METADATA,sha256=mNj4i-YegKoN65-SuObYXWzIG2AcDLZwvhC9WOaqC_Y,331
+librecoding-0.5.0.dist-info/WHEEL,sha256=W3fkpkm7-wf9vBI5Z-7s0eWkeM-spu78I8Neb98DeEg,87
+librecoding-0.5.0.dist-info/RECORD,,

librecoding-0.3.0.dist-info/RECORD

@@ -1,5 +0,0 @@
-librecoding/__init__.py,sha256=AcfE0XB4s3mv7vZRq1ndw-i3JrV1ZYQycd5gZZ_dwLE,1129
-librecoding/core.py,sha256=UO5R_K8a-XwaJAhwl3cybnCSbk2tkpcnqrT0d-PIdTA,180
-librecoding-0.3.0.dist-info/METADATA,sha256=hD3JW0D4NfZ7IMIYzBsmIJLm3CzUYAcZyyjQVHPf6t4,331
-librecoding-0.3.0.dist-info/WHEEL,sha256=W3fkpkm7-wf9vBI5Z-7s0eWkeM-spu78I8Neb98DeEg,87
-librecoding-0.3.0.dist-info/RECORD,,