pydantic-ai 0.4.1__tar.gz → 0.4.3__tar.gz

{pydantic_ai-0.4.1 → pydantic_ai-0.4.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydantic-ai
-Version: 0.4.1
+Version: 0.4.3
 Summary: Agent Framework / shim to use Pydantic with LLMs
 Project-URL: Homepage, https://ai.pydantic.dev
 Project-URL: Source, https://github.com/pydantic/pydantic-ai
@@ -28,11 +28,11 @@ Classifier: Topic :: Internet
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: >=3.9
-Requires-Dist: pydantic-ai-slim[anthropic,bedrock,cli,cohere,evals,google,groq,mcp,mistral,openai,vertexai]==0.4.1
+Requires-Dist: pydantic-ai-slim[anthropic,bedrock,cli,cohere,evals,google,groq,huggingface,mcp,mistral,openai,vertexai]==0.4.3
 Provides-Extra: a2a
-Requires-Dist: fasta2a==0.4.1; extra == 'a2a'
+Requires-Dist: fasta2a>=0.4.1; extra == 'a2a'
 Provides-Extra: examples
-Requires-Dist: pydantic-ai-examples==0.4.1; extra == 'examples'
+Requires-Dist: pydantic-ai-examples==0.4.3; extra == 'examples'
 Provides-Extra: logfire
 Requires-Dist: logfire>=3.11.0; extra == 'logfire'
 Description-Content-Type: text/markdown

{pydantic_ai-0.4.1 → pydantic_ai-0.4.3}/pyproject.toml

@@ -46,13 +46,13 @@ requires-python = ">=3.9"
 
 [tool.hatch.metadata.hooks.uv-dynamic-versioning]
 dependencies = [
-    "pydantic-ai-slim[openai,vertexai,google,groq,anthropic,mistral,cohere,bedrock,cli,mcp,evals]=={{ version }}",
+    "pydantic-ai-slim[openai,vertexai,google,groq,anthropic,mistral,cohere,bedrock,huggingface,cli,mcp,evals]=={{ version }}",
 ]
 
 [tool.hatch.metadata.hooks.uv-dynamic-versioning.optional-dependencies]
 examples = ["pydantic-ai-examples=={{ version }}"]
 logfire = ["logfire>=3.11.0"]
-a2a = ["fasta2a=={{ version }}"]
+a2a = ["fasta2a>=0.4.1"]
 
 [project.urls]
 Homepage = "https://ai.pydantic.dev"
@@ -64,11 +64,11 @@ Changelog = "https://github.com/pydantic/pydantic-ai/releases"
 pai = "pydantic_ai._cli:cli_exit" # TODO remove this when clai has been out for a while
 
 [tool.uv.sources]
+pydantic-ai = { workspace = true }
 pydantic-ai-slim = { workspace = true }
 pydantic-evals = { workspace = true }
 pydantic-graph = { workspace = true }
 pydantic-ai-examples = { workspace = true }
-fasta2a = { workspace = true }
 
 [tool.uv.workspace]
 members = [
@@ -78,7 +78,6 @@ members = [
     "mcp-run-python",
     "clai",
     "examples",
-    "fasta2a",
 ]
 
 [dependency-groups]
@@ -109,7 +108,6 @@ include = [
     "pydantic_evals/**/*.py",
     "pydantic_graph/**/*.py",
     "mcp-run-python/**/*.py",
-    "fasta2a/**/*.py",
     "examples/**/*.py",
     "clai/**/*.py",
     "tests/**/*.py",
@@ -128,7 +126,6 @@ extend-select = [
     "TID251",
 ]
 flake8-quotes = { inline-quotes = "single", multiline-quotes = "double" }
-isort = { combine-as-imports = true, known-first-party = ["pydantic_ai"] }
 mccabe = { max-complexity = 15 }
 ignore = [
     "D100", # ignore missing docstring in module
@@ -138,6 +135,12 @@ ignore = [
     "D107", # ignore missing docstring in __init__ methods
 ]
 
+[tool.ruff.lint.isort]
+combine-as-imports = true
+known-first-party = ["pydantic_ai"]
+# weird issue with ruff thinking fasta2a is still editable
+known-third-party = ["fasta2a"]
+
 [tool.ruff.lint.pydocstyle]
 convention = "google"
 
@@ -167,7 +170,6 @@ include = [
     "pydantic_evals",
     "pydantic_graph",
     "mcp-run-python",
-    "fasta2a",
     "tests",
     "examples",
     "clai",
@@ -191,10 +193,7 @@ files = "tests/typed_agent.py,tests/typed_graph.py"
 strict = true
 
 [tool.pytest.ini_options]
-testpaths = [
-    "tests",
-    "docs/.hooks"
-]
+testpaths = ["tests", "docs/.hooks"]
 xfail_strict = true
 filterwarnings = [
     "error",
@@ -218,8 +217,6 @@ include = [
     "pydantic_ai_slim/**/*.py",
     "pydantic_evals/**/*.py",
     "pydantic_graph/**/*.py",
-    # TODO(Marcelo): Add 100% coverage for A2A.
-    # "fasta2a/**/*.py",
     "tests/**/*.py",
 ]
 omit = [
@@ -228,6 +225,20 @@ omit = [
     "pydantic_ai_slim/pydantic_ai/ext/aci.py", # aci-sdk requires Python 3.10+ so cannot be added as an (optional) dependency
 ]
 branch = true
+# Disable include-ignored warnings as --source is enabled automatically causing a self conflict as per:
+# https://github.com/pytest-dev/pytest-cov/issues/532
+# https://github.com/pytest-dev/pytest-cov/issues/369
+# This prevents coverage being generated by pytest-cov which has direct editor support in VS Code,
+# making it super useful to check coverage while writing tests.
+disable_warnings = ["include-ignored"]
+
+[tool.coverage.paths]
+# Allow CI run assets to be downloaded an replicated locally.
+source = [
+    ".",
+    "/home/runner/work/pydantic-ai/pydantic-ai",
+    "/System/Volumes/Data/home/runner/work/pydantic-ai/pydantic-ai"
+]
 
 # https://coverage.readthedocs.io/en/latest/config.html#report
 [tool.coverage.report]

{pydantic_ai-0.4.1 → pydantic_ai-0.4.3}/tests/conftest.py

@@ -316,6 +316,11 @@ def openrouter_api_key() -> str:
     return os.getenv('OPENROUTER_API_KEY', 'mock-api-key')
 
 
+@pytest.fixture(scope='session')
+def huggingface_api_key() -> str:
+    return os.getenv('HF_TOKEN', 'hf_token')
+
+
 @pytest.fixture(scope='session')
 def heroku_inference_key() -> str:
     return os.getenv('HEROKU_INFERENCE_KEY', 'mock-api-key')
@@ -398,6 +403,7 @@ def model(
     groq_api_key: str,
     co_api_key: str,
     gemini_api_key: str,
+    huggingface_api_key: str,
     bedrock_provider: BedrockProvider,
 ) -> Model:  # pragma: lax no cover
     try:
@@ -440,6 +446,14 @@ def model(
             from pydantic_ai.models.bedrock import BedrockConverseModel
 
             return BedrockConverseModel('us.amazon.nova-micro-v1:0', provider=bedrock_provider)
+        elif request.param == 'huggingface':
+            from pydantic_ai.models.huggingface import HuggingFaceModel
+            from pydantic_ai.providers.huggingface import HuggingFaceProvider
+
+            return HuggingFaceModel(
+                'Qwen/Qwen2.5-72B-Instruct',
+                provider=HuggingFaceProvider(provider_name='nebius', api_key=huggingface_api_key),
+            )
         else:
             raise ValueError(f'Unknown model: {request.param}')
     except ImportError: