janito 2.6.0__py3-none-any.whl → 2.6.1__py3-none-any.whl

janito/__init__.py

@@ -1,6 +1,7 @@
-"""
-janito: A Python package for managing and interacting with Large Language Model (LLM) providers and their APIs.
-Provides a CLI for credential management and an extensible driver system for LLMs.
-"""
-
-from ._version import __version__
+"""
+janito: A Python package for managing and interacting with Large Language Model (LLM) providers and their APIs.
+Provides a CLI for credential management and an extensible driver system for LLMs.
+"""
+
+from ._version import __version__
+

janito/__main__.py

@@ -1,4 +1,5 @@
-from .cli.main import main
-
-if __name__ == "__main__":
-    main()
+from .cli.main import main
+
+if __name__ == "__main__":
+    main()
+

janito/_version.py

@@ -1,57 +1,58 @@
-"""Version handling for Janito.
-Attempts to obtain the package version in the following order:
-1. If a janito.version module exists (generated when the package is built with
-   setuptools-scm), use the version attribute from that module.
-2. Ask importlib.metadata for the installed distribution version – works for
-   both regular and editable installs handled by pip.
-3. Fall back to calling setuptools_scm.get_version() directly, using the git
-   repository when running from source without an installed distribution.
-4. If everything else fails, return the literal string ``"unknown"`` so that
-   the application continues to work even when the version cannot be
-   determined.
-
-This layered approach guarantees that a meaningful version string is returned
-in most development and production scenarios while keeping Janito free from
-hard-coded version numbers.
-"""
-
-from __future__ import annotations
-
-import pathlib
-from importlib import metadata as importlib_metadata
-
-__all__ = ["__version__"]
-
-
-# 1. "janito.version" (generated at build time by setuptools-scm)
-try:
-    from . import version as _generated_version  # type: ignore
-
-    __version__: str = _generated_version.version  # pytype: disable=module-attr
-except ImportError:  # pragma: no cover – not available in editable installs
-
-    def _resolve_version() -> str:
-        """Resolve the version string using several fallbacks."""
-
-        # 2. importlib.metadata – works for both regular and `pip install -e`.
-        try:
-            return importlib_metadata.version("janito")
-        except importlib_metadata.PackageNotFoundError:
-            pass  # Not installed – probably running from a source checkout.
-
-        # 3. setuptools_scm – query the VCS metadata directly.
-        try:
-            from setuptools_scm import get_version  # Imported lazily.
-
-            package_root = pathlib.Path(__file__).resolve().parent.parent
-            return get_version(root=str(package_root), relative_to=__file__)
-        except Exception:  # pragma: no cover – any failure here falls through
-            # Either setuptools_scm is not available or this is not a git repo.
-            pass
-
-        # 4. Ultimate fallback – return a placeholder.
-        return "unknown"
-
-    __version__ = _resolve_version()
-
-
+"""Version handling for Janito.
+Attempts to obtain the package version in the following order:
+1. If a janito.version module exists (generated when the package is built with
+   setuptools-scm), use the version attribute from that module.
+2. Ask importlib.metadata for the installed distribution version – works for
+   both regular and editable installs handled by pip.
+3. Fall back to calling setuptools_scm.get_version() directly, using the git
+   repository when running from source without an installed distribution.
+4. If everything else fails, return the literal string ``"unknown"`` so that
+   the application continues to work even when the version cannot be
+   determined.
+
+This layered approach guarantees that a meaningful version string is returned
+in most development and production scenarios while keeping Janito free from
+hard-coded version numbers.
+"""
+
+from __future__ import annotations
+
+import pathlib
+from importlib import metadata as importlib_metadata
+
+__all__ = ["__version__"]
+
+
+# 1. "janito.version" (generated at build time by setuptools-scm)
+try:
+    from . import version as _generated_version  # type: ignore
+
+    __version__: str = _generated_version.version  # pytype: disable=module-attr
+except ImportError:  # pragma: no cover – not available in editable installs
+
+    def _resolve_version() -> str:
+        """Resolve the version string using several fallbacks."""
+
+        # 2. importlib.metadata – works for both regular and `pip install -e`.
+        try:
+            return importlib_metadata.version("janito")
+        except importlib_metadata.PackageNotFoundError:
+            pass  # Not installed – probably running from a source checkout.
+
+        # 3. setuptools_scm – query the VCS metadata directly.
+        try:
+            from setuptools_scm import get_version  # Imported lazily.
+
+            package_root = pathlib.Path(__file__).resolve().parent.parent
+            return get_version(root=str(package_root), relative_to=__file__)
+        except Exception:  # pragma: no cover – any failure here falls through
+            # Either setuptools_scm is not available or this is not a git repo.
+            pass
+
+        # 4. Ultimate fallback – return a placeholder.
+        return "unknown"
+
+    __version__ = _resolve_version()
+
+
+

janito/agent/setup_agent.py

@@ -1,13 +1,15 @@
 import importlib.resources
 import re
+import os
 import sys
 import time
 import warnings
 import threading
 from pathlib import Path
 from jinja2 import Template
-from rich import print as rich_print
+from pathlib import Path
 from queue import Queue
+from rich import print as rich_print
 from janito.tools import get_local_tools_adapter
 from janito.llm.agent import LLMAgent
 from janito.drivers.driver_registry import get_driver_class
@@ -15,11 +17,13 @@ from janito.platform_discovery import PlatformDiscovery
 from janito.tools.tool_base import ToolPermissions
 from janito.tools.permissions import get_global_allowed_permissions
 
-
 def _load_template_content(profile, templates_dir):
     """
     Loads the template content for the given profile from the specified directory or package resources.
+    If the profile template is not found in the default locations, tries to load from the user profiles directory ~/.janito/profiles.
     """
+
+
     template_filename = f"system_prompt_template_{profile}.txt.j2"
     template_path = templates_dir / template_filename
     if template_path.exists():
@@ -32,8 +36,14 @@ def _load_template_content(profile, templates_dir):
         ).open("r", encoding="utf-8") as file:
             return file.read(), template_path
     except (FileNotFoundError, ModuleNotFoundError, AttributeError):
+        # Try user profiles directory
+        user_profiles_dir = Path(os.path.expanduser("~/.janito/profiles"))
+        user_template_path = user_profiles_dir / profile
+        if user_template_path.exists():
+            with open(user_template_path, "r", encoding="utf-8") as file:
+                return file.read(), user_template_path
         raise FileNotFoundError(
-            f"[janito] Could not find profile-specific template '{template_filename}' in {template_path} nor in janito.agent.templates.profiles package."
+            f"[janito] Could not find profile-specific template '{template_filename}' in {template_path} nor in janito.agent.templates.profiles package nor in user profiles directory {user_template_path}."
         )
 
 

janito/cli/__init__.py

@@ -1,9 +1,10 @@
-# janito.cli package
-from .utils import format_tokens, format_generation_time
-
-__all__ = [
-    "setup_provider",
-    "setup_agent",
-    "format_tokens",
-    "format_generation_time",
-]
+# janito.cli package
+from .utils import format_tokens, format_generation_time
+
+__all__ = [
+    "setup_provider",
+    "setup_agent",
+    "format_tokens",
+    "format_generation_time",
+]
+

janito/cli/chat_mode/bindings.py

@@ -1,37 +1,38 @@
-"""
-Key bindings for Janito Chat CLI.
-"""
-
-from prompt_toolkit.key_binding import KeyBindings
-from janito.tools.permissions import get_global_allowed_permissions
-
-class KeyBindingsFactory:
-    @staticmethod
-    def create():
-        bindings = KeyBindings()
-
-        @bindings.add("c-y")
-        def _(event):
-            buf = event.app.current_buffer
-            buf.text = "Yes"
-            buf.validate_and_handle()
-
-        @bindings.add("c-n")
-        def _(event):
-            buf = event.app.current_buffer
-            buf.text = "No"
-            buf.validate_and_handle()
-
-        @bindings.add("f2")
-        def _(event):
-            buf = event.app.current_buffer
-            buf.text = "/restart"
-            buf.validate_and_handle()
-
-        @bindings.add("f12")
-        def _(event):
-            buf = event.app.current_buffer
-            buf.text = "Do It"
-            buf.validate_and_handle()
-
-        return bindings
+"""
+Key bindings for Janito Chat CLI.
+"""
+
+from prompt_toolkit.key_binding import KeyBindings
+from janito.tools.permissions import get_global_allowed_permissions
+
+class KeyBindingsFactory:
+    @staticmethod
+    def create():
+        bindings = KeyBindings()
+
+        @bindings.add("c-y")
+        def _(event):
+            buf = event.app.current_buffer
+            buf.text = "Yes"
+            buf.validate_and_handle()
+
+        @bindings.add("c-n")
+        def _(event):
+            buf = event.app.current_buffer
+            buf.text = "No"
+            buf.validate_and_handle()
+
+        @bindings.add("f2")
+        def _(event):
+            buf = event.app.current_buffer
+            buf.text = "/restart"
+            buf.validate_and_handle()
+
+        @bindings.add("f12")
+        def _(event):
+            buf = event.app.current_buffer
+            buf.text = "Do It"
+            buf.validate_and_handle()
+
+        return bindings
+

janito/cli/chat_mode/chat_entry.py

@@ -1,22 +1,23 @@
-"""
-Main entry point for the Janito Chat CLI.
-Handles the interactive chat loop and session startup.
-"""
-
-from rich.console import Console
-from prompt_toolkit.formatted_text import HTML
-from janito.cli.chat_mode.session import ChatSession
-
-
-def main(args=None):
-    console = Console()
-    console.clear()
-    from janito.version import __version__
-
-
-    session = ChatSession(console, args=args)
-    session.run()
-
-
-if __name__ == "__main__":
-    main()
+"""
+Main entry point for the Janito Chat CLI.
+Handles the interactive chat loop and session startup.
+"""
+
+from rich.console import Console
+from prompt_toolkit.formatted_text import HTML
+from janito.cli.chat_mode.session import ChatSession
+
+
+def main(args=None):
+    console = Console()
+    console.clear()
+    from janito.version import __version__
+
+
+    session = ChatSession(console, args=args)
+    session.run()
+
+
+if __name__ == "__main__":
+    main()
+

janito/cli/chat_mode/prompt_style.py

@@ -1,24 +1,25 @@
-from prompt_toolkit.styles import Style
-
-chat_shell_style = Style.from_dict(
-    {
-        "prompt": "bg:#2323af #ffffff bold",
-        "": "bg:#005fdd #ffffff",  # blue background for input area
-        "bottom-toolbar": "fg:#2323af bg:yellow",
-        "key-label": "bg:#ff9500 fg:#232323 bold",
-        "provider": "fg:#117fbf",
-        "model": "fg:#1f5fa9",
-        "role": "fg:#e87c32 bold",
-        "msg_count": "fg:#5454dd",
-        "session_id": "fg:#704ab9",
-        "tokens_total": "fg:#a022c7",
-        "tokens_in": "fg:#00af5f",
-        "tokens_out": "fg:#01814a",
-        "max-tokens": "fg:#888888",
-        
-        
-        "key-toggle-on": "bg:#ffd700 fg:#232323 bold",
-        "key-toggle-off": "bg:#444444 fg:#ffffff bold",
-        "cmd-label": "bg:#ff9500 fg:#232323 bold",
-    }
-)
+from prompt_toolkit.styles import Style
+
+chat_shell_style = Style.from_dict(
+    {
+        "prompt": "bg:#2323af #ffffff bold",
+        "": "bg:#005fdd #ffffff",  # blue background for input area
+        "bottom-toolbar": "fg:#2323af bg:yellow",
+        "key-label": "bg:#ff9500 fg:#232323 bold",
+        "provider": "fg:#117fbf",
+        "model": "fg:#1f5fa9",
+        "role": "fg:#e87c32 bold",
+        "msg_count": "fg:#5454dd",
+        "session_id": "fg:#704ab9",
+        "tokens_total": "fg:#a022c7",
+        "tokens_in": "fg:#00af5f",
+        "tokens_out": "fg:#01814a",
+        "max-tokens": "fg:#888888",
+        
+        
+        "key-toggle-on": "bg:#ffd700 fg:#232323 bold",
+        "key-toggle-off": "bg:#444444 fg:#ffffff bold",
+        "cmd-label": "bg:#ff9500 fg:#232323 bold",
+    }
+)
+