delete: api; split out trace and query api from single api file

weave/__init__.py

@@ -13,25 +13,21 @@
 _loading_builtins_token = _context_state.set_loading_built_ins()
 
 from weave.legacy import weave_types as types
-from .legacy import storage
-
-from .legacy.api import *
-from .trace.api import *
-
-from .legacy.errors import *
-
+from weave.legacy import storage
+from weave.legacy.api import *
+from weave.legacy.errors import *
 from weave.legacy import mappers_python_def
-
 from weave.legacy import wandb_api as _wandb_api
+from weave.legacy import context as _context
 
-from . import version
+from weave import version
 
 _wandb_api.init()
 
 # Ensure there is a client available for eager mode
-from weave.legacy import context as _context
 
 
+from weave.trace.api import *
 _context_state.clear_loading_built_ins(_loading_builtins_token)
 
 __version__ = version.VERSION

weave/deploy/modal/stub.py

@@ -26,7 +26,7 @@
 @stub.function(image=image, secret=Secret.from_dotenv(__file__))
 @asgi_app(label=safe_name(uri.name))
 def fastapi_app() -> FastAPI:
-    from weave import api
+    from weave.trace import api
     from weave.trace.serve_fastapi import object_method_app
 
     uri_ref = parse_uri(os.environ["MODEL_REF"])

weave/tests/legacy/test_api.py

@@ -1,6 +1,7 @@
 import shutil
 
-from ... import api as weave
+from weave.legacy import api as weave
+
 from ...legacy.show import _show_params
 
 

weave/tests/legacy/test_arrow.py

@@ -7,6 +7,7 @@
 import pytest
 from PIL import Image
 
+from weave.legacy import api as weave
 from weave.legacy import (
     box,
     context_state,
@@ -36,7 +37,6 @@
 from weave.legacy.ops_primitives import list_, make_list
 from weave.tests import list_arrow_test_helpers as lath
 
-from ... import api as weave
 from ...tests import tag_test_util as ttu
 from ...tests import weavejs_ops
 from ...tests.legacy import test_wb

weave/tests/legacy/test_arrow_vectorizer.py

@@ -5,6 +5,7 @@
 import pytest
 from pyarrow import compute as pc
 
+from weave.legacy import api as weave
 from weave.legacy import box, dispatch, errors, ops, weave_internal
 from weave.legacy import ops_arrow as arrow
 from weave.legacy import weave_types as types
@@ -18,8 +19,6 @@
 from weave.legacy.ops_domain import wb_domain_types as wdt
 from weave.legacy.ops_primitives import Boolean, Number, date, dict_, list_
 
-from ... import api as weave
-
 string_ops_test_cases = [
     ("eq-scalar", lambda x: x == "bc", [True, False, False]),
     ("ne-scalar", lambda x: x != "bc", [False, True, True]),

weave/tests/legacy/test_async.py

@@ -1,8 +1,6 @@
 import pytest
 
-from weave.legacy import async_demo, ops, runs
-
-from ... import api, storage
+from weave.legacy import api, async_demo, ops, runs, storage
 
 
 def test_run_basic():

weave/tests/legacy/test_basic_ops.py

@@ -1,10 +1,9 @@
+from weave.legacy import api as weave
 from weave.legacy import box, ops
 from weave.legacy.ops_primitives import number
 from weave.legacy.ops_primitives.string import *
 from weave.legacy.weave_internal import make_const_node
 
-from ... import api as weave
-
 
 def test_number_ops():
     nine = make_const_node(weave.types.Number(), 9)

weave/tests/legacy/test_compile.py

@@ -4,14 +4,13 @@
 import weave
 from weave.legacy import async_demo, compile, graph
 from weave.legacy import weave_types as types
+from weave.legacy.api import use
 from weave.legacy.dispatch import RuntimeOutputNode
 from weave.legacy.ops_arrow import to_arrow
 from weave.legacy.ops_arrow.vectorize import raise_on_python_bailout
 from weave.legacy.wandb_interface.wandb_stream_table import StreamTable
 from weave.legacy.weave_internal import const, define_fn, make_const_node
 
-from ...api import use
-
 
 def test_automatic_await_compile():
     twelve = async_demo.slowmult(3, 4, 0.01)

weave/tests/legacy/test_custom_types.py

@@ -1,10 +1,10 @@
 import pytest
 from PIL import Image
 
+from weave.legacy import api as weave
 from weave.legacy import context_state as _context
 from weave.legacy import ops_arrow
 
-from ... import api as weave
 from ... import errors
 from .. import geom
 

weave/tests/legacy/test_decorators.py

@@ -1,9 +1,8 @@
+from weave.legacy import api as weave
 from weave.legacy import storage
 from weave.legacy import weave_types as types
 from weave.legacy.decorator_op import op
 
-from ... import api as weave
-
 
 def test_function_op_name():
     @op()

weave/tests/legacy/test_derive_op.py

@@ -1,4 +1,5 @@
-from ... import api as weave
+from weave.legacy import api as weave
+
 from ...legacy import registry_mem
 
 

weave/tests/legacy/test_examples.py

@@ -1,10 +1,9 @@
 import math
 import typing
 
+from weave.legacy import api as weave
 from weave.legacy import context, context_state
 
-from ... import api as weave
-
 
 class XOnly(typing.TypedDict):
     x: float

weave/tests/legacy/test_execute.py

@@ -4,10 +4,9 @@
 import pytest
 
 import weave
-from weave.legacy import environment, execute, ops, weave_internal
+from weave.legacy import api, environment, execute, ops, weave_internal
 from weave.legacy import weave_types as types
 
-from ... import api
 from . import test_wb
 
 execute_test_count_op_run_count = 0

weave/tests/legacy/test_file.py

@@ -1,9 +1,7 @@
 import pytest
 
 import weave
-from weave.legacy import context_state, environment, ops
-
-from ... import api, errors
+from weave.legacy import api, context_state, environment, errors, ops
 
 
 def test_dir():

weave/tests/legacy/test_list_arrow_compat.py

@@ -3,6 +3,7 @@
 import numpy as np
 import pytest
 
+from weave.legacy import api as weave
 from weave.legacy import box, ops, weave_internal
 from weave.legacy import ops_arrow as arrow
 from weave.legacy import weave_types as types
@@ -13,7 +14,6 @@
 )
 from weave.legacy.ops_primitives import dict_, list_
 
-from ... import api as weave
 from ...tests import tag_test_util as ttu
 from .. import list_arrow_test_helpers as lath
 

weave/tests/legacy/test_media_user.py

@@ -1,10 +1,9 @@
 from PIL import Image
 
+from weave.legacy import api as weave
 from weave.legacy import context_state
 from weave.legacy.ops_primitives import geom as media_user
 
-from ... import api as weave
-
 
 def test_im_with_metadata():
     base_im = Image.linear_gradient("L").resize((32, 32))

weave/tests/legacy/test_mutations.py

@@ -1,7 +1,6 @@
+from weave.legacy import api as weave
 from weave.legacy import ops, storage, weave_internal
 
-from ... import api as weave
-
 
 def test_autocommit(cereal_csv):
     csv = ops.local_path(cereal_csv).readcsv()

weave/tests/legacy/test_node_ref.py

@@ -1,6 +1,6 @@
+from weave.legacy import api as weave
 from weave.legacy import graph
 
-from ... import api as weave
 from ...legacy import node_ref
 
 

weave/tests/legacy/test_op_def.py

@@ -5,10 +5,9 @@
 
 import pytest
 
+from weave.legacy import api as weave
 from weave.legacy import context_state
 
-from ... import api as weave
-
 _loading_builtins_token = context_state.set_loading_built_ins()
 
 

weave/tests/legacy/test_partial_object.py

@@ -1,11 +1,11 @@
+from weave.legacy import api as weave
 from weave.legacy import compile, ops
 from weave.legacy import weave_types as types
 from weave.legacy.language_features.tagging.tagged_value_type import TaggedValueType
 from weave.legacy.ops_domain import wb_domain_types as wdt
 from weave.legacy.ops_domain.project_ops import root_all_projects
 from weave.legacy.ops_domain.report_ops import root_all_reports
 
-from ... import api as weave
 from .test_wb import table_mock1_no_display_name
 
 

weave/tests/legacy/test_run_segment.py

@@ -6,13 +6,11 @@
 import pytest
 
 import weave
-from weave.legacy import ops, storage, weave_internal
+from weave.legacy import api, ops, storage, weave_internal
 from weave.legacy import weave_types as types
 from weave.legacy.ops_arrow import ArrowWeaveList, arrow_as_array
 from weave.legacy.ops_domain.run_segment import RunSegment
 
-from ... import api
-
 N_NUMERIC_METRICS = 99  # number of numerical columns in the metrics table
 
 

weave/tests/legacy/test_storage.py

@@ -7,6 +7,7 @@
 import pytest
 import wandb
 
+from weave.legacy import api as weave
 from weave.legacy import (
     artifact_mem,
     artifact_wandb,
@@ -18,7 +19,6 @@
 from weave.legacy.arrow import list_ as arrow
 from weave.legacy.weave_internal import make_const_node
 
-from ... import api as weave
 from ...legacy.weavejs_fixes import recursively_unwrap_unions
 from . import test_helpers
 

weave/tests/legacy/test_table_ops.py

@@ -3,6 +3,7 @@
 
 import pytest
 
+from weave.legacy import api as weave
 from weave.legacy import (
     box,
     context,
@@ -15,7 +16,6 @@
 from weave.legacy import weave_types as types
 from weave.legacy.ops_domain import table as table_ops
 
-from ... import api as weave
 from .. import weavejs_ops
 
 TABLE_TYPES = ["list", "pandas", "sql"]

weave/tests/legacy/test_trace.py

@@ -1,9 +1,9 @@
 import re
 
+from weave.legacy import api as weave
 from weave.legacy import graph, storage
 from weave.legacy.weave_internal import make_const_node
 
-from ... import api as weave
 from ...legacy import trace_legacy
 
 

weave/tests/legacy/test_wb_domain_types.py

@@ -1,9 +1,8 @@
+from weave.legacy import api as weave
 from weave.legacy import storage
 from weave.legacy import weave_types as types
 from weave.legacy.ops_domain import wb_domain_types as wdt
 
-from ... import api as weave
-
 
 def test_with_keys_assignability():
     org_type = wdt.OrgType

weave/tests/legacy/test_weavejs_fixes.py

@@ -3,6 +3,7 @@
 import pytest
 
 from weave.legacy import (
+    api,
     context_state,
     mappers_python,
     ops,
@@ -11,8 +12,6 @@
 )
 from weave.legacy import weave_types as types
 
-from ... import api
-
 
 @pytest.mark.skip(
     "calling custom ops with graph engine is broken right now. Not needed for weaveflow"

weave/tests/list_arrow_test_helpers.py

@@ -1,8 +1,7 @@
+from weave.legacy import api as weave
 from weave.legacy import ops_arrow as arrow
 from weave.legacy.ops_primitives import list_
 
-from .. import api as weave
-
 
 class ListLikeNodeInterface:
     @staticmethod

weave/tests/trace/test_server.py

@@ -6,7 +6,7 @@
 import pytest
 import requests
 
-from weave import api as weave
+from weave.legacy import api as weave
 from weave.legacy import client as _client
 from weave.legacy import context_state, ops
 from weave.legacy import server as _server

weave/trace/cli.py

@@ -5,10 +5,9 @@
 
 from weave import __version__
 from weave.deploy import gcp as google
+from weave.trace import api
 from weave.trace.refs import ObjectRef, parse_uri
 
-from . import api
-
 # from .model_server import app
 
 # TODO: does this work?