librecoding 0.6.0__tar.gz → 1.0.0__tar.gz

{librecoding-0.6.0 → librecoding-1.0.0}/PKG-INFO

@@ -1,14 +1,13 @@
 Metadata-Version: 2.5
 Name: librecoding
-Version: 0.6.0
-Summary: Agent library: prompt -> tools -> results in files2/
+Version: 1.0.0
+Summary: Minimal prompt/response library: prompt in console, answer saved to N.txt
 Author: ar
 License-Expression: MIT
-Keywords: agent,ai,librecoding
+Keywords: ai,librecoding,openrouter
 Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python :: 3
 Requires-Python: >=3.10
-Requires-Dist: openpyxl>=3.1
 Description-Content-Type: text/markdown
 
 # librecoding

{librecoding-0.6.0 → librecoding-1.0.0}/pyproject.toml

@@ -4,18 +4,18 @@ build-backend = "hatchling.build"
 
 [project]
 name = "librecoding"
-version = "0.6.0"
-description = "Agent library: prompt -> tools -> results in files2/"
+version = "1.0.0"
+description = "Minimal prompt/response library: prompt in console, answer saved to N.txt"
 readme = "README.md"
 requires-python = ">=3.10"
 license = "MIT"
 authors = [{ name = "ar" }]
-keywords = ["librecoding", "ai", "agent"]
+keywords = ["librecoding", "ai", "openrouter"]
 classifiers = [
     "Programming Language :: Python :: 3",
     "Operating System :: OS Independent",
 ]
-dependencies = ["openpyxl>=3.1"]
+dependencies = []
 
 [tool.hatch.build.targets.wheel]
 packages = ["src/librecoding"]

librecoding-1.0.0/src/librecoding/__init__.py

@@ -0,0 +1,187 @@
+"""librecoding 1.0.0 - только запрос и ответ.
+
+Промпт -> ответ в консоль и в 1.txt, 2.txt, 3.txt рядом с запущенным скриптом.
+Никаких папок files/, files2/ и никаких других создаваемых файлов.
+"""
+
+from __future__ import annotations
+
+import json
+import os
+import sys
+import urllib.error
+import urllib.request
+from pathlib import Path
+from types import ModuleType
+
+__version__ = "1.0.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 = "Введи промпт: "
+EXIT_WORDS = {"exit", "quit", "q", "выход", "стоп"}
+SYSTEM_PROMPT = "Ты полезный ассистент. Отвечай по-русски, кратко и по делу."
+
+_state = {"key": None, "model": DEFAULT_MODEL}
+
+
+def set_api_key(value):
+    """Задать ключ OpenRouter на время работы процесса."""
+    _state["key"] = str(value).strip()
+
+
+def set_model(value):
+    """Сменить модель, например deepseek/deepseek-chat."""
+    _state["model"] = str(value)
+
+
+def get_model():
+    return _state["model"]
+
+
+def _api_key():
+    """Ключ: 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("нет ключа OpenRouter, положи его в " + str(KEY_FILE))
+
+
+def caller_dir():
+    """Папка запущенного скрипта, а не библиотеки."""
+    path = getattr(sys.modules.get("__main__"), "__file__", None)
+    if path:
+        return Path(path).resolve().parent
+    arg = sys.argv[0] if sys.argv else ""
+    if arg and arg != "-c" and not arg.startswith("-"):
+        p = Path(arg).resolve()
+        if p.parent.is_dir():
+            return p.parent
+    return Path.cwd()
+
+
+def next_filename(directory=None):
+    """Следующее имя 1.txt, 2.txt, 3.txt ..."""
+    base = Path(directory) if directory else caller_dir()
+    nums = [int(p.stem) for p in base.glob("*.txt") if p.stem.isdigit()]
+    return base / (str(max(nums, default=0) + 1) + ".txt")
+
+
+def ask(prompt, model=None, temperature=0.7, timeout=300):
+    """Отправить промпт и вернуть текст. Ничего не печатает и не сохраняет."""
+    body = {
+        "model": model or _state["model"],
+        "messages": [
+            {"role": "system", "content": SYSTEM_PROMPT},
+            {"role": "user", "content": str(prompt)},
+        ],
+        "temperature": temperature,
+    }
+    req = urllib.request.Request(
+        API_URL,
+        data=json.dumps(body).encode("utf-8"),
+        headers={
+            "Authorization": "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:
+        detail = exc.read().decode("utf-8", "replace")[:300]
+        raise RuntimeError("OpenRouter HTTP " + str(exc.code) + ": " + detail) from None
+    except urllib.error.URLError as exc:
+        raise RuntimeError("нет сети или таймаут: " + str(exc.reason)) from None
+    try:
+        return data["choices"][0]["message"]["content"] or ""
+    except (KeyError, IndexError, TypeError):
+        raise RuntimeError("неожиданный ответ: " + str(data)[:300]) from None
+
+
+def save(text, filename=None):
+    """Сохранить текст в N.txt (или в filename) рядом со скриптом."""
+    base = caller_dir()
+    path = Path(filename) if filename else next_filename(base)
+    if not path.is_absolute():
+        path = base / path
+    path.write_text(str(text).rstrip() + "\n", encoding="utf-8")
+    return path
+
+
+def chat(prompt, model=None, save_answer=True, filename=None):
+    """Один промпт: напечатать ответ и сохранить его в N.txt."""
+    text = str(prompt).strip()
+    if not text:
+        return None
+    print("думаю...")
+    answer = (ask(text, model=model) or "").strip()
+    print(answer or "(пустой ответ)")
+    if save_answer and answer:
+        path = save(answer, filename)
+        print("сохранено: " + path.name)
+        return path
+    return None
+
+
+def interactive(prompt=None, model=None, save_answer=True, filename=None):
+    """Без аргумента цикл промпт-ответ. С аргументом один ответ.
+
+    Выход: пустой Enter, выход, exit, q, Ctrl+C.
+    """
+    if prompt is not None:
+        return chat(prompt, model=model, save_answer=save_answer, filename=filename)
+
+    last = None
+    while True:
+        try:
+            line = input(PROMPT)
+        except (EOFError, KeyboardInterrupt):
+            print()
+            break
+        line = line.strip()
+        if not line or line.lower() in EXIT_WORDS:
+            break
+        print()
+        try:
+            last = chat(line, model=model, save_answer=save_answer, filename=filename)
+        except Exception as exc:
+            print("Ошибка: " + str(exc), file=sys.stderr)
+        print()
+    return last
+
+
+class _LibreCodingModule(ModuleType):
+    """Модуль, который можно вызвать как функцию: librecoding()."""
+
+    def __call__(self, obj=None, *args, **kwargs):
+        if callable(obj) and not args and not kwargs:
+            return obj
+        try:
+            if isinstance(obj, str):
+                kwargs["prompt"] = obj
+            return interactive(*args, **kwargs)
+        except Exception as exc:
+            print("Ошибка: " + str(exc), file=sys.stderr)
+            if os.environ.get("LIBRECODING_DEBUG"):
+                raise
+            return None
+
+
+sys.modules[__name__].__class__ = _LibreCodingModule
+
+__all__ = [
+    "ask", "chat", "save", "interactive", "caller_dir", "next_filename",
+    "set_api_key", "set_model", "get_model",
+]

librecoding-0.6.0/src/librecoding/__init__.py

@@ -1,418 +0,0 @@
-"""librecoding - агент: промпт -> инструменты -> результат в files2/."""
-
-from __future__ import annotations
-
-import json
-import os
-import subprocess
-import sys
-import urllib.error
-import urllib.request
-from pathlib import Path
-from types import ModuleType
-
-from .core import greet
-
-__version__ = "0.6.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"
-IN_DIR = "files"
-OUT_DIR = "files2"
-PROMPT = "Введи промпт: "
-PAUSE_TEXT = "Нажми Enter, чтобы закрыть..."
-
-MAX_STEPS = 12
-TOOL_LIMIT = 12000
-
-SYSTEM_PROMPT = f"""Ты — автономный агент библиотеки librecoding.
-Рабочая папка — та, где лежит запущенный скрипт.
-
-Структура:
-  {IN_DIR}/   — входные файлы пользователя, только чтение
-  {OUT_DIR}/  — сюда складывай всё, что создаёшь
-
-Инструменты:
-  list_files()              — что есть в {IN_DIR}/
-  read_file(name)           — текст или предпросмотр xlsx/csv
-  write_file(name, content) — записать текстовый файл в {OUT_DIR}/
-  run_python(code)          — выполнить Python-код, cwd = рабочая папка,
-                              openpyxl уже установлен
-
-Правила:
-1. Сначала list_files(), если точные имена файлов неизвестны.
-2. Для xlsx/xls/csv НИКОГДА не выдумывай содержимое — только run_python.
-3. Результат сохраняй в {OUT_DIR}/ с тем же именем, что исходник, если не сказано иначе.
-4. Файлы в {IN_DIR}/ не перезаписывай и не удаляй.
-5. Не задавай уточняющих вопросов — делай разумное предположение и выполняй.
-6. Финальный ответ — коротко по-русски: что сделал, какие файлы создал. Без кода."""
-
-TOOLS = [
-    {
-        "type": "function",
-        "function": {
-            "name": "list_files",
-            "description": f"Показать файлы во входной папке {IN_DIR}/.",
-            "parameters": {"type": "object", "properties": {}, "required": []},
-        },
-    },
-    {
-        "type": "function",
-        "function": {
-            "name": "read_file",
-            "description": f"Прочитать файл из {IN_DIR}/ (текст или предпросмотр таблицы).",
-            "parameters": {
-                "type": "object",
-                "properties": {"name": {"type": "string", "description": "имя файла в files/"}},
-                "required": ["name"],
-            },
-        },
-    },
-    {
-        "type": "function",
-        "function": {
-            "name": "write_file",
-            "description": f"Записать текстовый файл в {OUT_DIR}/.",
-            "parameters": {
-                "type": "object",
-                "properties": {
-                    "name": {"type": "string"},
-                    "content": {"type": "string"},
-                },
-                "required": ["name", "content"],
-            },
-        },
-    },
-    {
-        "type": "function",
-        "function": {
-            "name": "run_python",
-            "description": (
-                "Выполнить Python-код в рабочей папке (cwd). Относительные пути: "
-                f"'{IN_DIR}/...' и '{OUT_DIR}/...'. openpyxl установлен. "
-                f"Результаты сохраняй в {OUT_DIR}/. Возвращает stdout/stderr."
-            ),
-            "parameters": {
-                "type": "object",
-                "properties": {"code": {"type": "string"}},
-                "required": ["code"],
-            },
-        },
-    },
-]
-
-_state = {"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-...')"
-    )
-
-
-# --- пути ---------------------------------------------------------------
-
-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 ensure_dirs() -> tuple[Path, Path]:
-    """Создать files/ и files2/ рядом со скриптом. Вернуть (in_dir, out_dir)."""
-    base = _caller_dir()
-    in_dir, out_dir = base / IN_DIR, base / OUT_DIR
-    in_dir.mkdir(parents=True, exist_ok=True)
-    out_dir.mkdir(parents=True, exist_ok=True)
-    return in_dir, out_dir
-
-
-def _safe(base: Path, name: str) -> Path:
-    """Не дать выйти за пределы папки через ../ или абсолютный путь."""
-    base = base.resolve()
-    p = (base / name).resolve()
-    if base != p and base not in p.parents:
-        raise ValueError(f"путь вне папки: {name}")
-    return p
-
-
-def next_filename(directory=None) -> Path:
-    """Следующее имя 1.txt, 2.txt, 3.txt ... (максимум + 1)."""
-    base = Path(directory) if directory else ensure_dirs()[1]
-    nums = [int(p.stem) for p in base.glob("*.txt") if p.stem.isdigit()]
-    return base / f"{max(nums, default=0) + 1}.txt"
-
-
-# --- сеть ---------------------------------------------------------------
-
-def _api(payload: dict, timeout: int = 300) -> dict:
-    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:
-            return 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
-
-
-def ask(prompt, model=None, temperature=0.7, timeout=300) -> str:
-    """Один запрос без инструментов. Возвращает текст ответа."""
-    data = _api(
-        {
-            "model": model or _state["model"],
-            "messages": [{"role": "user", "content": prompt}],
-            "temperature": temperature,
-        },
-        timeout=timeout,
-    )
-    try:
-        return data["choices"][0]["message"]["content"] or ""
-    except (KeyError, IndexError, TypeError):
-        raise RuntimeError(f"неожиданный ответ: {str(data)[:400]}") from None
-
-
-# --- инструменты --------------------------------------------------------
-
-def _tool_list() -> str:
-    in_dir, _ = ensure_dirs()
-    files = sorted(p for p in in_dir.iterdir() if p.is_file())
-    if not files:
-        return f"{IN_DIR}/ пуста"
-    return f"файлы в {IN_DIR}/:\n" + "\n".join(
-        f"  {p.name} ({p.stat().st_size} байт)" for p in files
-    )
-
-
-def _tool_read(name: str) -> str:
-    in_dir, _ = ensure_dirs()
-    p = _safe(in_dir, name)
-    if not p.is_file():
-        return f"нет файла {name}. Список: {[q.name for q in in_dir.iterdir() if q.is_file()]}"
-
-    suffix = p.suffix.lower()
-    if suffix in {".xlsx", ".xlsm"}:
-        try:
-            from openpyxl import load_workbook
-        except ImportError:
-            return "openpyxl недоступен — используй run_python"
-        wb = load_workbook(p, read_only=True, data_only=True)
-        parts = []
-        for ws in wb.worksheets[:3]:
-            parts.append(f"--- лист '{ws.title}' ---")
-            for row in ws.iter_rows(min_row=1, max_row=min(ws.max_row or 0, 15), values_only=True):
-                parts.append(" | ".join("" if c is None else str(c) for c in row))
-        wb.close()
-        return "\n".join(parts)[:TOOL_LIMIT]
-
-    if suffix in {".csv", ".tsv", ".txt", ".md", ".json", ".py", ".log", ".xml", ".yml", ".yaml"}:
-        text = p.read_text(encoding="utf-8", errors="replace")
-        if len(text) > TOOL_LIMIT:
-            return text[:TOOL_LIMIT] + "\n...[обрезано]"
-        return text
-
-    return f"бинарный файл ({suffix or 'без расширения'}) — работай с ним через run_python"
-
-
-def _tool_write(name: str, content: str) -> str:
-    _, out_dir = ensure_dirs()
-    p = _safe(out_dir, name)
-    p.parent.mkdir(parents=True, exist_ok=True)
-    p.write_text(content, encoding="utf-8")
-    return f"записано: {OUT_DIR}/{p.relative_to(out_dir)}"
-
-
-def _tool_python(code: str, timeout: int = 180) -> str:
-    try:
-        proc = subprocess.run(
-            [sys.executable, "-c", code],
-            cwd=_caller_dir(),
-            capture_output=True,
-            text=True,
-            timeout=timeout,
-        )
-    except subprocess.TimeoutExpired:
-        return f"таймаут {timeout}s — код выполнялся слишком долго"
-    out = (proc.stdout or "")[-TOOL_LIMIT:]
-    err = (proc.stderr or "")[-4000:]
-    res = f"exit={proc.returncode}\nstdout:\n{out or '(пусто)'}"
-    if err.strip():
-        res += f"\nstderr:\n{err}"
-    return res
-
-
-def _dispatch(name: str, args: dict) -> str:
-    try:
-        if name == "list_files":
-            return _tool_list()
-        if name == "read_file":
-            return _tool_read(str(args.get("name", "")))
-        if name == "write_file":
-            return _tool_write(str(args.get("name", "")), str(args.get("content", "")))
-        if name == "run_python":
-            return _tool_python(str(args.get("code", "")))
-        return f"неизвестный инструмент: {name}"
-    except Exception as exc:
-        return f"ОШИБКА: {type(exc).__name__}: {exc}"
-
-
-# --- агент --------------------------------------------------------------
-
-def run_agent(prompt: str, model=None, max_steps=MAX_STEPS, verbose=True) -> str:
-    """Цикл: модель -> инструменты -> ответ. Возвращает финальный текст."""
-    mdl = model or _state["model"]
-    messages = [
-        {"role": "system", "content": SYSTEM_PROMPT},
-        {"role": "user", "content": prompt},
-    ]
-
-    for step in range(1, max_steps + 1):
-        data = _api({"model": mdl, "messages": messages, "tools": TOOLS})
-        try:
-            msg = data["choices"][0]["message"]
-        except (KeyError, IndexError, TypeError):
-            raise RuntimeError(f"неожиданный ответ: {str(data)[:400]}") from None
-
-        calls = msg.get("tool_calls") or []
-        if not calls:
-            return (msg.get("content") or "").strip()
-
-        assistant = {"role": "assistant", "content": msg.get("content") or ""}
-        assistant["tool_calls"] = calls
-        messages.append(assistant)
-
-        for tc in calls:
-            fn = (tc.get("function") or {}).get("name", "?")
-            raw = (tc.get("function") or {}).get("arguments") or "{}"
-            try:
-                args = json.loads(raw)
-            except json.JSONDecodeError:
-                args = {}
-            if verbose:
-                hint = args.get("name") or str(args.get("code", ""))[:45].replace("\n", " ")
-                print(f"  [{step}] {fn}({hint})")
-            result = _dispatch(fn, args)
-            messages.append(
-                {
-                    "role": "tool",
-                    "tool_call_id": tc.get("id", ""),
-                    "content": result[:TOOL_LIMIT],
-                }
-            )
-
-    return f"агент не уложился в {max_steps} шагов"
-
-
-# --- ввод/вывод ---------------------------------------------------------
-
-def pause(prompt: str = PAUSE_TEXT) -> None:
-    """Ждёт Enter. Больше не вызывается автоматом, оставлена для ручного использования."""
-    if not sys.stdin.isatty():
-        return
-    try:
-        input(prompt)
-    except (EOFError, KeyboardInterrupt):
-        print()
-
-
-def interactive(prompt=None, model=None, filename=None, save_answer=True, verbose=True):
-    """Создать files/ и files2/, принять промпт, выполнить агента, сохранить ответ в files2/N.txt."""
-    in_dir, out_dir = ensure_dirs()
-
-    if prompt is None:
-        if not sys.stdin.isatty():
-            raise RuntimeError("нужен интерактивный терминал или аргумент prompt")
-        try:
-            prompt = input(PROMPT).strip()
-        except (EOFError, KeyboardInterrupt):
-            print()
-            return None
-
-    prompt = str(prompt).strip()
-    if not prompt:
-        print("Пустой промпт — выход.")
-        return None
-
-    print(f"думаю... ({model or _state['model']})")
-    answer = run_agent(prompt, model=model, verbose=verbose)
-
-    target = None
-    if save_answer and answer:
-        target = out_dir / (filename or next_filename(out_dir).name)
-        target.write_text(answer + "\n", encoding="utf-8")
-
-    produced = sorted(p.name for p in out_dir.iterdir() if p.is_file())
-    print(f"готово. {OUT_DIR}/: {', '.join(produced) if produced else 'пусто'}")
-    return target
-
-
-# --- магия: модуль, который можно вызвать -------------------------------
-
-class _LibreCodingModule(ModuleType):
-    def __call__(self, obj=None, *args, **kwargs):
-        if callable(obj) and not args and not kwargs:
-            return obj                      # @librecoding -> режим декоратора
-        try:
-            if isinstance(obj, str):        # librecoding("готовый промпт")
-                kwargs["prompt"] = obj
-            return interactive(*args, **kwargs)
-        except Exception as exc:
-            print(f"Ошибка: {exc}", file=sys.stderr)
-            if os.environ.get("LIBRECODING_DEBUG"):
-                raise
-            return None
-
-
-sys.modules[__name__].__class__ = _LibreCodingModule
-
-__all__ = [
-    "greet", "ask", "run_agent", "interactive", "pause", "ensure_dirs", "next_filename",
-    "set_api_key", "set_model", "get_model",
-]

librecoding-0.6.0/src/librecoding/core.py

@@ -1,5 +0,0 @@
-def greet(name: str = "мир") -> str:
-    """Печатает и возвращает приветствие."""
-    msg = f"Привет, {name}!"
-    print(msg)
-    return msg