yaicli 0.4.0__py3-none-any.whl → 0.5.1__py3-none-any.whl

yaicli/roles.py

@@ -1,276 +0,0 @@
-import json
-from pathlib import Path
-from typing import Any, Dict, Optional, Union
-
-import typer
-from rich.console import Console
-from rich.prompt import Prompt
-from rich.table import Table
-
-from yaicli.config import cfg
-from yaicli.console import get_console
-from yaicli.const import DEFAULT_ROLES, ROLES_DIR, DefaultRoleNames
-from yaicli.exceptions import RoleAlreadyExistsError, RoleCreationError
-from yaicli.utils import detect_os, detect_shell, option_callback
-
-
-class Role:
-    def __init__(
-        self, name: str, prompt: str, variables: Optional[Dict[str, Any]] = None, filepath: Optional[str] = None
-    ):
-        self.name = name
-        self._prompt = prompt
-        if not variables:
-            variables = {"_os": detect_os(cfg), "_shell": detect_shell(cfg)}
-        self.variables = variables
-        self.filepath = filepath
-
-        self.prompt = self._prompt.format(**self.variables)
-
-    def to_dict(self) -> Dict[str, Any]:
-        """Convert Role to dictionary for serialization"""
-        return {
-            "name": self.name,
-            "prompt": self._prompt,
-        }
-
-    @classmethod
-    def from_dict(cls, role_id: str, data: Dict[str, Any], filepath: Optional[str] = None) -> "Role":
-        """Create Role object from dictionary"""
-        return cls(
-            name=data.get("name", role_id),
-            prompt=data.get("prompt", ""),
-            variables=data.get("variables", {}),
-            filepath=filepath,
-        )
-
-    def __str__(self):
-        return f"Role(name={self.name}, prompt={self.prompt[:30]}...)"
-
-
-class RoleManager:
-    roles_dir: Path = ROLES_DIR
-    console: Console = get_console()
-    _roles: Optional[Dict[str, Role]] = None
-
-    def __new__(cls):
-        """Singleton class for RoleManager"""
-        if not hasattr(cls, "instance"):
-            cls.instance = super().__new__(cls)
-        return cls.instance
-
-    @property
-    def roles(self) -> Dict[str, Role]:
-        if self._roles is None:
-            self._roles = self._load_roles()
-        return self._roles
-
-    def _load_roles(self) -> Dict[str, Role]:
-        """Load all role configurations"""
-        roles: Dict[str, Role] = {}
-        self.roles_dir.mkdir(parents=True, exist_ok=True)
-
-        # Check if any role files exist
-        role_files: list[Path] = list(self.roles_dir.glob("*.json"))
-
-        if not role_files:
-            # Fast path: no existing roles, just create defaults
-            for role_id, role_config in DEFAULT_ROLES.items():
-                role_file = self.roles_dir / f"{role_id}.json"
-                filepath = str(role_file)
-                roles[role_id] = Role.from_dict(role_id, role_config, filepath)
-                with role_file.open("w", encoding="utf-8") as f:
-                    json.dump(role_config, f, indent=2)
-            return roles
-
-        # Load existing role files
-        for role_file in role_files:
-            role_id = role_file.stem
-            filepath = str(role_file)
-            try:
-                with role_file.open("r", encoding="utf-8") as f:
-                    role_data = json.load(f)
-                    roles[role_id] = Role.from_dict(role_id, role_data, filepath)
-            except Exception as e:
-                self.console.print(f"Error loading role {role_id}: {e}", style="red")
-
-        # Ensure default roles exist
-        for role_id, role_config in DEFAULT_ROLES.items():
-            if role_id not in roles:
-                # Default role doesn't exist locally, create it
-                role_file = self.roles_dir / f"{role_id}.json"
-                filepath = str(role_file)
-                roles[role_id] = Role.from_dict(role_id, role_config, filepath)
-                with role_file.open("w", encoding="utf-8") as f:
-                    json.dump(role_config, f, indent=2)
-            else:
-                # TODO: Maybe not necessary
-                # Check if the local role's prompt differs from the built-in one
-                local_role = roles[role_id]
-                builtin_prompt = role_config.get("prompt", "")
-
-                # Only show warning if ROLE_MODIFY_WARNING is enabled
-                if cfg["ROLE_MODIFY_WARNING"]:
-                    if local_role._prompt != builtin_prompt:
-                        self.console.print(
-                            f"[yellow]Warning:[/yellow] Local role '{role_id}' has a different prompt than the built-in role.",
-                            style="yellow",
-                        )
-                        self.console.print(
-                            "To reset to the built-in version, delete the local role file with: "
-                            f'[bold]ai --delete-role "{role_id}"[/bold]',
-                            style="dim",
-                        )
-                        self.console.print(
-                            "To disable this warning, set [bold]ROLE_MODIFY_WARNING=false[/bold] in your config file.",
-                            style="dim",
-                        )
-
-        return roles
-
-    @classmethod
-    @option_callback
-    def print_list_option(cls, _: Any):
-        """Print the list of roles.
-        This method is a cli option callback.
-        """
-        table = Table(show_header=True, show_footer=False)
-        table.add_column("Name", style="dim")
-        table.add_column("Filepath", style="dim")
-        for file in sorted(cls.roles_dir.glob("*.json"), key=lambda f: f.stat().st_mtime):
-            table.add_row(file.stem, str(file))
-        cls.console.print(table)
-        cls.console.print("Use `ai --show-role <name>` to view a role.", style="dim")
-
-    def list_roles(self) -> list:
-        """List all available roles info"""
-        roles_list = []
-        for role_id, role in sorted(self.roles.items()):
-            roles_list.append(
-                {
-                    "id": role_id,
-                    "name": role.name,
-                    "prompt": role.prompt,
-                    "is_default": role_id in DEFAULT_ROLES,
-                    "filepath": role.filepath,
-                }
-            )
-        return roles_list
-
-    @classmethod
-    @option_callback
-    def show_role_option(cls, name: str):
-        """Show a role's prompt.
-        This method is a cli option callback.
-        """
-        self = cls()
-        role = self.get_role(name)
-        if not role:
-            self.console.print(f"Role '{name}' does not exist", style="red")
-            return
-        self.console.print(role.prompt)
-
-    def get_role(self, role_id: str) -> Optional[Role]:
-        """Get role by ID"""
-        return self.roles.get(role_id)
-
-    @classmethod
-    def check_id_ok(cls, param: typer.CallbackParam, role_id: str):
-        """Check if role exists by ID.
-        This method is a cli option callback.
-        If role does not exist, exit with error.
-        """
-        if not role_id or role_id == param.default:
-            return role_id
-        self = cls()
-        if not self.role_exists(role_id):
-            self.console.print(f"Role '{role_id}' does not exist", style="red")
-            raise typer.Abort()
-        return role_id
-
-    def role_exists(self, role_id: str) -> bool:
-        """Check if role exists"""
-        return role_id in self.roles
-
-    def save_role(self, role_id: str, role: Role) -> None:
-        """Save role configuration"""
-        try:
-            self.roles[role_id] = role
-            role_file = self.roles_dir / f"{role_id}.json"
-            role.filepath = str(role_file)
-            with role_file.open("w", encoding="utf-8") as f:
-                json.dump(role.to_dict(), f, indent=2)
-        except Exception as e:
-            raise RoleCreationError(f"RoleCreationError {e}") from e
-
-    @classmethod
-    @option_callback
-    def create_role_option(cls, name: str):
-        """Create a new role and save it to file.
-        This method is a cli option callback.
-        """
-        self = cls()
-        if name in self.roles:
-            self.console.print(f"Role '{name}' already exists", style="yellow")
-            return
-        description = Prompt.ask("Enter role description")
-
-        # Format the prompt as "You are {role_id}, {description}"
-        prompt = f"You are {name}, {description}"
-
-        role = Role(name=name, prompt=prompt)
-        self.create_role(name, role)
-        self.console.print(f"Role '{name}' created successfully", style="green")
-
-    def create_role(self, role_id: str, role: Union[Role, Dict[str, Any]]) -> None:
-        """Create a new role and save it to file"""
-        if role_id in self.roles:
-            raise RoleAlreadyExistsError(f"Role '{role_id}' already exists")
-        if isinstance(role, dict):
-            if "name" not in role or "prompt" not in role:
-                raise RoleCreationError("Role must have 'name' and 'prompt' keys")
-            # Convert dict to Role object
-            role = Role.from_dict(role_id, role)
-        self.save_role(role_id, role)
-
-    @classmethod
-    @option_callback
-    def delete_role_option(cls, name: str):
-        """Delete a role and its file.
-        This method is a cli option callback.
-        """
-        self = cls()
-        if self.delete_role(name):
-            self.console.print(f"Role '{name}' deleted successfully", style="green")
-
-    def delete_role(self, role_id: str) -> bool:
-        """Delete a role and its file"""
-        if role_id not in self.roles:
-            self.console.print(f"Role '{role_id}' does not exist", style="red")
-            return False
-
-        try:
-            role = self.roles[role_id]
-            if role.filepath:
-                Path(role.filepath).unlink(missing_ok=True)
-            del self.roles[role_id]
-            return True
-        except Exception as e:
-            self.console.print(f"Error deleting role: {e}", style="red")
-            return False
-
-    def get_system_prompt(self, role_id: str) -> str:
-        """Get prompt from file by role ID"""
-        role = self.get_role(role_id)
-        if not role:
-            # Fall back to default role if specified role doesn't exist
-            self.console.print(f"Role {role_id} not found, using default role", style="yellow")
-            role = self.get_role(DefaultRoleNames.DEFAULT)
-            if not role:
-                # Last resort fallback
-                default_config = DEFAULT_ROLES[DefaultRoleNames.DEFAULT]
-                role = Role.from_dict(DefaultRoleNames.DEFAULT, default_config)
-
-        # Create a copy of the role with system variables
-        system_role = Role(name=role.name, prompt=role._prompt)
-        return system_role.prompt

yaicli-0.4.0.dist-info/RECORD

@@ -1,23 +0,0 @@
-pyproject.toml,sha256=qBb73j9dLZ13fGtQDN81nxfqfpUVs5HNDq5hyfwxOtQ,1540
-yaicli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-yaicli/chat_manager.py,sha256=I7BAMz91FLYT6x69wbomtAGLx0WsoTwS4Wo0MgP6P9I,10644
-yaicli/cli.py,sha256=kqUXSDQej4lMefI2QQpM-Do01Be6CUphh65_Nu7sDck,22948
-yaicli/config.py,sha256=nfnrMz_Q78oHqDVvWSrQRGFyBl-q5MScvnYlQRWDQ_g,6515
-yaicli/console.py,sha256=291F4hGksJtxYpg_mehepCIJ-eB2MaDNIyv1JAMgJ1Y,1985
-yaicli/const.py,sha256=WjNzJAP37XizJYyd-LBTWxZAfJ2Nj7mj-eECPMOr3Do,6927
-yaicli/entry.py,sha256=Yp0Z--x-7dowrz-h8hJJ4_BoCzuDjS11NcM8YgFzUoY,7460
-yaicli/exceptions.py,sha256=ndedSdE0uaxxHrWN944BkbhMfRMSMxGDfmqmCKCGJco,924
-yaicli/history.py,sha256=s-57X9FMsaQHF7XySq1gGH_jpd_cHHTYafYu2ECuG6M,2472
-yaicli/printer.py,sha256=nXpralD5qZJQga3OTdEPhj22g7UoF-4mJbZeOtWXojo,12430
-yaicli/render.py,sha256=mB1OT9859_PTwI9f-KY802lPaeQXKRw6ls_5jN21jWc,511
-yaicli/roles.py,sha256=LdPqpdBKQr_eX7PV7iwRtFT0_trmWXVvZUmb3OMhK0w,10582
-yaicli/utils.py,sha256=MLvb-C5n19AD9Z1nW4Z3Z43ZKNH8STxQmNDnL7mq26E,4490
-yaicli/providers/__init__.py,sha256=zYXcnV8HdNeLEPRJEeSgSboNuphkdA1PrbB8MkDErAk,1243
-yaicli/providers/base.py,sha256=Q6QoxrIVqDYcUp1-FY5O-LT33rKSb1jeXGnrbd4Xlog,1870
-yaicli/providers/cohere.py,sha256=m_HIHm6NEd1VG4wCmDKYn3KHo2jzC98FsfxRzjWR8-M,5411
-yaicli/providers/openai.py,sha256=CKNN3wDsnCEOwZAMrARi43-AV65WgJ7AvHONgvpLZw8,8541
-yaicli-0.4.0.dist-info/METADATA,sha256=oeusTm4BZ1Dvow0IF_c2urp_onnV41OgE7QzwcuP09c,43917
-yaicli-0.4.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-yaicli-0.4.0.dist-info/entry_points.txt,sha256=iMhGm3btBaqrknQoF6WCg5sdx69ZyNSC73tRpCcbcLw,63
-yaicli-0.4.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-yaicli-0.4.0.dist-info/RECORD,,