split: make separate trace and legacy utils

weave/trace/api.py

@@ -6,7 +6,8 @@
 import time
 from typing import Any, Callable, Iterator, Optional, Union
 
-from weave.legacy.weave import urls, util
+from weave.legacy.weave import urls
+from weave.trace import util
 from weave.trace.call_context import get_current_call
 from weave.trace.client_context import weave_client as weave_client_context
 

weave/trace/feedback.py

@@ -5,8 +5,7 @@
 
 from rich.table import Table
 
-from weave.legacy.weave import util
-from weave.trace import rich_pydantic_util
+from weave.trace import rich_pydantic_util, util
 from weave.trace.client_context import weave_client as weave_client_context
 from weave.trace.refs import parse_uri
 from weave.trace.rich_container import AbstractRichContainer

weave/trace/rich_pydantic_util.py

@@ -6,7 +6,7 @@
 from rich.console import Console
 from rich.table import Table
 
-from weave.legacy.weave import util
+from weave.trace import util
 
 
 def dict_to_table(d: dict[str, Any]) -> Table:

weave/trace/util.py

@@ -94,6 +94,31 @@ def run(self) -> None:
         self.context.run(super().run)
 
 
+def is_colab():  # type: ignore
+    import importlib
+
+    spec = importlib.util.find_spec("google.colab")
+    return bool(spec)
+
+
+def is_notebook() -> bool:
+    if is_colab():  # type: ignore[no-untyped-call]
+        return True
+    try:
+        from IPython import get_ipython
+    except ImportError:
+        return False
+    else:
+        ip = get_ipython()
+        if ip is None:
+            return False
+        if "IPKernelApp" not in ip.config:
+            return False
+        # if "VSCODE_PID" in os.environ:
+        #     return False
+    return True
+
+
 # rename for cleaner export
 ThreadPoolExecutor = ContextAwareThreadPoolExecutor
 Thread = ContextAwareThread