Merge branch 'pydantic:main' into main

pydantic · Jan 22, 2025 · 7c6f716 · 7c6f716
2 parents 243db14 + 3b524b7
commit 7c6f716
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/pydantic_ai_slim/pydantic_ai/models/__init__.py b/pydantic_ai_slim/pydantic_ai/models/__init__.py
@@ -61,6 +61,7 @@
     'mistral:codestral-latest',
     'mistral:mistral-moderation-latest',
     'ollama:codellama',
+    'ollama:deepseek-r1',
     'ollama:gemma',
     'ollama:gemma2',
     'ollama:llama3',

diff --git a/pydantic_ai_slim/pydantic_ai/models/ollama.py b/pydantic_ai_slim/pydantic_ai/models/ollama.py
@@ -26,6 +26,7 @@
 
 CommonOllamaModelNames = Literal[
     'codellama',
+    'deepseek-r1',
     'gemma',
     'gemma2',
     'llama3',