janito 2.27.1__py3-none-any.whl → 2.28.0__py3-none-any.whl

janito/providers/alibaba/model_info.py

@@ -12,7 +12,7 @@ MODEL_SPECS = {
         max_cot=8192,
     ),
     "qwen-plus": LLMModelInfo(
-        name="qwen-plus", 
+        name="qwen-plus",
         context=131072,
         max_response=8192,
         category="Alibaba Qwen Plus Model (OpenAI-compatible)",
@@ -23,7 +23,7 @@ MODEL_SPECS = {
     ),
     "qwen-max": LLMModelInfo(
         name="qwen-max",
-        context=32768, 
+        context=32768,
         max_response=8192,
         category="Alibaba Qwen Max Model (OpenAI-compatible)",
         driver="OpenAIModelDriver",
@@ -31,7 +31,6 @@ MODEL_SPECS = {
         thinking=False,
         max_cot=8192,
     ),
-
     "qwen3-coder-plus": LLMModelInfo(
         name="qwen3-coder-plus",
         context=1048576,
@@ -52,7 +51,6 @@ MODEL_SPECS = {
         thinking=False,
         max_cot=65536,
     ),
-    
     # Qwen3 1M context models (July 2025 update)
     "qwen3-235b-a22b-thinking-2507": LLMModelInfo(
         name="qwen3-235b-a22b-thinking-2507",
@@ -94,4 +92,4 @@ MODEL_SPECS = {
         thinking=False,
         max_cot=32768,
     ),
-}
+}

janito/providers/alibaba/provider.py

@@ -17,7 +17,9 @@ class AlibabaProvider(LLMProvider):
     NAME = "alibaba"
     MAINTAINER = "João Pinto <janito@ikignosis.org>"
     MODEL_SPECS = MODEL_SPECS
-    DEFAULT_MODEL = "qwen3-235b-a22b-instruct-2507"  # 129k context, general-purpose model
+    DEFAULT_MODEL = (
+        "qwen3-235b-a22b-instruct-2507"  # 129k context, general-purpose model
+    )
 
     def __init__(
         self, auth_manager: LLMAuthManager = None, config: LLMDriverConfig = None

janito/providers/cerebras/__init__.py

@@ -1 +1 @@
-# Cerebras provider package
+# Cerebras provider package

janito/providers/cerebras/model_info.py

@@ -11,11 +11,8 @@ MODEL_SPECS = {
         driver="CerebrasModelDriver",
         other={
             "description": "Qwen 3 32B model for general instruction following",
-            "pricing": {
-                "input_per_1k_tokens": 0.0002,
-                "output_per_1k_tokens": 0.0006
-            }
-        }
+            "pricing": {"input_per_1k_tokens": 0.0002, "output_per_1k_tokens": 0.0006},
+        },
     ),
     "qwen-3-235b-a22b-instruct-2507": LLMModelInfo(
         name="qwen-3-235b-a22b-instruct-2507",
@@ -25,11 +22,8 @@ MODEL_SPECS = {
         driver="CerebrasModelDriver",
         other={
             "description": "Qwen 3 235B A22B instruction-tuned model (preview)",
-            "pricing": {
-                "input_per_1k_tokens": 0.001,
-                "output_per_1k_tokens": 0.003
-            }
-        }
+            "pricing": {"input_per_1k_tokens": 0.001, "output_per_1k_tokens": 0.003},
+        },
     ),
     "qwen-3-235b-a22b-thinking-2507": LLMModelInfo(
         name="qwen-3-235b-a22b-thinking-2507",
@@ -39,11 +33,8 @@ MODEL_SPECS = {
         driver="CerebrasModelDriver",
         other={
             "description": "Qwen 3 235B A22B thinking model for reasoning tasks (preview)",
-            "pricing": {
-                "input_per_1k_tokens": 0.001,
-                "output_per_1k_tokens": 0.003
-            }
-        }
+            "pricing": {"input_per_1k_tokens": 0.001, "output_per_1k_tokens": 0.003},
+        },
     ),
     "qwen-3-coder-480b": LLMModelInfo(
         name="qwen-3-coder-480b",
@@ -53,11 +44,8 @@ MODEL_SPECS = {
         driver="CerebrasModelDriver",
         other={
             "description": "Qwen 3 Coder 480B model for programming tasks (preview)",
-            "pricing": {
-                "input_per_1k_tokens": 0.002,
-                "output_per_1k_tokens": 0.006
-            }
-        }
+            "pricing": {"input_per_1k_tokens": 0.002, "output_per_1k_tokens": 0.006},
+        },
     ),
     "gpt-oss-120b": LLMModelInfo(
         name="gpt-oss-120b",
@@ -67,10 +55,7 @@ MODEL_SPECS = {
         driver="CerebrasModelDriver",
         other={
             "description": "GPT-OSS 120B open-source model (preview)",
-            "pricing": {
-                "input_per_1k_tokens": 0.0008,
-                "output_per_1k_tokens": 0.0024
-            }
-        }
-    )
-}
+            "pricing": {"input_per_1k_tokens": 0.0008, "output_per_1k_tokens": 0.0024},
+        },
+    ),
+}

janito/providers/cerebras/provider.py

@@ -12,14 +12,16 @@ from .model_info import MODEL_SPECS
 
 class CerebrasProvider(LLMProvider):
     """Cerebras Inference API provider."""
-    
+
     name = "cerebras"
     NAME = "cerebras"
     DEFAULT_MODEL = "qwen-3-coder-480b"
     MAINTAINER = "João Pinto <janito@ikignosis.org>"
     MODEL_SPECS = MODEL_SPECS
 
-    def __init__(self, auth_manager: LLMAuthManager = None, config: LLMDriverConfig = None):
+    def __init__(
+        self, auth_manager: LLMAuthManager = None, config: LLMDriverConfig = None
+    ):
         """Initialize Cerebras provider with optional configuration."""
         super().__init__()
         self._tools_adapter = get_local_tools_adapter()
@@ -32,10 +34,10 @@ class CerebrasProvider(LLMProvider):
         self._initialize_config(auth_manager, config)
         self._setup_model_config()
         self.fill_missing_device_info(self._driver_config)
-        
+
         if not self.available:
             return
-            
+
         self._initialize_config(None, None)
         self._driver_config.base_url = "https://api.cerebras.ai/v1"
 
@@ -72,11 +74,11 @@ class CerebrasProvider(LLMProvider):
             # Set context length
             if hasattr(model_spec, "context") and model_spec.context:
                 self._driver_config.context_length = model_spec.context
-            
+
             # Set max tokens based on model spec
             if hasattr(model_spec, "max_response") and model_spec.max_response:
                 self._driver_config.max_tokens = model_spec.max_response
-            
+
             # Set max completion tokens if thinking is supported
             if getattr(model_spec, "thinking_supported", False):
                 max_cot = getattr(model_spec, "max_cot", None)
@@ -130,10 +132,10 @@ class CerebrasProvider(LLMProvider):
                 name: model_info.to_dict()
                 for name, model_info in self.MODEL_SPECS.items()
             }
-        
+
         if model_name in self.MODEL_SPECS:
             return self.MODEL_SPECS[model_name].to_dict()
-        
+
         return None
 
     def execute_tool(self, tool_name: str, event_bus, *args, **kwargs):
@@ -142,4 +144,4 @@ class CerebrasProvider(LLMProvider):
 
 
 # Register the provider
-LLMProviderRegistry.register(CerebrasProvider.name, CerebrasProvider)
+LLMProviderRegistry.register(CerebrasProvider.name, CerebrasProvider)

janito/providers/mistral/__init__.py

@@ -1 +1 @@
-# Codestral provider module
+# Codestral provider module

janito/providers/mistral/model_info.py

@@ -78,4 +78,4 @@ MODEL_SPECS = {
         open="mistral",
         driver="OpenAIModelDriver",
     ),
-}
+}

janito/providers/mistral/provider.py

@@ -121,4 +121,4 @@ class MistralProvider(LLMProvider):
         return self._tools_adapter.execute_by_name(tool_name, *args, **kwargs)
 
 
-LLMProviderRegistry.register(MistralProvider.NAME, MistralProvider)
+LLMProviderRegistry.register(MistralProvider.NAME, MistralProvider)

janito/providers/moonshot/__init__.py

@@ -0,0 +1 @@
+# Moonshot provider package

janito/providers/{moonshotai → moonshot}/model_info.py

@@ -1,6 +1,6 @@
 from janito.llm.model import LLMModelInfo
 
-MOONSHOTAI_MODEL_SPECS = {
+MOONSHOT_MODEL_SPECS = {
     "kimi-k2-0711-preview": LLMModelInfo(
         name="kimi-k2-0711-preview",
         context=128000,
@@ -9,7 +9,7 @@ MOONSHOTAI_MODEL_SPECS = {
         max_response=4096,
         thinking_supported=False,
         default_temp=0.2,
-        open="moonshotai",
+        open="moonshot",
         driver="OpenAIModelDriver",
     ),
     "kimi-k2-turbo-preview": LLMModelInfo(
@@ -20,7 +20,7 @@ MOONSHOTAI_MODEL_SPECS = {
         max_response=4096,
         thinking_supported=False,
         default_temp=0.2,
-        open="moonshotai",
+        open="moonshot",
         driver="OpenAIModelDriver",
     ),
 }

janito/providers/{moonshotai → moonshot}/provider.py

@@ -4,14 +4,14 @@ from janito.llm.driver_config import LLMDriverConfig
 from janito.drivers.openai.driver import OpenAIModelDriver
 from janito.tools import get_local_tools_adapter
 from janito.providers.registry import LLMProviderRegistry
-from .model_info import MOONSHOTAI_MODEL_SPECS
+from .model_info import MOONSHOT_MODEL_SPECS
 
 
-class MoonshotAIProvider(LLMProvider):
-    name = "moonshotai"
-    NAME = "moonshotai"
+class MoonshotProvider(LLMProvider):
+    name = "moonshot"
+    NAME = "moonshot"
     MAINTAINER = "João Pinto <janito@ikignosis.org>"
-    MODEL_SPECS = MOONSHOTAI_MODEL_SPECS
+    MODEL_SPECS = MOONSHOT_MODEL_SPECS
     DEFAULT_MODEL = "kimi-k2-turbo-preview"
 
     def __init__(
@@ -34,7 +34,7 @@ class MoonshotAIProvider(LLMProvider):
         if not self._api_key:
             from janito.llm.auth_utils import handle_missing_api_key
 
-            handle_missing_api_key(self.name, "MOONSHOTAI_API_KEY")
+            handle_missing_api_key(self.name, "MOONSHOT_API_KEY")
 
         self._driver_config = config or LLMDriverConfig(model=None)
         if not self._driver_config.model:
@@ -69,7 +69,7 @@ class MoonshotAIProvider(LLMProvider):
     def driver(self) -> OpenAIModelDriver:
         if not self.available:
             raise ImportError(
-                f"MoonshotAIProvider unavailable: {self.unavailable_reason}"
+                f"MoonshotProvider unavailable: {self.unavailable_reason}"
             )
         return self._driver
 
@@ -101,4 +101,4 @@ class MoonshotAIProvider(LLMProvider):
         return self._tools_adapter.execute_by_name(tool_name, *args, **kwargs)
 
 
-LLMProviderRegistry.register(MoonshotAIProvider.NAME, MoonshotAIProvider)
+LLMProviderRegistry.register(MoonshotProvider.NAME, MoonshotProvider)

janito/providers/openai/provider.py

@@ -17,7 +17,9 @@ class OpenAIProvider(LLMProvider):
     NAME = "openai"
     MAINTAINER = "João Pinto <janito@ikignosis.org>"
     MODEL_SPECS = MODEL_SPECS
-    DEFAULT_MODEL = "gpt-5"  # Options: gpt-4.1, gpt-4o, o3-mini, o4-mini, gpt-5, gpt-5-nano
+    DEFAULT_MODEL = (
+        "gpt-4.1"  # Options: gpt-4.1, gpt-4o, o3-mini, o4-mini, gpt-5, gpt-5-nano
+    )
 
     def __init__(
         self, auth_manager: LLMAuthManager = None, config: LLMDriverConfig = None

janito/report_events.py

@@ -14,7 +14,6 @@ class ReportSubtype(Enum):
     PROGRESS = "progress"
 
 
-
 class ReportAction(Enum):
     READ = "READ"
     CREATE = "CREATE"

janito/tools/adapters/local/create_file.py

@@ -25,7 +25,7 @@ class CreateFileTool(ToolBase):
             - "✅ Successfully created the file at ..."
 
     Note: Syntax validation is automatically performed after this operation.
-    
+
     Security: This tool includes loop protection to prevent excessive file creation operations.
     Maximum 5 calls per 10 seconds for the same file path.
     """

janito/tools/adapters/local/fetch_url.py

@@ -68,19 +68,21 @@ class FetchUrlTool(ToolBase):
             {}
         )  # In-memory session cache - lifetime matches tool instance
         self._load_cache()
-        
+
         # Browser-like session with cookies and headers
         self.session = requests.Session()
-        self.session.headers.update({
-            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
-            'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8',
-            'Accept-Language': 'en-US,en;q=0.5',
-            'Accept-Encoding': 'gzip, deflate, br',
-            'DNT': '1',
-            'Connection': 'keep-alive',
-            'Upgrade-Insecure-Requests': '1',
-        })
-        
+        self.session.headers.update(
+            {
+                "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
+                "Accept": "text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8",
+                "Accept-Language": "en-US,en;q=0.5",
+                "Accept-Encoding": "gzip, deflate, br",
+                "DNT": "1",
+                "Connection": "keep-alive",
+                "Upgrade-Insecure-Requests": "1",
+            }
+        )
+
         # Load cookies from disk if they exist
         self.cookies_file = self.cache_dir / "cookies.json"
         self._load_cookies()
@@ -120,12 +122,14 @@ class FetchUrlTool(ToolBase):
         try:
             cookies_data = []
             for cookie in self.session.cookies:
-                cookies_data.append({
-                    'name': cookie.name,
-                    'value': cookie.value,
-                    'domain': cookie.domain,
-                    'path': cookie.path
-                })
+                cookies_data.append(
+                    {
+                        "name": cookie.name,
+                        "value": cookie.value,
+                        "domain": cookie.domain,
+                        "path": cookie.path,
+                    }
+                )
             with open(self.cookies_file, "w", encoding="utf-8") as f:
                 json.dump(cookies_data, f, indent=2)
         except IOError:
@@ -170,8 +174,14 @@ class FetchUrlTool(ToolBase):
         }
         self._save_cache()
 
-    def _fetch_url_content(self, url: str, timeout: int = 10, headers: Optional[Dict[str, str]] = None, 
-                          cookies: Optional[Dict[str, str]] = None, follow_redirects: bool = True) -> str:
+    def _fetch_url_content(
+        self,
+        url: str,
+        timeout: int = 10,
+        headers: Optional[Dict[str, str]] = None,
+        cookies: Optional[Dict[str, str]] = None,
+        follow_redirects: bool = True,
+    ) -> str:
         """Fetch URL content and handle HTTP errors.
 
         Implements two-tier caching:
@@ -224,23 +234,23 @@ class FetchUrlTool(ToolBase):
             request_headers = self.session.headers.copy()
             if headers:
                 request_headers.update(headers)
-            
+
             # Merge custom cookies
             if cookies:
                 self.session.cookies.update(cookies)
 
             response = self.session.get(
-                url, 
-                timeout=timeout, 
+                url,
+                timeout=timeout,
                 headers=request_headers,
-                allow_redirects=follow_redirects
+                allow_redirects=follow_redirects,
             )
             response.raise_for_status()
             content = response.text
-            
+
             # Save cookies after successful request
             self._save_cookies()
-            
+
             # Cache successful responses in session cache
             self.session_cache[url] = content
             return content
@@ -354,8 +364,11 @@ class FetchUrlTool(ToolBase):
         # Check if we should save to file
         if save_to_file:
             html_content = self._fetch_url_content(
-                url, timeout=timeout, headers=headers, cookies=cookies, 
-                follow_redirects=follow_redirects
+                url,
+                timeout=timeout,
+                headers=headers,
+                cookies=cookies,
+                follow_redirects=follow_redirects,
             )
             if html_content.startswith("Warning:"):
                 return html_content
@@ -380,8 +393,11 @@ class FetchUrlTool(ToolBase):
 
         # Normal processing path
         html_content = self._fetch_url_content(
-            url, timeout=timeout, headers=headers, cookies=cookies, 
-            follow_redirects=follow_redirects
+            url,
+            timeout=timeout,
+            headers=headers,
+            cookies=cookies,
+            follow_redirects=follow_redirects,
         )
         if html_content.startswith("Warning:"):
             return html_content

janito/tools/adapters/local/python_command_run.py

@@ -32,7 +32,8 @@ class PythonCommandRunTool(ToolBase):
             return tr("Warning: Empty code provided. Operation skipped.")
         if not silent:
             self.report_action(
-                tr("🐍 Running: python -c ...\n{code}\n", code=code), ReportAction.EXECUTE
+                tr("🐍 Running: python -c ...\n{code}\n", code=code),
+                ReportAction.EXECUTE,
             )
             self.report_stdout("\n")
         else:

janito/tools/adapters/local/python_file_run.py

@@ -28,6 +28,7 @@ class PythonFileRunTool(ToolBase):
 
     def run(self, path: str, timeout: int = 60, silent: bool = False) -> str:
         from janito.tools.path_utils import expand_path
+
         path = expand_path(path)
         if not silent:
             self.report_action(

janito/tools/adapters/local/run_powershell_command.py

@@ -43,7 +43,7 @@ class RunPowershellCommandTool(ToolBase):
         if require_confirmation:
             self.report_warning(
                 tr("⚠️ Confirmation requested, but no handler (auto-confirmed)."),
-                ReportAction.EXECUTE
+                ReportAction.EXECUTE,
             )
             return True  # Auto-confirm for now
         return True

janito/tools/adapters/local/search_text/core.py

@@ -98,7 +98,7 @@ class SearchTextTool(ToolBase):
         if max_depth > 0:
             info_str += tr(" [max_depth={max_depth}]", max_depth=max_depth)
         if count_only:
-            info_str += " [count]"
+            info_str += " [count-only]"
         self.report_action(info_str, ReportAction.READ)
         if os.path.isfile(search_path):
             dir_output, dir_limit_reached, per_file_counts = self._handle_file(

janito/tools/adapters/local/validate_file_syntax/jinja2_validator.py

@@ -8,40 +8,43 @@ def validate_jinja2(path: str) -> str:
     """Validate Jinja2 template syntax."""
     try:
         from jinja2 import Environment, TemplateSyntaxError
-        
+
         with open(path, "r", encoding="utf-8") as f:
             content = f.read()
-        
+
         # Create a Jinja2 environment and try to parse the template
         env = Environment()
         try:
             env.parse(content)
             return tr("✅ Syntax OK")
         except TemplateSyntaxError as e:
-            line_num = getattr(e, 'lineno', 0)
-            return tr("⚠️ Warning: Syntax error: {error} at line {line}", 
-                     error=str(e), line=line_num)
+            line_num = getattr(e, "lineno", 0)
+            return tr(
+                "⚠️ Warning: Syntax error: {error} at line {line}",
+                error=str(e),
+                line=line_num,
+            )
         except Exception as e:
             return tr("⚠️ Warning: Syntax error: {error}", error=str(e))
-            
+
     except ImportError:
         # If jinja2 is not available, just check basic structure
         try:
             with open(path, "r", encoding="utf-8") as f:
                 content = f.read()
-            
+
             # Basic checks for common Jinja2 syntax issues
             open_tags = content.count("{%")
             close_tags = content.count("%}")
             open_vars = content.count("{{")
             close_vars = content.count("}}")
-            
+
             if open_tags != close_tags:
                 return tr("⚠️ Warning: Syntax error: Mismatched Jinja2 tags")
             if open_vars != close_vars:
                 return tr("⚠️ Warning: Syntax error: Mismatched Jinja2 variables")
-                
+
             return tr("✅ Syntax OK (basic validation)")
-            
+
         except Exception as e:
-            return tr("⚠️ Warning: Syntax error: {error}", error=str(e))
+            return tr("⚠️ Warning: Syntax error: {error}", error=str(e))

janito/tools/base.py

@@ -1,11 +1,12 @@
 class BaseTool:
     """Base class for all tools."""
+
     tool_name: str = ""
-    
+
     def __init__(self):
         if not self.tool_name:
             self.tool_name = self.__class__.__name__.lower()
-    
+
     def run(self, *args, **kwargs) -> str:
         """Execute the tool."""
-        raise NotImplementedError
+        raise NotImplementedError