revengelibrary 0.1.5__tar.gz → 0.1.6__tar.gz

{revengelibrary-0.1.5 → revengelibrary-0.1.6}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: revengelibrary
-Version: 0.1.5
-Summary: Python chat library and CLI for free LLM models via OpenRouter.
+Version: 0.1.6
+Summary: Не нейросеть
 Author: revengebibliotek contributors
 License: MIT
 Project-URL: Homepage, https://github.com/example/revengelibrary

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

@@ -4,8 +4,8 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "revengelibrary"
-version = "0.1.5"
-description = "Python chat library and CLI for free LLM models via OpenRouter."
+version = "0.1.6"
+description = "Не нейросеть"
 readme = "README.md"
 requires-python = ">=3.9"
 license = { text = "MIT" }

{revengelibrary-0.1.5 → revengelibrary-0.1.6}/revengelibrary/__init__.py

@@ -1,3 +1,4 @@
+from .agents import DEFAULT_AGENT, AgentProfile, get_agent, list_agents
 from .chat import (
     APIError,
     DEFAULT_MEMORY_FILE,
@@ -12,5 +13,9 @@ __all__ = [
     "DEFAULT_OPENROUTER_API_KEY",
     "DEFAULT_MODEL",
     "DEFAULT_MEMORY_FILE",
+    "DEFAULT_AGENT",
+    "AgentProfile",
+    "list_agents",
+    "get_agent",
 ]
-__version__ = "0.1.5"
+__version__ = "0.1.6"

revengelibrary-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-0.1.5 → revengelibrary-0.1.6}/revengelibrary/chat.py

@@ -176,7 +176,7 @@ class FreeNeuroChatClient:
         value = file_path if file_path is not None else self.memory_file
         if not value:
             return None
-        return Path(value).expanduser()
+        return _normalize_memory_path(value)
 
     @staticmethod
     def _normalize_messages(data: Any) -> list[dict[str, str]]:
@@ -205,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-0.1.5 → revengelibrary-0.1.6}/revengelibrary/cli.py

@@ -2,7 +2,9 @@ from __future__ import annotations
 
 import argparse
 import os
+from pathlib import Path
 
+from .agents import DEFAULT_AGENT, get_agent, list_agents
 from .chat import (
     APIError,
     DEFAULT_MEMORY_FILE,
@@ -29,8 +31,18 @@ def _build_parser() -> argparse.ArgumentParser:
     )
     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",
@@ -47,11 +59,24 @@ def main() -> int:
     parser = _build_parser()
     args = parser.parse_args()
 
+    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 or DEFAULT_OPENROUTER_API_KEY,
         model=args.model,
-        system_prompt=args.system,
-        memory_file=args.memory_file,
+        system_prompt=system_prompt,
+        memory_file=memory_file,
     )
 
     print(
@@ -89,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-0.1.5 → revengelibrary-0.1.6}/revengelibrary.egg-info/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: revengelibrary
-Version: 0.1.5
-Summary: Python chat library and CLI for free LLM models via OpenRouter.
+Version: 0.1.6
+Summary: Не нейросеть
 Author: revengebibliotek contributors
 License: MIT
 Project-URL: Homepage, https://github.com/example/revengelibrary

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

@@ -3,6 +3,7 @@ MANIFEST.in
 README.md
 pyproject.toml
 revengelibrary/__init__.py
+revengelibrary/agents.py
 revengelibrary/chat.py
 revengelibrary/cli.py
 revengelibrary/memory_store.json