move to legacy: urls

weave/init_message.py

@@ -1,6 +1,6 @@
 import typing
 
-from weave import urls
+from weave.legacy import urls
 
 if typing.TYPE_CHECKING:
     import packaging.version  # type: ignore[import-not-found]

weave/legacy/artifact_wandb.py

@@ -15,11 +15,11 @@
 
 from weave import (
     filesystem,
-    urls,
     util,
 )
 from weave import weave_types as types
 from weave.legacy import (
+    urls,
     errors,
     engine_trace,
     artifact_fs,

weave/legacy/context.py

@@ -4,8 +4,8 @@
 import typing
 from urllib.parse import urlparse
 
-from weave import urls, util
-from weave.legacy import client, context_state
+from weave import util
+from weave.legacy import urls, client, context_state
 from weave.legacy.client_interface import ClientInterface
 
 

weave/trace/api.py

@@ -7,13 +7,13 @@
 from typing import Any, Callable, Iterator, Optional, Union
 
 from weave.client_context import weave_client as weave_client_context
+from weave.legacy import urls
 from weave.trace.call_context import get_current_call
 
 # TODO: type_serializers is imported here to trigger registration of the image serializer.
 # There is probably a better place for this, but including here for now to get the fix in.
 from .. import (
     type_serializers,  # noqa: F401
-    urls,
     util,
     weave_init,
 )

weave/trace/weave_client.py

@@ -9,8 +9,9 @@
 import pydantic
 from requests import HTTPError
 
-from weave import urls, version
+from weave import version
 from weave.client_context import weave_client as weave_client_context
+from weave.legacy import urls
 from weave.trace import call_context, trace_sentry
 from weave.trace.exception import exception_to_json_str
 from weave.trace.feedback import FeedbackQuery, RefFeedbackQuery