From 257446aee01cc8a377804e3cdfbaa23fe37f774f Mon Sep 17 00:00:00 2001 From: Andrew Truong Date: Thu, 22 Aug 2024 22:59:52 -0400 Subject: [PATCH] move to legacy: weave_types --- weave/{ => legacy}/weave_pydantic.py | 4 ++-- weave/serve_fastapi.py | 3 ++- weave/weave_types.py | 2 +- 3 files changed, 5 insertions(+), 4 deletions(-) rename weave/{ => legacy}/weave_pydantic.py (93%) diff --git a/weave/weave_pydantic.py b/weave/legacy/weave_pydantic.py similarity index 93% rename from weave/weave_pydantic.py rename to weave/legacy/weave_pydantic.py index 19d200ad2fd1..086feb8d2fa4 100644 --- a/weave/weave_pydantic.py +++ b/weave/legacy/weave_pydantic.py @@ -1,7 +1,7 @@ from pydantic import BaseModel, create_model -from . import weave_types as types -from .legacy import infer_types +from .. import weave_types as types +from . import infer_types def weave_type_to_pydantic( diff --git a/weave/serve_fastapi.py b/weave/serve_fastapi.py index d6b23431349d..ca63901d03b4 100644 --- a/weave/serve_fastapi.py +++ b/weave/serve_fastapi.py @@ -17,7 +17,8 @@ from weave.trace.op import Op from weave.trace.refs import ObjectRef -from . import errors, weave_pydantic +from . import errors +from .legacy import weave_pydantic key_cache: cache.LruTimeWindowCache[str, typing.Optional[bool]] = ( cache.LruTimeWindowCache(datetime.timedelta(minutes=5)) diff --git a/weave/weave_types.py b/weave/weave_types.py index 990ec2586a86..bee5df4bd115 100644 --- a/weave/weave_types.py +++ b/weave/weave_types.py @@ -1057,7 +1057,7 @@ def property_types(self) -> dict[str, Type]: @classmethod def typeclass_of_class(cls, check_class): - from . import weave_pydantic + from .legacy import weave_pydantic if not issubclass(check_class, pydantic.BaseModel): return cls