revengelibrary 0.1.0__py3-none-any.whl → 0.1.6__py3-none-any.whl

revengelibrary/__init__.py

@@ -1,4 +1,21 @@
-from .chat import FreeNeuroChatClient, APIError
+from .agents import DEFAULT_AGENT, AgentProfile, get_agent, list_agents
+from .chat import (
+    APIError,
+    DEFAULT_MEMORY_FILE,
+    DEFAULT_MODEL,
+    DEFAULT_OPENROUTER_API_KEY,
+    FreeNeuroChatClient,
+)
 
-__all__ = ["FreeNeuroChatClient", "APIError"]
-__version__ = "0.1.0"
+__all__ = [
+    "FreeNeuroChatClient",
+    "APIError",
+    "DEFAULT_OPENROUTER_API_KEY",
+    "DEFAULT_MODEL",
+    "DEFAULT_MEMORY_FILE",
+    "DEFAULT_AGENT",
+    "AgentProfile",
+    "list_agents",
+    "get_agent",
+]
+__version__ = "0.1.6"

revengelibrary/agents.py

@@ -0,0 +1,60 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+
+DEFAULT_AGENT = "general"
+
+
+@dataclass(frozen=True)
+class AgentProfile:
+    name: str
+    title: str
+    system_prompt: str
+
+
+AGENT_PROFILES: dict[str, AgentProfile] = {
+    "general": AgentProfile(
+        name="general",
+        title="Универсальный ассистент",
+        system_prompt="You are a helpful assistant.",
+    ),
+    "frontend": AgentProfile(
+        name="frontend",
+        title="Senior Frontend + Mobile Design",
+        system_prompt=(
+            "Ты senior frontend-разработчик с сильной экспертизой в дизайне "
+            "мобильных приложений. Пиши ответы как практический инженер: "
+            "mobile-first, доступность, понятная архитектура компонентов, "
+            "чистый UI/UX, адаптив под iOS/Android, хорошая типографика и "
+            "согласованная дизайн-система."
+        ),
+    ),
+    "backend": AgentProfile(
+        name="backend",
+        title="Backend Engineer",
+        system_prompt=(
+            "Ты senior backend-разработчик. Предлагай надежные API-контракты, "
+            "чистую архитектуру, контроль ошибок, безопасность и масштабируемость."
+        ),
+    ),
+    "qa": AgentProfile(
+        name="qa",
+        title="QA Engineer",
+        system_prompt=(
+            "Ты senior QA-инженер. Фокус: тест-кейсы, граничные условия, "
+            "регрессия, воспроизводимость багов и риски релиза."
+        ),
+    ),
+}
+
+
+def list_agents() -> list[AgentProfile]:
+    return list(AGENT_PROFILES.values())
+
+
+def get_agent(agent_name: str) -> AgentProfile:
+    name = (agent_name or DEFAULT_AGENT).strip().lower()
+    if name not in AGENT_PROFILES:
+        available = ", ".join(sorted(AGENT_PROFILES))
+        raise ValueError(f"Unknown agent '{agent_name}'. Available: {available}")
+    return AGENT_PROFILES[name]

revengelibrary/chat.py

@@ -1,10 +1,19 @@
 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."""
@@ -14,18 +23,28 @@ class APIError(RuntimeError):
 class FreeNeuroChatClient:
     """Small wrapper around OpenRouter's OpenAI-compatible chat endpoint."""
 
-    api_key: str
-    model: str = "meta-llama/llama-3.1-8b-instruct:free"
+    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:
-            raise ValueError("api_key is required")
-        if self.system_prompt and not self.messages:
+            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."""
@@ -36,12 +55,13 @@ class FreeNeuroChatClient:
         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,
-        )
+        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(
@@ -51,6 +71,7 @@ class FreeNeuroChatClient:
         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:
@@ -58,6 +79,37 @@ class FreeNeuroChatClient:
         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 {
@@ -65,6 +117,85 @@ class FreeNeuroChatClient:
             "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 _normalize_memory_path(value)
+
+    @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:
@@ -74,3 +205,10 @@ class FreeNeuroChatClient:
             return content
         except (KeyError, TypeError, IndexError):
             raise APIError(f"Unexpected API response: {data}") from None
+
+
+def _normalize_memory_path(value: str) -> Path:
+    path = Path(value).expanduser()
+    if path.suffix:
+        return path
+    return path.with_suffix(".json")

revengelibrary/cli.py

@@ -2,8 +2,16 @@ from __future__ import annotations
 
 import argparse
 import os
+from pathlib import Path
 
-from .chat import APIError, FreeNeuroChatClient
+from .agents import DEFAULT_AGENT, get_agent, list_agents
+from .chat import (
+    APIError,
+    DEFAULT_MEMORY_FILE,
+    DEFAULT_MODEL,
+    DEFAULT_OPENROUTER_API_KEY,
+    FreeNeuroChatClient,
+)
 
 
 def _build_parser() -> argparse.ArgumentParser:
@@ -13,18 +21,36 @@ 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(
         "--system",
-        default="You are a helpful assistant.",
-        help="System prompt.",
+        default=None,
+        help="System prompt override. If omitted, selected agent prompt is used.",
+    )
+    parser.add_argument(
+        "--agent",
+        default=os.getenv("REVENGELIBRARY_AGENT", DEFAULT_AGENT),
+        help="Agent role. Use --list-agents to view available roles.",
+    )
+    parser.add_argument(
+        "--list-agents",
+        action="store_true",
+        help="Show available agent roles and exit.",
+    )
+    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 +59,30 @@ 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.")
+    if args.list_agents:
+        for profile in list_agents():
+            print(f"{profile.name}: {profile.title}")
+        return 0
+
+    try:
+        agent = get_agent(args.agent)
+    except ValueError as exc:
+        parser.error(str(exc))
+
+    system_prompt = args.system if args.system else agent.system_prompt
+    memory_file = _agent_memory_file(args.memory_file, agent.name)
 
     client = FreeNeuroChatClient(
-        api_key=args.api_key,
+        api_key=args.api_key or DEFAULT_OPENROUTER_API_KEY,
         model=args.model,
-        system_prompt=args.system,
+        system_prompt=system_prompt,
+        memory_file=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 +99,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)
@@ -69,5 +114,17 @@ def main() -> int:
             print(f"unexpected error: {exc}")
 
 
+def _agent_memory_file(memory_file: str | None, agent_name: str) -> str | None:
+    if not memory_file:
+        return None
+    if agent_name == DEFAULT_AGENT:
+        return memory_file
+
+    path = Path(memory_file).expanduser()
+    suffix = path.suffix or ".json"
+    stem = path.stem if path.suffix else path.name
+    return str(path.with_name(f"{stem}__{agent_name}{suffix}"))
+
+
 if __name__ == "__main__":
     raise SystemExit(main())

revengelibrary/memory_store.json

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

revengelibrary-0.1.6.dist-info/METADATA

@@ -0,0 +1,36 @@
+Metadata-Version: 2.4
+Name: revengelibrary
+Version: 0.1.6
+Summary: Не нейросеть
+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.6.dist-info/RECORD

@@ -0,0 +1,11 @@
+revengelibrary/__init__.py,sha256=2eUolZgyW421WD9ABtUVRSIsauzY7IZJ-6YMveHcoi8,448
+revengelibrary/agents.py,sha256=GTrO_rsONJiSiIE09turpERnE34vYBLWkLtp5dzDj0o,2230
+revengelibrary/chat.py,sha256=1UyJ9HZNea_oglsYMVqg3VdR2ASJvn4IPTBdUjNQ03k,7542
+revengelibrary/cli.py,sha256=oCJlIIwZf97CtIG4jOMOk55EFSFR7_nYfeqQ2ONwMuI,3741
+revengelibrary/memory_store.json,sha256=N1F-Xz3GaBn2H1p7uKzhkhKCQV8QVR0t76XD6wmFtXA,3
+revengelibrary-0.1.6.dist-info/licenses/LICENSE,sha256=UgqQ8UtVfIOP8-clGkGAylmb9AUbT4-Ue49N_WjIpi4,1073
+revengelibrary-0.1.6.dist-info/METADATA,sha256=UnzJRFbbSLm_FKE9fsmnIKBsUaNPfTxIM96IxYn379w,822
+revengelibrary-0.1.6.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+revengelibrary-0.1.6.dist-info/entry_points.txt,sha256=299WMDffRchuwIlqEIjW-DwHYkBvRZQwGjiASIVofYE,59
+revengelibrary-0.1.6.dist-info/top_level.txt,sha256=gfUsAxA-IgliQakMoupLAAjrTEYtUJurH5n29eHhUBE,15
+revengelibrary-0.1.6.dist-info/RECORD,,

revengelibrary-0.1.0.dist-info/METADATA

@@ -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.dist-info/RECORD

@@ -1,9 +0,0 @@
-revengelibrary/__init__.py,sha256=fTxpG-6tD8L7mRvBv19uO7mZUgI0VYdTe7NS0V7TFQY,117
-revengelibrary/chat.py,sha256=tjwxG01Tg7FSp850px-15jP6Tuygo5U3w_oSLBrzZgE,2582
-revengelibrary/cli.py,sha256=bXiwnGuJFChOT6cxBOkT8FGtFed0LIrCdc8yIND4oZk,1961
-revengelibrary-0.1.0.dist-info/licenses/LICENSE,sha256=UgqQ8UtVfIOP8-clGkGAylmb9AUbT4-Ue49N_WjIpi4,1073
-revengelibrary-0.1.0.dist-info/METADATA,sha256=pFHcFRcvos2tVXN0D7xGtgwz4-miIMryB1eBbkCVXSU,2137
-revengelibrary-0.1.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-revengelibrary-0.1.0.dist-info/entry_points.txt,sha256=299WMDffRchuwIlqEIjW-DwHYkBvRZQwGjiASIVofYE,59
-revengelibrary-0.1.0.dist-info/top_level.txt,sha256=gfUsAxA-IgliQakMoupLAAjrTEYtUJurH5n29eHhUBE,15
-revengelibrary-0.1.0.dist-info/RECORD,,