arize-phoenix 4.10.2rc2__py3-none-any.whl → 4.12.0__py3-none-any.whl

phoenix/server/thread_server.py

@@ -4,7 +4,7 @@ from threading import Thread
 from time import sleep, time
 from typing import Generator
 
-from fastapi import FastAPI
+from starlette.applications import Starlette
 from uvicorn import Config, Server
 from uvicorn.config import LoopSetupType
 
@@ -24,7 +24,7 @@ class ThreadServer(Server):
 
     def __init__(
         self,
-        app: FastAPI,
+        app: Starlette,
         host: str,
         port: int,
         root_path: str,

phoenix/session/client.py

@@ -51,6 +51,9 @@ from phoenix.trace.otel import encode_span_to_otlp
 
 logger = logging.getLogger(__name__)
 
+
+DEFAULT_TIMEOUT_IN_SECONDS = 5
+
 DatasetAction: TypeAlias = Literal["create", "append"]
 
 
@@ -120,6 +123,7 @@ class Client(TraceDataExtractor):
         project_name: Optional[str] = None,
         # Deprecated
         stop_time: Optional[datetime] = None,
+        timeout: Optional[int] = DEFAULT_TIMEOUT_IN_SECONDS,
     ) -> Optional[Union[pd.DataFrame, List[pd.DataFrame]]]:
         """
         Queries spans from the Phoenix server or active session based on specified criteria.
@@ -159,6 +163,7 @@ class Client(TraceDataExtractor):
                 "limit": limit,
                 "root_spans_only": root_spans_only,
             },
+            timeout=timeout,
         )
         if response.status_code == 404:
             logger.info("No spans found.")

phoenix/session/data_extractor.py

@@ -9,6 +9,7 @@ from phoenix.trace.dsl import SpanQuery
 from phoenix.trace.trace_dataset import TraceDataset
 
 DEFAULT_SPAN_LIMIT = 1000
+DEFAULT_TIMEOUT_IN_SECONDS = 5
 
 
 class TraceDataExtractor(ABC):
@@ -26,6 +27,7 @@ class TraceDataExtractor(ABC):
         limit: Optional[int] = DEFAULT_SPAN_LIMIT,
         root_spans_only: Optional[bool] = None,
         project_name: Optional[str] = None,
+        timeout: Optional[int] = DEFAULT_TIMEOUT_IN_SECONDS,
     ) -> Optional[Union[pd.DataFrame, List[pd.DataFrame]]]: ...
 
     def get_spans_dataframe(
@@ -37,6 +39,7 @@ class TraceDataExtractor(ABC):
         limit: Optional[int] = DEFAULT_SPAN_LIMIT,
         root_spans_only: Optional[bool] = None,
         project_name: Optional[str] = None,
+        timeout: Optional[int] = DEFAULT_TIMEOUT_IN_SECONDS,
     ) -> Optional[pd.DataFrame]:
         return cast(
             Optional[pd.DataFrame],
@@ -47,6 +50,7 @@ class TraceDataExtractor(ABC):
                 limit=limit,
                 root_spans_only=root_spans_only,
                 project_name=project_name,
+                timeout=timeout,
             ),
         )
 
@@ -59,8 +63,23 @@ class TraceDataExtractor(ABC):
     def get_trace_dataset(
         self,
         project_name: Optional[str] = None,
+        *,
+        start_time: Optional[datetime] = None,
+        end_time: Optional[datetime] = None,
+        limit: Optional[int] = DEFAULT_SPAN_LIMIT,
+        root_spans_only: Optional[bool] = None,
+        timeout: Optional[int] = DEFAULT_TIMEOUT_IN_SECONDS,
     ) -> Optional[TraceDataset]:
-        if (dataframe := self.get_spans_dataframe(project_name=project_name)) is None:
+        if (
+            dataframe := self.get_spans_dataframe(
+                project_name=project_name,
+                start_time=start_time,
+                end_time=end_time,
+                limit=limit,
+                root_spans_only=root_spans_only,
+                timeout=timeout,
+            )
+        ) is None:
             return None
         evaluations = self.get_evaluations(project_name=project_name)
         return TraceDataset(dataframe=dataframe, evaluations=evaluations)

phoenix/session/session.py

@@ -75,6 +75,9 @@ global _session_working_dir
 _session_working_dir: Optional["TemporaryDirectory[str]"] = None
 
 
+DEFAULT_TIMEOUT_IN_SECONDS = 5
+
+
 class NotebookEnvironment(Enum):
     COLAB = "colab"
     LOCAL = "local"
@@ -152,6 +155,7 @@ class Session(TraceDataExtractor, ABC):
         project_name: Optional[str] = None,
         # Deprecated fields
         stop_time: Optional[datetime] = None,
+        timeout: Optional[int] = DEFAULT_TIMEOUT_IN_SECONDS,
     ) -> Optional[Union[pd.DataFrame, List[pd.DataFrame]]]:
         """
         Queries the spans in the project based on the provided parameters.

phoenix/trace/attributes.py

@@ -33,6 +33,7 @@ from typing import (
     cast,
 )
 
+import numpy as np
 from openinference.semconv import trace
 from openinference.semconv.trace import DocumentAttributes, SpanAttributes
 from typing_extensions import assert_never
@@ -307,7 +308,7 @@ def _flatten_mapping(
                     json_string_attributes=json_string_attributes,
                     separator=separator,
                 )
-        elif isinstance(value, Sequence) and recurse_on_sequence:
+        elif (isinstance(value, Sequence) or isinstance(value, np.ndarray)) and recurse_on_sequence:
             yield from _flatten_sequence(
                 value,
                 prefix=prefixed_key,

phoenix/trace/schemas.py

@@ -39,6 +39,7 @@ class SpanKind(Enum):
     AGENT = "AGENT"
     RERANKER = "RERANKER"
     EVALUATOR = "EVALUATOR"
+    GUARDRAIL = "GUARDRAIL"
     UNKNOWN = "UNKNOWN"
 
     def __str__(self) -> str:

phoenix/trace/span_json_decoder.py

@@ -81,7 +81,7 @@ def json_to_span(data: Dict[str, Any]) -> Any:
                 timestamp=datetime.fromisoformat(event["timestamp"]),
             )
             for event in (
-                data["events"] if isinstance(data["events"], list) else json.loads(data["events"])
+                json.loads(data["events"]) if isinstance(data["events"], str) else data["events"]
             )
         ]
         data["conversation"] = (

phoenix/version.py

@@ -1 +1 @@
-__version__ = "4.10.2rc2"
+__version__ = "4.12.0"

phoenix/server/api/routers/v1/utils.py

@@ -1,94 +0,0 @@
-from typing import Any, Dict, Generic, List, Optional, TypedDict, Union
-
-from pydantic import BaseModel
-from typing_extensions import TypeAlias, TypeVar, assert_never
-
-StatusCode: TypeAlias = int
-DataType = TypeVar("DataType")
-Responses: TypeAlias = Dict[
-    Union[int, str], Dict[str, Any]
-]  # input type for the `responses` parameter of a fastapi route
-
-
-class StatusCodeWithDescription(TypedDict):
-    """
-    A duck type for a status code with a description detailing under what
-    conditions the status code is raised.
-    """
-
-    status_code: StatusCode
-    description: str
-
-
-class RequestBody(BaseModel, Generic[DataType]):
-    # A generic request type accepted by V1 routes.
-    #
-    # Don't use """ for this docstring or it will be included as a description
-    # in the generated OpenAPI schema.
-    data: DataType
-
-
-class ResponseBody(BaseModel, Generic[DataType]):
-    # A generic response type returned by V1 routes.
-    #
-    # Don't use """ for this docstring or it will be included as a description
-    # in the generated OpenAPI schema.
-
-    data: DataType
-
-
-class PaginatedResponseBody(BaseModel, Generic[DataType]):
-    # A generic paginated response type returned by V1 routes.
-    #
-    # Don't use """ for this docstring or it will be included as a description
-    # in the generated OpenAPI schema.
-
-    data: List[DataType]
-    next_cursor: Optional[str]
-
-
-def add_errors_to_responses(
-    errors: List[Union[StatusCode, StatusCodeWithDescription]],
-    /,
-    *,
-    responses: Optional[Responses] = None,
-) -> Responses:
-    """
-    Creates or updates a patch for an OpenAPI schema's `responses` section to
-    include status codes in the generated OpenAPI schema.
-    """
-    output_responses: Responses = responses or {}
-    for error in errors:
-        status_code: int
-        description: Optional[str] = None
-        if isinstance(error, StatusCode):
-            status_code = error
-        elif isinstance(error, dict):
-            status_code = error["status_code"]
-            description = error["description"]
-        else:
-            assert_never(error)
-        if status_code not in output_responses:
-            output_responses[status_code] = {
-                "content": {"text/plain": {"schema": {"type": "string"}}}
-            }
-        if description:
-            output_responses[status_code]["description"] = description
-    return output_responses
-
-
-def add_text_csv_content_to_responses(
-    status_code: StatusCode, /, *, responses: Optional[Responses] = None
-) -> Responses:
-    """
-    Creates or updates a patch for an OpenAPI schema's `responses` section to
-    ensure that the response for the given status code is marked as text/csv in
-    the generated OpenAPI schema.
-    """
-    output_responses: Responses = responses or {}
-    if status_code not in output_responses:
-        output_responses[status_code] = {}
-    output_responses[status_code]["content"] = {
-        "text/csv": {"schema": {"type": "string", "contentMediaType": "text/csv"}}
-    }
-    return output_responses