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

{arize_phoenix-3.10.0.dist-info → arize_phoenix-3.11.1.dist-info}/METADATA

@@ -1,7 +1,7 @@
 Metadata-Version: 2.1
 Name: arize-phoenix
-Version: 3.10.0
-Summary: ML Observability in your notebook
+Version: 3.11.1
+Summary: AI Observability and Evaluation
 Project-URL: Documentation, https://docs.arize.com/phoenix/
 Project-URL: Issues, https://github.com/Arize-ai/phoenix/issues
 Project-URL: Source, https://github.com/Arize-ai/phoenix

{arize_phoenix-3.10.0.dist-info → arize_phoenix-3.11.1.dist-info}/RECORD

@@ -4,14 +4,14 @@ phoenix/datetime_utils.py,sha256=D955QLrkgrrSdUM6NyqbCeAu2SMsjhR5rHVQEsVUdng,277
 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=iKJAvc5i0UTDDSSefTGL0Tq-kWQ4S3OJJgvyaQfQNF8,23
+phoenix/version.py,sha256=UuKR3QjWk9sJtn2huDFsPdSgtx1CgiEl2B4BUtVRd58,23
 phoenix/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/core/embedding_dimension.py,sha256=zKGbcvwOXgLf-yrJBpQyKtd-LEOPRKHnUToyAU8Owis,87
 phoenix/core/model.py,sha256=C-kDATyJEgP-oqYVKOiQM76Ljs66F6VZdT93_b8kTGk,4725
 phoenix/core/model_schema.py,sha256=lQaTvKS34yurHOJ53YD020uURLfgG3dqKC1NLQftOjA,50222
 phoenix/core/model_schema_adapter.py,sha256=3GkyzqUST4fYi-Bgs8qAam5hwMCdQRZTDLjZ9Bnzdm4,8268
 phoenix/core/project.py,sha256=5bitRrQ-aG70k21ht0HUDJZkMl3dpI_HyLmBO7rLB-c,24673
-phoenix/core/traces.py,sha256=enZbpGW6Z7huxYhrFEd3x5LV13r24SkA8jl6oZdyh6o,5650
+phoenix/core/traces.py,sha256=OnvOo4epNGMptOF1-O2FQgDV1c9TywQq5yqvj9ph3Mw,2902
 phoenix/datasets/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/datasets/dataset.py,sha256=scKVZ7zc6Dpc_ntt-pWhzY-KWqOJEwKePuyNnKSVTGE,30515
 phoenix/datasets/errors.py,sha256=cGp9vxnw4SewFoWBV3ZGMkhE0Kh73lPIv3Ppz_H_RoA,8261
@@ -103,13 +103,13 @@ phoenix/server/api/types/MimeType.py,sha256=VIUQBqveSctiRFStJjDUdZ85TeKhwNNEosMC
 phoenix/server/api/types/Model.py,sha256=IFhO2nouLQoMPNhl8vnm7nEXXSpr14CL3ihw8RnVWNY,7826
 phoenix/server/api/types/NumericRange.py,sha256=afEjgF97Go_OvmjMggbPBt-zGM8IONewAyEiKEHRds0,192
 phoenix/server/api/types/PerformanceMetric.py,sha256=W92B7OghEOgzFvmY0LCqpgavHaQggTGshdgfD0yqHX4,350
-phoenix/server/api/types/Project.py,sha256=EArem8TsGV0PLUS_Vm-z45gY6syzcnJPlY1Iu0FI0eQ,8109
+phoenix/server/api/types/Project.py,sha256=MqzQicNm7IHiJom5IXJvbgnOqCBHZ8k_wGTB5rchtkU,8126
 phoenix/server/api/types/PromptResponse.py,sha256=Q8HKtpp8GpUOcxPCzZpkkokidDd6u0aZOv_SuPZZd5Q,630
 phoenix/server/api/types/Retrieval.py,sha256=OhMK2ncjoyp5h1yjKhjlKpoTbQrMHuxmgSFw-AO1rWw,285
 phoenix/server/api/types/ScalarDriftMetricEnum.py,sha256=IUAcRPpgL41WdoIgK6cNk2Te38SspXGyEs-S1fY23_A,232
 phoenix/server/api/types/Segments.py,sha256=B6UUWjalZONjWjl_l61A6USPSu15ICXRgzZ4m3vA1yw,2921
 phoenix/server/api/types/SortDir.py,sha256=OUpXhlCzCxPoXSDkJJygEs9Rw9pMymfaZUG5zPTrw4Y,152
-phoenix/server/api/types/Span.py,sha256=mIczY_mMa-g7lOjvA2btAxtsD4lVIHAbxtMx38Hn5q4,12378
+phoenix/server/api/types/Span.py,sha256=1o7DeeR7Ytj7XgXrS9Oy5N7ZjSOMXxEGCINs0sjh6Lc,12291
 phoenix/server/api/types/TimeSeries.py,sha256=QbLfxHnwYsMsirpq4tx9us6ha7YtAVzK4m8mAL3fMt0,5200
 phoenix/server/api/types/UMAPPoints.py,sha256=8l9RJXi308qty4MdHb2pBbiU6ZuLbrRRxXNbPhXoxKI,1639
 phoenix/server/api/types/ValidationResult.py,sha256=pHwdYk4J7SJ5xhlWWHg_6qWkfk4rjOx-bSkGHvkDE3Q,142
@@ -127,7 +127,7 @@ phoenix/server/static/apple-touch-icon-76x76.png,sha256=CT_xT12I0u2i0WU8JzBZBuOQ
 phoenix/server/static/apple-touch-icon.png,sha256=fOfpjqGpWYbJ0eAurKsyoZP1EAs6ZVooBJ_SGk2ZkDs,3801
 phoenix/server/static/favicon.ico,sha256=bY0vvCKRftemZfPShwZtE93DiiQdaYaozkPGwNFr6H8,34494
 phoenix/server/static/index.css,sha256=KKGpx4iwF91VGRm0YN-4cn8oC-oIqC6HecoPf0x3ZM8,1885
-phoenix/server/static/index.js,sha256=NTBw2dg0KXyBdHg_1oGbpaKLD1kx1ZzGYMuMJpHIv2U,3175475
+phoenix/server/static/index.js,sha256=OZGKVWu4fPQLtzthHYlixwpbUSsEsOIHzVkb2cFKHD0,3175609
 phoenix/server/static/modernizr.js,sha256=mvK-XtkNqjOral-QvzoqsyOMECXIMu5BQwSVN_wcU9c,2564
 phoenix/server/templates/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/server/templates/index.html,sha256=lO2wGA5XsftPg03rw_VcyaYf_4vegtlWbIT5ms4fA_c,1982
@@ -135,11 +135,11 @@ phoenix/session/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 phoenix/session/client.py,sha256=9qlybwGGdPHql7n6Mye8feuh5_5AMq34xSajvcPn2yg,5853
 phoenix/session/data_extractor.py,sha256=0Kf-2mKY_YbYoD2fZkAYpKdFgsXrC3OKQ5d2iZsGgAI,1947
 phoenix/session/evaluation.py,sha256=YCv1XkWHi7vM_W5V7rorrrAxadv78wuMPeCVJvf5-oE,5444
-phoenix/session/session.py,sha256=elU_198oGobkmbAL_nACpzBCYCnCaEbYgja00toU7mc,24472
-phoenix/trace/__init__.py,sha256=4d_MqzUIFmlY9WWcFeTONJ4xL5mPGoWZaPM2TJ0ZDBQ,266
+phoenix/session/session.py,sha256=1RV6FBNqNDDrPFYU2qLffmI4RCRKy22KCLZFcH9oqJA,24556
+phoenix/trace/__init__.py,sha256=qMipNV47ahKsOOz8uQpoI4VnNRnjc5uh_0cJpL1ABvY,706
 phoenix/trace/errors.py,sha256=wB1z8qdPckngdfU-TORToekvg3344oNFAA83_hC2yFY,180
 phoenix/trace/evaluation_conventions.py,sha256=t8jydM3U0-T5YpiQKRJ3tWdWGlHtzKyttYdw-ddvPOk,1048
-phoenix/trace/exporter.py,sha256=O-9Arn-S_B9Me-jy4Qa84y6lvxKNoa8pczrsamPl3_Q,4871
+phoenix/trace/exporter.py,sha256=vh2RO1CpP143HxIX94KV0qks8p1x66RE3Tgf8kcBCCg,4519
 phoenix/trace/fixtures.py,sha256=HOjuYOB_xtR7JUeLz7WpEroiGj4E5_SxVLSjBYUy8RQ,7055
 phoenix/trace/otel.py,sha256=mpuj_eOdBUPAcslpxk8XZflb9kkzpMJ8X9acJS8ACPA,15322
 phoenix/trace/schemas.py,sha256=6PVPnQIIDsWijwCwU19TGKG3xhFCYFY9K_5IqR5dWF4,5904
@@ -147,28 +147,26 @@ phoenix/trace/span_evaluations.py,sha256=T67grfU71iANMM0g6dp2OQ_ZjtkON2viZz95_U9
 phoenix/trace/span_json_decoder.py,sha256=IAFakPRqSMYxTPKYFMiXYxm7U-FipdN8_xbvapDS0Qc,3131
 phoenix/trace/span_json_encoder.py,sha256=LKITl8Evuv3Qb1_EPW4Kg4Yu9BU9vyE1WCe8P2rE1-s,1776
 phoenix/trace/trace_dataset.py,sha256=RpHIfZLbMmULOIb-fKXJkQLhIdC0sJlAOTjlyJppMYA,13776
-phoenix/trace/tracer.py,sha256=9V3eFwKd-eAyMZ9ZzX8HyrGZEVlWx0Ni-_v1LG3J1kY,3640
 phoenix/trace/utils.py,sha256=7LurVGXn245cjj4MJsc7v6jq4DSJkpK6YGBfIaSywuw,1307
 phoenix/trace/dsl/__init__.py,sha256=WIQIjJg362XD3s50OsPJJ0xbDsGp41bSv7vDllLrPuA,144
 phoenix/trace/dsl/filter.py,sha256=paLpcSMnHdgCfcvcroaqOoCe2retAZ5ocp_5cNTnv9s,14167
 phoenix/trace/dsl/helpers.py,sha256=CP6WaJpP7_WdI1Yoip7tDWcYDXZRg6xgNBRS2gojjMc,1728
 phoenix/trace/dsl/missing.py,sha256=BWPOHr2_tBkPDgVeq8GVXXVbNbJiBelu4NtwHBg6mTE,1435
 phoenix/trace/dsl/query.py,sha256=k0guhWBEo6L7ZJH5FJs2-iGSnWXdUUqu09gd-8M4CGg,14783
-phoenix/trace/langchain/__init__.py,sha256=aTKMFmEOgjx_6dnyplalgYi7PQnetablwrwpcUZqcGE,764
-phoenix/trace/langchain/instrumentor.py,sha256=95jtReVDtpymfK8eUGybevBJQWpbMnL9dg6XBgXmOQM,1420
-phoenix/trace/langchain/tracer.py,sha256=gq5RDgFexw-4qpwf701hGsZGcT6XdPjRIbI5aZdBI2I,1425
+phoenix/trace/langchain/__init__.py,sha256=F37GfD1pd5Kuw7R7iRUM1zXXpO8xEcycNZh5dwqBXNk,109
+phoenix/trace/langchain/instrumentor.py,sha256=iy4seCcuTZutNmq0HFmdjIEFfUlg0zDAYXKPY8ApEwQ,1302
 phoenix/trace/llama_index/__init__.py,sha256=4fpR5702Qh2t5TaXIx584EkA-BveCPftXPOKvI0Oi3I,105
-phoenix/trace/llama_index/callback.py,sha256=EWHcg2IckOWMVUi1n1CmcDXRoMDoLTgGZ1LkqP2_fCg,4499
+phoenix/trace/llama_index/callback.py,sha256=1U9-Br1aQHVNJH9geFeSfwa3S4lxQLvxXm_5Sv0oSaM,4381
 phoenix/trace/openai/__init__.py,sha256=J3G0uqCxGdksUpaQVHds_Egv2drvh8UEqoLjiQAOveg,79
-phoenix/trace/openai/instrumentor.py,sha256=OJGwuLXVChFmMcuojQrfBFsXvx4YsAsGhgwtgLfEX4c,1318
+phoenix/trace/openai/instrumentor.py,sha256=Z7iN3lV6bS6lVEohQmJyaEApRVszcrVy0AffG-_7Ju4,1200
 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=8w1Ivw0KO9YKWrhcdnO73cSVqP9VHAp0pSfsi_oDiuQ,672
 phoenix/utilities/error_handling.py,sha256=7b5rpGFj9EWZ8yrZK1IHvxB89suWk3lggDayUQcvZds,1946
 phoenix/utilities/logging.py,sha256=lDXd6EGaamBNcQxL4vP1au9-i_SXe0OraUDiJOcszSw,222
-arize_phoenix-3.10.0.dist-info/METADATA,sha256=ZQfN5x4w8Pd7EdgjWDKFr_2JnQjl8MSr-ZpRx-OoeJ8,29082
-arize_phoenix-3.10.0.dist-info/WHEEL,sha256=TJPnKdtrSue7xZ_AVGkp9YXcvDrobsjBds1du3Nx6dc,87
-arize_phoenix-3.10.0.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
-arize_phoenix-3.10.0.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
-arize_phoenix-3.10.0.dist-info/RECORD,,
+arize_phoenix-3.11.1.dist-info/METADATA,sha256=CxX1LnZOSawfGoh1H2Q222Rw7PWanWDpxUUzbAKWMAo,29080
+arize_phoenix-3.11.1.dist-info/WHEEL,sha256=TJPnKdtrSue7xZ_AVGkp9YXcvDrobsjBds1du3Nx6dc,87
+arize_phoenix-3.11.1.dist-info/licenses/IP_NOTICE,sha256=JBqyyCYYxGDfzQ0TtsQgjts41IJoa-hiwDrBjCb9gHM,469
+arize_phoenix-3.11.1.dist-info/licenses/LICENSE,sha256=HFkW9REuMOkvKRACuwLPT0hRydHb3zNg-fdFt94td18,3794
+arize_phoenix-3.11.1.dist-info/RECORD,,

phoenix/core/traces.py

@@ -1,10 +1,9 @@
 import weakref
 from collections import defaultdict
-from datetime import datetime
 from queue import SimpleQueue
 from threading import RLock, Thread
 from types import MethodType
-from typing import DefaultDict, Iterable, Iterator, Optional, Tuple, Union
+from typing import DefaultDict, Iterator, Optional, Tuple, Union
 
 from typing_extensions import assert_never
 
@@ -13,10 +12,9 @@ from phoenix.core.project import (
     DEFAULT_PROJECT_NAME,
     END_OF_QUEUE,
     Project,
-    WrappedSpan,
     _ProjectName,
 )
-from phoenix.trace.schemas import Span, SpanID, TraceID
+from phoenix.trace.schemas import Span
 
 _SpanItem = Tuple[Span, _ProjectName]
 _EvalItem = Tuple[pb.Evaluation, _ProjectName]
@@ -45,76 +43,6 @@ class Traces:
             projects = tuple(self._projects.items())
         yield from projects
 
-    def get_trace(self, trace_id: TraceID) -> Iterator[WrappedSpan]:
-        with self._lock:
-            if not (project := self._projects.get(DEFAULT_PROJECT_NAME)):
-                return
-        yield from project.get_trace(trace_id)
-
-    def get_spans(
-        self,
-        start_time: Optional[datetime] = None,
-        stop_time: Optional[datetime] = None,
-        root_spans_only: Optional[bool] = False,
-        span_ids: Optional[Iterable[SpanID]] = None,
-    ) -> Iterator[WrappedSpan]:
-        with self._lock:
-            if not (project := self._projects.get(DEFAULT_PROJECT_NAME)):
-                return
-        yield from project.get_spans(start_time, stop_time, root_spans_only, span_ids)
-
-    def get_num_documents(self, span_id: SpanID) -> int:
-        with self._lock:
-            if not (project := self._projects.get(DEFAULT_PROJECT_NAME)):
-                return 0
-        return project.get_num_documents(span_id)
-
-    def root_span_latency_ms_quantiles(self, *probabilities: float) -> Iterator[Optional[float]]:
-        """Root span latency quantiles in milliseconds"""
-        with self._lock:
-            if not (project := self._projects.get(DEFAULT_PROJECT_NAME)):
-                for _ in probabilities:
-                    yield None
-                return
-        for probability in probabilities:
-            yield project.root_span_latency_ms_quantiles(probability)
-
-    def get_descendant_spans(self, span_id: SpanID) -> Iterator[WrappedSpan]:
-        with self._lock:
-            if not (project := self._projects.get(DEFAULT_PROJECT_NAME)):
-                return
-        yield from project.get_descendant_spans(span_id)
-
-    @property
-    def last_updated_at(self) -> Optional[datetime]:
-        with self._lock:
-            if not (project := self._projects.get(DEFAULT_PROJECT_NAME)):
-                return None
-        return project.last_updated_at
-
-    @property
-    def span_count(self) -> int:
-        """Total number of spans"""
-        project_name = DEFAULT_PROJECT_NAME
-        with self._lock:
-            if not (project := self._projects.get(project_name)):
-                return 0
-        return project.span_count()
-
-    @property
-    def token_count_total(self) -> int:
-        with self._lock:
-            if not (project := self._projects.get(DEFAULT_PROJECT_NAME)):
-                return 0
-        return project.token_count_total
-
-    @property
-    def right_open_time_range(self) -> Tuple[Optional[datetime], Optional[datetime]]:
-        with self._lock:
-            if not (project := self._projects.get(DEFAULT_PROJECT_NAME)):
-                return None, None
-        return project.right_open_time_range
-
     def put(
         self,
         item: Union[Span, pb.Evaluation],

phoenix/server/api/types/Project.py

@@ -112,7 +112,7 @@ class Project(Node):
             spans = filter(predicate, spans)
         if sort:
             spans = sort(spans, evals=project)
-        data = list(map(to_gql_span, spans))
+        data = [to_gql_span(span, project) for span in spans]
         return connection_from_list(data=data, args=args)
 
     @strawberry.field(

phoenix/server/api/types/Span.py

@@ -10,7 +10,7 @@ from strawberry import ID, UNSET
 from strawberry.types import Info
 
 import phoenix.trace.schemas as trace_schema
-from phoenix.core.project import DEFAULT_PROJECT_NAME, WrappedSpan
+from phoenix.core.project import DEFAULT_PROJECT_NAME, Project, WrappedSpan
 from phoenix.metrics.retrieval_metrics import RetrievalMetrics
 from phoenix.server.api.context import Context
 from phoenix.server.api.types.DocumentRetrievalMetrics import DocumentRetrievalMetrics
@@ -95,6 +95,7 @@ class SpanEvent:
 
 @strawberry.type
 class Span:
+    project: strawberry.Private[Project]
     name: str
     status_code: SpanStatusCode
     status_message: str
@@ -185,14 +186,10 @@ class Span:
         info: Info[Context, None],
         evaluation_name: Optional[str] = UNSET,
     ) -> List[DocumentRetrievalMetrics]:
-        if (
-            not self.num_documents
-            or not (traces := info.context.traces)
-            or not (project := traces.get_project(DEFAULT_PROJECT_NAME))
-        ):
+        if not self.num_documents:
             return []
         span_id = SpanID(str(self.context.span_id))
-        all_document_evaluation_names = project.get_document_evaluation_names(span_id)
+        all_document_evaluation_names = self.project.get_document_evaluation_names(span_id)
         if not all_document_evaluation_names:
             return []
         if evaluation_name is UNSET:
@@ -203,7 +200,7 @@ class Span:
             evaluation_names = [evaluation_name]
         retrieval_metrics = []
         for name in evaluation_names:
-            evaluation_scores = project.get_document_evaluation_scores(
+            evaluation_scores = self.project.get_document_evaluation_scores(
                 span_id=span_id,
                 evaluation_name=name,
                 num_documents=self.num_documents,
@@ -223,20 +220,20 @@ class Span:
         self,
         info: Info[Context, None],
     ) -> List["Span"]:
-        if (traces := info.context.traces) is None:
-            return []
         return [
-            to_gql_span(span) for span in traces.get_descendant_spans(SpanID(self.context.span_id))
+            to_gql_span(span, self.project)
+            for span in self.project.get_descendant_spans(SpanID(self.context.span_id))
         ]
 
 
-def to_gql_span(span: WrappedSpan) -> "Span":
+def to_gql_span(span: WrappedSpan, project: Project) -> "Span":
     events: List[SpanEvent] = list(map(SpanEvent.from_event, span.events))
     input_value = cast(Optional[str], span.attributes.get(INPUT_VALUE))
     output_value = cast(Optional[str], span.attributes.get(OUTPUT_VALUE))
     retrieval_documents = span.attributes.get(RETRIEVAL_DOCUMENTS)
     num_documents = len(retrieval_documents) if isinstance(retrieval_documents, Sized) else None
     return Span(
+        project=project,
         name=span.name,
         status_code=SpanStatusCode(span.status_code),
         status_message=span.status_message,