donkit-llm 0.1.2__tar.gz → 0.1.3__tar.gz

{donkit_llm-0.1.2 → donkit_llm-0.1.3}/PKG-INFO

@@ -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 → donkit_llm-0.1.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "donkit-llm"
-version = "0.1.2"
+version = "0.1.3"
 description = "Unified LLM model implementations for Donkit (OpenAI, Azure OpenAI, Claude, Vertex AI, Ollama)"
 authors = ["Donkit AI <opensource@donkit.ai>"]
 license = "MIT"

{donkit_llm-0.1.2 → donkit_llm-0.1.3}/src/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-0.1.2 → donkit_llm-0.1.3}/src/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":