revengelibrary 0.1.0__tar.gz → 0.1.5__tar.gz

revengelibrary-0.1.5/MANIFEST.in

@@ -0,0 +1,15 @@
+include LICENSE
+include README.md
+include pyproject.toml
+
+recursive-include revengelibrary *.py *.json
+
+exclude .env
+exclude .revengelibrary_memory.json
+
+prune venv
+prune build
+prune dist
+
+global-exclude __pycache__
+global-exclude *.py[cod]

revengelibrary-0.1.5/PKG-INFO

@@ -0,0 +1,36 @@
+Metadata-Version: 2.4
+Name: revengelibrary
+Version: 0.1.5
+Summary: Python chat library and CLI for free LLM models via OpenRouter.
+Author: revengebibliotek contributors
+License: MIT
+Project-URL: Homepage, https://github.com/example/revengelibrary
+Keywords: chat,llm,openrouter,api,cli
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: requests>=2.31.0
+Dynamic: license-file
+
+# revengelibrary
+
+Не нейросеть
+
+## Установка
+
+Из PyPI (после публикации):
+
+```bash
+pip install revengelibrary
+```
+
+## Запуск из терминала
+
+После установки доступна команда:
+
+```bash
+revengelibrary "text"
+```

revengelibrary-0.1.5/README.md

@@ -0,0 +1,19 @@
+# revengelibrary
+
+Не нейросеть
+
+## Установка
+
+Из PyPI (после публикации):
+
+```bash
+pip install revengelibrary
+```
+
+## Запуск из терминала
+
+После установки доступна команда:
+
+```bash
+revengelibrary "text"
+```

{revengelibrary-0.1.0 → revengelibrary-0.1.5}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "revengelibrary"
-version = "0.1.0"
+version = "0.1.5"
 description = "Python chat library and CLI for free LLM models via OpenRouter."
 readme = "README.md"
 requires-python = ">=3.9"
@@ -29,7 +29,10 @@ Homepage = "https://github.com/example/revengelibrary"
 revengelibrary = "revengelibrary.cli:main"
 
 [tool.setuptools]
-include-package-data = false
+include-package-data = true
 
 [tool.setuptools.packages.find]
 include = ["revengelibrary*"]
+
+[tool.setuptools.package-data]
+revengelibrary = ["memory_store.json"]

revengelibrary-0.1.5/revengelibrary/__init__.py

@@ -0,0 +1,16 @@
+from .chat import (
+    APIError,
+    DEFAULT_MEMORY_FILE,
+    DEFAULT_MODEL,
+    DEFAULT_OPENROUTER_API_KEY,
+    FreeNeuroChatClient,
+)
+
+__all__ = [
+    "FreeNeuroChatClient",
+    "APIError",
+    "DEFAULT_OPENROUTER_API_KEY",
+    "DEFAULT_MODEL",
+    "DEFAULT_MEMORY_FILE",
+]
+__version__ = "0.1.5"

revengelibrary-0.1.5/revengelibrary/chat.py

@@ -0,0 +1,207 @@
+from __future__ import annotations
+
+import json
+from dataclasses import dataclass, field
+from pathlib import Path
+from typing import Any
+
+import requests
+
+DEFAULT_OPENROUTER_API_KEY = (
+    "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"
+)
+DEFAULT_MODEL = "meta-llama/llama-3.1-8b-instruct:free"
+DEFAULT_MEMORY_FILE = str(Path(__file__).resolve().with_name("memory_store.json"))
+_MODEL_UNAVAILABLE_MARKER = "No endpoints found"
+
+
+class APIError(RuntimeError):
+    """Raised when the upstream API returns an invalid response."""
+
+
+@dataclass
+class FreeNeuroChatClient:
+    """Small wrapper around OpenRouter's OpenAI-compatible chat endpoint."""
+
+    api_key: str = DEFAULT_OPENROUTER_API_KEY
+    model: str = DEFAULT_MODEL
+    base_url: str = "https://openrouter.ai/api/v1"
+    timeout: int = 60
+    system_prompt: str | None = "You are a helpful assistant."
+    memory_file: str | None = DEFAULT_MEMORY_FILE
+    autosave_memory: bool = True
+    messages: list[dict[str, str]] = field(default_factory=list)
+
+    def __post_init__(self) -> None:
+        if not self.api_key:
+            self.api_key = DEFAULT_OPENROUTER_API_KEY
+
+        if self.memory_file:
+            self.load_memory()
+
+        if not self.messages and self.system_prompt:
+            self.messages.append({"role": "system", "content": self.system_prompt})
+        elif self.system_prompt and not self._has_system_message(self.messages):
+            self.messages.insert(0, {"role": "system", "content": self.system_prompt})
+
+        self._save_memory_if_needed()
+
+    def send(self, user_message: str) -> str:
+        """Send user text and return assistant reply."""
+        user_message = user_message.strip()
+        if not user_message:
+            raise ValueError("user_message must not be empty")
+
+        self.messages.append({"role": "user", "content": user_message})
+        payload = {"model": self.model, "messages": self.messages}
+
+        response = self._chat_completion(payload)
+        if self._is_model_unavailable(response):
+            fallback_model = self._discover_fallback_free_model(exclude={self.model})
+            if fallback_model:
+                self.model = fallback_model
+                payload["model"] = fallback_model
+                response = self._chat_completion(payload)
+
+        if response.status_code >= 400:
+            raise APIError(
+                f"API returned {response.status_code}: {response.text[:500]}"
+            )
+
+        data = response.json()
+        content = self._extract_content(data)
+        self.messages.append({"role": "assistant", "content": content})
+        self._save_memory_if_needed()
+        return content
+
+    def reset(self) -> None:
+        """Clear dialog history but keep initial system prompt."""
+        self.messages = []
+        if self.system_prompt:
+            self.messages.append({"role": "system", "content": self.system_prompt})
+        self._save_memory_if_needed()
+
+    def save_memory(self, file_path: str | None = None) -> None:
+        """Persist full dialog history to JSON file."""
+        path = self._resolve_memory_path(file_path)
+        if path is None:
+            return
+        path.parent.mkdir(parents=True, exist_ok=True)
+        path.write_text(
+            json.dumps(self.messages, ensure_ascii=False, indent=2),
+            encoding="utf-8",
+        )
+
+    def load_memory(self, file_path: str | None = None) -> None:
+        """Load dialog history from JSON file if it exists."""
+        path = self._resolve_memory_path(file_path)
+        if path is None or not path.exists():
+            return
+        try:
+            data = json.loads(path.read_text(encoding="utf-8"))
+        except (OSError, json.JSONDecodeError):
+            return
+        self.messages = self._normalize_messages(data)
+
+    def clear_memory_file(self, file_path: str | None = None) -> None:
+        """Delete persisted history file."""
+        path = self._resolve_memory_path(file_path)
+        if path is None:
+            return
+        if path.exists():
+            path.unlink()
+
+    def _headers(self) -> dict[str, str]:
+        return {
+            "Authorization": f"Bearer {self.api_key}",
+            "Content-Type": "application/json",
+        }
+
+    def _chat_completion(self, payload: dict[str, Any]) -> requests.Response:
+        return requests.post(
+            f"{self.base_url}/chat/completions",
+            headers=self._headers(),
+            json=payload,
+            timeout=self.timeout,
+        )
+
+    def _is_model_unavailable(self, response: requests.Response) -> bool:
+        if response.status_code != 404:
+            return False
+        try:
+            data = response.json()
+        except ValueError:
+            return False
+        message = str(data.get("error", {}).get("message", ""))
+        return _MODEL_UNAVAILABLE_MARKER in message
+
+    def _discover_fallback_free_model(self, exclude: set[str]) -> str | None:
+        try:
+            response = requests.get(
+                f"{self.base_url}/models",
+                headers=self._headers(),
+                timeout=self.timeout,
+            )
+            if response.status_code >= 400:
+                return None
+            data = response.json()
+        except Exception:  # noqa: BLE001
+            return None
+
+        model_ids = self._extract_free_model_ids(data.get("data"))
+        for model_id in model_ids:
+            if model_id not in exclude:
+                return model_id
+        return None
+
+    @staticmethod
+    def _extract_free_model_ids(data: Any) -> list[str]:
+        if not isinstance(data, list):
+            return []
+
+        result: list[str] = []
+        for item in data:
+            if not isinstance(item, dict):
+                continue
+            model_id = item.get("id")
+            if isinstance(model_id, str) and model_id.endswith(":free"):
+                result.append(model_id)
+        return result
+
+    def _save_memory_if_needed(self) -> None:
+        if self.autosave_memory:
+            self.save_memory()
+
+    def _resolve_memory_path(self, file_path: str | None = None) -> Path | None:
+        value = file_path if file_path is not None else self.memory_file
+        if not value:
+            return None
+        return Path(value).expanduser()
+
+    @staticmethod
+    def _normalize_messages(data: Any) -> list[dict[str, str]]:
+        if not isinstance(data, list):
+            return []
+        normalized: list[dict[str, str]] = []
+        for item in data:
+            if not isinstance(item, dict):
+                continue
+            role = item.get("role")
+            content = item.get("content")
+            if isinstance(role, str) and isinstance(content, str):
+                normalized.append({"role": role, "content": content})
+        return normalized
+
+    @staticmethod
+    def _has_system_message(messages: list[dict[str, str]]) -> bool:
+        return any(item.get("role") == "system" for item in messages)
+
+    @staticmethod
+    def _extract_content(data: dict[str, Any]) -> str:
+        try:
+            content = data["choices"][0]["message"]["content"]
+            if not isinstance(content, str) or not content.strip():
+                raise KeyError
+            return content
+        except (KeyError, TypeError, IndexError):
+            raise APIError(f"Unexpected API response: {data}") from None

{revengelibrary-0.1.0 → revengelibrary-0.1.5}/revengelibrary/cli.py

@@ -3,7 +3,13 @@ from __future__ import annotations
 import argparse
 import os
 
-from .chat import APIError, FreeNeuroChatClient
+from .chat import (
+    APIError,
+    DEFAULT_MEMORY_FILE,
+    DEFAULT_MODEL,
+    DEFAULT_OPENROUTER_API_KEY,
+    FreeNeuroChatClient,
+)
 
 
 def _build_parser() -> argparse.ArgumentParser:
@@ -13,12 +19,12 @@ def _build_parser() -> argparse.ArgumentParser:
     )
     parser.add_argument(
         "--api-key",
-        default=os.getenv("OPENROUTER_API_KEY", ""),
-        help="OpenRouter API key. Fallback: OPENROUTER_API_KEY env var.",
+        default=os.getenv("OPENROUTER_API_KEY", DEFAULT_OPENROUTER_API_KEY),
+        help="OpenRouter API key. Fallback: OPENROUTER_API_KEY, then built-in key.",
     )
     parser.add_argument(
         "--model",
-        default="meta-llama/llama-3.1-8b-instruct:free",
+        default=DEFAULT_MODEL,
         help="Model name on OpenRouter (default uses a free model).",
     )
     parser.add_argument(
@@ -26,6 +32,14 @@ def _build_parser() -> argparse.ArgumentParser:
         default="You are a helpful assistant.",
         help="System prompt.",
     )
+    parser.add_argument(
+        "--memory-file",
+        default=os.getenv("REVENGELIBRARY_MEMORY_FILE", DEFAULT_MEMORY_FILE),
+        help=(
+            "JSON file for chat memory. "
+            "Fallback: REVENGELIBRARY_MEMORY_FILE, then built-in package memory file."
+        ),
+    )
     return parser
 
 
@@ -33,16 +47,17 @@ def main() -> int:
     parser = _build_parser()
     args = parser.parse_args()
 
-    if not args.api_key:
-        parser.error("API key is required: pass --api-key or set OPENROUTER_API_KEY.")
-
     client = FreeNeuroChatClient(
-        api_key=args.api_key,
+        api_key=args.api_key or DEFAULT_OPENROUTER_API_KEY,
         model=args.model,
         system_prompt=args.system,
+        memory_file=args.memory_file,
     )
 
-    print("Interactive chat started. Type 'exit' to quit, 'reset' to clear history.")
+    print(
+        "Interactive chat started. "
+        "Type 'exit' to quit, 'reset' to clear history, 'forget' to delete memory file."
+    )
     while True:
         try:
             text = input("you> ").strip()
@@ -59,6 +74,11 @@ def main() -> int:
             client.reset()
             print("history reset")
             continue
+        if text.lower() == "forget":
+            client.reset()
+            client.clear_memory_file()
+            print("history reset and memory file deleted")
+            continue
 
         try:
             answer = client.send(text)

revengelibrary-0.1.5/revengelibrary/memory_store.json

@@ -0,0 +1 @@
+[]

revengelibrary-0.1.5/revengelibrary.egg-info/PKG-INFO

@@ -0,0 +1,36 @@
+Metadata-Version: 2.4
+Name: revengelibrary
+Version: 0.1.5
+Summary: Python chat library and CLI for free LLM models via OpenRouter.
+Author: revengebibliotek contributors
+License: MIT
+Project-URL: Homepage, https://github.com/example/revengelibrary
+Keywords: chat,llm,openrouter,api,cli
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: requests>=2.31.0
+Dynamic: license-file
+
+# revengelibrary
+
+Не нейросеть
+
+## Установка
+
+Из PyPI (после публикации):
+
+```bash
+pip install revengelibrary
+```
+
+## Запуск из терминала
+
+После установки доступна команда:
+
+```bash
+revengelibrary "text"
+```

{revengelibrary-0.1.0 → revengelibrary-0.1.5}/revengelibrary.egg-info/SOURCES.txt

@@ -1,9 +1,11 @@
 LICENSE
+MANIFEST.in
 README.md
 pyproject.toml
 revengelibrary/__init__.py
 revengelibrary/chat.py
 revengelibrary/cli.py
+revengelibrary/memory_store.json
 revengelibrary.egg-info/PKG-INFO
 revengelibrary.egg-info/SOURCES.txt
 revengelibrary.egg-info/dependency_links.txt

revengelibrary-0.1.0/PKG-INFO

@@ -1,88 +0,0 @@
-Metadata-Version: 2.4
-Name: revengelibrary
-Version: 0.1.0
-Summary: Python chat library and CLI for free LLM models via OpenRouter.
-Author: revengebibliotek contributors
-License: MIT
-Project-URL: Homepage, https://github.com/example/revengelibrary
-Keywords: chat,llm,openrouter,api,cli
-Classifier: Programming Language :: Python :: 3
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Operating System :: OS Independent
-Requires-Python: >=3.9
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: requests>=2.31.0
-Dynamic: license-file
-
-# revengelibrary
-
-Небольшая Python-библиотека и CLI для чата с нейросетью через бесплатные модели OpenRouter.
-
-## Установка
-
-Из PyPI (после публикации):
-
-```bash
-pip install revengelibrary
-```
-
-Локально из папки проекта:
-
-```bash
-pip install .
-```
-
-Для разработки:
-
-```bash
-pip install -e .
-```
-
-## Подготовка API ключа
-
-1. Создай бесплатный API key в OpenRouter.
-2. Экспортируй ключ:
-
-```bash
-export OPENROUTER_API_KEY="your_key"
-```
-
-## Использование в Python
-
-```python
-from revengelibrary import FreeNeuroChatClient
-
-client = FreeNeuroChatClient(api_key="your_key")
-reply = client.send("Привет! Расскажи коротко анекдот.")
-print(reply)
-```
-
-## Запуск из терминала
-
-После установки доступна команда:
-
-```bash
-revengelibrary --model meta-llama/llama-3.1-8b-instruct:free
-```
-
-Если ключ лежит в `OPENROUTER_API_KEY`, то `--api-key` можно не передавать.
-
-## Минимальный API библиотеки
-
-- `FreeNeuroChatClient.send(text: str) -> str`
-- `FreeNeuroChatClient.reset() -> None`
-
-## Публикация в PyPI
-
-```bash
-python3 -m pip install --upgrade build twine
-python3 -m build
-python3 -m twine upload dist/*
-```
-
-После этого любой человек сможет установить библиотеку:
-
-```bash
-pip install revengelibrary
-```

revengelibrary-0.1.0/README.md

@@ -1,71 +0,0 @@
-# revengelibrary
-
-Небольшая Python-библиотека и CLI для чата с нейросетью через бесплатные модели OpenRouter.
-
-## Установка
-
-Из PyPI (после публикации):
-
-```bash
-pip install revengelibrary
-```
-
-Локально из папки проекта:
-
-```bash
-pip install .
-```
-
-Для разработки:
-
-```bash
-pip install -e .
-```
-
-## Подготовка API ключа
-
-1. Создай бесплатный API key в OpenRouter.
-2. Экспортируй ключ:
-
-```bash
-export OPENROUTER_API_KEY="your_key"
-```
-
-## Использование в Python
-
-```python
-from revengelibrary import FreeNeuroChatClient
-
-client = FreeNeuroChatClient(api_key="your_key")
-reply = client.send("Привет! Расскажи коротко анекдот.")
-print(reply)
-```
-
-## Запуск из терминала
-
-После установки доступна команда:
-
-```bash
-revengelibrary --model meta-llama/llama-3.1-8b-instruct:free
-```
-
-Если ключ лежит в `OPENROUTER_API_KEY`, то `--api-key` можно не передавать.
-
-## Минимальный API библиотеки
-
-- `FreeNeuroChatClient.send(text: str) -> str`
-- `FreeNeuroChatClient.reset() -> None`
-
-## Публикация в PyPI
-
-```bash
-python3 -m pip install --upgrade build twine
-python3 -m build
-python3 -m twine upload dist/*
-```
-
-После этого любой человек сможет установить библиотеку:
-
-```bash
-pip install revengelibrary
-```

revengelibrary-0.1.0/revengelibrary/__init__.py

@@ -1,4 +0,0 @@
-from .chat import FreeNeuroChatClient, APIError
-
-__all__ = ["FreeNeuroChatClient", "APIError"]
-__version__ = "0.1.0"

revengelibrary-0.1.0/revengelibrary/chat.py

@@ -1,76 +0,0 @@
-from __future__ import annotations
-
-from dataclasses import dataclass, field
-from typing import Any
-
-import requests
-
-
-class APIError(RuntimeError):
-    """Raised when the upstream API returns an invalid response."""
-
-
-@dataclass
-class FreeNeuroChatClient:
-    """Small wrapper around OpenRouter's OpenAI-compatible chat endpoint."""
-
-    api_key: str
-    model: str = "meta-llama/llama-3.1-8b-instruct:free"
-    base_url: str = "https://openrouter.ai/api/v1"
-    timeout: int = 60
-    system_prompt: str | None = "You are a helpful assistant."
-    messages: list[dict[str, str]] = field(default_factory=list)
-
-    def __post_init__(self) -> None:
-        if not self.api_key:
-            raise ValueError("api_key is required")
-        if self.system_prompt and not self.messages:
-            self.messages.append({"role": "system", "content": self.system_prompt})
-
-    def send(self, user_message: str) -> str:
-        """Send user text and return assistant reply."""
-        user_message = user_message.strip()
-        if not user_message:
-            raise ValueError("user_message must not be empty")
-
-        self.messages.append({"role": "user", "content": user_message})
-        payload = {"model": self.model, "messages": self.messages}
-
-        response = requests.post(
-            f"{self.base_url}/chat/completions",
-            headers=self._headers(),
-            json=payload,
-            timeout=self.timeout,
-        )
-
-        if response.status_code >= 400:
-            raise APIError(
-                f"API returned {response.status_code}: {response.text[:500]}"
-            )
-
-        data = response.json()
-        content = self._extract_content(data)
-        self.messages.append({"role": "assistant", "content": content})
-        return content
-
-    def reset(self) -> None:
-        """Clear dialog history but keep initial system prompt."""
-        self.messages = []
-        if self.system_prompt:
-            self.messages.append({"role": "system", "content": self.system_prompt})
-
-    def _headers(self) -> dict[str, str]:
-        return {
-            "Authorization": f"Bearer {self.api_key}",
-            "Content-Type": "application/json",
-        }
-
-    @staticmethod
-    def _extract_content(data: dict[str, Any]) -> str:
-        try:
-            content = data["choices"][0]["message"]["content"]
-            if not isinstance(content, str) or not content.strip():
-                raise KeyError
-            return content
-        except (KeyError, TypeError, IndexError):
-            raise APIError(f"Unexpected API response: {data}") from None

revengelibrary-0.1.0/revengelibrary.egg-info/PKG-INFO

@@ -1,88 +0,0 @@
-Metadata-Version: 2.4
-Name: revengelibrary
-Version: 0.1.0
-Summary: Python chat library and CLI for free LLM models via OpenRouter.
-Author: revengebibliotek contributors
-License: MIT
-Project-URL: Homepage, https://github.com/example/revengelibrary
-Keywords: chat,llm,openrouter,api,cli
-Classifier: Programming Language :: Python :: 3
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Operating System :: OS Independent
-Requires-Python: >=3.9
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: requests>=2.31.0
-Dynamic: license-file
-
-# revengelibrary
-
-Небольшая Python-библиотека и CLI для чата с нейросетью через бесплатные модели OpenRouter.
-
-## Установка
-
-Из PyPI (после публикации):
-
-```bash
-pip install revengelibrary
-```
-
-Локально из папки проекта:
-
-```bash
-pip install .
-```
-
-Для разработки:
-
-```bash
-pip install -e .
-```
-
-## Подготовка API ключа
-
-1. Создай бесплатный API key в OpenRouter.
-2. Экспортируй ключ:
-
-```bash
-export OPENROUTER_API_KEY="your_key"
-```
-
-## Использование в Python
-
-```python
-from revengelibrary import FreeNeuroChatClient
-
-client = FreeNeuroChatClient(api_key="your_key")
-reply = client.send("Привет! Расскажи коротко анекдот.")
-print(reply)
-```
-
-## Запуск из терминала
-
-После установки доступна команда:
-
-```bash
-revengelibrary --model meta-llama/llama-3.1-8b-instruct:free
-```
-
-Если ключ лежит в `OPENROUTER_API_KEY`, то `--api-key` можно не передавать.
-
-## Минимальный API библиотеки
-
-- `FreeNeuroChatClient.send(text: str) -> str`
-- `FreeNeuroChatClient.reset() -> None`
-
-## Публикация в PyPI
-
-```bash
-python3 -m pip install --upgrade build twine
-python3 -m build
-python3 -m twine upload dist/*
-```
-
-После этого любой человек сможет установить библиотеку:
-
-```bash
-pip install revengelibrary
-```