librecoding 0.4.0__tar.gz → 0.5.0__tar.gz

{librecoding-0.4.0 → librecoding-0.5.0}/PKG-INFO

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

{librecoding-0.4.0 → librecoding-0.5.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "librecoding"
-version = "0.4.0"
+version = "0.5.0"
 description = "Free and open-source coding library"
 readme = "README.md"
 requires-python = ">=3.10"

{librecoding-0.4.0 → librecoding-0.5.0}/src/librecoding/__init__.py

@@ -1,4 +1,4 @@
-"""librecoding - вызываемый модуль: спрашивает промпт, шлёт в OpenRouter, пишет .txt."""
+"""librecoding - вызываемый модуль: спрашивает промпт, шлёт в OpenRouter, сохраняет ответ в N.txt."""
 
 from __future__ import annotations
 
@@ -7,13 +7,12 @@ import os
 import sys
 import urllib.error
 import urllib.request
-from datetime import datetime
 from pathlib import Path
 from types import ModuleType
 
 from .core import greet
 
-__version__ = "0.4.0"
+__version__ = "0.5.0"
 
 API_URL = "https://openrouter.ai/api/v1/chat/completions"
 DEFAULT_MODEL = "deepseek/deepseek-v4.1-flash"
@@ -24,6 +23,8 @@ PAUSE_TEXT = "Нажми Enter, чтобы закрыть..."
 _state: dict = {"key": None, "model": DEFAULT_MODEL}
 
 
+# --- настройки ----------------------------------------------------------
+
 def set_api_key(key: str) -> None:
     """Задать ключ OpenRouter на время работы процесса."""
     _state["key"] = key.strip()
@@ -62,6 +63,8 @@ def _api_key() -> str:
     )
 
 
+# --- сеть ---------------------------------------------------------------
+
 def ask(prompt, model=None, temperature=0.7, timeout=180) -> str:
     """Отправить промпт в OpenRouter, вернуть текст ответа."""
     payload = {
@@ -96,6 +99,8 @@ def ask(prompt, model=None, temperature=0.7, timeout=180) -> str:
         raise RuntimeError(f"неожиданный ответ: {str(data)[:400]}") from None
 
 
+# --- ввод/вывод ---------------------------------------------------------
+
 def pause(prompt: str = PAUSE_TEXT) -> None:
     """Ждёт Enter, чтобы терминал не закрылся. Молчит без tty."""
     if not sys.stdin.isatty():
@@ -116,8 +121,15 @@ def _caller_dir() -> Path:
     return Path.cwd()
 
 
-def interactive(prompt=None, filename=None, model=None):
-    """Спросить промпт -> запросить OpenRouter -> сохранить .txt рядом со скриптом."""
+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("нужен интерактивный терминал: запусти файл в консоли")
 
@@ -132,33 +144,31 @@ def interactive(prompt=None, filename=None, model=None):
         print("Пустой промпт — выход.")
         return None
 
-    mdl = model or _state["model"]
-    print(f"[{mdl}] думаю...")
-    answer = ask(prompt, model=mdl)
-
-    print("\n--- ответ ---")
-    print(answer.strip())
-    print("--- конец ---")
-
-    name = filename or f"response_{datetime.now():%Y-%m-%d_%H-%M-%S}.txt"
-    path = _caller_dir() / name
-    path.write_text(
-        f"Модель: {mdl}\n"
-        f"Дата: {datetime.now():%Y-%m-%d %H:%M:%S}\n\n"
-        f"Промпт:\n{prompt}\n\n"
-        f"Ответ:\n{answer.strip()}\n",
-        encoding="utf-8",
-    )
+    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):
         if callable(obj) and not args and not kwargs:
-            return obj
+            return obj                      # @librecoding -> режим декоратора
         try:
-            if isinstance(obj, str):
+            if isinstance(obj, str):        # librecoding("готовый промпт")
                 kwargs["prompt"] = obj
             interactive(*args, **kwargs)
         except Exception as exc:
@@ -172,6 +182,6 @@ class _LibreCodingModule(ModuleType):
 sys.modules[__name__].__class__ = _LibreCodingModule
 
 __all__ = [
-    "greet", "ask", "interactive", "pause",
+    "greet", "ask", "interactive", "pause", "next_filename",
     "set_api_key", "set_model", "get_model",
 ]