bridgekit 0.3.10__tar.gz → 0.3.11__tar.gz

{bridgekit-0.3.10/bridgekit.egg-info → bridgekit-0.3.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bridgekit
-Version: 0.3.10
+Version: 0.3.11
 Summary: AI tools that make you a better data scientist, not a redundant one.
 License: MIT
 Project-URL: Homepage, https://usebridgekit.com
@@ -25,6 +25,10 @@ Provides-Extra: openai
 Requires-Dist: openai>=1.0.0; extra == "openai"
 Provides-Extra: gemini
 Requires-Dist: google-generativeai>=0.3.0; extra == "gemini"
+Provides-Extra: ollama
+Requires-Dist: ollama>=0.3.0; extra == "ollama"
+Provides-Extra: openrouter
+Requires-Dist: openai>=1.0.0; extra == "openrouter"
 Provides-Extra: dev
 Requires-Dist: pytest>=7.0.0; extra == "dev"
 Requires-Dist: pytest-mock>=3.0.0; extra == "dev"
@@ -87,6 +91,18 @@ pip install bridgekit[gemini]
 export GOOGLE_API_KEY=your_key_here
 ```
 
+**Ollama (local models, free, no API key):**
+```bash
+pip install bridgekit[ollama]
+# Make sure Ollama is running locally: https://ollama.com
+```
+
+**OpenRouter (free and paid models via one API):**
+```bash
+pip install bridgekit[openrouter]
+export OPENROUTER_API_KEY=your_key_here
+```
+
 ---
 
 ## Getting Started
@@ -528,7 +544,7 @@ high-confidence growth lever worth testing immediately.
 
 ## Multi-Provider Support
 
-Bridgekit now supports multiple AI providers so you're not locked into one API. You can use Anthropic, OpenAI, or Google Gemini models with any tool.
+Bridgekit now supports multiple AI providers so you're not locked into one API. You can use Anthropic, OpenAI, Google Gemini, local models via Ollama, or any model on OpenRouter with any tool.
 
 **Using different providers:**
 
@@ -541,21 +557,35 @@ print(evaluate("Your analysis here", provider="openai"))
 # Use Google Gemini (default model: gemini-1.5-pro)
 print(plan("Your question here", provider="gemini"))
 
+# Use a local model via Ollama - free, private, no API key (default model: llama3.2)
+print(evaluate("Your analysis here", provider="ollama"))
+
+# Use OpenRouter to access a wide range of models, including free tiers
+print(evaluate("Your analysis here", provider="openrouter", model="deepseek/deepseek-v4-flash-0731:free"))
+
 # Use specific model
 print(redteam("Your analysis here", model="gpt-4-turbo"))
 print(ask("Your question here", source="reports/", model="claude-3-opus-20240229"))
 ```
 
+**Ollama setup:** install and start [Ollama](https://ollama.com) locally, then pull a model (e.g. `ollama pull llama3.2`). By default Bridgekit connects to `http://localhost:11434`; set `OLLAMA_HOST` to point at a different host.
+
 **Provider auto-detection:**
 Bridgekit automatically detects the provider from model names:
 - Models starting with "claude" → Anthropic
 - Models starting with "gpt" → OpenAI  
 - Models starting with "gemini" → Google Gemini
+- Model names containing "/" (e.g. `deepseek/deepseek-v4-flash-0731:free`) → OpenRouter
+- Common local model families (e.g. "llama", "mistral", "mixtral", "gemma", "phi", "qwen") → Ollama
 
 **Default models by provider:**
 - Anthropic: `claude-opus-4-8`
 - OpenAI: `gpt-4o`
 - Gemini: `gemini-1.5-pro`
+- Ollama: `llama3.2`
+- OpenRouter: `deepseek/deepseek-v4-flash-0731:free`
+
+> **Note:** OpenRouter's free-tier model lineup rotates over time. Check the current list at [openrouter.ai/models?max_price=0](https://openrouter.ai/models?max_price=0) if a `:free` model stops working, and pass `model=` explicitly with whichever slug is currently free.
 
 All tools support the same `provider` and `model` parameters:
 - `evaluate(text, provider=None, model=None, system_prompt=None)`

bridgekit-0.3.10/PKG-INFO → bridgekit-0.3.11/README.md

@@ -1,35 +1,3 @@
-Metadata-Version: 2.4
-Name: bridgekit
-Version: 0.3.10
-Summary: AI tools that make you a better data scientist, not a redundant one.
-License: MIT
-Project-URL: Homepage, https://usebridgekit.com
-Project-URL: Issues, https://github.com/getbridgekit/bridgekit/issues
-Keywords: data science,AI,analysis,evaluation,anthropic
-Classifier: Development Status :: 3 - Alpha
-Classifier: Intended Audience :: Science/Research
-Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
-Classifier: Programming Language :: Python :: 3
-Classifier: License :: OSI Approved :: MIT License
-Requires-Python: >=3.9
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: anthropic>=0.20.0
-Requires-Dist: chromadb>=0.4.0
-Requires-Dist: sentence-transformers>=2.0.0
-Requires-Dist: pypdf>=3.0.0
-Requires-Dist: python-docx>=1.0.0
-Requires-Dist: python-pptx>=0.6.0
-Requires-Dist: nbformat>=5.0.0
-Provides-Extra: openai
-Requires-Dist: openai>=1.0.0; extra == "openai"
-Provides-Extra: gemini
-Requires-Dist: google-generativeai>=0.3.0; extra == "gemini"
-Provides-Extra: dev
-Requires-Dist: pytest>=7.0.0; extra == "dev"
-Requires-Dist: pytest-mock>=3.0.0; extra == "dev"
-Dynamic: license-file
-
 <p align="center">
   <img src="https://raw.githubusercontent.com/getbridgekit/bridgekit/main/assets/logo.png" alt="Bridgekit" width="200"/>
 </p>
@@ -87,6 +55,18 @@ pip install bridgekit[gemini]
 export GOOGLE_API_KEY=your_key_here
 ```
 
+**Ollama (local models, free, no API key):**
+```bash
+pip install bridgekit[ollama]
+# Make sure Ollama is running locally: https://ollama.com
+```
+
+**OpenRouter (free and paid models via one API):**
+```bash
+pip install bridgekit[openrouter]
+export OPENROUTER_API_KEY=your_key_here
+```
+
 ---
 
 ## Getting Started
@@ -528,7 +508,7 @@ high-confidence growth lever worth testing immediately.
 
 ## Multi-Provider Support
 
-Bridgekit now supports multiple AI providers so you're not locked into one API. You can use Anthropic, OpenAI, or Google Gemini models with any tool.
+Bridgekit now supports multiple AI providers so you're not locked into one API. You can use Anthropic, OpenAI, Google Gemini, local models via Ollama, or any model on OpenRouter with any tool.
 
 **Using different providers:**
 
@@ -541,21 +521,35 @@ print(evaluate("Your analysis here", provider="openai"))
 # Use Google Gemini (default model: gemini-1.5-pro)
 print(plan("Your question here", provider="gemini"))
 
+# Use a local model via Ollama - free, private, no API key (default model: llama3.2)
+print(evaluate("Your analysis here", provider="ollama"))
+
+# Use OpenRouter to access a wide range of models, including free tiers
+print(evaluate("Your analysis here", provider="openrouter", model="deepseek/deepseek-v4-flash-0731:free"))
+
 # Use specific model
 print(redteam("Your analysis here", model="gpt-4-turbo"))
 print(ask("Your question here", source="reports/", model="claude-3-opus-20240229"))
 ```
 
+**Ollama setup:** install and start [Ollama](https://ollama.com) locally, then pull a model (e.g. `ollama pull llama3.2`). By default Bridgekit connects to `http://localhost:11434`; set `OLLAMA_HOST` to point at a different host.
+
 **Provider auto-detection:**
 Bridgekit automatically detects the provider from model names:
 - Models starting with "claude" → Anthropic
 - Models starting with "gpt" → OpenAI  
 - Models starting with "gemini" → Google Gemini
+- Model names containing "/" (e.g. `deepseek/deepseek-v4-flash-0731:free`) → OpenRouter
+- Common local model families (e.g. "llama", "mistral", "mixtral", "gemma", "phi", "qwen") → Ollama
 
 **Default models by provider:**
 - Anthropic: `claude-opus-4-8`
 - OpenAI: `gpt-4o`
 - Gemini: `gemini-1.5-pro`
+- Ollama: `llama3.2`
+- OpenRouter: `deepseek/deepseek-v4-flash-0731:free`
+
+> **Note:** OpenRouter's free-tier model lineup rotates over time. Check the current list at [openrouter.ai/models?max_price=0](https://openrouter.ai/models?max_price=0) if a `:free` model stops working, and pass `model=` explicitly with whichever slug is currently free.
 
 All tools support the same `provider` and `model` parameters:
 - `evaluate(text, provider=None, model=None, system_prompt=None)`

{bridgekit-0.3.10 → bridgekit-0.3.11}/bridgekit/config.py

@@ -7,25 +7,39 @@ class Provider(Enum):
     ANTHROPIC = "anthropic"
     OPENAI = "openai"
     GEMINI = "gemini"
+    OLLAMA = "ollama"
+    OPENROUTER = "openrouter"
 
 
 # Default models for each provider
 DEFAULT_MODELS = {
     Provider.ANTHROPIC: "claude-opus-4-8",
     Provider.OPENAI: "gpt-4o",
-    Provider.GEMINI: "gemini-1.5-pro"
+    Provider.GEMINI: "gemini-1.5-pro",
+    Provider.OLLAMA: "llama3.2",
+    Provider.OPENROUTER: "deepseek/deepseek-v4-flash-0731:free",
 }
 
 # Legacy support
 DEFAULT_MODEL = DEFAULT_MODELS[Provider.ANTHROPIC]
 
+# Common local model family names served by Ollama. Used to infer the
+# provider from a bare model name when no explicit provider is given.
+OLLAMA_MODEL_PREFIXES = (
+    "llama", "mistral", "mixtral", "gemma", "phi", "qwen",
+    "codellama", "vicuna", "deepseek", "tinyllama",
+)
 
-def require_api_key(provider: Provider = Provider.ANTHROPIC) -> str:
+
+def require_api_key(provider: Provider = Provider.ANTHROPIC) -> Optional[str]:
     """Return the API key for the specified provider from the environment, or raise a clear error.
 
     Each Bridgekit tool calls this before constructing a client so
     users get the same friendly message instead of whatever the SDK surfaces
     when the key is missing.
+
+    Ollama runs locally and doesn't use an API key, so this returns None
+    for that provider instead of raising.
     """
     if provider == Provider.ANTHROPIC:
         api_key = os.environ.get("ANTHROPIC_API_KEY")
@@ -48,6 +62,15 @@ def require_api_key(provider: Provider = Provider.ANTHROPIC) -> str:
                 "GOOGLE_API_KEY not found. Set it with: export GOOGLE_API_KEY=your_key_here"
             )
         return api_key
+    elif provider == Provider.OLLAMA:
+        return None
+    elif provider == Provider.OPENROUTER:
+        api_key = os.environ.get("OPENROUTER_API_KEY")
+        if not api_key:
+            raise EnvironmentError(
+                "OPENROUTER_API_KEY not found. Set it with: export OPENROUTER_API_KEY=your_key_here"
+            )
+        return api_key
     else:
         raise ValueError(f"Unsupported provider: {provider}")
 
@@ -73,7 +96,13 @@ def parse_provider(provider: Optional[str] = None, model: Optional[str] = None)
             return Provider.OPENAI
         elif model.startswith("gemini"):
             return Provider.GEMINI
-    
+        elif "/" in model:
+            # OpenRouter model names use a "vendor/model" format,
+            # e.g. "deepseek/deepseek-v4-flash-0731:free"
+            return Provider.OPENROUTER
+        elif model.startswith(OLLAMA_MODEL_PREFIXES):
+            return Provider.OLLAMA
+
     # Default to Anthropic for backward compatibility
     return Provider.ANTHROPIC
 

{bridgekit-0.3.10 → bridgekit-0.3.11}/bridgekit/providers.py

@@ -1,5 +1,6 @@
 """Provider client factory for Bridgekit multi-provider support."""
 
+import os
 from typing import Union, Dict, Any, List
 from .config import Provider, require_api_key, get_default_model
 
@@ -39,11 +40,14 @@ class AnthropicClient(BaseProviderClient):
 
 class OpenAIClient(BaseProviderClient):
     """OpenAI provider client."""
-    
-    def __init__(self, api_key: str):
+
+    def __init__(self, api_key: str, base_url: str = None):
         super().__init__(api_key)
         import openai
-        self.client = openai.OpenAI(api_key=api_key)
+        if base_url:
+            self.client = openai.OpenAI(api_key=api_key, base_url=base_url)
+        else:
+            self.client = openai.OpenAI(api_key=api_key)
     
     def create_message(self, system_prompt: str, user_message: str, model: str, max_tokens: int = 1024) -> str:
         """Create a message using OpenAI's API."""
@@ -86,6 +90,35 @@ class GeminiClient(BaseProviderClient):
         return response.text
 
 
+class OllamaClient(BaseProviderClient):
+    """Ollama provider client for locally-hosted models. No API key required."""
+
+    def __init__(self, api_key: str = None):
+        super().__init__(api_key)
+        import ollama
+        host = os.environ.get("OLLAMA_HOST", "http://localhost:11434")
+        self.client = ollama.Client(host=host)
+
+    def create_message(self, system_prompt: str, user_message: str, model: str, max_tokens: int = 1024) -> str:
+        """Create a message using a locally-hosted Ollama model."""
+        response = self.client.chat(
+            model=model,
+            messages=[
+                {"role": "system", "content": system_prompt},
+                {"role": "user", "content": user_message}
+            ],
+            options={"num_predict": max_tokens}
+        )
+        return response["message"]["content"]
+
+
+class OpenRouterClient(OpenAIClient):
+    """OpenRouter provider client. OpenAI-compatible API routing to many models, including free tiers."""
+
+    def __init__(self, api_key: str):
+        super().__init__(api_key, base_url="https://openrouter.ai/api/v1")
+
+
 def create_client(provider: Provider, model: str = None) -> BaseProviderClient:
     """Create a client for the specified provider."""
     if model is None:
@@ -99,6 +132,10 @@ def create_client(provider: Provider, model: str = None) -> BaseProviderClient:
         return OpenAIClient(api_key)
     elif provider == Provider.GEMINI:
         return GeminiClient(api_key)
+    elif provider == Provider.OLLAMA:
+        return OllamaClient(api_key)
+    elif provider == Provider.OPENROUTER:
+        return OpenRouterClient(api_key)
     else:
         raise ValueError(f"Unsupported provider: {provider}")
 

bridgekit-0.3.10/README.md → bridgekit-0.3.11/bridgekit.egg-info/PKG-INFO

@@ -1,3 +1,39 @@
+Metadata-Version: 2.4
+Name: bridgekit
+Version: 0.3.11
+Summary: AI tools that make you a better data scientist, not a redundant one.
+License: MIT
+Project-URL: Homepage, https://usebridgekit.com
+Project-URL: Issues, https://github.com/getbridgekit/bridgekit/issues
+Keywords: data science,AI,analysis,evaluation,anthropic
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Science/Research
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: anthropic>=0.20.0
+Requires-Dist: chromadb>=0.4.0
+Requires-Dist: sentence-transformers>=2.0.0
+Requires-Dist: pypdf>=3.0.0
+Requires-Dist: python-docx>=1.0.0
+Requires-Dist: python-pptx>=0.6.0
+Requires-Dist: nbformat>=5.0.0
+Provides-Extra: openai
+Requires-Dist: openai>=1.0.0; extra == "openai"
+Provides-Extra: gemini
+Requires-Dist: google-generativeai>=0.3.0; extra == "gemini"
+Provides-Extra: ollama
+Requires-Dist: ollama>=0.3.0; extra == "ollama"
+Provides-Extra: openrouter
+Requires-Dist: openai>=1.0.0; extra == "openrouter"
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0.0; extra == "dev"
+Requires-Dist: pytest-mock>=3.0.0; extra == "dev"
+Dynamic: license-file
+
 <p align="center">
   <img src="https://raw.githubusercontent.com/getbridgekit/bridgekit/main/assets/logo.png" alt="Bridgekit" width="200"/>
 </p>
@@ -55,6 +91,18 @@ pip install bridgekit[gemini]
 export GOOGLE_API_KEY=your_key_here
 ```
 
+**Ollama (local models, free, no API key):**
+```bash
+pip install bridgekit[ollama]
+# Make sure Ollama is running locally: https://ollama.com
+```
+
+**OpenRouter (free and paid models via one API):**
+```bash
+pip install bridgekit[openrouter]
+export OPENROUTER_API_KEY=your_key_here
+```
+
 ---
 
 ## Getting Started
@@ -496,7 +544,7 @@ high-confidence growth lever worth testing immediately.
 
 ## Multi-Provider Support
 
-Bridgekit now supports multiple AI providers so you're not locked into one API. You can use Anthropic, OpenAI, or Google Gemini models with any tool.
+Bridgekit now supports multiple AI providers so you're not locked into one API. You can use Anthropic, OpenAI, Google Gemini, local models via Ollama, or any model on OpenRouter with any tool.
 
 **Using different providers:**
 
@@ -509,21 +557,35 @@ print(evaluate("Your analysis here", provider="openai"))
 # Use Google Gemini (default model: gemini-1.5-pro)
 print(plan("Your question here", provider="gemini"))
 
+# Use a local model via Ollama - free, private, no API key (default model: llama3.2)
+print(evaluate("Your analysis here", provider="ollama"))
+
+# Use OpenRouter to access a wide range of models, including free tiers
+print(evaluate("Your analysis here", provider="openrouter", model="deepseek/deepseek-v4-flash-0731:free"))
+
 # Use specific model
 print(redteam("Your analysis here", model="gpt-4-turbo"))
 print(ask("Your question here", source="reports/", model="claude-3-opus-20240229"))
 ```
 
+**Ollama setup:** install and start [Ollama](https://ollama.com) locally, then pull a model (e.g. `ollama pull llama3.2`). By default Bridgekit connects to `http://localhost:11434`; set `OLLAMA_HOST` to point at a different host.
+
 **Provider auto-detection:**
 Bridgekit automatically detects the provider from model names:
 - Models starting with "claude" → Anthropic
 - Models starting with "gpt" → OpenAI  
 - Models starting with "gemini" → Google Gemini
+- Model names containing "/" (e.g. `deepseek/deepseek-v4-flash-0731:free`) → OpenRouter
+- Common local model families (e.g. "llama", "mistral", "mixtral", "gemma", "phi", "qwen") → Ollama
 
 **Default models by provider:**
 - Anthropic: `claude-opus-4-8`
 - OpenAI: `gpt-4o`
 - Gemini: `gemini-1.5-pro`
+- Ollama: `llama3.2`
+- OpenRouter: `deepseek/deepseek-v4-flash-0731:free`
+
+> **Note:** OpenRouter's free-tier model lineup rotates over time. Check the current list at [openrouter.ai/models?max_price=0](https://openrouter.ai/models?max_price=0) if a `:free` model stops working, and pass `model=` explicitly with whichever slug is currently free.
 
 All tools support the same `provider` and `model` parameters:
 - `evaluate(text, provider=None, model=None, system_prompt=None)`

{bridgekit-0.3.10 → bridgekit-0.3.11}/bridgekit.egg-info/requires.txt

@@ -13,5 +13,11 @@ pytest-mock>=3.0.0
 [gemini]
 google-generativeai>=0.3.0
 
+[ollama]
+ollama>=0.3.0
+
 [openai]
 openai>=1.0.0
+
+[openrouter]
+openai>=1.0.0

{bridgekit-0.3.10 → bridgekit-0.3.11}/pyproject.toml

@@ -7,7 +7,7 @@ include = ["bridgekit*"]
 
 [project]
 name = "bridgekit"
-version = "0.3.10"
+version = "0.3.11"
 description = "AI tools that make you a better data scientist, not a redundant one."
 readme = "README.md"
 requires-python = ">=3.9"
@@ -33,6 +33,8 @@ dependencies = [
 [project.optional-dependencies]
 openai = ["openai>=1.0.0"]
 gemini = ["google-generativeai>=0.3.0"]
+ollama = ["ollama>=0.3.0"]
+openrouter = ["openai>=1.0.0"]
 dev = [
     "pytest>=7.0.0",
     "pytest-mock>=3.0.0",

bridgekit-0.3.11/tests/test_providers.py

@@ -0,0 +1,214 @@
+import os
+import sys
+import types
+import pytest
+from unittest.mock import MagicMock, patch
+
+from bridgekit.config import parse_provider, require_api_key, Provider
+from bridgekit.providers import OllamaClient, OpenRouterClient, create_client
+
+
+class TestParseProvider:
+    """parse_provider() returns the correct Provider enum."""
+
+    def test_explicit_anthropic(self):
+        assert parse_provider(provider="anthropic") == Provider.ANTHROPIC
+
+    def test_explicit_openai(self):
+        assert parse_provider(provider="openai") == Provider.OPENAI
+
+    def test_explicit_gemini(self):
+        assert parse_provider(provider="gemini") == Provider.GEMINI
+
+    def test_infer_anthropic_from_model(self):
+        assert parse_provider(model="claude-opus-4-6") == Provider.ANTHROPIC
+
+    def test_infer_openai_from_model(self):
+        assert parse_provider(model="gpt-4o") == Provider.OPENAI
+
+    def test_infer_gemini_from_model(self):
+        assert parse_provider(model="gemini-1.5-pro") == Provider.GEMINI
+
+    def test_infer_openrouter_from_model(self):
+        assert parse_provider(model="deepseek/deepseek-v4-flash-0731:free") == Provider.OPENROUTER
+
+    def test_infer_ollama_from_model(self):
+        assert parse_provider(model="llama3.2") == Provider.OLLAMA
+
+    def test_infer_ollama_from_mistral_model(self):
+        assert parse_provider(model="mistral") == Provider.OLLAMA
+
+    def test_defaults_to_anthropic(self):
+        assert parse_provider() == Provider.ANTHROPIC
+
+    def test_raises_on_unsupported_provider(self):
+        with pytest.raises(ValueError, match="Unsupported provider"):
+            parse_provider(provider="cohere")
+
+    def test_provider_takes_precedence_over_model(self):
+        assert parse_provider(provider="openai", model="claude-opus-4-6") == Provider.OPENAI
+
+    def test_explicit_ollama(self):
+        assert parse_provider(provider="ollama") == Provider.OLLAMA
+
+    def test_explicit_openrouter(self):
+        assert parse_provider(provider="openrouter") == Provider.OPENROUTER
+
+
+class TestRequireApiKey:
+    """require_api_key() returns the correct key or raises EnvironmentError."""
+
+    def test_returns_anthropic_key(self):
+        with patch.dict(os.environ, {"ANTHROPIC_API_KEY": "sk-ant-test"}, clear=True):
+            assert require_api_key(Provider.ANTHROPIC) == "sk-ant-test"
+
+    def test_returns_openai_key(self):
+        with patch.dict(os.environ, {"OPENAI_API_KEY": "sk-openai-test"}, clear=True):
+            assert require_api_key(Provider.OPENAI) == "sk-openai-test"
+
+    def test_returns_gemini_key(self):
+        with patch.dict(os.environ, {"GOOGLE_API_KEY": "gemini-test"}, clear=True):
+            assert require_api_key(Provider.GEMINI) == "gemini-test"
+
+    def test_raises_when_anthropic_key_missing(self):
+        env = {k: v for k, v in os.environ.items() if k != "ANTHROPIC_API_KEY"}
+        with patch.dict(os.environ, env, clear=True):
+            with pytest.raises(EnvironmentError, match="ANTHROPIC_API_KEY"):
+                require_api_key(Provider.ANTHROPIC)
+
+    def test_raises_when_openai_key_missing(self):
+        env = {k: v for k, v in os.environ.items() if k != "OPENAI_API_KEY"}
+        with patch.dict(os.environ, env, clear=True):
+            with pytest.raises(EnvironmentError, match="OPENAI_API_KEY"):
+                require_api_key(Provider.OPENAI)
+
+    def test_raises_when_gemini_key_missing(self):
+        env = {k: v for k, v in os.environ.items() if k != "GOOGLE_API_KEY"}
+        with patch.dict(os.environ, env, clear=True):
+            with pytest.raises(EnvironmentError, match="GOOGLE_API_KEY"):
+                require_api_key(Provider.GEMINI)
+
+    def test_returns_openrouter_key(self):
+        with patch.dict(os.environ, {"OPENROUTER_API_KEY": "sk-or-test"}, clear=True):
+            assert require_api_key(Provider.OPENROUTER) == "sk-or-test"
+
+    def test_raises_when_openrouter_key_missing(self):
+        env = {k: v for k, v in os.environ.items() if k != "OPENROUTER_API_KEY"}
+        with patch.dict(os.environ, env, clear=True):
+            with pytest.raises(EnvironmentError, match="OPENROUTER_API_KEY"):
+                require_api_key(Provider.OPENROUTER)
+
+    def test_ollama_does_not_require_a_key(self):
+        env = {k: v for k, v in os.environ.items() if k != "OLLAMA_API_KEY"}
+        with patch.dict(os.environ, env, clear=True):
+            assert require_api_key(Provider.OLLAMA) is None
+
+
+class TestOllamaClient:
+    """OllamaClient talks to a local Ollama server and needs no API key."""
+
+    def _fake_ollama_module(self):
+        fake_module = types.ModuleType("ollama")
+        fake_module.Client = MagicMock()
+        return fake_module
+
+    def test_no_api_key_required(self):
+        fake_ollama = self._fake_ollama_module()
+        with patch.dict(sys.modules, {"ollama": fake_ollama}):
+            client = OllamaClient()
+            assert client.api_key is None
+
+    def test_uses_default_host_when_env_var_not_set(self):
+        fake_ollama = self._fake_ollama_module()
+        env = {k: v for k, v in os.environ.items() if k != "OLLAMA_HOST"}
+        with patch.dict(os.environ, env, clear=True), patch.dict(sys.modules, {"ollama": fake_ollama}):
+            OllamaClient()
+            fake_ollama.Client.assert_called_once_with(host="http://localhost:11434")
+
+    def test_uses_ollama_host_env_var_when_set(self):
+        fake_ollama = self._fake_ollama_module()
+        with patch.dict(os.environ, {"OLLAMA_HOST": "http://remote-box:11434"}), \
+                patch.dict(sys.modules, {"ollama": fake_ollama}):
+            OllamaClient()
+            fake_ollama.Client.assert_called_once_with(host="http://remote-box:11434")
+
+    def test_create_message_returns_response_content(self):
+        fake_ollama = self._fake_ollama_module()
+        mock_client_instance = MagicMock()
+        mock_client_instance.chat.return_value = {"message": {"content": "hello from llama"}}
+        fake_ollama.Client.return_value = mock_client_instance
+
+        with patch.dict(sys.modules, {"ollama": fake_ollama}):
+            client = OllamaClient()
+            result = client.create_message("system prompt", "user message", model="llama3.2", max_tokens=50)
+
+        assert result == "hello from llama"
+        mock_client_instance.chat.assert_called_once_with(
+            model="llama3.2",
+            messages=[
+                {"role": "system", "content": "system prompt"},
+                {"role": "user", "content": "user message"}
+            ],
+            options={"num_predict": 50}
+        )
+
+
+def _fake_openai_module():
+    """A stand-in for the `openai` package so these tests don't require it
+    to actually be installed - it's an optional extra, and CI only installs
+    `.[dev]`."""
+    fake_module = types.ModuleType("openai")
+    fake_module.OpenAI = MagicMock()
+    return fake_module
+
+
+class TestOpenRouterClient:
+    """OpenRouterClient reuses OpenAI's SDK pointed at OpenRouter's base URL."""
+
+    def test_uses_openrouter_base_url(self):
+        fake_openai = _fake_openai_module()
+        with patch.dict(sys.modules, {"openai": fake_openai}):
+            OpenRouterClient(api_key="or-test-key")
+        fake_openai.OpenAI.assert_called_once_with(api_key="or-test-key", base_url="https://openrouter.ai/api/v1")
+
+    def test_create_message_returns_response_content(self):
+        fake_openai = _fake_openai_module()
+        mock_response = MagicMock()
+        mock_response.choices[0].message.content = "response text"
+        mock_client_instance = MagicMock()
+        mock_client_instance.chat.completions.create.return_value = mock_response
+        fake_openai.OpenAI.return_value = mock_client_instance
+
+        with patch.dict(sys.modules, {"openai": fake_openai}):
+            client = OpenRouterClient(api_key="or-test-key")
+            result = client.create_message(
+                "system prompt", "user message", model="deepseek/deepseek-v4-flash-0731:free", max_tokens=100
+            )
+
+        assert result == "response text"
+        mock_client_instance.chat.completions.create.assert_called_once_with(
+            model="deepseek/deepseek-v4-flash-0731:free",
+            messages=[
+                {"role": "system", "content": "system prompt"},
+                {"role": "user", "content": "user message"}
+            ],
+            max_tokens=100
+        )
+
+
+class TestCreateClientRouting:
+    """create_client() routes to the right client class for the new providers."""
+
+    def test_routes_ollama(self):
+        fake_ollama = types.ModuleType("ollama")
+        fake_ollama.Client = MagicMock()
+        with patch.dict(sys.modules, {"ollama": fake_ollama}):
+            client = create_client(Provider.OLLAMA, model="llama3.2")
+        assert isinstance(client, OllamaClient)
+
+    def test_routes_openrouter(self):
+        fake_openai = _fake_openai_module()
+        with patch.dict(os.environ, {"OPENROUTER_API_KEY": "sk-or-test"}, clear=True), \
+                patch.dict(sys.modules, {"openai": fake_openai}):
+            client = create_client(Provider.OPENROUTER, model="deepseek/deepseek-v4-flash-0731:free")
+        assert isinstance(client, OpenRouterClient)

bridgekit-0.3.10/tests/test_providers.py

@@ -1,71 +0,0 @@
-import os
-import pytest
-from unittest.mock import patch
-
-from bridgekit.config import parse_provider, require_api_key, Provider
-
-
-class TestParseProvider:
-    """parse_provider() returns the correct Provider enum."""
-
-    def test_explicit_anthropic(self):
-        assert parse_provider(provider="anthropic") == Provider.ANTHROPIC
-
-    def test_explicit_openai(self):
-        assert parse_provider(provider="openai") == Provider.OPENAI
-
-    def test_explicit_gemini(self):
-        assert parse_provider(provider="gemini") == Provider.GEMINI
-
-    def test_infer_anthropic_from_model(self):
-        assert parse_provider(model="claude-opus-4-6") == Provider.ANTHROPIC
-
-    def test_infer_openai_from_model(self):
-        assert parse_provider(model="gpt-4o") == Provider.OPENAI
-
-    def test_infer_gemini_from_model(self):
-        assert parse_provider(model="gemini-1.5-pro") == Provider.GEMINI
-
-    def test_defaults_to_anthropic(self):
-        assert parse_provider() == Provider.ANTHROPIC
-
-    def test_raises_on_unsupported_provider(self):
-        with pytest.raises(ValueError, match="Unsupported provider"):
-            parse_provider(provider="cohere")
-
-    def test_provider_takes_precedence_over_model(self):
-        assert parse_provider(provider="openai", model="claude-opus-4-6") == Provider.OPENAI
-
-
-class TestRequireApiKey:
-    """require_api_key() returns the correct key or raises EnvironmentError."""
-
-    def test_returns_anthropic_key(self):
-        with patch.dict(os.environ, {"ANTHROPIC_API_KEY": "sk-ant-test"}, clear=True):
-            assert require_api_key(Provider.ANTHROPIC) == "sk-ant-test"
-
-    def test_returns_openai_key(self):
-        with patch.dict(os.environ, {"OPENAI_API_KEY": "sk-openai-test"}, clear=True):
-            assert require_api_key(Provider.OPENAI) == "sk-openai-test"
-
-    def test_returns_gemini_key(self):
-        with patch.dict(os.environ, {"GOOGLE_API_KEY": "gemini-test"}, clear=True):
-            assert require_api_key(Provider.GEMINI) == "gemini-test"
-
-    def test_raises_when_anthropic_key_missing(self):
-        env = {k: v for k, v in os.environ.items() if k != "ANTHROPIC_API_KEY"}
-        with patch.dict(os.environ, env, clear=True):
-            with pytest.raises(EnvironmentError, match="ANTHROPIC_API_KEY"):
-                require_api_key(Provider.ANTHROPIC)
-
-    def test_raises_when_openai_key_missing(self):
-        env = {k: v for k, v in os.environ.items() if k != "OPENAI_API_KEY"}
-        with patch.dict(os.environ, env, clear=True):
-            with pytest.raises(EnvironmentError, match="OPENAI_API_KEY"):
-                require_api_key(Provider.OPENAI)
-
-    def test_raises_when_gemini_key_missing(self):
-        env = {k: v for k, v in os.environ.items() if k != "GOOGLE_API_KEY"}
-        with patch.dict(os.environ, env, clear=True):
-            with pytest.raises(EnvironmentError, match="GOOGLE_API_KEY"):
-                require_api_key(Provider.GEMINI)