janito 2.12.0__py3-none-any.whl → 2.14.0__py3-none-any.whl

janito/agent/setup_agent.py

@@ -12,7 +12,7 @@ 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
+
 from janito.platform_discovery import PlatformDiscovery
 from janito.tools.tool_base import ToolPermissions
 from janito.tools.permissions import get_global_allowed_permissions

janito/cli/cli_commands/list_drivers.py

@@ -0,0 +1,137 @@
+"""
+CLI Command: List available LLM drivers and their dependencies
+"""
+
+import importlib
+import sys
+from pathlib import Path
+from rich.console import Console
+from rich.table import Table
+from rich.panel import Panel
+from rich.text import Text
+
+console = Console()
+
+
+def get_driver_info():
+    """Get information about all available drivers."""
+    drivers = []
+    
+    # Define known driver modules
+    driver_modules = [
+        ("janito.drivers.openai.driver", "OpenAIModelDriver"),
+        ("janito.drivers.azure_openai.driver", "AzureOpenAIModelDriver"),
+        ("janito.drivers.zai.driver", "ZAIModelDriver"),
+    ]
+    
+    for module_path, class_name in driver_modules:
+        try:
+            # Import the module
+            module = importlib.import_module(module_path)
+            driver_class = getattr(module, class_name)
+            
+            # Get availability info
+            available = getattr(driver_class, 'available', True)
+            unavailable_reason = getattr(driver_class, 'unavailable_reason', None)
+            
+            # Get dependencies from module imports
+            dependencies = []
+            module_file = Path(module.__file__)
+            
+            # Read module file to detect imports
+            with open(module_file, 'r', encoding='utf-8') as f:
+                content = f.read()
+                
+            # Simple dependency detection
+            if 'import openai' in content or 'from openai' in content:
+                dependencies.append('openai')
+            if 'import zai' in content or 'from zai' in content:
+                dependencies.append('zai')
+            if 'import anthropic' in content or 'from anthropic' in content:
+                dependencies.append('anthropic')
+            if 'import google' in content or 'from google' in content:
+                dependencies.append('google-generativeai')
+            
+
+            
+            # Remove duplicates while preserving order
+            seen = set()
+            dependencies = [dep for dep in dependencies if not (dep in seen or seen.add(dep))]
+            
+            # Check if dependencies are available
+            dep_status = []
+            for dep in dependencies:
+                try:
+                    importlib.import_module(dep)
+                    dep_status.append(f"✅ {dep}")
+                except ImportError:
+                    dep_status.append(f"❌ {dep}")
+            
+            if not dependencies:
+                dep_status = ["No external dependencies"]
+            
+            drivers.append({
+                'name': class_name,
+                'available': available,
+                'reason': unavailable_reason,
+                'dependencies': dep_status
+            })
+            
+        except (ImportError, AttributeError) as e:
+            drivers.append({
+                'name': class_name,
+                'module': module_path,
+                'available': False,
+                'reason': str(e),
+                'dependencies': ["❌ Module not found"]
+            })
+    
+    return drivers
+
+
+def handle_list_drivers(args=None):
+    """List all available LLM drivers with their status and dependencies."""
+    drivers = get_driver_info()
+    
+    if not drivers:
+        console.print("[red]No drivers found[/red]")
+        return
+    
+    # Create table
+    table = Table(title="Available LLM Drivers")
+    table.add_column("Driver", style="cyan", no_wrap=True)
+    table.add_column("Status", style="bold")
+    table.add_column("Dependencies", style="yellow")
+    
+    for driver in drivers:
+        name = driver['name']
+        
+        if driver['available']:
+            status = "[green]✅ Available[/green]"
+            if driver['reason']:
+                status = f"[yellow]⚠️ Available ({driver['reason']})[/yellow]"
+        else:
+            status = f"[red]❌ Unavailable[/red]"
+            if driver['reason']:
+                status = f"[red]❌ {driver['reason']}[/red]"
+        
+        deps = "\n".join(driver['dependencies'])
+        
+        table.add_row(name, status, deps)
+    
+    console.print(table)
+    
+    # Installation help
+    # Get unique missing dependencies
+    missing_deps = set()
+    for driver in drivers:
+        for dep_status in driver['dependencies']:
+            if dep_status.startswith('❌'):
+                missing_deps.add(dep_status[2:].strip())
+    
+    if missing_deps:
+        console.print(f"\n[dim]💡 Install missing deps: pip install {' '.join(sorted(missing_deps))}[/dim]")
+
+
+if __name__ == "__main__":
+    handle_list_drivers()

janito/cli/core/getters.py

@@ -7,6 +7,7 @@ from janito.cli.cli_commands.list_models import handle_list_models
 from janito.cli.cli_commands.list_tools import handle_list_tools
 from janito.cli.cli_commands.show_config import handle_show_config
 from janito.cli.cli_commands.list_config import handle_list_config
+from janito.cli.cli_commands.list_drivers import handle_list_drivers
 from functools import partial
 from janito.provider_registry import ProviderRegistry
 
@@ -17,6 +18,7 @@ GETTER_KEYS = [
     "list_models",
     "list_tools",
     "list_config",
+    "list_drivers",
 ]
 
 
@@ -42,7 +44,10 @@ def handle_getter(args, config_mgr=None):
         "list_profiles": partial(handle_list_profiles, args),
         "show_config": partial(handle_show_config, args),
         "list_config": partial(handle_list_config, args),
+        "list_drivers": partial(handle_list_drivers, args),
     }
     for arg in GETTER_KEYS:
         if getattr(args, arg, False) and arg in GETTER_DISPATCH:
-            return GETTER_DISPATCH[arg]()
+            GETTER_DISPATCH[arg]()
+            import sys
+            sys.exit(0)

janito/cli/main_cli.py

@@ -117,6 +117,10 @@ definition = [
         ["--list-providers"],
         {"action": "store_true", "help": "List supported LLM providers"},
     ),
+    (
+        ["--list-drivers"],
+        {"action": "store_true", "help": "List available LLM drivers and their dependencies"},
+    ),
     (
         ["-l", "--list-models"],
         {"action": "store_true", "help": "List all supported models"},
@@ -206,6 +210,7 @@ GETTER_KEYS = [
     "list_models",
     "list_tools",
     "list_config",
+    "list_drivers",
 ]
 
 
@@ -321,14 +326,8 @@ class JanitoCLI:
         if run_mode == RunMode.SET:
             if self._run_set_mode():
                 return
-        # Special handling: provider is not required for list_providers, list_tools, show_config
-        if run_mode == RunMode.GET and (
-            self.args.list_providers
-            or self.args.list_tools
-            or self.args.list_profiles
-            or self.args.show_config
-            or self.args.list_config
-        ):
+        # Special handling: provider is not required for list_providers, list_tools, show_config, list_drivers
+        if run_mode == RunMode.GET:
             self._maybe_print_verbose_provider_model()
             handle_getter(self.args)
             return
@@ -364,8 +363,7 @@ class JanitoCLI:
                 agent_role,
                 verbose_tools=self.args.verbose_tools,
             )
-        elif run_mode == RunMode.GET:
-            handle_getter(self.args)
+
 
     def _run_set_mode(self):
         if handle_api_key_set(self.args):

janito/drivers/zai/driver.py

@@ -12,10 +12,18 @@ from janito.llm.driver_input import DriverInput
 from janito.driver_events import RequestFinished, RequestStatus, RateLimitRetry
 from janito.llm.message_parts import TextMessagePart, FunctionCallMessagePart
 
-import openai
+try:
+    import openai
+    available = True
+    unavailable_reason = None
+except ImportError:
+    available = False
+    unavailable_reason = "openai module not installed"
 
 
 class ZAIModelDriver(LLMDriver):
+    available = available
+    unavailable_reason = unavailable_reason
     def _get_message_from_result(self, result):
         """Extract the message object from the provider result (Z.AI-specific)."""
         if hasattr(result, "choices") and result.choices:
@@ -52,15 +60,15 @@ class ZAIModelDriver(LLMDriver):
         # Z.AI-specific parameters
         if config.model:
             api_kwargs["model"] = config.model
-        # Prefer max_completion_tokens if present, else fallback to max_tokens (for backward compatibility)
-        if (
+        # Use max_tokens for Z.ai SDK compatibility
+        if hasattr(config, "max_tokens") and config.max_tokens is not None:
+            api_kwargs["max_tokens"] = int(config.max_tokens)
+        elif (
             hasattr(config, "max_completion_tokens")
             and config.max_completion_tokens is not None
         ):
-            api_kwargs["max_completion_tokens"] = int(config.max_completion_tokens)
-        elif hasattr(config, "max_tokens") and config.max_tokens is not None:
-            # For models that do not support 'max_tokens', map to 'max_completion_tokens'
-            api_kwargs["max_completion_tokens"] = int(config.max_tokens)
+            # Fallback to max_completion_tokens if max_tokens not set
+            api_kwargs["max_tokens"] = int(config.max_completion_tokens)
         for p in (
             "temperature",
             "top_p",
@@ -260,10 +268,6 @@ class ZAIModelDriver(LLMDriver):
             api_key_display = str(config.api_key)
             if api_key_display and len(api_key_display) > 8:
                 api_key_display = api_key_display[:4] + "..." + api_key_display[-4:]
-            client_kwargs = {
-                "api_key": config.api_key,
-                "base_url": "https://api.z.ai/api/paas/v4",
-            }
 
             # HTTP debug wrapper
             if os.environ.get("ZAI_DEBUG_HTTP", "0") == "1":
@@ -280,7 +284,14 @@ class ZAIModelDriver(LLMDriver):
                     flush=True,
                 )
 
-            client = openai.OpenAI(**client_kwargs)
+            # Use OpenAI SDK for Z.AI API compatibility
+            try:
+                import openai
+            except ImportError:
+                raise ImportError("openai module is not available. Please install it with: pip install openai")
+            client = openai.OpenAI(
+                api_key=config.api_key, base_url="https://api.z.ai/api/paas/v4/"
+            )
             return client
         except Exception as e:
             print(

janito/providers/zai/model_info.py

@@ -18,15 +18,15 @@ MODEL_SPECS = {
             "output_cost_per_1k": 0.0015,
         },
     ),
-    "glm-4": LLMModelInfo(
-        name="glm-4",
+    "glm-4.5-air": LLMModelInfo(
+        name="glm-4.5-air",
         context=128000,
         max_input=128000,
-        max_cot="N/A",
+        max_cot=4096,
         max_response=4096,
-        thinking_supported=False,
+        thinking_supported=True,
         other={
-            "description": "Z.AI's GLM-4 model for general purpose tasks",
+            "description": "Z.AI's GLM-4.5-Air model - compact and efficient version",
             "supports_tools": True,
             "supports_images": True,
             "supports_audio": False,
@@ -35,21 +35,4 @@ MODEL_SPECS = {
             "output_cost_per_1k": 0.0009,
         },
     ),
-    "glm-4v": LLMModelInfo(
-        name="glm-4v",
-        context=128000,
-        max_input=128000,
-        max_cot="N/A",
-        max_response=4096,
-        thinking_supported=False,
-        other={
-            "description": "Z.AI's GLM-4V vision model for image understanding",
-            "supports_tools": True,
-            "supports_images": True,
-            "supports_audio": False,
-            "supports_video": False,
-            "input_cost_per_1k": 0.0004,
-            "output_cost_per_1k": 0.0012,
-        },
-    ),
 }

janito/providers/zai/provider.py

@@ -17,7 +17,7 @@ class ZAIProvider(LLMProvider):
     NAME = "zai"
     MAINTAINER = "João Pinto <janito@ikignosis.org>"
     MODEL_SPECS = MODEL_SPECS
-    DEFAULT_MODEL = "glm-4.5"  # Options: glm-4.5, glm-4, glm-4v
+    DEFAULT_MODEL = "glm-4.5-air"  # Options: glm-4.5, glm-4.5-air
 
     def __init__(
         self, auth_manager: LLMAuthManager = None, config: LLMDriverConfig = None
@@ -33,6 +33,8 @@ class ZAIProvider(LLMProvider):
         # crash with an AttributeError when it tries to access `self._tools_adapter`.
         self._tools_adapter = get_local_tools_adapter()
         self._driver = None
+        # Initialize _driver_config to avoid AttributeError
+        self._driver_config = LLMDriverConfig(model=None)
 
     def _setup_available(self, auth_manager, config):
         self.auth_manager = auth_manager or LLMAuthManager()
@@ -43,6 +45,7 @@ class ZAIProvider(LLMProvider):
             )
             print(f"  janito --set-api-key YOUR_API_KEY -p {self.name}")
             print(f"Or set the ZAI_API_KEY environment variable.")
+            self._tools_adapter = get_local_tools_adapter()
             return
 
         self._tools_adapter = get_local_tools_adapter()

janito/providers/zai/schema_generator.py

@@ -4,7 +4,7 @@ Generate OpenAI-compatible tool schemas for Z.AI API.
 
 import inspect
 from typing import get_type_hints, Dict, Any, Optional, List, Union
-from janito.tools import Tool
+from janito.tools.tool_base import ToolBase
 
 
 def generate_tool_schemas(tool_classes):
@@ -35,14 +35,16 @@ def generate_tool_schema(tool_class):
     Returns:
         OpenAI-compatible tool schema dict
     """
-    if not issubclass(tool_class, Tool):
+    if not issubclass(tool_class, ToolBase):
         return None
 
     tool_instance = tool_class()
 
-    # Get the execute method
+    # Get the execute or run method
     execute_method = getattr(tool_class, "execute", None)
-    if not execute_method:
+    if execute_method is None:
+        execute_method = getattr(tool_class, "run", None)
+    if execute_method is None:
         return None
 
     # Get method signature and type hints
@@ -78,7 +80,7 @@ def generate_tool_schema(tool_class):
     schema = {
         "type": "function",
         "function": {
-            "name": tool_class.__name__,
+            "name": getattr(tool_instance, "tool_name", tool_class.__name__),
             "description": getattr(
                 tool_instance, "description", f"Execute {tool_class.__name__}"
             ),

{janito-2.12.0.dist-info → janito-2.14.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: janito
-Version: 2.12.0
+Version: 2.14.0
 Summary: A new Python package called janito.
 Author-email: João Pinto <janito@ikignosis.org>
 Project-URL: Homepage, https://github.com/ikignosis/janito

{janito-2.12.0.dist-info → janito-2.14.0.dist-info}/RECORD

@@ -20,7 +20,7 @@ janito/provider_registry.py,sha256=l0jJZ74KIebOSYXPiy7uqH8d48pckR_WTyAO4iQF98o,6
 janito/report_events.py,sha256=q4OR_jTZNfcqaQF_fzTjgqo6_VlUIxSGWfhpT4nJWcw,938
 janito/shell.bak.zip,sha256=hznHbmgfkAkjuQDJ3w73XPQh05yrtUZQxLmtGbanbYU,22
 janito/utils.py,sha256=eXSsMgM69YyzahgCNrJQLcEbB8ssLI1MQqaa20ONxbE,376
-janito/agent/setup_agent.py,sha256=XiES3D7mhAEZZ05Otoceo2TJ2Z12H5mDp-5_wJzJqAU,11664
+janito/agent/setup_agent.py,sha256=ibi1oHyAFkf0FHgN6QCHXF8tz8IfM79ugAMlVKttWYw,11605
 janito/agent/templates/profiles/system_prompt_template_Developer_with_Python_Tools.txt.j2,sha256=28TITVITH4RTdOwPpNZFSygm6OSpFb_Jr4mHprrLBhc,2584
 janito/agent/templates/profiles/system_prompt_template_developer.txt.j2,sha256=zj0ZA17iYt-6c0usgjUw_cLKnb5qDuixpxS9et5ECyw,2272
 janito/agent/templates/profiles/system_prompt_template_model_conversation_without_tools_or_context.txt.j2,sha256=dTV9aF8ji2r9dzi-l4b9r95kHrbKmjvnRxk5cVpopN4,28
@@ -28,7 +28,7 @@ janito/cli/__init__.py,sha256=xaPDOrWphBbCR63Xpcx_yfpXSJIlCaaICc4j2qpWqrM,194
 janito/cli/config.py,sha256=HkZ14701HzIqrvaNyDcDhGlVHfpX_uHlLp2rHmhRm_k,872
 janito/cli/console.py,sha256=gJolqzWL7jEPLxeuH-CwBDRFpXt976KdZOEAB2tdBDs,64
 janito/cli/main.py,sha256=s5odou0txf8pzTf1ADk2yV7T5m8B6cejJ81e7iu776U,312
-janito/cli/main_cli.py,sha256=lPY3080rvDvDPQE_lhAwEuzrUeQ2Fg-DfEt4J-bWblE,14381
+janito/cli/main_cli.py,sha256=figOoRlmegu6fyl-bmDbnYY5EQIcOEdJ6h2K6Qe1E7Q,14273
 janito/cli/prompt_core.py,sha256=F68J4Xl6jZMYFN4oBBYZFj15Jp-HTYoLub4bw2XpNRU,11648
 janito/cli/prompt_handler.py,sha256=SnPTlL64noeAMGlI08VBDD5IDD8jlVMIYA4-fS8zVLg,215
 janito/cli/prompt_setup.py,sha256=1s5yccFaWMgDkUjkvnTYGEWJAFPJ6hIiqwbrLfzWxMI,2038
@@ -73,6 +73,7 @@ janito/cli/chat_mode/shell/session/__init__.py,sha256=uTYE_QpZFEn7v9QE5o1LdulpCW
 janito/cli/chat_mode/shell/session/history.py,sha256=tYav6GgjAZkvWhlI_rfG6OArNqW6Wn2DTv39Hb20QYc,1262
 janito/cli/chat_mode/shell/session/manager.py,sha256=MwD9reHsRaly0CyRB-S1JJ0wPKz2g8Xdj2VvlU35Hgc,1001
 janito/cli/cli_commands/list_config.py,sha256=oiQEGaGPjwjG-PrOcakpNMbbqISTsBEs7rkGH3ceQsI,1179
+janito/cli/cli_commands/list_drivers.py,sha256=u7o0xk4V5iScC4iPM_a5rFIiRqHtCNZse4ilszAZ1B0,4715
 janito/cli/cli_commands/list_models.py,sha256=_rqHz89GsNLcH0GGkFqPue7ah4ZbN9YHm0lEP30Aa-A,1111
 janito/cli/cli_commands/list_profiles.py,sha256=9-HV2EbtP2AdubbMoakjbu7Oq4Ss9UDyO7Eb6CC52wI,2681
 janito/cli/cli_commands/list_providers.py,sha256=v8OQ8ULnuzNuvgkeKqGXGj69eOiavAlPGhzfR0zavhg,185
@@ -84,7 +85,7 @@ janito/cli/cli_commands/show_config.py,sha256=eYMcuvU-d7mvvuctbQacZFERqcKHEnxaRR
 janito/cli/cli_commands/show_system_prompt.py,sha256=9ZJGW7lIGJ9LX2JZiWVEm4AbaD0qEQO7LF89jPgk52I,5232
 janito/cli/core/__init__.py,sha256=YH95fhgY9yBX8RgqX9dSrEkl4exjV0T4rbmJ6xUpG-Y,196
 janito/cli/core/event_logger.py,sha256=1X6lR0Ax7AgF8HlPWFoY5Ystuu7Bh4ooTo78vXzeGB0,2008
-janito/cli/core/getters.py,sha256=AO34OBhh3f1Sx2mWVYQIvH-4fcmXG7b2471XdKNZdYs,1856
+janito/cli/core/getters.py,sha256=glUtg8K_q0vMAaLG91J9JRq5f26nJLGbDVghSNx9s28,2050
 janito/cli/core/model_guesser.py,sha256=jzkkiQ-J2buT2Omh6jYZHa8-zCJxqKQBL08Z58pe1_o,1741
 janito/cli/core/runner.py,sha256=3vP92XEUzzHeOWbMHg82iISsXVUAM7y8YKWGNSIMyA8,8337
 janito/cli/core/setters.py,sha256=PD3aT1y1q8XWQVtRNfrU0dtlW4JGdn6BMJyP7FCQWhc,4623
@@ -93,13 +94,12 @@ janito/cli/single_shot_mode/__init__.py,sha256=Ct99pKe9tINzVW6oedZJfzfZQKWpXz-we
 janito/cli/single_shot_mode/handler.py,sha256=U70X7c9MHbmj1vQlTI-Ao2JvRprpLbPh9wL5gAMbEhs,3790
 janito/docs/GETTING_STARTED.md,sha256=EbXV7B3XxjSy1E0XQJFOVITVbTmZBVB7pjth2Mb4_rg,2835
 janito/drivers/dashscope.bak.zip,sha256=9Pv4Xyciju8jO1lEMFVgYXexoZkxmDO3Ig6vw3ODfL8,4936
-janito/drivers/driver_registry.py,sha256=sbij7R71JJqJVeMfmaU-FKsEuZVO8oEn6Qp8020hdZw,773
 janito/drivers/openai_responses.bak.zip,sha256=E43eDCHGa2tCtdjzj_pMnWDdnsOZzj8BJTR5tJp8wcM,13352
 janito/drivers/azure_openai/driver.py,sha256=rec2D4DDuMjdnbGNIsrnB0oiwuxL_zBykJeUGa-PffI,4074
 janito/drivers/openai/README.md,sha256=bgPdaYX0pyotCoJ9t3cJbYM-teQ_YM1DAFEKLCMP32Q,666
 janito/drivers/openai/driver.py,sha256=O0AAp-aF3TKQLp_FSsRWm_QDG_mKliLlpDjf09fWzl4,19061
 janito/drivers/zai/__init__.py,sha256=rleES3ZJEslJ8M02TdTPyxHKXxA4-e2fDJa6yjuzY8s,22
-janito/drivers/zai/driver.py,sha256=jI_ddVK_vT-cRTpZ_8piXSJEFZ6zLTeRyT0Jzxjtd70,18926
+janito/drivers/zai/driver.py,sha256=YqdOBMltsjWq8nFttW61wSVpppS-YQh_F-nhVdIvZ_o,19245
 janito/event_bus/__init__.py,sha256=VG6GOhKMBh0O_92D-zW8a3YitJPKDajGgPiFezTXlNE,77
 janito/event_bus/bus.py,sha256=LokZbAdwcWhWOyKSp7H3Ism57x4EZhxlRPjl3NE4UKU,2847
 janito/event_bus/event.py,sha256=MtgcBPD7cvCuubiLIyo-BWcsNSO-941HLk6bScHTJtQ,427
@@ -143,9 +143,9 @@ janito/providers/openai/model_info.py,sha256=cz08O26Ychm-aP3T8guJRqpR96Im9Cwtgl2
 janito/providers/openai/provider.py,sha256=U9Bp9g2KQ58J6-B5vDgsXM05xASsgaWQOofewC7hiXs,5145
 janito/providers/openai/schema_generator.py,sha256=hTqeLcPTR8jeKn5DUUpo7b-EZ-V-g1WwXiX7MbHnFzE,2234
 janito/providers/zai/__init__.py,sha256=qtIr9_QBFaXG8xB6cRDGhS7se6ir11CWseI9azLMRBo,24
-janito/providers/zai/model_info.py,sha256=8lwwoqwuKEkrMvXWyt1iq_H_Bf0GzZk37byjSzs_YDo,1708
-janito/providers/zai/provider.py,sha256=mzGICaeg1NX_PghH17TDMxz_fKLeF4QHq98eVTuRUes,5234
-janito/providers/zai/schema_generator.py,sha256=xIc1U_AYKX0cIU89RyJ63mo6rk-Mebx5CQ-qFfmd4ZQ,3934
+janito/providers/zai/model_info.py,sha256=ldwD8enpxXv1G-YsDw4YJn31YsVueQ4vj5HgoYvnPxo,1183
+janito/providers/zai/provider.py,sha256=9RLVchSUzUlcsugd6fKEB1jZbWvJsAjKsb5XaxM5Tdo,5417
+janito/providers/zai/schema_generator.py,sha256=0kuxbrWfNKGO9fzxFStqOpTT09ldyI6vYxeDxFN4ku8,4091
 janito/tools/DOCSTRING_STANDARD.txt,sha256=VLPwNgjxRVD_xZSSVvUZ4H-4bBwM-VKh_RyfzYQsYSs,1735
 janito/tools/README.md,sha256=5HkLpF5k4PENJER7SlDPRXj0yo9mpHvAHW4uuzhq4ak,115
 janito/tools/__init__.py,sha256=W1B39PztC2UF7PS2WyLH6el32MFOETMlN1-LurOROCg,1171
@@ -207,9 +207,9 @@ janito/tools/adapters/local/validate_file_syntax/ps1_validator.py,sha256=TeIkPt0
 janito/tools/adapters/local/validate_file_syntax/python_validator.py,sha256=BfCO_K18qy92m-2ZVvHsbEU5e11OPo1pO9Vz4G4616E,130
 janito/tools/adapters/local/validate_file_syntax/xml_validator.py,sha256=AijlsP_PgNuC8ZbGsC5vOTt3Jur76otQzkd_7qR0QFY,284
 janito/tools/adapters/local/validate_file_syntax/yaml_validator.py,sha256=TgyI0HRL6ug_gBcWEm5TGJJuA4E34ZXcIzMpAbv3oJs,155
-janito-2.12.0.dist-info/licenses/LICENSE,sha256=GSAKapQH5ZIGWlpQTA7v5YrfECyaxaohUb1vJX-qepw,1090
-janito-2.12.0.dist-info/METADATA,sha256=28P5foSbPIT1QiWcLpVgWVSKLmzgG2JpwEJY8aEbsPE,16365
-janito-2.12.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-janito-2.12.0.dist-info/entry_points.txt,sha256=wIo5zZxbmu4fC-ZMrsKD0T0vq7IqkOOLYhrqRGypkx4,48
-janito-2.12.0.dist-info/top_level.txt,sha256=m0NaVCq0-ivxbazE2-ND0EA9Hmuijj_OGkmCbnBcCig,7
-janito-2.12.0.dist-info/RECORD,,
+janito-2.14.0.dist-info/licenses/LICENSE,sha256=GSAKapQH5ZIGWlpQTA7v5YrfECyaxaohUb1vJX-qepw,1090
+janito-2.14.0.dist-info/METADATA,sha256=l3vlitZrrayIG-WdOza2JqsTeOBEgtH_Md1FZXz_GSw,16365
+janito-2.14.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+janito-2.14.0.dist-info/entry_points.txt,sha256=wIo5zZxbmu4fC-ZMrsKD0T0vq7IqkOOLYhrqRGypkx4,48
+janito-2.14.0.dist-info/top_level.txt,sha256=m0NaVCq0-ivxbazE2-ND0EA9Hmuijj_OGkmCbnBcCig,7
+janito-2.14.0.dist-info/RECORD,,

janito/drivers/driver_registry.py

@@ -1,27 +0,0 @@
-# janito/drivers/driver_registry.py
-"""
-DriverRegistry: Maps driver string names to class objects for use by providers.
-"""
-
-from typing import Dict, Type
-
-# --- Import driver classes ---
-from janito.drivers.azure_openai.driver import AzureOpenAIModelDriver
-from janito.drivers.openai.driver import OpenAIModelDriver
-
-_DRIVER_REGISTRY: Dict[str, Type] = {
-    "AzureOpenAIModelDriver": AzureOpenAIModelDriver,
-    "OpenAIModelDriver": OpenAIModelDriver,
-}
-
-
-def get_driver_class(name: str):
-    """Get the driver class by string name."""
-    try:
-        return _DRIVER_REGISTRY[name]
-    except KeyError:
-        raise ValueError(f"No driver found for name: {name}")
-
-
-def register_driver(name: str, cls: type):
-    _DRIVER_REGISTRY[name] = cls