donkit-llm 0.1.2__py3-none-any.whl → 0.1.3__py3-none-any.whl

donkit/llm/donkit_model.py

@@ -27,7 +27,7 @@ class DonkitModel(LLMModelAbstract):
         self,
         base_url: str,
         api_token: str,
-        provider: str,
+        provider: str = "default",
         model_name: str | None = None,
         project_id: str | None = None,
     ):
@@ -37,7 +37,8 @@ class DonkitModel(LLMModelAbstract):
         Args:
             base_url: Base URL for the API Gateway
             api_token: API token for authentication
-            provider: The LLM provider name (e.g., "openai", "anthropic", "vertex")
+            provider: The LLM provider name
+                (e.g., "openai", "anthropic", "vertex", "azure_openai", "ollama", "default")
             model_name: The specific model identifier (e.g., "gpt-4o", "claude-3-opus")
             project_id: The project ID for the gateway
         """

donkit/llm/factory.py

@@ -151,7 +151,7 @@ class ModelFactory:
 
     @staticmethod
     def create_donkit_model(
-        model_name: str,
+        model_name: str | None,
         api_key: str,
         base_url: str = "http://localhost:9017",
         provider: str = "default",
@@ -162,8 +162,8 @@ class ModelFactory:
             model_name: Name of the model to use
             api_key: API key for authentication
             base_url: Base URL of the RagOps API Gateway
-            provider: Provider to use (default: vertex)
-
+            provider: Provider to use e.g.:
+                vertex, openai, azure_openai, ollama, default
         Returns:
             DonkitModel instance
         """
@@ -197,7 +197,7 @@ class ModelFactory:
                 "gemini": "gemini-2.5-flash",
                 "vertex": "gemini-2.5-flash",
                 "ollama": "mistral",
-                "donkit": "default",
+                "donkit": None,
             }
             model_name = default_models.get(provider, "default")
         if provider == "openai":

{donkit_llm-0.1.2.dist-info → donkit_llm-0.1.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: donkit-llm
-Version: 0.1.2
+Version: 0.1.3
 Summary: Unified LLM model implementations for Donkit (OpenAI, Azure OpenAI, Claude, Vertex AI, Ollama)
 License: MIT
 Author: Donkit AI

{donkit_llm-0.1.2.dist-info → donkit_llm-0.1.3.dist-info}/RECORD

@@ -1,12 +1,12 @@
 donkit/llm/__init__.py,sha256=0w5hPdaJDpzL1EpnBZm-7WV1Rz1OUdcCMcnxNRpXdiM,1357
 donkit/llm/claude_model.py,sha256=9UjNkACc6wHFus2gOXLKOi9yjX2FkP3cpQ9zFZEcXWU,16650
-donkit/llm/donkit_model.py,sha256=uxrD7QBBovw8WjvKVu0IoaEMOv68tCKSneio3w826b0,8432
-donkit/llm/factory.py,sha256=hvvjwEccwiLzYooRJpkcnV2YteQKr6eqwolxsX8fr50,9144
+donkit/llm/donkit_model.py,sha256=rEPxBW6k_BhIBF4XkgLzibVfwW6OJHiX89yMadcJkY4,8497
+donkit/llm/factory.py,sha256=IVz9fY_XMDRTgLCCUhBOxKtOcGdzuwFIrUlG2QLD1PE,9195
 donkit/llm/gemini_model.py,sha256=2uLoZr9HjUf1wxiZRGLQFcURCutsB2SV9f-1VaR6kGI,14413
 donkit/llm/model_abstract.py,sha256=aOgYh3I96PsxSxnkIJ1ETx5UFeRxozCD1c44wiKoBSs,8191
 donkit/llm/ollama_integration.py,sha256=WXeV2xNxP7gd1JyMsHMKaQOjvH7QYkLIPs7pmTPWFrg,13236
 donkit/llm/openai_model.py,sha256=P8gBw_WqZiclAt6QvZLZ8Q1-HCMCgS6O_lKGI1YDHOI,23372
 donkit/llm/vertex_model.py,sha256=HavW0iam3EmnlznB9KXADxo5SgaNCdwMjzgp5AOdoOU,29323
-donkit_llm-0.1.2.dist-info/METADATA,sha256=V4GHADO9LUFsxmlQlYvbmuqDzCJ5qUlpgowbq-A_TuM,742
-donkit_llm-0.1.2.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-donkit_llm-0.1.2.dist-info/RECORD,,
+donkit_llm-0.1.3.dist-info/METADATA,sha256=H65OHtU9OK0lrsV9dmbMKA16SR0CzPIFTaDX6GJVn_E,742
+donkit_llm-0.1.3.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+donkit_llm-0.1.3.dist-info/RECORD,,