arize-phoenix 3.10.0__py3-none-any.whl → 3.11.1__py3-none-any.whl

phoenix/session/session.py

@@ -6,6 +6,7 @@ from abc import ABC, abstractmethod
 from collections import UserList
 from datetime import datetime
 from enum import Enum
+from importlib.util import find_spec
 from pathlib import Path
 from tempfile import TemporaryDirectory
 from typing import (
@@ -619,9 +620,11 @@ def _is_colab() -> bool:
 
 def _is_sagemaker() -> bool:
     """Determines whether this is in a SageMaker notebook"""
+    if find_spec("sagemaker") is None:
+        return False
     try:
-        import sagemaker  # type: ignore # noqa: F401
-    except ImportError:
+        _get_sagemaker_notebook_base_url()
+    except Exception:
         return False
     try:
         from IPython.core.getipython import get_ipython

phoenix/trace/__init__.py

@@ -1,3 +1,6 @@
+import contextlib
+from typing import Iterator
+
 from .span_evaluations import DocumentEvaluations, Evaluations, SpanEvaluations, TraceEvaluations
 from .trace_dataset import TraceDataset
 
@@ -8,3 +11,16 @@ __all__ = [
     "DocumentEvaluations",
     "TraceEvaluations",
 ]
+
+
+@contextlib.contextmanager
+def suppress_tracing() -> Iterator[None]:
+    """Context manager to pause OpenTelemetry instrumentation."""
+    try:
+        from opentelemetry.context import _SUPPRESS_INSTRUMENTATION_KEY, attach, detach, set_value
+    except ImportError:
+        yield
+        return
+    token = attach(set_value(_SUPPRESS_INSTRUMENTATION_KEY, True))
+    yield
+    detach(token)

phoenix/trace/exporter.py

@@ -4,10 +4,9 @@ import weakref
 from queue import SimpleQueue
 from threading import Thread
 from types import MethodType
-from typing import Any, Optional, Union
+from typing import Any, Optional
 from urllib.parse import urljoin
 
-import opentelemetry.proto.trace.v1.trace_pb2 as otlp
 import requests
 from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
 from requests import Session
@@ -15,15 +14,13 @@ from typing_extensions import TypeAlias, assert_never
 
 import phoenix.trace.v1 as pb
 from phoenix.config import get_env_collector_endpoint, get_env_host, get_env_port
-from phoenix.trace.otel import encode
-from phoenix.trace.schemas import Span
 
 logger = logging.getLogger(__name__)
 logger.addHandler(logging.NullHandler())
 
 END_OF_QUEUE = None  # sentinel value for queue termination
 
-Message: TypeAlias = Union[otlp.Span, pb.Evaluation]
+Message: TypeAlias = pb.Evaluation
 
 
 class NoOpExporter:
@@ -52,7 +49,7 @@ class HttpExporter:
         port: Optional[int] = None,
     ) -> None:
         """
-        Span/Evaluation Exporter using HTTP.
+        Evaluation Exporter using HTTP.
 
         Parameters
         ----------
@@ -90,10 +87,8 @@ class HttpExporter:
         weakref.finalize(self, self._queue.put, END_OF_QUEUE)
         self._start_consumer()
 
-    def export(self, item: Union[Span, pb.Evaluation]) -> None:
-        if isinstance(item, Span):
-            self._queue.put(encode(item))
-        elif isinstance(item, pb.Evaluation):
+    def export(self, item: pb.Evaluation) -> None:
+        if isinstance(item, pb.Evaluation):
             self._queue.put(item)
         else:
             logger.exception(f"unrecognized item type: {type(item)}")
@@ -121,8 +116,6 @@ class HttpExporter:
             logger.exception(e)
 
     def _url(self, message: Message) -> str:
-        if isinstance(message, otlp.Span):
-            return urljoin(self._base_url, "/v1/spans")
         if isinstance(message, pb.Evaluation):
             return urljoin(self._base_url, "/v1/evaluations")
         logger.exception(f"unrecognized message type: {type(message)}")

phoenix/trace/langchain/__init__.py

@@ -1,26 +1,3 @@
-import sys
-from typing import Any
+from phoenix.trace.langchain.instrumentor import LangChainInstrumentor
 
-import phoenix.trace.langchain.instrumentor as _instrumentor
-import phoenix.trace.langchain.tracer as _tracer
-
-_DUMMY = "OpenInferenceTracer"
-
-
-class _Deprecation:
-    __all__ = ("LangChainInstrumentor", _DUMMY)
-
-    def __getattr__(self, name: str) -> Any:
-        if name == "tracer":
-            return _tracer
-        if name == "instrumentor":
-            return _instrumentor
-        if name == _DUMMY:
-            return getattr(_tracer, name)
-        if name == "LangChainInstrumentor":
-            return _instrumentor.LangChainInstrumentor
-        raise AttributeError(f"module {__name__} has no attribute {name}")
-
-
-# See e.g. https://stackoverflow.com/a/7668273
-sys.modules[__name__] = _Deprecation()  # type: ignore
+__all__ = ("LangChainInstrumentor",)

phoenix/trace/langchain/instrumentor.py

@@ -11,7 +11,6 @@ from opentelemetry.sdk.trace.export import SimpleSpanProcessor
 
 from phoenix.config import get_env_project_name
 from phoenix.trace.exporter import _OpenInferenceExporter
-from phoenix.trace.tracer import _show_deprecation_warnings
 
 logger = logging.getLogger(__name__)
 
@@ -21,7 +20,6 @@ __all__ = ("LangChainInstrumentor",)
 
 class LangChainInstrumentor(Instrumentor):
     def __init__(self, *args: Any, **kwargs: Any) -> None:
-        _show_deprecation_warnings(self, *args, **kwargs)
         if find_spec("langchain_core") is None:
             raise PackageNotFoundError(
                 "Missing `langchain-core`. Install with `pip install langchain-core`."

phoenix/trace/llama_index/callback.py

@@ -11,7 +11,6 @@ from opentelemetry.sdk.trace.export import SimpleSpanProcessor
 from phoenix.config import get_env_project_name
 from phoenix.trace.errors import IncompatibleLibraryVersionError
 from phoenix.trace.exporter import _OpenInferenceExporter
-from phoenix.trace.tracer import _show_deprecation_warnings
 
 logger = logging.getLogger(__name__)
 
@@ -95,7 +94,6 @@ class OpenInferenceTraceCallbackHandler(_OpenInferenceTraceCallbackHandler):
     """
 
     def __init__(self, *args: Any, **kwargs: Any) -> None:
-        _show_deprecation_warnings(self, *args, **kwargs)
         tracer_provider = trace_sdk.TracerProvider(
             resource=Resource({ResourceAttributes.PROJECT_NAME: get_env_project_name()})
         )

phoenix/trace/openai/instrumentor.py

@@ -11,14 +11,12 @@ from opentelemetry.sdk.trace.export import SimpleSpanProcessor
 
 from phoenix.config import get_env_project_name
 from phoenix.trace.exporter import _OpenInferenceExporter
-from phoenix.trace.tracer import _show_deprecation_warnings
 
 logger = logging.getLogger(__name__)
 
 
 class OpenAIInstrumentor(Instrumentor):
     def __init__(self, *args: Any, **kwargs: Any) -> None:
-        _show_deprecation_warnings(self, *args, **kwargs)
         if find_spec("openai") is None:
             raise PackageNotFoundError("Missing `openai`. Install with `pip install openai`.")
         super().__init__()

phoenix/version.py

@@ -1 +1 @@
-__version__ = "3.10.0"
+__version__ = "3.11.1"

phoenix/trace/langchain/tracer.py

@@ -1,47 +0,0 @@
-"""
-This module is defunct and will be removed in the future. It's currently
-maintaining a dummy class to avoid breaking any import code.
-"""
-
-import logging
-import sys
-from typing import Any, Iterator
-
-from phoenix.trace.schemas import Span
-
-logger = logging.getLogger(__name__)
-
-_DUMMY = "OpenInferenceTracer"
-_DEPRECATION_MESSAGE = (
-    f"`{__name__}.{_DUMMY}` is a defunct class in the current version of Phoenix, "
-    "and will be removed in the future. For a migration guide, see "
-    "https://github.com/Arize-ai/phoenix/blob/main/MIGRATION.md"
-)
-
-
-class _DummyObject:
-    def __init__(self, *_: Any, **__: Any) -> None:
-        logger.warning(_DEPRECATION_MESSAGE)
-
-    def get_spans(self) -> Iterator[Span]:
-        logger.warning(_DEPRECATION_MESSAGE)
-        logger.warning(
-            "`.get_spans()` is a defunct method that does nothing, and will be removed "
-            "in the future. For a migration guide, see "
-            "https://github.com/Arize-ai/phoenix/blob/main/MIGRATION.md"
-        )
-        return iter(())
-
-
-class _DefunctModule:
-    __all__ = (_DUMMY,)
-
-    def __getattr__(self, name: str) -> Any:
-        if name == _DUMMY:
-            logger.warning(_DEPRECATION_MESSAGE)
-            return _DummyObject
-        raise AttributeError(f"module {__name__} has no attribute {name}")
-
-
-# See e.g. https://stackoverflow.com/a/7668273
-sys.modules[__name__] = _DefunctModule()  # type: ignore

phoenix/trace/tracer.py

@@ -1,99 +0,0 @@
-"""
-This module is defunct and will be removed in the future. It's currently
-maintaining a dummy class to avoid breaking any import code.
-"""
-
-import logging
-import sys
-from typing import Any, Iterator, Protocol
-
-from phoenix.trace.exporter import HttpExporter
-
-logger = logging.getLogger(__name__)
-
-
-class SpanExporter(Protocol):
-    def export(self, _: Any) -> None: ...
-
-
-_DEFUNCT_MSG = (
-    "`phoenix.trace.tracer.Tracer` is defunct in the current version of Phoenix, "
-    "and will be removed in the future. For a migration guide, see "
-    "https://github.com/Arize-ai/phoenix/blob/main/MIGRATION.md"
-)
-
-_USE_ENV_MSG = """
-Setting the Phoenix endpoint via HttpExporter() is no longer supported.
-Please use environment variables instead:
-  - os.environ["PHOENIX_HOST"] = "127.0.0.1"
-  - os.environ["PHOENIX_PORT"] = "54321"
-  - os.environ["PHOENIX_COLLECTOR_ENDPOINT"] = "http://127.0.0.1:54321
-For a migration guide, see https://github.com/Arize-ai/phoenix/blob/main/MIGRATION.md
-"""
-
-_ON_APPEND_DEPRECATION_MSG = (
-    "OpenInference has been updated for full OpenTelemetry compliance. The ability to set "
-    "`on_append` callbacks are removed. For a migration guide, see "
-    "https://github.com/Arize-ai/phoenix/blob/main/MIGRATION.md"
-)
-
-
-def _show_deprecation_warnings(obj: object, *args: Any, **kwargs: Any) -> None:
-    if args or kwargs:
-        logger.warning(
-            f"{obj.__class__.__name__}() no longer takes any arguments. "
-            "The arguments provided has been ignored. For a migration guide, "
-            "see https://github.com/Arize-ai/phoenix/blob/main/MIGRATION.md"
-        )
-        if any(callable(arg) for arg in args) or "callback" in kwargs:
-            logger.warning(
-                "The `callback` argument is defunct and no longer has any effect. "
-                "If you need access to spans for processing, some options include "
-                "exporting spans from Phoenix or adding a SpanProcessor to the "
-                "OpenTelemetry TracerProvider. For a migration guide, "
-                "see https://github.com/Arize-ai/phoenix/blob/main/MIGRATION.md"
-            )
-        if any(isinstance(arg, HttpExporter) for arg in args):
-            logger.warning(_USE_ENV_MSG)
-
-
-class Tracer:
-    _exporter: Any
-
-    def __init__(self, exporter: Any = None, on_append: Any = None) -> None:
-        logger.warning(_DEFUNCT_MSG)
-        if exporter is not None:
-            logger.warning(_USE_ENV_MSG)
-        if on_append is not None:
-            logger.warning(_ON_APPEND_DEPRECATION_MSG)
-
-    def create_span(self, *_: Any, **__: Any) -> Any:
-        logger.warning(_DEFUNCT_MSG)
-
-    def get_spans(self) -> Iterator[Any]:
-        logger.warning(_DEFUNCT_MSG)
-        logger.warning(
-            ".get_spans() is a defunct method that does nothing. "
-            "It will be removed in the future. For a migration guide, "
-            "see https://github.com/Arize-ai/phoenix/blob/main/MIGRATION.md"
-        )
-        return iter(())
-
-
-class _DefunctModule:
-    __all__ = ("Tracer", "SpanExporter")
-
-    def __getattr__(self, name: str) -> Any:
-        if name == "Tracer":
-            logger.warning(_DEFUNCT_MSG)
-            return Tracer
-        if name == "SpanExporter":
-            logger.warning("`SpanExporter` is defunct and will be removed in the future.")
-            return SpanExporter
-        if name == "_show_deprecation_warnings":
-            return _show_deprecation_warnings
-        raise AttributeError(f"module {__name__} has no attribute {name}")
-
-
-# See e.g. https://stackoverflow.com/a/7668273
-sys.modules[__name__] = _DefunctModule()  # type: ignore