Skip to content

Commit 703f6e1

Browse files
authored
fix: add 'openrouter' string to OpenAIModel provider param (#1792)
1 parent e402772 commit 703f6e1

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

pydantic_ai_slim/pydantic_ai/models/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -491,7 +491,7 @@ def infer_model(model: Model | KnownModelName | str) -> Model:
491491
from .cohere import CohereModel
492492

493493
return CohereModel(model_name, provider=provider)
494-
elif provider in ('deepseek', 'openai', 'azure'):
494+
elif provider in ('deepseek', 'openai', 'azure', 'openrouter'):
495495
from .openai import OpenAIModel
496496

497497
return OpenAIModel(model_name, provider=provider)

pydantic_ai_slim/pydantic_ai/models/openai.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -170,7 +170,7 @@ def __init__(
170170
self,
171171
model_name: OpenAIModelName,
172172
*,
173-
provider: Literal['openai', 'deepseek', 'azure'] | Provider[AsyncOpenAI] = 'openai',
173+
provider: Literal['openai', 'deepseek', 'azure', 'openrouter'] | Provider[AsyncOpenAI] = 'openai',
174174
system_prompt_role: OpenAISystemPromptRole | None = None,
175175
):
176176
"""Initialize an OpenAI model.

0 commit comments

Comments
 (0)