arize-phoenix 5.5.2__py3-none-any.whl → 5.6.0__py3-none-any.whl

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

@@ -1,6 +1,6 @@
 import gzip
 import zlib
-from typing import Any, Dict, List, Literal, Optional
+from typing import Any, Literal, Optional
 
 from fastapi import APIRouter, BackgroundTasks, Header, HTTPException, Query
 from google.protobuf.message import DecodeError
@@ -110,7 +110,7 @@ class TraceAnnotation(V1RoutesBaseModel):
     result: Optional[TraceAnnotationResult] = Field(
         default=None, description="The result of the annotation"
     )
-    metadata: Optional[Dict[str, Any]] = Field(
+    metadata: Optional[dict[str, Any]] = Field(
         default=None, description="Metadata for the annotation"
     )
 
@@ -128,15 +128,15 @@ class TraceAnnotation(V1RoutesBaseModel):
         )
 
 
-class AnnotateTracesRequestBody(RequestBody[List[TraceAnnotation]]):
-    data: List[TraceAnnotation] = Field(description="The trace annotations to be upserted")
+class AnnotateTracesRequestBody(RequestBody[list[TraceAnnotation]]):
+    data: list[TraceAnnotation] = Field(description="The trace annotations to be upserted")
 
 
 class InsertedTraceAnnotation(V1RoutesBaseModel):
     id: str = Field(description="The ID of the inserted trace annotation")
 
 
-class AnnotateTracesResponseBody(ResponseBody[List[InsertedTraceAnnotation]]):
+class AnnotateTracesResponseBody(ResponseBody[list[InsertedTraceAnnotation]]):
     pass
 
 

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

@@ -1,4 +1,4 @@
-from typing import Any, Dict, Generic, List, Optional, TypedDict, TypeVar, Union
+from typing import Any, Generic, Optional, TypedDict, TypeVar, Union
 
 from typing_extensions import TypeAlias, assert_never
 
@@ -6,8 +6,8 @@ from .pydantic_compat import V1RoutesBaseModel
 
 StatusCode: TypeAlias = int
 DataType = TypeVar("DataType")
-Responses: TypeAlias = Dict[
-    Union[int, str], Dict[str, Any]
+Responses: TypeAlias = dict[
+    Union[int, str], dict[str, Any]
 ]  # input type for the `responses` parameter of a fastapi route
 
 
@@ -44,12 +44,12 @@ class PaginatedResponseBody(V1RoutesBaseModel, Generic[DataType]):
     # Don't use """ for this docstring or it will be included as a description
     # in the generated OpenAPI schema.
 
-    data: List[DataType]
+    data: list[DataType]
     next_cursor: Optional[str]
 
 
 def add_errors_to_responses(
-    errors: List[Union[StatusCode, StatusCodeWithDescription]],
+    errors: list[Union[StatusCode, StatusCodeWithDescription]],
     /,
     *,
     responses: Optional[Responses] = None,