From 25d0be77b242b0cd075e6874f110d4f3b55bda65 Mon Sep 17 00:00:00 2001
From: Andrew Truong <itsandrewtruong@gmail.com>
Date: Wed, 21 Aug 2024 08:58:47 -0400
Subject: [PATCH] stream_data_interfaces

---
 weave/engine_trace.py                        | 3 ++-
 weave/legacy/monitoring/langchain.py         | 2 +-
 weave/legacy/monitoring/monitor.py           | 3 ++-
 weave/legacy/ops_domain/trace_tree.py        | 2 +-
 weave/legacy/run_streamtable_span.py         | 2 +-
 weave/{ => legacy}/stream_data_interfaces.py | 0
 6 files changed, 7 insertions(+), 5 deletions(-)
 rename weave/{ => legacy}/stream_data_interfaces.py (100%)

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