janito 2.3.0__py3-none-any.whl → 2.3.1__py3-none-any.whl

janito/llm/auth.py

@@ -1,63 +1,63 @@
-"""
-LLMAuthManager: Handles authentication credentials for LLM providers, persisted in ~/.janito/auth.json or a custom path.
-"""
-
-import os
-import json
-from typing import Dict, Optional
-
-
-class LLMAuthManager:
-    """
-    Manages authentication tokens, API keys, or credentials for LLM providers.
-    Persists credentials in ~/.janito/auth.json or a custom path.
-    """
-
-    def __init__(self, auth_file: Optional[str] = None):
-        if auth_file is not None:
-            self._auth_file = os.path.expanduser(auth_file)
-        else:
-            self._auth_file = os.path.expanduser("~/.janito/auth.json")
-        self._credentials: Dict[str, str] = {}
-        self._load_credentials()
-
-    def _load_credentials(self):
-        if os.path.exists(self._auth_file):
-            try:
-                with open(self._auth_file, "r") as f:
-                    self._credentials = json.load(f)
-            except Exception:
-                self._credentials = {}
-        else:
-            self._credentials = {}
-
-    def _save_credentials(self):
-        os.makedirs(os.path.dirname(self._auth_file), exist_ok=True)
-        with open(self._auth_file, "w") as f:
-            json.dump(self._credentials, f, indent=2)
-            f.write("\n")
-
-    def set_credentials(self, provider_name: str, credentials: str) -> None:
-        """
-        Store credentials for a given provider and persist to disk. Raises ValueError if provider is unknown.
-        """
-        from janito.providers.registry import LLMProviderRegistry
-
-        if provider_name not in LLMProviderRegistry.list_providers():
-            raise ValueError(f"Unknown provider: {provider_name}")
-        self._credentials[provider_name] = credentials
-        self._save_credentials()
-
-    def get_credentials(self, provider_name: str) -> Optional[str]:
-        """
-        Retrieve credentials for a given provider.
-        """
-        return self._credentials.get(provider_name)
-
-    def remove_credentials(self, provider_name: str) -> None:
-        """
-        Remove credentials for a given provider and update disk.
-        """
-        if provider_name in self._credentials:
-            del self._credentials[provider_name]
-            self._save_credentials()
+"""
+LLMAuthManager: Handles authentication credentials for LLM providers, persisted in ~/.janito/auth.json or a custom path.
+"""
+
+import os
+import json
+from typing import Dict, Optional
+
+
+class LLMAuthManager:
+    """
+    Manages authentication tokens, API keys, or credentials for LLM providers.
+    Persists credentials in ~/.janito/auth.json or a custom path.
+    """
+
+    def __init__(self, auth_file: Optional[str] = None):
+        if auth_file is not None:
+            self._auth_file = os.path.expanduser(auth_file)
+        else:
+            self._auth_file = os.path.expanduser("~/.janito/auth.json")
+        self._credentials: Dict[str, str] = {}
+        self._load_credentials()
+
+    def _load_credentials(self):
+        if os.path.exists(self._auth_file):
+            try:
+                with open(self._auth_file, "r") as f:
+                    self._credentials = json.load(f)
+            except Exception:
+                self._credentials = {}
+        else:
+            self._credentials = {}
+
+    def _save_credentials(self):
+        os.makedirs(os.path.dirname(self._auth_file), exist_ok=True)
+        with open(self._auth_file, "w") as f:
+            json.dump(self._credentials, f, indent=2)
+            f.write("\n")
+
+    def set_credentials(self, provider_name: str, credentials: str) -> None:
+        """
+        Store credentials for a given provider and persist to disk. Raises ValueError if provider is unknown.
+        """
+        from janito.providers.registry import LLMProviderRegistry
+
+        if provider_name not in LLMProviderRegistry.list_providers():
+            raise ValueError(f"Unknown provider: {provider_name}")
+        self._credentials[provider_name] = credentials
+        self._save_credentials()
+
+    def get_credentials(self, provider_name: str) -> Optional[str]:
+        """
+        Retrieve credentials for a given provider.
+        """
+        return self._credentials.get(provider_name)
+
+    def remove_credentials(self, provider_name: str) -> None:
+        """
+        Remove credentials for a given provider and update disk.
+        """
+        if provider_name in self._credentials:
+            del self._credentials[provider_name]
+            self._save_credentials()

janito/llm/driver_config_builder.py

@@ -1,34 +1,34 @@
-from typing import Type, Dict, Any
-from janito.llm.driver_config import LLMDriverConfig
-
-
-def build_llm_driver_config(
-    config: Dict[str, Any], driver_class: Type
-) -> LLMDriverConfig:
-    """
-    Build an LLMDriverConfig instance for the given driver class based on its declared driver_fields.
-    Only fills fields missing from given config; does not overwrite fields already provided.
-    Any config fields not in driver_fields or LLMDriverConfig fields go into .extra.
-    """
-    driver_fields = getattr(driver_class, "driver_fields", None)
-    if driver_fields is None:
-        driver_fields = set(LLMDriverConfig.__dataclass_fields__.keys()) - {
-            "model",
-            "extra",
-        }
-    base_info = {}
-    extra = {}
-    for k, v in (config or {}).items():
-        if k in driver_fields and k in LLMDriverConfig.__dataclass_fields__:
-            base_info[k] = v
-        else:
-            extra[k] = v
-    # Only set missing fields, do NOT overwrite those from CLI/user
-    for field in driver_fields:
-        if field not in base_info and field in LLMDriverConfig.__dataclass_fields__:
-            base_info[field] = (
-                None  # Optional: replace None with provider/driver default if wanted
-            )
-    return LLMDriverConfig(
-        model=config.get("model") or config.get("model_name"), extra=extra, **base_info
-    )
+from typing import Type, Dict, Any
+from janito.llm.driver_config import LLMDriverConfig
+
+
+def build_llm_driver_config(
+    config: Dict[str, Any], driver_class: Type
+) -> LLMDriverConfig:
+    """
+    Build an LLMDriverConfig instance for the given driver class based on its declared driver_fields.
+    Only fills fields missing from given config; does not overwrite fields already provided.
+    Any config fields not in driver_fields or LLMDriverConfig fields go into .extra.
+    """
+    driver_fields = getattr(driver_class, "driver_fields", None)
+    if driver_fields is None:
+        driver_fields = set(LLMDriverConfig.__dataclass_fields__.keys()) - {
+            "model",
+            "extra",
+        }
+    base_info = {}
+    extra = {}
+    for k, v in (config or {}).items():
+        if k in driver_fields and k in LLMDriverConfig.__dataclass_fields__:
+            base_info[k] = v
+        else:
+            extra[k] = v
+    # Only set missing fields, do NOT overwrite those from CLI/user
+    for field in driver_fields:
+        if field not in base_info and field in LLMDriverConfig.__dataclass_fields__:
+            base_info[field] = (
+                None  # Optional: replace None with provider/driver default if wanted
+            )
+    return LLMDriverConfig(
+        model=config.get("model") or config.get("model_name"), extra=extra, **base_info
+    )

janito/llm/driver_input.py

@@ -1,12 +1,12 @@
-from dataclasses import dataclass, field
-from typing import Optional
-import threading
-from janito.llm.driver_config import LLMDriverConfig
-from janito.conversation_history import LLMConversationHistory
-
-
-@dataclass
-class DriverInput:
-    config: LLMDriverConfig
-    conversation_history: LLMConversationHistory
-    cancel_event: Optional[threading.Event] = field(default=None)
+from dataclasses import dataclass, field
+from typing import Optional
+import threading
+from janito.llm.driver_config import LLMDriverConfig
+from janito.conversation_history import LLMConversationHistory
+
+
+@dataclass
+class DriverInput:
+    config: LLMDriverConfig
+    conversation_history: LLMConversationHistory
+    cancel_event: Optional[threading.Event] = field(default=None)

janito/llm/message_parts.py

@@ -1,60 +1,60 @@
-import attr
-
-
-class MessagePart:
-    """
-    Base class for all driver message parts.
-    """
-
-    type: str
-
-
-@attr.s(auto_attribs=True, kw_only=True)
-class TextMessagePart(MessagePart):
-    content: str = ""
-
-
-@attr.s(auto_attribs=True, kw_only=True)
-class InlineDataMessagePart(MessagePart):
-    content: bytes = b""
-
-
-@attr.s(auto_attribs=True, kw_only=True)
-class FileDataMessagePart(MessagePart):
-    content: str = ""
-
-
-@attr.s(auto_attribs=True, kw_only=True)
-class VideoMetadataMessagePart(MessagePart):
-    content: dict = attr.Factory(dict)
-
-
-@attr.s(auto_attribs=True, kw_only=True)
-class CodeExecutionResultMessagePart(MessagePart):
-    content: str = ""
-    stdout: str = ""
-    stderr: str = ""
-
-
-@attr.s(auto_attribs=True, kw_only=True)
-class ExecutableCodeMessagePart(MessagePart):
-    content: str = ""
-
-
-@attr.s(auto_attribs=True, kw_only=True)
-class FunctionCallMessagePart(MessagePart):
-    tool_call_id: str = ""
-    function: object = (
-        None  # Should match OpenAI SDK structure (with arguments as JSON string)
-    )
-
-
-@attr.s(auto_attribs=True, kw_only=True)
-class FunctionResponseMessagePart(MessagePart):
-    name: str = ""
-    content: dict = attr.Factory(dict)
-
-
-@attr.s(auto_attribs=True, kw_only=True)
-class ThoughtMessagePart(MessagePart):
-    content: bool = False
+import attr
+
+
+class MessagePart:
+    """
+    Base class for all driver message parts.
+    """
+
+    type: str
+
+
+@attr.s(auto_attribs=True, kw_only=True)
+class TextMessagePart(MessagePart):
+    content: str = ""
+
+
+@attr.s(auto_attribs=True, kw_only=True)
+class InlineDataMessagePart(MessagePart):
+    content: bytes = b""
+
+
+@attr.s(auto_attribs=True, kw_only=True)
+class FileDataMessagePart(MessagePart):
+    content: str = ""
+
+
+@attr.s(auto_attribs=True, kw_only=True)
+class VideoMetadataMessagePart(MessagePart):
+    content: dict = attr.Factory(dict)
+
+
+@attr.s(auto_attribs=True, kw_only=True)
+class CodeExecutionResultMessagePart(MessagePart):
+    content: str = ""
+    stdout: str = ""
+    stderr: str = ""
+
+
+@attr.s(auto_attribs=True, kw_only=True)
+class ExecutableCodeMessagePart(MessagePart):
+    content: str = ""
+
+
+@attr.s(auto_attribs=True, kw_only=True)
+class FunctionCallMessagePart(MessagePart):
+    tool_call_id: str = ""
+    function: object = (
+        None  # Should match OpenAI SDK structure (with arguments as JSON string)
+    )
+
+
+@attr.s(auto_attribs=True, kw_only=True)
+class FunctionResponseMessagePart(MessagePart):
+    name: str = ""
+    content: dict = attr.Factory(dict)
+
+
+@attr.s(auto_attribs=True, kw_only=True)
+class ThoughtMessagePart(MessagePart):
+    content: bool = False

janito/llm/model.py

@@ -1,38 +1,38 @@
-from dataclasses import dataclass, field
-from typing import Any, Optional
-
-
-@dataclass
-class LLMModelInfo:
-    name: str
-    context: Any = "N/A"
-    max_input: Any = "N/A"
-    max_cot: Any = "N/A"
-    max_response: Any = "N/A"
-    thinking_supported: Any = "N/A"
-    default_temp: float = 0.2
-    open: Optional[Any] = None
-    category: Optional[str] = None
-    driver: Optional[str] = None
-    # This enables arbitrary provider-specific metadata
-    other: dict = field(default_factory=dict)
-
-    def to_dict(self) -> dict:
-        d = self.__dict__.copy()
-        if not self.open:
-            d.pop("open")
-        if not self.category:
-            d.pop("category")
-        if not self.driver:
-            d.pop("driver")
-        if not self.other:
-            d.pop("other")
-        return d
-
-    @staticmethod
-    def get_model_info(model_specs):
-        """
-        Standard get_model_info implementation for all providers:
-        returns a list of model info dicts, one per model in the given MODEL_SPECS dict.
-        """
-        return [m.to_dict() for m in model_specs.values()]
+from dataclasses import dataclass, field
+from typing import Any, Optional
+
+
+@dataclass
+class LLMModelInfo:
+    name: str
+    context: Any = "N/A"
+    max_input: Any = "N/A"
+    max_cot: Any = "N/A"
+    max_response: Any = "N/A"
+    thinking_supported: Any = "N/A"
+    default_temp: float = 0.2
+    open: Optional[Any] = None
+    category: Optional[str] = None
+    driver: Optional[str] = None
+    # This enables arbitrary provider-specific metadata
+    other: dict = field(default_factory=dict)
+
+    def to_dict(self) -> dict:
+        d = self.__dict__.copy()
+        if not self.open:
+            d.pop("open")
+        if not self.category:
+            d.pop("category")
+        if not self.driver:
+            d.pop("driver")
+        if not self.other:
+            d.pop("other")
+        return d
+
+    @staticmethod
+    def get_model_info(model_specs):
+        """
+        Standard get_model_info implementation for all providers:
+        returns a list of model info dicts, one per model in the given MODEL_SPECS dict.
+        """
+        return [m.to_dict() for m in model_specs.values()]