test

docs/scripts/generate_python_sdk_docs.py

@@ -265,10 +265,8 @@ def main():
     from weave import feedback
     from weave import weave_client as client
     from weave.trace import util
-    from weave.trace_server import (
-        remote_http_trace_server,
-        trace_server_interface,
-    )
+    from weave.trace.trace_server_bindings import remote_http_trace_server
+    from weave.trace_server import trace_server_interface
     from weave.trace_server.interface import query
 
     module_root_path = weave.__file__.split("/weave/__init__.py")[0]

weave/conftest.py

@@ -17,14 +17,9 @@
 from weave.legacy import client as client_legacy
 from weave.legacy import context_state, io_service, serialize
 from weave.legacy.language_features.tagging.tag_store import isolated_tagging_context
-from weave.trace_server import (
-    clickhouse_trace_server_batched,
-    remote_http_trace_server,
-    sqlite_trace_server,
-)
-from weave.trace_server import (
-    trace_server_interface as tsi,
-)
+from weave.trace.trace_server_bindings import remote_http_trace_server
+from weave.trace_server import clickhouse_trace_server_batched, sqlite_trace_server
+from weave.trace_server import trace_server_interface as tsi
 
 from . import autopatch, environment, logs
 from .tests import fixture_fakewandb

weave/trace_server/tests/test_remote_http_trace_server.py

@@ -5,9 +5,11 @@
 import requests
 from pydantic import ValidationError
 
+from weave.trace.trace_server_bindings.remote_http_trace_server import (
+    RemoteHTTPTraceServer,
+)
 from weave.trace_server import trace_server_interface as tsi
 from weave.trace_server.ids import generate_id
-from weave.trace_server.remote_http_trace_server import RemoteHTTPTraceServer
 
 
 def generate_start(id) -> tsi.StartedCallSchemaForInsert: