stream_data_interfaces

wandb · Aug 21, 2024 · 15af662 · 15af662
1 parent 63a4745
commit 15af662
Show file tree

Hide file tree

Showing 6 changed files with 7 additions and 5 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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 → weave/legacy/stream_data_interfaces.py b/weave/stream_data_interfaces.py → weave/legacy/stream_data_interfaces.py