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

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/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

@@ -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/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.dist-info → revengelibrary-0.1.6.dist-info}/METADATA

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

@@ -1,10 +0,0 @@
-revengelibrary/__init__.py,sha256=HWDxVkOjWf65-SM4JIm0p3aNyDoEByCuCjTJCiFgBD4,299
-revengelibrary/chat.py,sha256=a1crCvwW9L3w3YYstE_OFtCM4khrPoGtr8eZFL6-EnQ,7374
-revengelibrary/cli.py,sha256=zllHMGN8EHJCSzMI5kpT8wk2-h9sD_38ysJ9zY6NxO0,2543
-revengelibrary/memory_store.json,sha256=N1F-Xz3GaBn2H1p7uKzhkhKCQV8QVR0t76XD6wmFtXA,3
-revengelibrary-0.1.5.dist-info/licenses/LICENSE,sha256=UgqQ8UtVfIOP8-clGkGAylmb9AUbT4-Ue49N_WjIpi4,1073
-revengelibrary-0.1.5.dist-info/METADATA,sha256=bqiA6Mi7yblgWF42hGOXMO2gDgcExiPNUIEL-U4O6gQ,862
-revengelibrary-0.1.5.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-revengelibrary-0.1.5.dist-info/entry_points.txt,sha256=299WMDffRchuwIlqEIjW-DwHYkBvRZQwGjiASIVofYE,59
-revengelibrary-0.1.5.dist-info/top_level.txt,sha256=gfUsAxA-IgliQakMoupLAAjrTEYtUJurH5n29eHhUBE,15
-revengelibrary-0.1.5.dist-info/RECORD,,