Wrap pydantic.dataclass; split interfaces.py into parts

.gitignore

@@ -21,6 +21,9 @@ __pycache__
 /*_data.json
 /*_embeddings.bin
 
+# pytest configuration
+pytest.local.ini
+
 # Coverage
 .coverage
 .coverage.*

typeagent/knowpro/answer_response_schema.py

@@ -3,7 +3,8 @@
 
 from typing import Literal, Annotated
 from typing_extensions import Doc
-from pydantic.dataclasses import dataclass
+
+from .dataclasses import dataclass
 
 AnswerType = Literal[
     "NoAnswer",  # If question cannot be accurately answered from [ANSWER CONTEXT]

typeagent/knowpro/dataclasses.py

@@ -0,0 +1,35 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
+"""Compatibility helpers for pydantic dataclasses."""
+
+from collections.abc import Callable
+from typing import Any, cast, overload
+
+from typing_extensions import dataclass_transform
+
+from pydantic.dataclasses import dataclass as _pydantic_dataclass
+
+from .field_helpers import CamelCaseField
+
+
+@overload
+def dataclass[T](__cls: type[T], /, **kwargs: Any) -> type[T]: ...
+
+
+@overload
+def dataclass[T](**kwargs: Any) -> Callable[[type[T]], type[T]]: ...
+
+
+@dataclass_transform(field_specifiers=(CamelCaseField,))
+def dataclass[T](
+    __cls: type[T] | None = None, /, **kwargs: Any
+) -> Callable[[type[T]], type[T]] | type[T]:
+    """Wrapper that preserves pydantic behavior while informing type-checkers."""
+
+    def wrap(cls: type[T]) -> type[T]:
+        return cast(type[T], _pydantic_dataclass(cls, **kwargs))
+
+    if __cls is None:
+        return wrap
+
+    return wrap(__cls)

typeagent/knowpro/date_time_schema.py

@@ -1,10 +1,11 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
 
-from pydantic.dataclasses import dataclass
 from typing import Annotated
 from typing_extensions import Doc
 
+from .dataclasses import dataclass
+
 
 @dataclass
 class DateVal: