donkit-llm 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

donkit/llm/factory.py

@@ -121,7 +121,7 @@ class ModelFactory:
     @staticmethod
     def create_model(
         provider: Literal[
-            "openai", "azure_openai", "claude", "claude_vertex", "vertex"
+            "openai", "azure_openai", "claude", "claude_vertex", "vertex", "ollama"
         ],
         model_name: str,
         credentials: dict,
@@ -160,5 +160,12 @@ class ModelFactory:
                 location=credentials.get("location", "us-central1"),
                 credentials=credentials.get("credentials"),
             )
+        elif provider == "ollama":
+            # Ollama uses OpenAI-compatible API
+            return ModelFactory.create_openai_model(
+                model_name=model_name,
+                api_key=credentials.get("api_key", "ollama"),
+                base_url=credentials.get("base_url"),
+            )
         else:
             raise ValueError(f"Unknown provider: {provider}")

donkit/llm/model_abstract.py

@@ -159,7 +159,7 @@ class LLMModelAbstract(ABC):
 
     def supports_capability(self, capability: ModelCapability) -> bool:
         """Check if this model supports a specific capability."""
-        return bool(self.capabilities & capability)
+        return capability in list(self.capabilities)
 
     @abstractmethod
     async def generate(self, request: GenerateRequest) -> GenerateResponse:

donkit/llm/openai_model.py

@@ -61,17 +61,11 @@ class OpenAIModel(LLMModelAbstract):
             | ModelCapability.STREAMING
             | ModelCapability.STRUCTURED_OUTPUT
             | ModelCapability.TOOL_CALLING
+            | ModelCapability.VISION
+            | ModelCapability.MULTIMODAL_INPUT
         )
 
         model_lower = self._model_name.lower()
-
-        # Vision models (GPT-4o, GPT-4 Turbo, GPT-5, etc.)
-        if any(
-            x in model_lower
-            for x in ["gpt-4o", "gpt-4-turbo", "gpt-4-vision", "gpt-5", "o1", "o3"]
-        ):
-            caps |= ModelCapability.VISION | ModelCapability.MULTIMODAL_INPUT
-
         # Audio models
         if "audio" in model_lower:
             caps |= ModelCapability.AUDIO_INPUT | ModelCapability.MULTIMODAL_INPUT

{donkit_llm-0.1.0.dist-info → donkit_llm-0.1.1.dist-info}/METADATA

@@ -1,7 +1,7 @@
 Metadata-Version: 2.3
 Name: donkit-llm
-Version: 0.1.0
-Summary: Unified LLM model implementations for Donkit (OpenAI, Azure OpenAI, Claude, Vertex AI)
+Version: 0.1.1
+Summary: Unified LLM model implementations for Donkit (OpenAI, Azure OpenAI, Claude, Vertex AI, Ollama)
 License: MIT
 Author: Donkit AI
 Author-email: opensource@donkit.ai

donkit_llm-0.1.1.dist-info/RECORD

@@ -0,0 +1,9 @@
+donkit/llm/__init__.py,sha256=VLnsTe_cB_FkKQbutGkEGVXOGevbrfgICRjqTXq4nsA,1193
+donkit/llm/claude_model.py,sha256=RCvqaFSIaWJvcmW7y706Kc8l6YKdP0xR3chYRR-otV8,16589
+donkit/llm/factory.py,sha256=Hn64gpI-H-jny4cHI0iaG_bRbRmyvk5cGVT7uU3wrDk,5991
+donkit/llm/model_abstract.py,sha256=5mVXvdcP-LHXR17cGfpCqc32O391F1rJY62ygdI9PBo,7862
+donkit/llm/openai_model.py,sha256=oR4iEECFadKOMN_DO-t11vdedpqD7f_DWrjEOAK0p98,19200
+donkit/llm/vertex_model.py,sha256=z14briPYZt3qmxJZdJYvPKyGSHZxLdNKiF1N2_FNoZc,17066
+donkit_llm-0.1.1.dist-info/METADATA,sha256=f1CV_hXEtJn9byW8-QmQ1dYqCArgs8OCM5mVzK_tHnc,676
+donkit_llm-0.1.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+donkit_llm-0.1.1.dist-info/RECORD,,

donkit_llm-0.1.0.dist-info/RECORD

@@ -1,9 +0,0 @@
-donkit/llm/__init__.py,sha256=VLnsTe_cB_FkKQbutGkEGVXOGevbrfgICRjqTXq4nsA,1193
-donkit/llm/claude_model.py,sha256=RCvqaFSIaWJvcmW7y706Kc8l6YKdP0xR3chYRR-otV8,16589
-donkit/llm/factory.py,sha256=PRRIY3sMOxgBTX-5Ecci62TWW9k9MP7lyhHyoo4dZeM,5676
-donkit/llm/model_abstract.py,sha256=-v_hDMzKS6_sZ3kWY6WUIP1LoAvXQvn7YZC9Hy4lWWo,7861
-donkit/llm/openai_model.py,sha256=nGTPjzsldILsFG6GlU5ZNwsTfaWT9rsApvpNpxLhbAc,19395
-donkit/llm/vertex_model.py,sha256=z14briPYZt3qmxJZdJYvPKyGSHZxLdNKiF1N2_FNoZc,17066
-donkit_llm-0.1.0.dist-info/METADATA,sha256=rIuOQ0ZALJXKIxqMbqA0_lJn2puEaHfV5TXnegO-CaY,668
-donkit_llm-0.1.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-donkit_llm-0.1.0.dist-info/RECORD,,