arize-phoenix 4.10.2rc2__py3-none-any.whl → 4.11.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/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.11.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