Skip to content

Commit f80040a

Browse files
committed
LCORE-1831: update redaction test imports to use models.config
1 parent b404cdf commit f80040a

3 files changed

Lines changed: 6 additions & 6 deletions

File tree

tests/unit/pydantic_ai_lightspeed/capabilities/redaction/test_capability.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,10 @@
1313
from pydantic_ai.models import ModelRequestContext
1414
from pytest_mock import MockerFixture
1515

16+
from models.config import (
17+
RedactionConfig,
18+
RedactionRule,
19+
)
1620
from pydantic_ai_lightspeed.capabilities.redaction.capability import (
1721
PiiRedactionCapability,
1822
_redact_content_item,
@@ -23,10 +27,6 @@
2327
_redact_response,
2428
_redact_user_prompt_part,
2529
)
26-
from pydantic_ai_lightspeed.capabilities.redaction.config import (
27-
RedactionConfig,
28-
RedactionRule,
29-
)
3030
from pydantic_ai_lightspeed.capabilities.redaction.core import (
3131
CompiledPatterns,
3232
)

tests/unit/pydantic_ai_lightspeed/capabilities/redaction/test_config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
import pytest
66
from pydantic import ValidationError
77

8-
from pydantic_ai_lightspeed.capabilities.redaction.config import (
8+
from models.config import (
99
RedactionConfig,
1010
RedactionRule,
1111
)

tests/unit/pydantic_ai_lightspeed/capabilities/redaction/test_core.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,9 +5,9 @@
55
import pytest
66
from pydantic import ValidationError
77

8+
from models.config import RedactionResult
89
from pydantic_ai_lightspeed.capabilities.redaction.core import (
910
CompiledPatterns,
10-
RedactionResult,
1111
redact_text,
1212
)
1313

0 commit comments

Comments
 (0)