arize-phoenix 2.11.1__py3-none-any.whl → 3.0.1__py3-none-any.whl

{arize_phoenix-2.11.1.dist-info → arize_phoenix-3.0.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: arize-phoenix
-Version: 2.11.1
+Version: 3.0.1
 Summary: ML Observability in your notebook
 Project-URL: Documentation, https://docs.arize.com/phoenix/
 Project-URL: Issues, https://github.com/Arize-ai/phoenix/issues
@@ -20,6 +20,10 @@ Requires-Dist: ddsketch
 Requires-Dist: hdbscan<1.0.0,>=0.8.33
 Requires-Dist: jinja2
 Requires-Dist: numpy
+Requires-Dist: openinference-instrumentation-langchain
+Requires-Dist: openinference-instrumentation-llama-index
+Requires-Dist: openinference-instrumentation-openai
+Requires-Dist: opentelemetry-exporter-otlp
 Requires-Dist: opentelemetry-proto
 Requires-Dist: opentelemetry-sdk
 Requires-Dist: pandas
@@ -50,10 +54,10 @@ Requires-Dist: llama-index>=0.9.14; extra == 'dev'
 Requires-Dist: nbqa; extra == 'dev'
 Requires-Dist: pandas-stubs<=2.0.2.230605; extra == 'dev'
 Requires-Dist: pre-commit; extra == 'dev'
-Requires-Dist: pytest; extra == 'dev'
 Requires-Dist: pytest-asyncio; extra == 'dev'
 Requires-Dist: pytest-cov; extra == 'dev'
 Requires-Dist: pytest-lazy-fixture; extra == 'dev'
+Requires-Dist: pytest==7.4.4; extra == 'dev'
 Requires-Dist: ruff==0.1.5; extra == 'dev'
 Requires-Dist: strawberry-graphql[debug-server]==0.208.2; extra == 'dev'
 Provides-Extra: experimental
@@ -103,22 +107,22 @@ Phoenix provides MLOps and LLMOps insights at lightning speed with zero-config o
 
 **Table of Contents**
 
-- [Installation](#installation)
-- [LLM Traces](#llm-traces)
-  - [Tracing with LlamaIndex](#tracing-with-llamaindex)
-  - [Tracing with LangChain](#tracing-with-langchain)
-- [LLM Evals](#llm-evals)
-- [Embedding Analysis](#embedding-analysis)
-  - [UMAP-based Exploratory Data Analysis](#umap-based-exploratory-data-analysis)
-  - [Cluster-driven Drift and Performance Analysis](#cluster-driven-drift-and-performance-analysis)
-  - [Exportable Clusters](#exportable-clusters)
-- [Retrieval-Augmented Generation Analysis](#retrieval-augmented-generation-analysis)
-- [Structured Data Analysis](#structured-data-analysis)
-- [Deploying Phoenix](#deploying-phoenix)
-- [Breaking Changes](#breaking-changes)
-- [Community](#community)
-- [Thanks](#thanks)
-- [Copyright, Patent, and License](#copyright-patent-and-license)
+-   [Installation](#installation)
+-   [LLM Traces](#llm-traces)
+    -   [Tracing with LlamaIndex](#tracing-with-llamaindex)
+    -   [Tracing with LangChain](#tracing-with-langchain)
+-   [LLM Evals](#llm-evals)
+-   [Embedding Analysis](#embedding-analysis)
+    -   [UMAP-based Exploratory Data Analysis](#umap-based-exploratory-data-analysis)
+    -   [Cluster-driven Drift and Performance Analysis](#cluster-driven-drift-and-performance-analysis)
+    -   [Exportable Clusters](#exportable-clusters)
+-   [Retrieval-Augmented Generation Analysis](#retrieval-augmented-generation-analysis)
+-   [Structured Data Analysis](#structured-data-analysis)
+-   [Deploying Phoenix](#deploying-phoenix)
+-   [Breaking Changes](#breaking-changes)
+-   [Community](#community)
+-   [Thanks](#thanks)
+-   [Copyright, Patent, and License](#copyright-patent-and-license)
 
 ## Installation
 
@@ -219,9 +223,8 @@ session = px.launch_app()
 
 from phoenix.trace.langchain import OpenInferenceTracer, LangChainInstrumentor
 
-# If no exporter is specified, the tracer will export to the locally running Phoenix server
-tracer = OpenInferenceTracer()
-LangChainInstrumentor(tracer).instrument()
+# By default, the traces will be exported to the locally running Phoenix server.
+LangChainInstrumentor().instrument()
 
 # Initialize your LangChain application
 from langchain.chains import RetrievalQA
@@ -456,7 +459,7 @@ For more information on deploying Phoenix, see the [Phoenix Deployment Guide](ht
 
 ## Breaking Changes
 
--   **v1.0.0** - Phoenix now exclusively supports the `openai>=1.0.0` sdk. If you are using an older version of the OpenAI SDK, you can continue to use `arize-phoenix==0.1.1`. However, we recommend upgrading to the latest version of the OpenAI SDK as it contains many improvements. If you are using Phoenix with LlamaIndex and and LangChain, you will have to upgrade to the versions of these packages that support the OpenAI `1.0.0` SDK as well (`llama-index>=0.8.64`, `langchain>=0.0.334`)
+see the [migration guide](./MIGRATION.md) for a list of breaking changes.
 
 ## Community
 

{arize_phoenix-2.11.1.dist-info → arize_phoenix-3.0.1.dist-info}/RECORD

@@ -1,10 +1,10 @@
 phoenix/__init__.py,sha256=MSamw7OMCIcCtrJdvJYMvQmC6O0uyehfPAXXELsllZM,1451
-phoenix/config.py,sha256=RbQw8AkVyI4SSo5CD520AjUNcwkDNOGZA6_ErE48R7A,3454
+phoenix/config.py,sha256=RT4UR5VH8JUAKQhfmlkpxq2kmBSA8G8MODrC3GXWyaI,3507
 phoenix/datetime_utils.py,sha256=D955QLrkgrrSdUM6NyqbCeAu2SMsjhR5rHVQEsVUdng,2773
 phoenix/exceptions.py,sha256=X5k9ipUDfwSCwZB-H5zFJLas86Gf9tAx0W4l5TZxp5k,108
 phoenix/py.typed,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 phoenix/services.py,sha256=f6AeyKTuOpy9RCcTCjVH3gx5nYZhbTMFOuv1WSUOB5o,4992
-phoenix/version.py,sha256=BYffn7HbxSh9b7MJ8cUy0hnjLOsSSOSFFKXqE11oMj4,23
+phoenix/version.py,sha256=E3P6AbnCwaWk6ndR1zNqlOTVebX9z5rv9voltc71dos,22
 phoenix/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/core/embedding_dimension.py,sha256=zKGbcvwOXgLf-yrJBpQyKtd-LEOPRKHnUToyAU8Owis,87
 phoenix/core/evals.py,sha256=gJyqQzpud5YjtoY8h4pgXvHDsdubGfqmEewLuZHPPmQ,10224
@@ -139,39 +139,38 @@ phoenix/session/session.py,sha256=mrTAQkUFMGjN65ApYqRjhGXExbfxu2ODJcJ14MSDOLE,21
 phoenix/trace/__init__.py,sha256=4d_MqzUIFmlY9WWcFeTONJ4xL5mPGoWZaPM2TJ0ZDBQ,266
 phoenix/trace/errors.py,sha256=DbXSJnNErV7305tKv7pUWLD6jcVHJ6EBdSu4mZJ6IM4,112
 phoenix/trace/evaluation_conventions.py,sha256=t8jydM3U0-T5YpiQKRJ3tWdWGlHtzKyttYdw-ddvPOk,1048
-phoenix/trace/exporter.py,sha256=7Ir-6gEwcTrkT78190DysVkIS6WSDTiVzsO7PfdFOCI,4411
+phoenix/trace/exporter.py,sha256=O-9Arn-S_B9Me-jy4Qa84y6lvxKNoa8pczrsamPl3_Q,4871
 phoenix/trace/fixtures.py,sha256=LokNedhbGYxpzXznteO4m5QehvNYjzvoh231-CMJQeY,7113
-phoenix/trace/otel.py,sha256=9oum5RPCsEZvKg41mEy8aKDcXHBwtR-P9eeqEXp-ts4,14642
+phoenix/trace/otel.py,sha256=lJoTvkCSFiPT4dFRLMZZgZfsOoTSATctW9lQk9J30ZI,14821
 phoenix/trace/schemas.py,sha256=QDBlM94faceTnloY_ZVo2BHWWnE5UGymefC0jmfj4ew,6011
 phoenix/trace/semantic_conventions.py,sha256=u6NG85ZhbreriZr8cqJaddldM_jUcew7JilszY7JUk8,4652
 phoenix/trace/span_evaluations.py,sha256=FvY9YRnKuYIzCa-H9P5SuDaI2DeqGnVCjNgclC2v3HA,12909
 phoenix/trace/span_json_decoder.py,sha256=nrIPkcgbCcNML-0OSjWC6fxIfBEMiP0n67yM_m-vegg,3068
 phoenix/trace/span_json_encoder.py,sha256=C5y7rkyOcV08oJC5t8TZqVxsKCZMJKad7bBQzAgLoDs,1763
 phoenix/trace/trace_dataset.py,sha256=KW0TzmhlKuX8PUPLV172iTK08myYE0QXUC75KiIqJ7k,13204
-phoenix/trace/tracer.py,sha256=AoYyWRco-EcvK7TASmZO0z-nJEm3cXlG9lhTWDTz4VU,3691
+phoenix/trace/tracer.py,sha256=JDKlyvjy6AsQmaA60ycJ1hKXoUQU61jqPx3nvYr8xUc,3647
 phoenix/trace/utils.py,sha256=7LurVGXn245cjj4MJsc7v6jq4DSJkpK6YGBfIaSywuw,1307
 phoenix/trace/dsl/__init__.py,sha256=WIQIjJg362XD3s50OsPJJ0xbDsGp41bSv7vDllLrPuA,144
 phoenix/trace/dsl/filter.py,sha256=VwlzL4KsIoyYjjCSKXlOSyVm49WyYRlkK5zedXqZlcc,13136
 phoenix/trace/dsl/helpers.py,sha256=qxRS65NFUFVpLDd8IZVrZUH891MamLeXyfIIN1BJJ_c,1552
 phoenix/trace/dsl/missing.py,sha256=BWPOHr2_tBkPDgVeq8GVXXVbNbJiBelu4NtwHBg6mTE,1435
 phoenix/trace/dsl/query.py,sha256=W6_e1o_JbucGWWv6pSUQFyIHcTuj7QuCBSuKXodRfCE,14773
-phoenix/trace/langchain/__init__.py,sha256=vAjrmrreetV7L5IL8VH_9efG9VJunJTgT0iKyWqjFbc,148
-phoenix/trace/langchain/instrumentor.py,sha256=HkNKbFNclTYjRXBM8qU4qvZHdyw06J9bhwgE7JnqbNI,1323
-phoenix/trace/langchain/tracer.py,sha256=1Oz3orSDpZX1pZKwtZbeM_f9tiAhQb7Of8ARjRlKVQY,16827
+phoenix/trace/langchain/__init__.py,sha256=aTKMFmEOgjx_6dnyplalgYi7PQnetablwrwpcUZqcGE,764
+phoenix/trace/langchain/instrumentor.py,sha256=rmqdaEh2rwyZnaddns4RC2WQnj7TbCnJjETgoiwWvzs,1163
+phoenix/trace/langchain/tracer.py,sha256=fSjtPc5VxaZK63x3ob3DAW7eM2kfOZWgbSQ9dmhhtsw,1424
 phoenix/trace/llama_index/__init__.py,sha256=wCcQgD9CG5TA8i-1XsSed4ZzwHTUmqZwegQAV_FqEng,178
-phoenix/trace/llama_index/callback.py,sha256=cSa5whoaMDdBc7W2QSWWatMoNL-wKU2fozkP8prpUMQ,27563
+phoenix/trace/llama_index/callback.py,sha256=MHgUNJSG6ecxZ8SZ_9LdcKnAAWmNI-U5pySFyXqubjM,1694
 phoenix/trace/llama_index/debug_callback.py,sha256=SKToD9q_QADSGTJ5lhilqRVKaUnUSRXUvURCzN4by2U,1367
-phoenix/trace/llama_index/streaming.py,sha256=yt_kB0LJK6lGdARtivmEmkZgbnzFUqIHfSN0hjYbTpM,3248
 phoenix/trace/openai/__init__.py,sha256=J3G0uqCxGdksUpaQVHds_Egv2drvh8UEqoLjiQAOveg,79
-phoenix/trace/openai/instrumentor.py,sha256=H1T2_1uqeH2lKCKeMmirEUl6PRtHQlQTXfsLR_hwDFM,24948
+phoenix/trace/openai/instrumentor.py,sha256=Lre96YfPxs5iM3frZfwdAh94LUpyYw_RZpSF0XmPtKw,1061
 phoenix/trace/v1/__init__.py,sha256=-IbAD0ruESMjvQLvGAg9CTfjBUATFDx1OXseDPis6-0,88
 phoenix/trace/v1/evaluation_pb2.py,sha256=8sXvv2BW_vqD30MOMbmkeE2zpmm7ncik21kl3e-HzeQ,2254
 phoenix/trace/v1/evaluation_pb2.pyi,sha256=cCbbx06gwQmaH14s3J1X25TtaARh-k1abbxQdQCXGm8,4500
 phoenix/utilities/__init__.py,sha256=3TVirVnjIGyaCFuJCqeZO4tjlzQ_chZgYM0itIwsEpE,656
 phoenix/utilities/error_handling.py,sha256=7b5rpGFj9EWZ8yrZK1IHvxB89suWk3lggDayUQcvZds,1946
 phoenix/utilities/logging.py,sha256=lDXd6EGaamBNcQxL4vP1au9-i_SXe0OraUDiJOcszSw,222
-arize_phoenix-2.11.1.dist-info/METADATA,sha256=wdQLjJ9OUXcNT1d-0MA1L-mKs6HhPPZ1Rc6B1FN4wZk,28766
-arize_phoenix-2.11.1.dist-info/WHEEL,sha256=TJPnKdtrSue7xZ_AVGkp9YXcvDrobsjBds1du3Nx6dc,87
-arize_phoenix-2.11.1.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
-arize_phoenix-2.11.1.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
-arize_phoenix-2.11.1.dist-info/RECORD,,
+arize_phoenix-3.0.1.dist-info/METADATA,sha256=MIwob2psMophui1KBYcjRPTmKinVvD9YIY0aDqRE5V0,28556
+arize_phoenix-3.0.1.dist-info/WHEEL,sha256=TJPnKdtrSue7xZ_AVGkp9YXcvDrobsjBds1du3Nx6dc,87
+arize_phoenix-3.0.1.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
+arize_phoenix-3.0.1.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
+arize_phoenix-3.0.1.dist-info/RECORD,,

phoenix/config.py

@@ -12,7 +12,7 @@ ENV_PHOENIX_COLLECTOR_ENDPOINT = "PHOENIX_COLLECTOR_ENDPOINT"
 The endpoint traces and evals are sent to. This must be set if the Phoenix
 server is running on a remote instance.
 """
-ENV_WORKING_DIR = "PHOENIX_WORKING_DIR"
+ENV_PHOENIX_WORKING_DIR = "PHOENIX_WORKING_DIR"
 """
 The directory in which to save, load, and export datasets. This directory must
 be accessible by both the Phoenix server and the notebook environment.
@@ -45,7 +45,7 @@ def get_working_dir() -> Path:
     """
     Get the working directory for saving, loading, and exporting datasets.
     """
-    working_dir_str = os.getenv(ENV_WORKING_DIR)
+    working_dir_str = os.getenv(ENV_PHOENIX_WORKING_DIR)
     if working_dir_str is not None:
         return Path(working_dir_str)
     # Fall back to ~/.phoenix if PHOENIX_WORKING_DIR is not set
@@ -77,7 +77,9 @@ except Exception as e:
         f"⚠️ Failed to initialize the working directory at {WORKING_DIR} due to an error: {str(e)}"
     )
     print("⚠️ While phoenix will still run, you will not be able to save, load, or export data")
-    print("ℹ️ To change, set the `{ENV_WORKING_DIR}` environment variable before importing phoenix.")
+    print(
+        f"ℹ️ To change, set the `{ENV_PHOENIX_WORKING_DIR}` environment variable before importing phoenix."  # noqa: E501
+    )
 
 
 def get_exported_files(directory: Path) -> List[Path]:

phoenix/trace/exporter.py

@@ -9,6 +9,7 @@ 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
 from typing_extensions import TypeAlias, assert_never
 
@@ -30,6 +31,19 @@ class NoOpExporter:
         pass
 
 
+class _OpenInferenceExporter(OTLPSpanExporter):
+    def __init__(self) -> None:
+        host = get_env_host()
+        if host == "0.0.0.0":
+            host = "127.0.0.1"
+        endpoint = urljoin(
+            get_env_collector_endpoint() or f"http://{host}:{get_env_port()}",
+            "/v1/traces",
+        )
+        _warn_if_phoenix_is_not_running(endpoint)
+        super().__init__(endpoint)
+
+
 class HttpExporter:
     def __init__(
         self,
@@ -62,7 +76,7 @@ class HttpExporter:
             or get_env_collector_endpoint()
             or f"http://{'127.0.0.1' if self._host == '0.0.0.0' else self._host}:{self._port}"
         )
-        self._warn_if_phoenix_is_not_running()
+        _warn_if_phoenix_is_not_running(self._base_url)
         self._session = Session()
         weakref.finalize(self, self._session.close)
         self._session.headers.update(
@@ -114,11 +128,12 @@ class HttpExporter:
         logger.exception(f"unrecognized message type: {type(message)}")
         assert_never(message)
 
-    def _warn_if_phoenix_is_not_running(self) -> None:
-        try:
-            requests.get(urljoin(self._base_url, "arize_phoenix_version")).raise_for_status()
-        except Exception:
-            logger.warning(
-                f"Arize Phoenix is not running on {self._base_url}. Launch Phoenix "
-                f"with `import phoenix as px; px.launch_app()`"
-            )
+
+def _warn_if_phoenix_is_not_running(endpoint: str) -> None:
+    try:
+        requests.get(urljoin(endpoint, "/arize_phoenix_version")).raise_for_status()
+    except Exception:
+        logger.warning(
+            f"Arize Phoenix is not running on {endpoint}. Launch Phoenix "
+            f"with `import phoenix as px; px.launch_app()`"
+        )

phoenix/trace/langchain/__init__.py

@@ -1,4 +1,26 @@
-from .instrumentor import LangChainInstrumentor
-from .tracer import OpenInferenceTracer
+import sys
+from typing import Any
 
-__all__ = ["OpenInferenceTracer", "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

phoenix/trace/langchain/instrumentor.py

@@ -1,37 +1,31 @@
-from typing import Any, Optional
+import logging
+from importlib.metadata import PackageNotFoundError
+from importlib.util import find_spec
+from typing import Any
 
-from .tracer import OpenInferenceTracer
+from openinference.instrumentation.langchain import LangChainInstrumentor as Instrumentor
+from opentelemetry.sdk import trace as trace_sdk
+from opentelemetry.sdk.trace.export import SimpleSpanProcessor
 
+from phoenix.trace.exporter import _OpenInferenceExporter
+from phoenix.trace.tracer import _show_deprecation_warnings
 
-class LangChainInstrumentor:
-    """
-    Instruments the OpenInferenceTracer for LangChain automatically by patching the
-    BaseCallbackManager in LangChain.
-    """
+logger = logging.getLogger(__name__)
 
-    def __init__(self, tracer: Optional[OpenInferenceTracer] = None) -> None:
-        self._tracer = tracer if tracer is not None else OpenInferenceTracer()
 
-    def instrument(self) -> None:
-        try:
-            from langchain.callbacks.base import BaseCallbackManager
-        except ImportError:
-            # Raise a cleaner error if LangChain is not installed
-            raise ImportError(
-                "LangChain is not installed. Please install LangChain first to use the instrumentor"
-            )
-
-        source_init = BaseCallbackManager.__init__
+__all__ = ("LangChainInstrumentor",)
 
-        # Keep track of the source init so we can tell if the patching occurred
-        self._source_callback_manager_init = source_init
 
-        tracer = self._tracer
-
-        # Patch the init method of the BaseCallbackManager to add the tracer
-        # to all callback managers
-        def patched_init(self: BaseCallbackManager, *args: Any, **kwargs: Any) -> None:
-            source_init(self, *args, **kwargs)
-            self.add_handler(tracer, True)
+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`."
+            )
+        super().__init__()
 
-        BaseCallbackManager.__init__ = patched_init
+    def instrument(self) -> None:
+        tracer_provider = trace_sdk.TracerProvider()
+        tracer_provider.add_span_processor(SimpleSpanProcessor(_OpenInferenceExporter()))
+        super().instrument(skip_dep_check=True, tracer_provider=tracer_provider)