move to legacy: weave_types

examples/experimental/skip_test/branching_runs.ipynb

@@ -13,7 +13,7 @@
     "from weave.legacy.ops_domain import RunSegment\n",
     "from weave.legacy import storage\n",
     "from weave import publish, type_of\n",
-    "from weave.weave_types import List\n",
+    "from weave.legacy.weave_types import List\n",
     "import typing\n",
     "import time\n",
     "import sys\n",

mypy.ini

@@ -151,7 +151,7 @@ disallow_untyped_calls = False
 disallow_untyped_defs = False
 disallow_untyped_calls = False
 
-[mypy-weave.weave_types]
+[mypy-weave.legacy.weave_types]
 disallow_untyped_defs = False
 disallow_untyped_calls = False
 

weave/__init__.py

@@ -12,7 +12,7 @@
 
 _loading_builtins_token = _context_state.set_loading_built_ins()
 
-from . import weave_types as types
+from weave.legacy import weave_types as types
 from .legacy import storage
 
 from .legacy.api import *

weave/legacy/_dict_utils.py

@@ -4,7 +4,7 @@
 from weave.legacy.language_features.tagging import tag_store, tagged_value_type
 
 from weave.legacy import errors
-from .. import weave_types as types
+from weave.legacy import weave_types as types
 
 
 def typeddict_pick_output_type(input_types):

weave/legacy/api.py

@@ -21,7 +21,7 @@
 from ..trace import weave_client as _weave_client
 
 # exposed as part of api
-from .. import weave_types as types
+from weave.legacy import weave_types as types
 
 # needed to enable automatic numpy serialization
 from . import types_numpy as _types_numpy

weave/legacy/arrow/arrow.py

@@ -12,7 +12,7 @@
 py_type = type
 
 from weave.legacy import errors
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 from weave.legacy import artifact_fs, partial_object
 
 

weave/legacy/arrow/arrow_tags.py

@@ -3,7 +3,7 @@
 import pyarrow as pa
 from pyarrow import compute as pc
 
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 from weave.legacy.arrow import convert
 from weave.legacy.arrow.arrow import offsets_starting_at_zero
 from weave.legacy.language_features.tagging import (

weave/legacy/arrow/concat.py

@@ -13,7 +13,7 @@
 from pyarrow import compute as pc
 
 from weave.legacy import errors
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 from weave.legacy.arrow.list_ import (
     ArrowWeaveList,
     ArrowWeaveListGeneric,

weave/legacy/arrow/constructors.py

@@ -4,7 +4,7 @@
 import pyarrow as pa
 
 from weave.legacy import errors
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 from weave.legacy import artifact_base, box
 from weave.legacy.arrow import arrow_tags, convert
 from weave.legacy.arrow.arrow import ArrowWeaveListType, arrow_as_array

weave/legacy/arrow/convert.py

@@ -5,7 +5,7 @@
 
 from weave.legacy import api as api
 from weave import weave_internal
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 from weave.legacy import errors, arrow_util, artifact_base, artifact_mem, box, mappers_arrow
 from weave.legacy.arrow.arrow import (
     ArrowWeaveListType,

weave/legacy/arrow/list_.py

@@ -13,7 +13,7 @@
 from weave import (
     weave_internal,
 )
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 from weave.legacy import (
     errors,
     _dict_utils,

weave/legacy/artifact_base.py

@@ -7,7 +7,7 @@
 
 import typing
 
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 from weave.legacy import ref_base
 
 

weave/legacy/artifact_fs.py

@@ -7,7 +7,7 @@
 import typing
 
 from weave.legacy import errors
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 from weave.legacy import artifact_base, file_base, object_context, ref_base, uris
 from weave.legacy.language_features.tagging import tag_store
 from weave.trace import ref_util

weave/legacy/artifact_local.py

@@ -10,7 +10,7 @@
 from datetime import datetime
 
 from weave.legacy import filesystem
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 from weave.legacy import (
     artifact_fs,
     artifact_wandb,

weave/legacy/artifact_mem.py

@@ -1,7 +1,7 @@
 import typing
 
 from . import artifact_base
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 from weave.legacy import errors
 from weave.legacy import ref_base
 

weave/legacy/artifact_wandb.py

@@ -13,7 +13,7 @@
 from wandb.apis.public import api as wb_public
 from wandb.sdk.lib.hashutil import b64_to_hex_id, hex_to_b64_id
 
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 from weave.legacy import (
     filesystem,
     urls,

weave/legacy/client.py

@@ -1,4 +1,4 @@
-from weave import weave_types
+from weave.legacy import weave_types
 from weave.legacy import storage
 
 

weave/legacy/codify.py

@@ -6,7 +6,7 @@
 
 import black
 
-from weave import weave_types
+from weave.legacy import weave_types
 from weave.legacy import storage, graph, registry_mem
 
 from . import codifiable_value_mixin
@@ -128,7 +128,7 @@ def _try_otc_using_dataclasses(obj: typing.Any) -> typing.Optional[str]:
     if class_type.__module__.startswith("weave.decorator_type") and issubclass(
         class_type, weave_types.Type
     ):
-        qualified_classpath = "weave.weave_types"
+        qualified_classpath = "weave.legacy.weave_types"
         qualified_classname = f"type_name_to_type('{class_type.name}')"
     else:
         qualified_classpath = _module_name_corrections(class_type.__module__)

weave/legacy/compile.py

@@ -8,7 +8,7 @@
 from weave import (
     weave_internal,
 )
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 from weave.legacy import (
     errors,
     engine_trace,

weave/legacy/compile_domain.py

@@ -3,7 +3,7 @@
 import graphql
 
 from weave.legacy import errors
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 from weave.legacy import gql_op_plugin, gql_to_weave, graph, op_args, stitch, registry_mem
 from weave.legacy.input_provider import InputAndStitchProvider
 

weave/legacy/debug_types.py

@@ -1,7 +1,7 @@
 import textwrap
 
 from weave.legacy.language_features.tagging import tagged_value_type
-from weave.weave_types import *
+from weave.legacy.weave_types import *
 
 
 def short_type(type: Type) -> str:

weave/legacy/decorator_arrow_op.py

@@ -1,6 +1,6 @@
 import typing
 
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 from weave.legacy import op_def
 from weave.legacy.arrow.arrow import ArrowWeaveListType
 from weave.legacy.language_features.tagging import tagged_value_type

weave/legacy/decorator_class.py

@@ -2,7 +2,7 @@
 import typing
 
 from weave.legacy import errors
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 from weave.legacy import context_state, derive_op, op_def, registry_mem
 
 # Contrary to the way it is read, the weave.class() decorator runs AFTER the

weave/legacy/decorator_op.py

@@ -4,7 +4,7 @@
 
 from typing_extensions import ParamSpec
 
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 from weave.legacy import context_state, derive_op, op_args, op_def, pyfunc_type_util, registry_mem
 
 if typing.TYPE_CHECKING:

weave/legacy/decorator_type.py

@@ -3,7 +3,7 @@
 import typing
 
 from weave.legacy import errors
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 from weave.legacy import context_state, decorator_class, object_type_ref_util, infer_types
 
 _py_type = type

weave/legacy/derive_op.py

@@ -10,7 +10,7 @@
     storage,
     weave_internal,
 )
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 from weave.legacy import (
     errors,
     parallelism,

weave/legacy/dispatch.py

@@ -6,7 +6,7 @@
 from dataclasses import dataclass
 
 from weave.legacy import util
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 from weave.legacy import graph, memo, op_args, op_def, pyfunc_type_util, registry_mem, errors
 from weave.legacy.language_features.tagging.is_tag_getter import is_tag_getter
 from weave.legacy.language_features.tagging.tagged_value_type import TaggedValueType

weave/legacy/eager.py

@@ -3,7 +3,7 @@
 from weave.legacy import context_state, graph
 
 from .. import weave_internal
-from .. import weave_types as types
+from weave.legacy import weave_types as types
 
 WeaveIterObjectType = typing.TypeVar("WeaveIterObjectType")
 

weave/legacy/ecosystem/langchain/lc.py

@@ -46,7 +46,7 @@
 from weave.legacy.artifact_fs import FilesystemArtifact
 from weave.legacy.ecosystem.langchain import util
 from weave.legacy.ops_domain import trace_tree
-from weave.weave_types import Type
+from weave.legacy.weave_types import Type
 
 
 class WeaveTracer(BaseTracer):

weave/legacy/execute.py

@@ -9,7 +9,7 @@
 import typing
 from collections.abc import Mapping
 
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 
 # Configuration
 # Libraries

weave/legacy/execute_fast.py

@@ -3,7 +3,7 @@
 from weave import (
     weave_internal,
 )
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 from weave.legacy import (
     errors,
     engine_trace,

weave/legacy/file_base.py

@@ -2,7 +2,7 @@
 import dataclasses
 import typing
 
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 
 TRACE_EXT = "trace.json"
 

weave/legacy/file_local.py

@@ -2,7 +2,7 @@
 import os
 import typing
 
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 
 from . import file_base, file_util
 

weave/legacy/gql_op_plugin.py

@@ -1,7 +1,7 @@
 import typing
 from dataclasses import dataclass
 
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 from weave.legacy import input_provider, op_args, op_def
 
 # A GQLOutputTypeFn is a function that is called during the refinement phase of the compile pass

weave/legacy/gql_to_weave.py

@@ -18,7 +18,7 @@
 )
 
 from weave.legacy import errors
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 from weave.legacy import gql_schema
 
 

weave/legacy/graph.py

@@ -2,7 +2,7 @@
 import json
 import typing
 
-from weave import weave_types
+from weave.legacy import weave_types
 from weave.legacy import uris, errors, storage
 
 if typing.TYPE_CHECKING:

weave/legacy/graph_debug.py

@@ -2,7 +2,7 @@
 import textwrap
 import typing
 
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 from weave.legacy import forward_graph, graph
 
 

weave/legacy/graph_mapper.py

@@ -1,5 +1,5 @@
 from weave.legacy import ref_base
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 from weave.legacy import graph, mappers, node_ref
 from weave.legacy import mappers_python_def as mappers_python
 

weave/legacy/infer_types.py

@@ -8,7 +8,7 @@
 
 from weave.legacy import errors, graph
 
-from .. import weave_types
+from weave.legacy import weave_types
 
 
 class TypedDictLike:

weave/legacy/language_autocall.py

@@ -15,7 +15,7 @@
 
 import typing
 
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 from weave.legacy import op_args
 
 

weave/legacy/language_features/tagging/is_tag_getter.py

@@ -1,6 +1,6 @@
 import typing
 
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 from weave.legacy import op_args
 from weave.legacy.language_features.tagging import tagged_value_type
 

weave/legacy/language_features/tagging/make_tag_getter_op.py

@@ -1,6 +1,6 @@
 import typing
 
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 from weave.legacy import context_state as _context_state
 from weave.legacy import decorator_op
 from weave.legacy.language_features.tagging import tag_store, tagged_value_type

weave/legacy/language_features/tagging/opdef_util.py

@@ -1,6 +1,6 @@
 import typing
 
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 from weave.legacy.language_features.tagging import tagged_value_type
 
 if typing.TYPE_CHECKING:

weave/legacy/language_features/tagging/process_opdef_output_type.py

@@ -6,7 +6,7 @@
 import typing
 
 from weave.legacy import registry_mem
-from weave import weave_types as types
+from weave.legacy import weave_types as types
 from weave.legacy import graph
 from weave.legacy.language_features.tagging.opdef_util import (
     should_flow_tags,