arize-phoenix 4.36.0__py3-none-any.whl → 5.0.0__py3-none-any.whl

phoenix/db/migrations/future_versions/cd164e83824f_users_and_tokens.py

@@ -1,293 +0,0 @@
-"""users and tokens
-
-Revision ID: cd164e83824f
-Revises: 10460e46d750
-Create Date: 2024-08-01 18:36:52.157604
-
-"""
-
-from datetime import datetime, timezone
-from typing import Any, Dict, List, Optional, Sequence, TypedDict, Union
-
-import sqlalchemy as sa
-from alembic import op
-from sqlalchemy import (
-    JSON,
-    TIMESTAMP,
-    CheckConstraint,
-    Dialect,
-    ForeignKey,
-    MetaData,
-    TypeDecorator,
-    func,
-    insert,
-)
-from sqlalchemy.dialects import postgresql
-from sqlalchemy.ext.asyncio.engine import AsyncConnection
-from sqlalchemy.ext.compiler import compiles
-from sqlalchemy.orm import (
-    DeclarativeBase,
-    Mapped,
-    mapped_column,
-)
-
-from phoenix.datetime_utils import normalize_datetime
-
-
-class JSONB(JSON):
-    # See https://docs.sqlalchemy.org/en/20/core/custom_types.html
-    __visit_name__ = "JSONB"
-
-
-@compiles(JSONB, "sqlite")  # type: ignore
-def _(*args: Any, **kwargs: Any) -> str:
-    # See https://docs.sqlalchemy.org/en/20/core/custom_types.html
-    return "JSONB"
-
-
-JSON_ = (
-    JSON()
-    .with_variant(
-        postgresql.JSONB(),  # type: ignore
-        "postgresql",
-    )
-    .with_variant(
-        JSONB(),
-        "sqlite",
-    )
-)
-
-
-class JsonDict(TypeDecorator[Dict[str, Any]]):
-    # See # See https://docs.sqlalchemy.org/en/20/core/custom_types.html
-    cache_ok = True
-    impl = JSON_
-
-    def process_bind_param(self, value: Optional[Dict[str, Any]], _: Dialect) -> Dict[str, Any]:
-        return value if isinstance(value, dict) else {}
-
-
-class JsonList(TypeDecorator[List[Any]]):
-    # See # See https://docs.sqlalchemy.org/en/20/core/custom_types.html
-    cache_ok = True
-    impl = JSON_
-
-    def process_bind_param(self, value: Optional[List[Any]], _: Dialect) -> List[Any]:
-        return value if isinstance(value, list) else []
-
-
-class UtcTimeStamp(TypeDecorator[datetime]):
-    # See # See https://docs.sqlalchemy.org/en/20/core/custom_types.html
-    cache_ok = True
-    impl = TIMESTAMP(timezone=True)
-
-    def process_bind_param(self, value: Optional[datetime], _: Dialect) -> Optional[datetime]:
-        return normalize_datetime(value)
-
-    def process_result_value(self, value: Optional[Any], _: Dialect) -> Optional[datetime]:
-        return normalize_datetime(value, timezone.utc)
-
-
-class ExperimentRunOutput(TypedDict, total=False):
-    task_output: Any
-
-
-class Base(DeclarativeBase):
-    # Enforce best practices for naming constraints
-    # https://alembic.sqlalchemy.org/en/latest/naming.html#integration-of-naming-conventions-into-operations-autogenerate
-    metadata = MetaData(
-        naming_convention={
-            "ix": "ix_%(table_name)s_%(column_0_N_name)s",
-            "uq": "uq_%(table_name)s_%(column_0_N_name)s",
-            "ck": "ck_%(table_name)s_`%(constraint_name)s`",
-            "fk": "fk_%(table_name)s_%(column_0_name)s_%(referred_table_name)s",
-            "pk": "pk_%(table_name)s",
-        }
-    )
-    type_annotation_map = {
-        Dict[str, Any]: JsonDict,
-        List[Dict[str, Any]]: JsonList,
-        ExperimentRunOutput: JsonDict,
-    }
-
-
-class UserRole(Base):
-    __tablename__ = "user_roles"
-    id: Mapped[int] = mapped_column(primary_key=True)
-    name: Mapped[str] = mapped_column(unique=True)
-
-
-class User(Base):
-    __tablename__ = "users"
-    id: Mapped[int] = mapped_column(primary_key=True)
-    user_role_id: Mapped[int] = mapped_column(
-        ForeignKey("user_roles.id"),
-        index=True,
-    )
-    username: Mapped[Optional[str]] = mapped_column(nullable=True, unique=True, index=True)
-    email: Mapped[str] = mapped_column(nullable=False, unique=True, index=True)
-    auth_method: Mapped[str] = mapped_column(
-        CheckConstraint("auth_method IN ('LOCAL')", name="valid_auth_method")
-    )
-    password_hash: Mapped[Optional[str]]
-    reset_password: Mapped[bool]
-    created_at: Mapped[datetime] = mapped_column(UtcTimeStamp, server_default=func.now())
-    updated_at: Mapped[datetime] = mapped_column(
-        UtcTimeStamp, server_default=func.now(), onupdate=func.now()
-    )
-    deleted_at: Mapped[Optional[datetime]] = mapped_column(UtcTimeStamp)
-
-
-# revision identifiers, used by Alembic.
-revision: str = "cd164e83824f"
-down_revision: Union[str, None] = "3be8647b87d8"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
-
-
-async def insert_roles_and_users(connection: AsyncConnection) -> None:
-    """
-    Populates the `user_roles` table and adds a system user and initial admin
-    user to the `users` table.
-    """
-    await connection.execute(
-        insert(UserRole).values([{"name": "SYSTEM"}, {"name": "ADMIN"}, {"name": "MEMBER"}])
-    )
-    system_user_role_id = sa.select(UserRole.id).where(UserRole.name == "SYSTEM").scalar_subquery()
-    admin_user_role_id = sa.select(UserRole.id).where(UserRole.name == "ADMIN").scalar_subquery()
-    await connection.execute(
-        insert(User).values(
-            [
-                {
-                    "user_role_id": system_user_role_id,
-                    "username": None,
-                    "email": "system@localhost",
-                    "auth_method": "LOCAL",
-                    "password_hash": None,
-                    "reset_password": False,
-                },
-                {
-                    "user_role_id": admin_user_role_id,
-                    "username": "admin",
-                    "email": "admin@localhost",
-                    "auth_method": "LOCAL",
-                    "password_hash": None,  # todo: replace this with the hashed PHOENIX_SECRET
-                    "reset_password": True,
-                },
-            ]
-        )
-    )
-
-
-def upgrade() -> None:
-    op.create_table(
-        "user_roles",
-        sa.Column("id", sa.Integer, primary_key=True),
-        sa.Column(
-            "name",
-            sa.String,
-            nullable=False,
-            unique=True,
-        ),
-    )
-    op.create_table(
-        "users",
-        sa.Column("id", sa.Integer, primary_key=True),
-        sa.Column(
-            "user_role_id",
-            sa.Integer,
-            sa.ForeignKey("user_roles.id"),
-            nullable=False,
-            index=True,
-        ),
-        sa.Column("username", sa.String, nullable=True, unique=True, index=True),
-        sa.Column("email", sa.String, nullable=False, unique=True, index=True),
-        sa.Column(
-            "auth_method",
-            sa.String,
-            sa.CheckConstraint("auth_method IN ('LOCAL')", "valid_auth_method"),
-            nullable=False,
-        ),
-        sa.Column("password_hash", sa.String, nullable=True),
-        sa.Column("reset_password", sa.Boolean, nullable=False),
-        sa.Column(
-            "created_at",
-            sa.TIMESTAMP(timezone=True),
-            nullable=False,
-            server_default=sa.func.now(),
-        ),
-        sa.Column(
-            "updated_at",
-            sa.TIMESTAMP(timezone=True),
-            nullable=False,
-            server_default=sa.func.now(),
-            onupdate=sa.func.now(),
-        ),
-        sa.Column(
-            "deleted_at",
-            sa.TIMESTAMP(timezone=True),
-            nullable=True,
-        ),
-    )
-    op.create_table(
-        "api_keys",
-        sa.Column("id", sa.Integer, primary_key=True),
-        sa.Column(
-            "user_id",
-            sa.Integer,
-            sa.ForeignKey("users.id"),
-            nullable=False,
-            index=True,
-        ),
-        sa.Column("name", sa.String, nullable=False),
-        sa.Column("description", sa.String, nullable=True),
-        sa.Column(
-            "created_at",
-            sa.TIMESTAMP(timezone=True),
-            nullable=False,
-            server_default=sa.func.now(),
-        ),
-        sa.Column(
-            "expires_at",
-            sa.TIMESTAMP(timezone=True),
-            nullable=True,
-        ),
-    )
-    op.create_table(
-        "audit_api_keys",
-        sa.Column("id", sa.Integer, primary_key=True),
-        sa.Column(
-            "api_key_id",
-            sa.Integer,
-            sa.ForeignKey("api_keys.id"),
-            nullable=False,
-            index=True,
-        ),
-        sa.Column(
-            "user_id",
-            sa.Integer,
-            sa.ForeignKey("users.id"),
-            nullable=False,
-            index=True,
-        ),
-        sa.Column(
-            "action",
-            sa.String,
-            sa.CheckConstraint("action IN ('CREATE', 'DELETE')", "valid_action"),
-            nullable=False,
-        ),
-        sa.Column(
-            "created_at",
-            sa.TIMESTAMP(timezone=True),
-            nullable=False,
-            server_default=sa.func.now(),
-        ),
-    )
-    op.run_async(insert_roles_and_users)
-
-
-def downgrade() -> None:
-    op.drop_table("audit_api_keys")
-    op.drop_table("api_keys")
-    op.drop_table("users")
-    op.drop_table("user_roles")

phoenix/db/migrations/versions/.gitignore

@@ -1 +0,0 @@
-cd164e83824f_users_and_tokens.py

phoenix/server/api/mutations/auth.py

@@ -1,18 +0,0 @@
-from typing import Any
-
-from strawberry import Info
-from strawberry.permission import BasePermission
-
-
-class IsAuthenticated(BasePermission):
-    message = "User is not authenticated"
-
-    async def has_permission(self, source: Any, info: Info, **kwargs: Any) -> bool:
-        return not info.context.read_only
-
-
-class HasSecret(BasePermission):
-    message = "Application secret is not set"
-
-    async def has_permission(self, source: Any, info: Info, **kwargs: Any) -> bool:
-        return info.context.secret is not None

phoenix/server/api/mutations/auth_mutations.py

@@ -1,65 +0,0 @@
-import asyncio
-from datetime import timedelta
-
-import strawberry
-from sqlalchemy import select
-from strawberry.types import Info
-
-from phoenix.auth import is_valid_password
-from phoenix.db import models
-from phoenix.server.api.context import Context
-from phoenix.server.api.exceptions import Unauthorized
-from phoenix.server.api.mutations.auth import HasSecret
-
-PHOENIX_ACCESS_TOKEN_COOKIE_NAME = "phoenix-access-token"
-PHOENIX_ACCESS_TOKEN_COOKIE_MAX_AGE_IN_SECONDS = int(timedelta(days=31).total_seconds())
-FAILED_LOGIN_MESSAGE = "login failed"
-
-
-@strawberry.input
-class LoginMutationInput:
-    email: str
-    password: str
-
-
-@strawberry.type
-class AuthMutationMixin:
-    @strawberry.mutation(permission_classes=[HasSecret])  # type: ignore
-    async def login(
-        self,
-        info: Info[Context, None],
-        input: LoginMutationInput,
-    ) -> None:
-        async with info.context.db() as session:
-            if (
-                user := await session.scalar(
-                    select(models.User).where(models.User.email == input.email)
-                )
-            ) is None or (password_hash := user.password_hash) is None:
-                raise Unauthorized(FAILED_LOGIN_MESSAGE)
-        secret = info.context.get_secret()
-        loop = asyncio.get_running_loop()
-        if not await loop.run_in_executor(
-            executor=None,
-            func=lambda: is_valid_password(
-                password=input.password, salt=secret, password_hash=password_hash
-            ),
-        ):
-            raise Unauthorized(FAILED_LOGIN_MESSAGE)
-        response = info.context.get_response()
-        response.set_cookie(
-            key=PHOENIX_ACCESS_TOKEN_COOKIE_NAME,
-            value="token",  # todo: compute access token
-            secure=True,
-            httponly=True,
-            samesite="strict",
-            max_age=PHOENIX_ACCESS_TOKEN_COOKIE_MAX_AGE_IN_SECONDS,
-        )
-
-    @strawberry.mutation(permission_classes=[HasSecret])  # type: ignore
-    async def logout(
-        self,
-        info: Info[Context, None],
-    ) -> None:
-        response = info.context.get_response()
-        response.delete_cookie(key=PHOENIX_ACCESS_TOKEN_COOKIE_NAME)

phoenix/server/static/assets/index-fq1-hCK4.js

@@ -1,100 +0,0 @@
-import{r as d,j as e,d5 as P,v as s,F,R as v,w as E,aQ as S,d6 as L,d7 as R,d8 as a,d9 as w,da as z,b as A,db as j}from"./vendor-BC3OPQuM.js";import{S as C,j as k,Z as $,U as _,t as I,a4 as O}from"./vendor-arizeai-NjB3cZzD.js";import{E as T,L as D,a as N,P as G,h as M,M as U,b as m,D as B,d as q,c as J,e as K,f as W,g as H,T as Q,p as V,i as g,j as Y,k as Z,l as u,m as X,n as h,o as b,q as ee,r as re,s as ae,t as te,v as oe,A as ne,S as se,F as le}from"./pages-CnTvEGEN.js";import{b9 as ie,d as ce,R as de,ba as pe,bb as me}from"./components-Dte7_KRd.js";import"./vendor-three-DwGkEfCM.js";import"./vendor-recharts-BXLYwcXF.js";import"./vendor-codemirror-gE_JCOgX.js";(function(){const n=document.createElement("link").relList;if(n&&n.supports&&n.supports("modulepreload"))return;for(const t of document.querySelectorAll('link[rel="modulepreload"]'))c(t);new MutationObserver(t=>{for(const o of t)if(o.type==="childList")for(const l of o.addedNodes)l.tagName==="LINK"&&l.rel==="modulepreload"&&c(l)}).observe(document,{childList:!0,subtree:!0});function i(t){const o={};return t.integrity&&(o.integrity=t.integrity),t.referrerPolicy&&(o.referrerPolicy=t.referrerPolicy),t.crossOrigin==="use-credentials"?o.credentials="include":t.crossOrigin==="anonymous"?o.credentials="omit":o.credentials="same-origin",o}function c(t){if(t.ep)return;t.ep=!0;const o=i(t);fetch(t.href,o)}})();const f="arize-phoenix-feature-flags",p={__CLEAR__:!0};function ge(){const r=localStorage.getItem(f);if(!r)return p;try{const n=JSON.parse(r);return Object.assign({},p,n)}catch{return p}}const x=d.createContext(null);function ue(){const r=v.useContext(x);if(r===null)throw new Error("useFeatureFlags must be used within a FeatureFlagsProvider");return r}function he(r){const[n,i]=d.useState(ge()),c=t=>{localStorage.setItem(f,JSON.stringify(t)),i(t)};return e(x.Provider,{value:{featureFlags:n,setFeatureFlags:c},children:e(be,{children:r.children})})}function be(r){const{children:n}=r,{featureFlags:i,setFeatureFlags:c}=ue(),[t,o]=d.useState(!1);return P("ctrl+shift+f",()=>o(!0)),s(F,{children:[n,e(_,{type:"modal",isDismissable:!0,onDismiss:()=>o(!1),children:t&&e(C,{title:"Feature Flags",children:e(k,{height:"size-1000",padding:"size-100",children:Object.keys(i).map(l=>e($,{isSelected:i[l],onChange:y=>c({...i,[l]:y}),children:l},l))})})})]})}function fe(){return e(S,{styles:r=>E`
-        body {
-          background-color: var(--ac-global-color-grey-75);
-          color: var(--ac-global-text-color-900);
-          font-family: "Roboto";
-          font-size: ${r.typography.sizes.medium.fontSize}px;
-          margin: 0;
-          #root,
-          #root > div[data-overlay-container="true"],
-          #root > div[data-overlay-container="true"] > .ac-theme {
-            height: 100vh;
-          }
-        }
-
-        /* Remove list styling */
-        ul {
-          display: block;
-          list-style-type: none;
-          margin-block-start: none;
-          margin-block-end: 0;
-          padding-inline-start: 0;
-          margin-block-start: 0;
-        }
-
-        /* A reset style for buttons */
-        .button--reset {
-          background: none;
-          border: none;
-          padding: 0;
-        }
-        /* this css class is added to html via modernizr @see modernizr.js */
-        .no-hiddenscroll {
-          /* Works on Firefox */
-          * {
-            scrollbar-width: thin;
-            scrollbar-color: var(--ac-global-color-grey-300)
-              var(--ac-global-color-grey-400);
-          }
-
-          /* Works on Chrome, Edge, and Safari */
-          *::-webkit-scrollbar {
-            width: 14px;
-          }
-
-          *::-webkit-scrollbar-track {
-            background: var(--ac-global-color-grey-100);
-          }
-
-          *::-webkit-scrollbar-thumb {
-            background-color: var(--ac-global-color-grey-75);
-            border-radius: 8px;
-            border: 1px solid var(--ac-global-color-grey-300);
-          }
-        }
-
-        :root {
-          --px-blue-color: ${r.colors.arizeBlue};
-
-          --px-flex-gap-sm: ${r.spacing.margin4}px;
-          --px-flex-gap-sm: ${r.spacing.margin8}px;
-
-          --px-section-background-color: ${r.colors.gray500};
-
-          /* An item is a typically something in a list */
-          --px-item-background-color: ${r.colors.gray800};
-          --px-item-border-color: ${r.colors.gray600};
-
-          --px-spacing-sm: ${r.spacing.padding4}px;
-          --px-spacing-med: ${r.spacing.padding8}px;
-          --px-spacing-lg: ${r.spacing.padding16}px;
-
-          --px-border-radius-med: ${r.borderRadius.medium}px;
-
-          --px-font-size-sm: ${r.typography.sizes.small.fontSize}px;
-          --px-font-size-med: ${r.typography.sizes.medium.fontSize}px;
-          --px-font-size-lg: ${r.typography.sizes.large.fontSize}px;
-
-          --px-gradient-bar-height: 8px;
-
-          --px-nav-collapsed-width: 45px;
-          --px-nav-expanded-width: 200px;
-        }
-
-        .ac-theme--dark {
-          --px-primary-color: #9efcfd;
-          --px-primary-color--transparent: rgb(158, 252, 253, 0.2);
-          --px-reference-color: #baa1f9;
-          --px-reference-color--transparent: #baa1f982;
-          --px-corpus-color: #92969c;
-          --px-corpus-color--transparent: #92969c63;
-        }
-        .ac-theme--light {
-          --px-primary-color: #00add0;
-          --px-primary-color--transparent: rgba(0, 173, 208, 0.2);
-          --px-reference-color: #4500d9;
-          --px-reference-color--transparent: rgba(69, 0, 217, 0.2);
-          --px-corpus-color: #92969c;
-          --px-corpus-color--transparent: #92969c63;
-        }
-      `})}const xe=L(R(s(a,{path:"/",errorElement:e(T,{}),children:[e(a,{path:"/login",element:e(D,{})}),s(a,{element:e(N,{}),children:[e(a,{path:"/profile",handle:{crumb:()=>"profile"},element:e(G,{})}),e(a,{index:!0,loader:M}),s(a,{path:"/model",handle:{crumb:()=>"model"},element:e(U,{}),children:[e(a,{index:!0,element:e(m,{})}),e(a,{element:e(m,{}),children:e(a,{path:"dimensions",children:e(a,{path:":dimensionId",element:e(B,{}),loader:q})})}),e(a,{path:"embeddings",children:e(a,{path:":embeddingDimensionId",element:e(J,{}),loader:K,handle:{crumb:r=>r.embedding.name}})})]}),s(a,{path:"/projects",handle:{crumb:()=>"projects"},element:e(W,{}),children:[e(a,{index:!0,element:e(H,{})}),s(a,{path:":projectId",element:e(Q,{}),loader:V,handle:{crumb:r=>r.project.name},children:[e(a,{index:!0,element:e(g,{})}),e(a,{element:e(g,{}),children:e(a,{path:"traces/:traceId",element:e(Y,{})})})]})]}),s(a,{path:"/datasets",handle:{crumb:()=>"datasets"},children:[e(a,{index:!0,element:e(Z,{})}),s(a,{path:":datasetId",loader:u,handle:{crumb:r=>r.dataset.name},children:[s(a,{element:e(X,{}),loader:u,children:[e(a,{index:!0,element:e(h,{}),loader:b}),e(a,{path:"experiments",element:e(h,{}),loader:b}),e(a,{path:"examples",element:e(ee,{}),loader:re,children:e(a,{path:":exampleId",element:e(ae,{})})})]}),e(a,{path:"compare",handle:{crumb:()=>"compare"},loader:te,element:e(oe,{})})]})]}),e(a,{path:"/apis",element:e(ne,{}),handle:{crumb:()=>"APIs"}}),e(a,{path:"/settings",element:e(se,{}),handle:{crumb:()=>"Settings"}})]})]})),{basename:window.Config.basename});function ye(){return e(w,{router:xe})}function Pe(){return e(le,{children:e(ie,{children:e(Fe,{})})})}function Fe(){const{theme:r}=ce();return e(O,{theme:r,children:e(z,{theme:I,children:s(A.RelayEnvironmentProvider,{environment:de,children:[e(fe,{}),e(he,{children:e(pe,{children:e(d.Suspense,{children:e(me,{children:e(ye,{})})})})})]})})})}const ve=document.getElementById("root"),Ee=j.createRoot(ve);Ee.render(e(Pe,{}));

phoenix/trace/langchain/__init__.py

@@ -1,3 +0,0 @@
-from phoenix.trace.langchain.instrumentor import LangChainInstrumentor
-
-__all__ = ("LangChainInstrumentor",)

phoenix/trace/langchain/instrumentor.py

@@ -1,34 +0,0 @@
-import logging
-from importlib.metadata import PackageNotFoundError
-from importlib.util import find_spec
-from typing import Any
-
-from openinference.instrumentation.langchain import LangChainInstrumentor as Instrumentor
-from openinference.semconv.resource import ResourceAttributes
-from opentelemetry.sdk import trace as trace_sdk
-from opentelemetry.sdk.resources import Resource
-from opentelemetry.sdk.trace.export import SimpleSpanProcessor
-
-from phoenix.config import get_env_project_name
-from phoenix.trace.exporter import _OpenInferenceExporter
-
-logger = logging.getLogger(__name__)
-
-__all__ = ("LangChainInstrumentor",)
-
-
-class LangChainInstrumentor(Instrumentor):
-    def __init__(self, *args: Any, **kwargs: Any) -> None:
-        if find_spec("langchain_core") is None:
-            raise PackageNotFoundError(
-                "Missing `langchain-core`. Install with `pip install langchain-core`."
-            )
-        super().__init__()
-
-    def instrument(self) -> None:
-        tracer_provider = trace_sdk.TracerProvider(
-            resource=Resource({ResourceAttributes.PROJECT_NAME: get_env_project_name()}),
-            span_limits=trace_sdk.SpanLimits(max_attributes=10_000),
-        )
-        tracer_provider.add_span_processor(SimpleSpanProcessor(_OpenInferenceExporter()))
-        super().instrument(skip_dep_check=True, tracer_provider=tracer_provider)

phoenix/trace/llama_index/__init__.py

@@ -1,3 +0,0 @@
-from .callback import OpenInferenceTraceCallbackHandler
-
-__all__ = ["OpenInferenceTraceCallbackHandler"]

phoenix/trace/llama_index/callback.py

@@ -1,102 +0,0 @@
-import logging
-from importlib.metadata import PackageNotFoundError, version
-from typing import Any, Optional, Tuple
-
-from openinference.semconv.resource import ResourceAttributes
-from opentelemetry import trace as trace_api
-from opentelemetry.sdk import trace as trace_sdk
-from opentelemetry.sdk.resources import Resource
-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
-
-logger = logging.getLogger(__name__)
-
-LLAMA_INDEX_MODERN_VERSION = (0, 10, 0)
-INSTRUMENTATION_MODERN_VERSION = (1, 0, 0)
-
-
-def _check_instrumentation_compatibility() -> bool:
-    llama_index_version_str = _get_version_if_installed("llama-index")
-    llama_index_installed = llama_index_version_str is not None
-    llama_index_core_version_str = _get_version_if_installed("llama-index-core")
-    llama_index_core_installed = modern_llama_index_installed = (
-        llama_index_core_version_str is not None
-    )
-    instrumentation_version_str = version("openinference-instrumentation-llama-index")
-    instrumentation_version = _parse_semantic_version(instrumentation_version_str)
-
-    if not llama_index_installed and not llama_index_core_installed:
-        raise PackageNotFoundError(
-            "Missing `llama_index`. "
-            "Install with `pip install llama-index` or "
-            "`pip install llama-index-core` for a minimal installation."
-        )
-    elif modern_llama_index_installed and instrumentation_version < INSTRUMENTATION_MODERN_VERSION:
-        raise IncompatibleLibraryVersionError(
-            f"llama-index-core v{llama_index_core_version_str} is not compatible with "
-            f"openinference-instrumentation-llama-index v{instrumentation_version_str}. "
-            "Please upgrade openinference-instrumentation-llama-index to at least 1.0.0 via "
-            "`pip install 'openinference-instrumentation-llama-index>=1.0.0'`."
-        )
-    elif (
-        llama_index_installed
-        and llama_index_version_str
-        and _parse_semantic_version(llama_index_version_str) < LLAMA_INDEX_MODERN_VERSION
-        and instrumentation_version >= INSTRUMENTATION_MODERN_VERSION
-    ):
-        raise IncompatibleLibraryVersionError(
-            f"llama-index v{llama_index_version_str} is not compatible with "
-            f"openinference-instrumentation-llama-index v{instrumentation_version_str}. "
-            "Please either migrate llama-index to at least 0.10.0 or downgrade "
-            "openinference-instrumentation-llama-index via "
-            "`pip install 'openinference-instrumentation-llama-index<1.0.0'`."
-        )
-    return True
-
-
-def _get_version_if_installed(package_name: str) -> Optional[str]:
-    """
-    Gets the version of the package if it is installed, otherwise, returns None.
-    """
-    try:
-        return version(package_name)
-    except PackageNotFoundError:
-        return None
-
-
-def _parse_semantic_version(semver_string: str) -> Tuple[int, ...]:
-    """
-    Parse a semantic version string into a tuple of integers.
-    """
-    return tuple(map(int, semver_string.split(".")[:3]))
-
-
-if _check_instrumentation_compatibility():
-    from openinference.instrumentation.llama_index._callback import (
-        OpenInferenceTraceCallbackHandler as _OpenInferenceTraceCallbackHandler,
-    )
-    from openinference.instrumentation.llama_index.version import (
-        __version__,
-    )
-
-
-class OpenInferenceTraceCallbackHandler(_OpenInferenceTraceCallbackHandler):
-    """Callback handler for storing LLM application trace data in OpenInference format.
-    OpenInference is an open standard for capturing and storing AI model
-    inferences. It enables production LLMapp servers to seamlessly integrate
-    with LLM observability solutions such as Arize and Phoenix.
-
-    For more information on the specification, see
-    https://github.com/Arize-ai/openinference
-    """
-
-    def __init__(self, *args: Any, **kwargs: Any) -> None:
-        tracer_provider = trace_sdk.TracerProvider(
-            resource=Resource({ResourceAttributes.PROJECT_NAME: get_env_project_name()}),
-            span_limits=trace_sdk.SpanLimits(max_attributes=10_000),
-        )
-        tracer_provider.add_span_processor(SimpleSpanProcessor(_OpenInferenceExporter()))
-        super().__init__(trace_api.get_tracer(__name__, __version__, tracer_provider))

phoenix/trace/openai/__init__.py

@@ -1,3 +0,0 @@
-from .instrumentor import OpenAIInstrumentor
-
-__all__ = ["OpenAIInstrumentor"]

phoenix/trace/openai/instrumentor.py

@@ -1,30 +0,0 @@
-import logging
-from importlib.metadata import PackageNotFoundError
-from importlib.util import find_spec
-from typing import Any
-
-from openinference.instrumentation.openai import OpenAIInstrumentor as Instrumentor
-from openinference.semconv.resource import ResourceAttributes
-from opentelemetry.sdk import trace as trace_sdk
-from opentelemetry.sdk.resources import Resource
-from opentelemetry.sdk.trace.export import SimpleSpanProcessor
-
-from phoenix.config import get_env_project_name
-from phoenix.trace.exporter import _OpenInferenceExporter
-
-logger = logging.getLogger(__name__)
-
-
-class OpenAIInstrumentor(Instrumentor):
-    def __init__(self, *args: Any, **kwargs: Any) -> None:
-        if find_spec("openai") is None:
-            raise PackageNotFoundError("Missing `openai`. Install with `pip install openai`.")
-        super().__init__()
-
-    def instrument(self) -> None:
-        tracer_provider = trace_sdk.TracerProvider(
-            resource=Resource({ResourceAttributes.PROJECT_NAME: get_env_project_name()}),
-            span_limits=trace_sdk.SpanLimits(max_attributes=10_000),
-        )
-        tracer_provider.add_span_processor(SimpleSpanProcessor(_OpenInferenceExporter()))
-        super().instrument(skip_dep_check=True, tracer_provider=tracer_provider)