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

janito/providers/zai/schema_generator.py

@@ -0,0 +1,135 @@
+"""
+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.tool_base import ToolBase
+
+
+def generate_tool_schemas(tool_classes):
+    """
+    Generate OpenAI-compatible tool schemas from tool classes.
+
+    Args:
+        tool_classes: List of Tool classes to generate schemas for
+
+    Returns:
+        List of OpenAI-compatible tool schemas
+    """
+    schemas = []
+    for tool_class in tool_classes:
+        schema = generate_tool_schema(tool_class)
+        if schema:
+            schemas.append(schema)
+    return schemas
+
+
+def generate_tool_schema(tool_class):
+    """
+    Generate an OpenAI-compatible tool schema from a Tool class.
+
+    Args:
+        tool_class: Tool class to generate schema for
+
+    Returns:
+        OpenAI-compatible tool schema dict
+    """
+    if not issubclass(tool_class, ToolBase):
+        return None
+
+    tool_instance = tool_class()
+
+    # Get the execute or run method
+    execute_method = getattr(tool_class, "execute", None)
+    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
+    try:
+        sig = inspect.signature(execute_method)
+        type_hints = get_type_hints(execute_method)
+    except (ValueError, TypeError):
+        return None
+
+    # Build parameters schema
+    properties = {}
+    required = []
+
+    for param_name, param in sig.parameters.items():
+        if param_name == "self":
+            continue
+
+        param_type = type_hints.get(param_name, str)
+        param_schema = python_type_to_json_schema(param_type)
+
+        # Add description if available
+        if hasattr(tool_instance, "get_parameter_description"):
+            desc = tool_instance.get_parameter_description(param_name)
+            if desc:
+                param_schema["description"] = desc
+
+        properties[param_name] = param_schema
+
+        # Check if required
+        if param.default == inspect.Parameter.empty:
+            required.append(param_name)
+
+    schema = {
+        "type": "function",
+        "function": {
+            "name": getattr(tool_instance, "tool_name", tool_class.__name__),
+            "description": getattr(
+                tool_instance, "description", f"Execute {tool_class.__name__}"
+            ),
+            "parameters": {
+                "type": "object",
+                "properties": properties,
+                "required": required,
+                "additionalProperties": False,
+            },
+        },
+    }
+
+    return schema
+
+
+def python_type_to_json_schema(python_type):
+    """
+    Convert Python type hints to JSON schema types.
+
+    Args:
+        python_type: Python type hint
+
+    Returns:
+        JSON schema dict
+    """
+    if python_type == str:
+        return {"type": "string"}
+    elif python_type == int:
+        return {"type": "integer"}
+    elif python_type == float:
+        return {"type": "number"}
+    elif python_type == bool:
+        return {"type": "boolean"}
+    elif hasattr(python_type, "__origin__"):
+        # Handle generic types
+        origin = python_type.__origin__
+        if origin == list or origin == List:
+            args = getattr(python_type, "__args__", (str,))
+            item_type = args[0] if args else str
+            return {"type": "array", "items": python_type_to_json_schema(item_type)}
+        elif origin == dict or origin == Dict:
+            return {"type": "object"}
+        elif origin == Union:
+            args = getattr(python_type, "__args__", ())
+            # Handle Optional types (Union with None)
+            if len(args) == 2 and type(None) in args:
+                non_none_type = args[0] if args[1] is type(None) else args[1]
+                schema = python_type_to_json_schema(non_none_type)
+                return schema
+
+    # Default fallback
+    return {"type": "string"}

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: janito
-Version: 2.10.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.10.0.dist-info → janito-2.14.0.dist-info}/RECORD

@@ -11,7 +11,7 @@ janito/exceptions.py,sha256=l4AepRdWwAuLp5fUygrBBgO9rpwgfV6JvY1afOdq2pw,913
 janito/formatting.py,sha256=SMvOmL6bst3KGcI7OLa5D4DE127fQYuHZS3oY8OPsn8,2031
 janito/formatting_token.py,sha256=9Pz0svhV0pyNuGRXSmVkGDaQC8N-koTkf50AJR_gtSo,2217
 janito/gitignore_utils.py,sha256=P3iF9fMWAomqULq2xsoqHtI9uNIFSPcagljrxZshmLw,4110
-janito/mkdocs.yml,sha256=ve0ipJDpOpetaVOtzJf2oSSCx441r1FDMv0v6iSSxhk,920
+janito/mkdocs.yml,sha256=hceV1mnCuj5Eo42jJQHSAhqHC4emiMExgnISiiwqSow,925
 janito/perf_singleton.py,sha256=g1h0Sdf4ydzegeEpJlMhQt4H0GQZ2hryXrdYOTL-b30,113
 janito/performance_collector.py,sha256=RYu4av16Trj3RljJZ8-2Gbn1KlGdJUosrcVFYtwviNI,6285
 janito/platform_discovery.py,sha256=JN3kC7hkxdvuj-AyrJTlbbDJjtNHke3fdlZDqGi_uz0,4621
@@ -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=MAAQpD7VyGHveref4XtF7WDI6Qew81yA65lu23V2LYk,14182
+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
@@ -39,7 +39,7 @@ janito/cli/chat_mode/bindings.py,sha256=odjc5_-YW1t2FRhBUNRNoBMoQIg5sMz3ktV7xG0A
 janito/cli/chat_mode/chat_entry.py,sha256=RFdPd23jsA2DMHRacpjAdwI_1dFBaWrtnwyQEgb2fHA,475
 janito/cli/chat_mode/prompt_style.py,sha256=vsqQ9xxmrYjj1pWuVe9CayQf39fo2EIXrkKPkflSVn4,805
 janito/cli/chat_mode/script_runner.py,sha256=wOwEn4bgmjqHqjTqtfyaSOnRPsGf4ZVW-YAWhEeqxXU,6507
-janito/cli/chat_mode/session.py,sha256=cqc4Ug3yGA8rR6vFRe5v6U4recqD0tzqgwcf2ASaPgs,12891
+janito/cli/chat_mode/session.py,sha256=o-Eh3oGAMpSdOj38xTubVi8_z3Pz3HUfGd-ZyeDlejw,13262
 janito/cli/chat_mode/session_profile_select.py,sha256=CJ2g3VbPGWfBNrNkYYX57oIJZJ-hIZBNGB-zcdjC9vk,5379
 janito/cli/chat_mode/toolbar.py,sha256=bJ9jPaTInp2gB3yjSVJp8mpNEFiOslzNhVaiqpXJhKc,3025
 janito/cli/chat_mode/shell/autocomplete.py,sha256=lE68MaVaodbA2VfUM0_YLqQVLBJAE_BJsd5cMtwuD-g,793
@@ -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,20 +85,21 @@ 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/runner.py,sha256=C7eePvhzIYHgsP95WGwltIhM_T4Kzyo14M7n12RevPQ,7693
+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
 janito/cli/core/unsetters.py,sha256=FEw9gCt0vRvoCt0kRSNfVB2tzi_TqppJIx2nHPP59-k,2012
 janito/cli/single_shot_mode/__init__.py,sha256=Ct99pKe9tINzVW6oedZJfzfZQKWpXz-weSSCn0hrwHY,115
 janito/cli/single_shot_mode/handler.py,sha256=U70X7c9MHbmj1vQlTI-Ao2JvRprpLbPh9wL5gAMbEhs,3790
 janito/docs/GETTING_STARTED.md,sha256=EbXV7B3XxjSy1E0XQJFOVITVbTmZBVB7pjth2Mb4_rg,2835
-janito/docs/PROVIDERS.md,sha256=ZJK6A2j7uA651K5ypDnm-UQsnorCZvcU4qUrBPxpf0Y,4775
 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=7KxJes82SNcuaTMFvG-Avv60FNjg9s3DiynTtocK6Ac,18947
+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=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
@@ -117,12 +119,12 @@ janito/llm/driver_input.py,sha256=Zq7IO4KdQPUraeIo6XoOaRy1IdQAyYY15RQw4JU30uA,38
 janito/llm/message_parts.py,sha256=QY_0kDjaxdoErDgKPRPv1dNkkYJuXIBmHWNLiOEKAH4,1365
 janito/llm/model.py,sha256=42hjcffZDTuzjAJoVhDcDqwIXm6rUmmi5UwTOYopf5w,1131
 janito/llm/provider.py,sha256=3FbhQPrWBSEoIdIi-5DWIh0DD_CM570EFf1NcuGyGko,7961
-janito/providers/__init__.py,sha256=-FHm_JuCSNGVhF61k9wlzJFxDBnTIl9L28UngTeQVa0,369
+janito/providers/__init__.py,sha256=P2r90SUduTqn0CumjpJ9yojx2BUKDVy136xdbA8I6VU,407
 janito/providers/dashscope.bak.zip,sha256=BwXxRmZreEivvRtmqbr5BR62IFVlNjAf4y6DrF2BVJo,5998
 janito/providers/registry.py,sha256=Ygwv9eVrTXOKhv0EKxSWQXO5WMHvajWE2Q_Lc3p7dKo,730
 janito/providers/alibaba/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-janito/providers/alibaba/model_info.py,sha256=oJOTw3yr6SuQItS9ZIAch2AtsKMosgZrdYygPhOSw9s,978
-janito/providers/alibaba/provider.py,sha256=1sMp02WvtBJMjMtQjpBNUMh_nOsLfHxzz0gbuCCMjqA,4280
+janito/providers/alibaba/model_info.py,sha256=evAW2CZUX9qgRAzDhZTp47ZNj4G1T7W66gkV60Nfan8,1264
+janito/providers/alibaba/provider.py,sha256=JOkR0pKCpuG1z5KQ35TaEw6Egfp7g1gU7XiT8aeZp-0,4304
 janito/providers/anthropic/model_info.py,sha256=m6pBh0Ia8_xC1KZ7ke_4HeHIFw7nWjnYVItnRpkCSWc,1206
 janito/providers/anthropic/provider.py,sha256=JS74pDs7gSpwvG0jY-MDO5rljO0JJOffSjaL1LK1YlE,3165
 janito/providers/azure_openai/model_info.py,sha256=TMSqEpQROIIYUGAyulYJ5xGhj7CbLoaKL_JXeLbXaG0,689
@@ -140,6 +142,10 @@ janito/providers/openai/__init__.py,sha256=f0m16-sIqScjL9Mp4A0CQBZx6H3PTEy0cnE08
 janito/providers/openai/model_info.py,sha256=cz08O26Ychm-aP3T8guJRqpR96Im9Cwtgl2iMgM7tJs,3384
 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=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
@@ -201,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.10.0.dist-info/licenses/LICENSE,sha256=GSAKapQH5ZIGWlpQTA7v5YrfECyaxaohUb1vJX-qepw,1090
-janito-2.10.0.dist-info/METADATA,sha256=I8q2g4vGHg3W7ctBlg3u3jSX6RrAR4RJoqOZJdcPBq4,16365
-janito-2.10.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-janito-2.10.0.dist-info/entry_points.txt,sha256=wIo5zZxbmu4fC-ZMrsKD0T0vq7IqkOOLYhrqRGypkx4,48
-janito-2.10.0.dist-info/top_level.txt,sha256=m0NaVCq0-ivxbazE2-ND0EA9Hmuijj_OGkmCbnBcCig,7
-janito-2.10.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/docs/PROVIDERS.md

@@ -1,224 +0,0 @@
-# Provider Configuration Guide
-
-This guide covers how to configure and use different LLM providers with Janito.
-
-## MoonshotAI (Recommended)
-
-**MoonshotAI** is the recommended default provider for Janito, offering excellent performance and competitive pricing.
-
-### Setup
-```bash
-# Set API key
-janito --set-api-key YOUR_API_KEY -p moonshotai
-
-# Set as default provider
-janito --set provider=moonshotai
-janito --set model=kimi-k1-8k
-```
-
-### Available Models
-
-- **kimi-k1-8k**: Fast, general-purpose model (8k context)
-- **kimi-k1-32k**: Extended context model (32k context)
-- **kimi-k1-128k**: Long context model (128k context)
-- **kimi-k2-turbo-preview**: Latest enhanced model
-
-### Environment Variables
-```bash
-export MOONSHOTAI_API_KEY=your_key_here
-```
-
-## OpenAI
-
-### Setup
-```bash
-# Set API key
-janito --set-api-key YOUR_API_KEY -p openai
-
-# Use specific model
-janito -p openai -m gpt-4 "Your prompt"
-```
-
-### Available Models
-
-- **gpt-4**: Most capable model
-- **gpt-4-turbo**: Faster, more efficient
-- **gpt-3.5-turbo**: Cost-effective option
-
-### Environment Variables
-```bash
-export OPENAI_API_KEY=your_key_here
-```
-
-## Anthropic
-
-### Setup
-```bash
-# Set API key
-janito --set-api-key YOUR_API_KEY -p anthropic
-
-# Use Claude models
-janito -p anthropic -m claude-3-5-sonnet-20241022 "Your prompt"
-```
-
-### Available Models
-
-- **claude-3-5-sonnet-20241022**: Most capable
-- **claude-3-opus-20240229**: High performance
-- **claude-3-haiku-20240307**: Fast and cost-effective
-
-### Environment Variables
-```bash
-export ANTHROPIC_API_KEY=your_key_here
-```
-
-## Google
-
-### Setup
-```bash
-# Set API key
-janito --set-api-key YOUR_API_KEY -p google
-
-# Use Gemini models
-janito -p google -m gemini-2.0-flash-exp "Your prompt"
-```
-
-### Available Models
-
-- **gemini-2.0-flash-exp**: Latest experimental model
-- **gemini-1.5-pro**: Production-ready
-- **gemini-1.5-flash**: Fast and efficient
-
-### Environment Variables
-```bash
-export GOOGLE_API_KEY=your_key_here
-```
-
-## Azure OpenAI
-
-### Setup
-```bash
-# Set configuration
-janito --set-api-key YOUR_API_KEY -p azure-openai
-janito --set azure_deployment_name=your_deployment_name -p azure-openai
-```
-
-### Configuration
-
-Requires both API key and deployment name:
-
-- **API Key**: Your Azure OpenAI key
-- **Deployment Name**: Your Azure deployment name
-- **Base URL**: Your Azure endpoint URL
-
-### Environment Variables
-```bash
-export AZURE_OPENAI_API_KEY=your_key_here
-export AZURE_OPENAI_ENDPOINT=https://your-resource.openai.azure.com/
-```
-
-## Other Providers
-
-Janito also supports these providers through OpenAI-compatible APIs:
-
-### Alibaba Cloud
-```bash
-janito --set-api-key YOUR_KEY -p alibaba
-```
-
-### DeepSeek
-```bash
-janito --set-api-key YOUR_KEY -p deepseek
-```
-
-### Groq
-```bash
-janito --set-api-key YOUR_KEY -p groq
-```
-
-### Mistral
-```bash
-janito --set-api-key YOUR_KEY -p mistral
-```
-
-## Configuration Management
-
-### Check Current Configuration
-```bash
-janito --show-config
-```
-
-### List All Providers
-```bash
-janito --list-providers
-```
-
-### List Models for a Provider
-```bash
-janito -p moonshotai --list-models
-janito -p openai --list-models
-```
-
-### Switch Providers
-```bash
-# Temporarily for one command
-janito -p openai -m gpt-4 "Your prompt"
-
-# Permanently as default
-janito --set provider=openai
-janito --set model=gpt-4
-```
-
-## Advanced Configuration
-
-### Custom Base URLs
-For OpenAI-compatible providers, you can set custom base URLs:
-
-```bash
-janito --set base_url=https://your-custom-endpoint.com -p openai
-```
-
-### Provider-Specific Settings
-Each provider can have custom settings:
-
-```bash
-# Set temperature for a specific provider/model
-janito --set temperature=0.7 -p moonshotai -m kimi-k1-8k
-
-# Set max tokens
-janito --set max_tokens=2000 -p openai -m gpt-4
-```
-
-## Troubleshooting
-
-### Provider Not Found
-```bash
-# Check if provider is registered
-janito --list-providers
-
-# Re-register provider
-janito --set-api-key YOUR_KEY -p PROVIDER_NAME
-```
-
-### API Key Issues
-```bash
-# Check current API key
-janito --show-config
-
-# Reset API key
-janito --set-api-key NEW_KEY -p PROVIDER_NAME
-```
-
-### Model Not Available
-```bash
-# List available models for provider
-janito -p PROVIDER_NAME --list-models
-```
-
-## Best Practices
-
-1. **Start with MoonshotAI**: It's the recommended default for good reason
-2. **Use environment variables**: For CI/CD and containerized environments
-3. **Test different models**: Each has different strengths and pricing
-4. **Monitor usage**: Keep track of API costs and rate limits
-5. **Use profiles**: Set up different configurations for different use cases

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