diff --git a/weave/engine_trace.py b/weave/engine_trace.py index 2039c714473..7423b8048ca 100644 --- a/weave/engine_trace.py +++ b/weave/engine_trace.py @@ -18,7 +18,8 @@ import time import typing -from . import environment, logs, stream_data_interfaces +from . import environment, logs +from .legacy import stream_data_interfaces # Thanks co-pilot! diff --git a/weave/legacy/monitoring/langchain.py b/weave/legacy/monitoring/langchain.py index f5e50205cf7..8e3afddc148 100644 --- a/weave/legacy/monitoring/langchain.py +++ b/weave/legacy/monitoring/langchain.py @@ -11,7 +11,7 @@ from langchain.callbacks.tracers.schemas import Run from wandb.sdk.data_types import trace_tree -from weave import stream_data_interfaces +from weave.legacy import stream_data_interfaces from weave.legacy.monitoring import StreamTable diff --git a/weave/legacy/monitoring/monitor.py b/weave/legacy/monitoring/monitor.py index b586e0f4565..69fe290a023 100644 --- a/weave/legacy/monitoring/monitor.py +++ b/weave/legacy/monitoring/monitor.py @@ -11,10 +11,11 @@ import typing import uuid -from weave import call_context, errors, stream_data_interfaces +from weave import call_context, errors from weave.legacy import ( graph, run_streamtable_span, + stream_data_interfaces ) from weave.legacy.wandb_interface.wandb_stream_table import StreamTable from weave.trace import context as trace_context diff --git a/weave/legacy/ops_domain/trace_tree.py b/weave/legacy/ops_domain/trace_tree.py index d9c4ae5d3f3..e4f618eafa8 100644 --- a/weave/legacy/ops_domain/trace_tree.py +++ b/weave/legacy/ops_domain/trace_tree.py @@ -12,7 +12,7 @@ from wandb.sdk.data_types.trace_tree import Span as WBSpan from weave import query_api as weave -from weave import stream_data_interfaces +from weave.legacy import stream_data_interfaces from weave import weave_types as types from weave.legacy import op_def from weave.legacy.decorator_op import op diff --git a/weave/legacy/run_streamtable_span.py b/weave/legacy/run_streamtable_span.py index f59249fc0e7..269187d8b0b 100644 --- a/weave/legacy/run_streamtable_span.py +++ b/weave/legacy/run_streamtable_span.py @@ -1,7 +1,7 @@ import typing from typing import Iterable -from weave import stream_data_interfaces +from weave.legacy import stream_data_interfaces from weave.eager import WeaveIter from weave.legacy import artifact_wandb, uris from weave.legacy.run import Run diff --git a/weave/stream_data_interfaces.py b/weave/legacy/stream_data_interfaces.py similarity index 100% rename from weave/stream_data_interfaces.py rename to weave/legacy/stream_data_interfaces.py