Jul 16, 2025 · Jun 4, 2025 · Jun 4, 2025 · Jun 4, 2025 · Jun 4, 2025 · Jun 9, 2025
diff --git a/pydantic_ai_slim/pydantic_ai/models/__init__.py b/pydantic_ai_slim/pydantic_ai/models/__init__.py
        'groq:llama-3.2-3b-preview',
        'groq:llama-3.2-11b-vision-preview',
        'groq:llama-3.2-90b-vision-preview',
        'heroku:claude-3-5-haiku',
        'heroku:claude-3-5-sonnet-latest',
        'heroku:claude-3-7-sonnet',
        'heroku:claude-4-sonnet',
        'heroku:claude-3-haiku',
        'huggingface:Qwen/QwQ-32B',
        'huggingface:Qwen/Qwen2.5-72B-Instruct',
        'huggingface:Qwen/Qwen3-235B-A22B',
diff --git a/pydantic_ai_slim/pydantic_ai/providers/__init__.py b/pydantic_ai_slim/pydantic_ai/providers/__init__.py
        from .cohere import CohereProvider

        return CohereProvider()

    elif provider == 'grok':
        from .grok import GrokProvider

        return GrokProvider()
    elif provider == 'fireworks':
        from .fireworks import FireworksProvider

        return FireworksProvider()
    elif provider == 'together':
        from .together import TogetherProvider

        return TogetherProvider()
    elif provider == 'heroku':
        from .heroku import HerokuProvider

        return HerokuProvider()
    elif provider == 'huggingface':
        from .huggingface import HuggingFaceProvider

diff --git a/tests/conftest.py b/tests/conftest.py
    return os.getenv('HF_TOKEN', 'mock-api-key') or os.getenv('HUGGINGFACE_API_KEY', 'mock-api-key')


 @pytest.fixture(scope='session')
 def heroku_inference_key() -> str:
    return os.getenv('HEROKU_INFERENCE_KEY', 'mock-api-key')


 @pytest.fixture(scope='session')
 def bedrock_provider():
    try:
diff --git a/tests/models/test_model_names.py b/tests/models/test_model_names.py
    bedrock_names = [f'bedrock:{n}' for n in get_model_names(BedrockModelName)]
    deepseek_names = ['deepseek:deepseek-chat', 'deepseek:deepseek-reasoner']
    huggingface_names = [f'huggingface:{n}' for n in get_model_names(HuggingFaceModelName)]
    heroku_names = get_heroku_model_names()
    extra_names = ['test']

    generated_names = sorted(
        + bedrock_names
        + deepseek_names
        + huggingface_names
        + heroku_names
        + extra_names
    )
Original file line number	Diff line number	Diff line change
Expand Up		@@ -211,6 +211,11 @@
		'groq:llama-3.2-3b-preview',
		'groq:llama-3.2-11b-vision-preview',
		'groq:llama-3.2-90b-vision-preview',
		'heroku:claude-3-5-haiku',
		'heroku:claude-3-5-sonnet-latest',
		'heroku:claude-3-7-sonnet',
		'heroku:claude-4-sonnet',
		'heroku:claude-3-haiku',
		'huggingface:Qwen/QwQ-32B',
		'huggingface:Qwen/Qwen2.5-72B-Instruct',
		'huggingface:Qwen/Qwen3-235B-A22B',
Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -95,6 +95,23 @@ def infer_provider(provider: str) -> Provider[Any]: # noqa: C901
		from .cohere import CohereProvider

		return CohereProvider()

		elif provider == 'grok':
		from .grok import GrokProvider

		return GrokProvider()
		elif provider == 'fireworks':
		from .fireworks import FireworksProvider

		return FireworksProvider()
		elif provider == 'together':
		from .together import TogetherProvider

		return TogetherProvider()
		elif provider == 'heroku':
		from .heroku import HerokuProvider

		return HerokuProvider()
		elif provider == 'huggingface':
		from .huggingface import HuggingFaceProvider

Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -286,6 +286,11 @@ def huggingface_api_key() -> str:
		return os.getenv('HF_TOKEN', 'mock-api-key') or os.getenv('HUGGINGFACE_API_KEY', 'mock-api-key')


		@pytest.fixture(scope='session')
		def heroku_inference_key() -> str:
		return os.getenv('HEROKU_INFERENCE_KEY', 'mock-api-key')


		@pytest.fixture(scope='session')
		def bedrock_provider():
		try:
Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -56,6 +56,7 @@ def get_model_names(model_name_type: Any) -> Iterator[str]:
		bedrock_names = [f'bedrock:{n}' for n in get_model_names(BedrockModelName)]
		deepseek_names = ['deepseek:deepseek-chat', 'deepseek:deepseek-reasoner']
		huggingface_names = [f'huggingface:{n}' for n in get_model_names(HuggingFaceModelName)]
		heroku_names = get_heroku_model_names()
		extra_names = ['test']

		generated_names = sorted(
Expand All		@@ -68,6 +69,7 @@ def get_model_names(model_name_type: Any) -> Iterator[str]:
		+ bedrock_names
		+ deepseek_names
		+ huggingface_names
		+ heroku_names
		+ extra_names
		)

Expand Down