move to trace: serve_fastapi

wandb · Aug 28, 2024 · f8759ac · f8759ac
1 parent e741c0d
commit f8759ac
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/weave/deploy/modal/stub.py b/weave/deploy/modal/stub.py
@@ -27,7 +27,7 @@
 @asgi_app(label=safe_name(uri.name))
 def fastapi_app() -> FastAPI:
     from weave import api
-    from weave.serve_fastapi import object_method_app
+    from weave.trace.serve_fastapi import object_method_app
 
     uri_ref = parse_uri(os.environ["MODEL_REF"])
     if not isinstance(uri_ref, ObjectRef):

diff --git a/weave/trace/api.py b/weave/trace/api.py
@@ -209,7 +209,7 @@ def serve(
 
     from weave.legacy import wandb_api
 
-    from ..serve_fastapi import object_method_app
+    from .serve_fastapi import object_method_app
 
     client = weave_client_context.require_weave_client()
     # if not isinstance(

diff --git a/weave/serve_fastapi.py → weave/trace/serve_fastapi.py b/weave/serve_fastapi.py → weave/trace/serve_fastapi.py
@@ -17,8 +17,8 @@
 from weave.trace.op import Op
 from weave.trace.refs import ObjectRef
 
-from . import errors
-from .legacy import weave_pydantic
+from .. import errors
+from ..legacy import weave_pydantic
 
 key_cache: cache.LruTimeWindowCache[str, typing.Optional[bool]] = (
     cache.LruTimeWindowCache(datetime.timedelta(minutes=5))