Skip to content

Commit f3fd801

Browse files
committed
Fix IsStr imports - use conftest version for type safety
1 parent 10c24a6 commit f3fd801

File tree

3 files changed

+4
-6
lines changed

3 files changed

+4
-6
lines changed

tests/test_agent.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010

1111
import httpx
1212
import pytest
13-
from dirty_equals import IsJson, IsStr
13+
from dirty_equals import IsJson
1414
from inline_snapshot import snapshot
1515
from pydantic import BaseModel, TypeAdapter, field_validator
1616
from pydantic_core import to_json
@@ -66,7 +66,7 @@
6666
from pydantic_ai.tools import DeferredToolRequests, DeferredToolResults, ToolDefinition, ToolDenied
6767
from pydantic_ai.usage import RequestUsage
6868

69-
from .conftest import IsDatetime, IsNow, TestEnv
69+
from .conftest import IsDatetime, IsNow, IsStr, TestEnv
7070

7171
pytestmark = pytest.mark.anyio
7272

tests/test_streaming.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@
1010
from typing import Any
1111

1212
import pytest
13-
from dirty_equals import IsStr
1413
from inline_snapshot import snapshot
1514
from pydantic import BaseModel
1615

@@ -51,7 +50,7 @@
5150
from pydantic_ai.usage import RequestUsage
5251
from pydantic_graph import End
5352

54-
from .conftest import IsDatetime, IsInt, IsNow
53+
from .conftest import IsDatetime, IsInt, IsNow, IsStr
5554

5655
pytestmark = pytest.mark.anyio
5756

tests/test_tools.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,6 @@
77
import pydantic_core
88
import pytest
99
from _pytest.logging import LogCaptureFixture
10-
from dirty_equals import IsStr
1110
from inline_snapshot import snapshot
1211
from pydantic import BaseModel, Field, TypeAdapter, WithJsonSchema
1312
from pydantic.json_schema import GenerateJsonSchema, JsonSchemaValue
@@ -39,7 +38,7 @@
3938
from pydantic_ai.tools import DeferredToolRequests, DeferredToolResults, ToolApproved, ToolDefinition, ToolDenied
4039
from pydantic_ai.usage import RequestUsage
4140

42-
from .conftest import IsDatetime
41+
from .conftest import IsDatetime, IsStr
4342

4443

4544
def test_tool_no_ctx():

0 commit comments

Comments
 (0)