arize-phoenix 3.16.1__py3-none-any.whl → 7.7.0__py3-none-any.whl

phoenix/server/api/input_types/CreateTraceAnnotationInput.py

@@ -0,0 +1,18 @@
+from typing import Optional
+
+import strawberry
+from strawberry.relay import GlobalID
+from strawberry.scalars import JSON
+
+from phoenix.server.api.types.AnnotatorKind import AnnotatorKind
+
+
+@strawberry.input
+class CreateTraceAnnotationInput:
+    trace_id: GlobalID
+    name: str
+    annotator_kind: AnnotatorKind
+    label: Optional[str] = None
+    score: Optional[float] = None
+    explanation: Optional[str] = None
+    metadata: JSON = strawberry.field(default_factory=dict)

phoenix/server/api/input_types/DataQualityMetricInput.py

@@ -8,7 +8,10 @@ from typing_extensions import Annotated
 from phoenix.core.model_schema import Column
 from phoenix.metrics import Metric
 from phoenix.metrics.mixins import UnaryOperator
-from phoenix.server.api.types.DataQualityMetric import DataQualityMetric
+from phoenix.server.api.types.DataQualityMetric import (
+    DATA_QUALITY_METRIC_FACTORIES,
+    DataQualityMetric,
+)
 
 
 @strawberry.input
@@ -26,7 +29,7 @@ class DataQualityMetricInput:
     metric_instance: strawberry.Private[Metric] = field(init=False)
 
     def __post_init__(self) -> None:
-        metric_instance = self.metric.value()
+        metric_instance = DATA_QUALITY_METRIC_FACTORIES[self.metric]()
         if isinstance(metric_instance, UnaryOperator):
             if not isinstance(self.column_name, str):
                 raise ValueError(f"column name must not be null for {self.metric.name}")

phoenix/server/api/input_types/DatasetExampleInput.py

@@ -0,0 +1,14 @@
+from typing import Optional
+
+import strawberry
+from strawberry import UNSET
+from strawberry.relay import GlobalID
+from strawberry.scalars import JSON
+
+
+@strawberry.input
+class DatasetExampleInput:
+    input: JSON
+    output: JSON
+    metadata: JSON
+    span_id: Optional[GlobalID] = UNSET

phoenix/server/api/input_types/DatasetSort.py

@@ -0,0 +1,17 @@
+from enum import Enum
+
+import strawberry
+
+from phoenix.server.api.types.SortDir import SortDir
+
+
+@strawberry.enum
+class DatasetColumn(Enum):
+    createdAt = "created_at"
+    name = "name"
+
+
+@strawberry.input(description="The sort key and direction for dataset connections")
+class DatasetSort:
+    col: DatasetColumn
+    dir: SortDir

phoenix/server/api/input_types/DatasetVersionSort.py

@@ -0,0 +1,16 @@
+from enum import Enum
+
+import strawberry
+
+from phoenix.server.api.types.SortDir import SortDir
+
+
+@strawberry.enum
+class DatasetVersionColumn(Enum):
+    createdAt = "created_at"
+
+
+@strawberry.input(description="The sort key and direction for dataset version connections")
+class DatasetVersionSort:
+    col: DatasetVersionColumn
+    dir: SortDir

phoenix/server/api/input_types/DeleteAnnotationsInput.py

@@ -0,0 +1,7 @@
+import strawberry
+from strawberry.relay import GlobalID
+
+
+@strawberry.input
+class DeleteAnnotationsInput:
+    annotation_ids: list[GlobalID]

phoenix/server/api/input_types/DeleteDatasetExamplesInput.py

@@ -0,0 +1,13 @@
+from typing import Optional
+
+import strawberry
+from strawberry import UNSET
+from strawberry.relay import GlobalID
+from strawberry.scalars import JSON
+
+
+@strawberry.input
+class DeleteDatasetExamplesInput:
+    example_ids: list[GlobalID]
+    dataset_version_description: Optional[str] = UNSET
+    dataset_version_metadata: Optional[JSON] = UNSET

phoenix/server/api/input_types/DeleteDatasetInput.py

@@ -0,0 +1,7 @@
+import strawberry
+from strawberry.relay import GlobalID
+
+
+@strawberry.input
+class DeleteDatasetInput:
+    dataset_id: GlobalID

phoenix/server/api/input_types/DeleteExperimentsInput.py

@@ -0,0 +1,7 @@
+import strawberry
+from strawberry.relay import GlobalID
+
+
+@strawberry.input
+class DeleteExperimentsInput:
+    experiment_ids: list[GlobalID]

phoenix/server/api/input_types/DimensionFilter.py

@@ -1,4 +1,4 @@
-from typing import List, Optional
+from typing import Optional
 
 import strawberry
 from strawberry import UNSET
@@ -52,9 +52,9 @@ class DimensionFilter:
 
     """
 
-    types: Optional[List[DimensionType]] = UNSET
-    shapes: Optional[List[DimensionShape]] = UNSET
-    data_types: Optional[List[DimensionDataType]] = UNSET
+    types: Optional[list[DimensionType]] = UNSET
+    shapes: Optional[list[DimensionShape]] = UNSET
+    data_types: Optional[list[DimensionDataType]] = UNSET
 
     def __post_init__(self) -> None:
         self.types = ensure_list(self.types)

phoenix/server/api/input_types/GenerativeModelInput.py

@@ -0,0 +1,17 @@
+from typing import Optional
+
+import strawberry
+from strawberry import UNSET
+
+from phoenix.server.api.types.GenerativeProvider import GenerativeProviderKey
+
+
+@strawberry.input
+class GenerativeModelInput:
+    provider_key: GenerativeProviderKey
+    name: str
+    """ The name of the model. Or the Deployment Name for Azure OpenAI models. """
+    endpoint: Optional[str] = UNSET
+    """ The endpoint to use for the model. Only required for Azure OpenAI models. """
+    api_version: Optional[str] = UNSET
+    """ The API version to use for the model. """

phoenix/server/api/input_types/Granularity.py

@@ -1,6 +1,6 @@
+from collections.abc import Iterator
 from datetime import datetime, timedelta
 from itertools import accumulate, repeat, takewhile
-from typing import Iterator
 
 import strawberry
 

phoenix/server/api/input_types/InvocationParameters.py

@@ -0,0 +1,162 @@
+from enum import Enum
+from typing import Annotated, Any, Mapping, Optional, Union
+
+import strawberry
+from strawberry import UNSET
+from strawberry.scalars import JSON
+
+
+@strawberry.enum
+class CanonicalParameterName(str, Enum):
+    TEMPERATURE = "temperature"
+    MAX_COMPLETION_TOKENS = "max_completion_tokens"
+    STOP_SEQUENCES = "stop_sequences"
+    TOP_P = "top_p"
+    RANDOM_SEED = "random_seed"
+    TOOL_CHOICE = "tool_choice"
+    RESPONSE_FORMAT = "response_format"
+
+
+@strawberry.enum
+class InvocationInputField(str, Enum):
+    value_int = "value_int"
+    value_float = "value_float"
+    value_bool = "value_bool"
+    value_string = "value_string"
+    value_json = "value_json"
+    value_string_list = "value_string_list"
+    value_boolean = "value_boolean"
+
+
+@strawberry.input
+class InvocationParameterInput:
+    invocation_name: str
+    canonical_name: Optional[CanonicalParameterName] = None
+    value_int: Optional[int] = UNSET
+    value_float: Optional[float] = UNSET
+    value_bool: Optional[bool] = UNSET
+    value_string: Optional[str] = UNSET
+    value_json: Optional[JSON] = UNSET
+    value_string_list: Optional[list[str]] = UNSET
+    value_boolean: Optional[bool] = UNSET
+
+
+@strawberry.interface
+class InvocationParameterBase:
+    invocation_name: str
+    canonical_name: Optional[CanonicalParameterName] = None
+    label: str
+    required: bool = False
+
+
+@strawberry.type
+class IntInvocationParameter(InvocationParameterBase):
+    invocation_input_field: InvocationInputField = InvocationInputField.value_int
+    default_value: Optional[int] = None
+
+
+@strawberry.type
+class FloatInvocationParameter(InvocationParameterBase):
+    invocation_input_field: InvocationInputField = InvocationInputField.value_float
+    default_value: Optional[float] = None
+
+
+@strawberry.type
+class BoundedFloatInvocationParameter(InvocationParameterBase):
+    invocation_input_field: InvocationInputField = InvocationInputField.value_float
+    default_value: Optional[float] = None
+    min_value: float
+    max_value: float
+
+
+@strawberry.type
+class StringInvocationParameter(InvocationParameterBase):
+    invocation_input_field: InvocationInputField = InvocationInputField.value_string
+    default_value: Optional[str] = None
+
+
+@strawberry.type
+class JSONInvocationParameter(InvocationParameterBase):
+    invocation_input_field: InvocationInputField = InvocationInputField.value_json
+    default_value: Optional[JSON] = None
+
+
+@strawberry.type
+class StringListInvocationParameter(InvocationParameterBase):
+    invocation_input_field: InvocationInputField = InvocationInputField.value_string_list
+    default_value: Optional[list[str]] = None
+
+
+@strawberry.type
+class BooleanInvocationParameter(InvocationParameterBase):
+    invocation_input_field: InvocationInputField = InvocationInputField.value_bool
+    default_value: Optional[bool] = None
+
+
+def extract_parameter(
+    param_def: InvocationParameterBase, param_input: InvocationParameterInput
+) -> Any:
+    if isinstance(param_def, IntInvocationParameter):
+        return (
+            param_input.value_int if param_input.value_int is not UNSET else param_def.default_value
+        )
+    elif isinstance(param_def, FloatInvocationParameter):
+        return (
+            param_input.value_float
+            if param_input.value_float is not UNSET
+            else param_def.default_value
+        )
+    elif isinstance(param_def, BoundedFloatInvocationParameter):
+        return (
+            param_input.value_float
+            if param_input.value_float is not UNSET
+            else param_def.default_value
+        )
+    elif isinstance(param_def, StringInvocationParameter):
+        return (
+            param_input.value_string
+            if param_input.value_string is not UNSET
+            else param_def.default_value
+        )
+    elif isinstance(param_def, JSONInvocationParameter):
+        return (
+            param_input.value_json
+            if param_input.value_json is not UNSET
+            else param_def.default_value
+        )
+    elif isinstance(param_def, StringListInvocationParameter):
+        return (
+            param_input.value_string_list
+            if param_input.value_string_list is not UNSET
+            else param_def.default_value
+        )
+    elif isinstance(param_def, BooleanInvocationParameter):
+        return (
+            param_input.value_bool
+            if param_input.value_bool is not UNSET
+            else param_def.default_value
+        )
+
+
+def validate_invocation_parameters(
+    parameters: list["InvocationParameter"],
+    input: Mapping[str, Any],
+) -> None:
+    for param_def in parameters:
+        if param_def.required and param_def.invocation_name not in input:
+            raise ValueError(f"Required parameter {param_def.invocation_name} not provided")
+
+
+# Create the union for output types
+InvocationParameter = Annotated[
+    Union[
+        IntInvocationParameter,
+        FloatInvocationParameter,
+        BoundedFloatInvocationParameter,
+        StringInvocationParameter,
+        JSONInvocationParameter,
+        StringListInvocationParameter,
+        BooleanInvocationParameter,
+    ],
+    strawberry.union("InvocationParameter"),
+]

phoenix/server/api/input_types/PatchAnnotationInput.py

@@ -0,0 +1,19 @@
+from typing import Optional
+
+import strawberry
+from strawberry import UNSET
+from strawberry.relay import GlobalID
+from strawberry.scalars import JSON
+
+from phoenix.server.api.types.AnnotatorKind import AnnotatorKind
+
+
+@strawberry.input
+class PatchAnnotationInput:
+    annotation_id: GlobalID
+    name: Optional[str] = UNSET
+    annotator_kind: Optional[AnnotatorKind] = UNSET
+    label: Optional[str] = UNSET
+    score: Optional[float] = UNSET
+    explanation: Optional[str] = UNSET
+    metadata: Optional[JSON] = UNSET

phoenix/server/api/input_types/PatchDatasetExamplesInput.py

@@ -0,0 +1,35 @@
+from typing import Optional
+
+import strawberry
+from strawberry import UNSET
+from strawberry.relay import GlobalID
+from strawberry.scalars import JSON
+
+
+@strawberry.input
+class DatasetExamplePatch:
+    """
+    Contains the information needed to apply a patch revision to a dataset example.
+    """
+
+    example_id: GlobalID
+    input: Optional[JSON] = UNSET
+    output: Optional[JSON] = UNSET
+    metadata: Optional[JSON] = UNSET
+
+    def is_empty(self) -> bool:
+        """
+        Non-empty patches have at least one field set.
+        """
+        return all(field is UNSET for field in (self.input, self.output, self.metadata))
+
+
+@strawberry.input
+class PatchDatasetExamplesInput:
+    """
+    Input type to the patchDatasetExamples mutation.
+    """
+
+    patches: list[DatasetExamplePatch]
+    version_description: Optional[str] = UNSET
+    version_metadata: Optional[JSON] = UNSET

phoenix/server/api/input_types/PatchDatasetInput.py

@@ -0,0 +1,14 @@
+from typing import Optional
+
+import strawberry
+from strawberry import UNSET
+from strawberry.relay import GlobalID
+from strawberry.scalars import JSON
+
+
+@strawberry.input
+class PatchDatasetInput:
+    dataset_id: GlobalID
+    name: Optional[str] = UNSET
+    description: Optional[str] = UNSET
+    metadata: Optional[JSON] = UNSET

phoenix/server/api/input_types/PerformanceMetricInput.py

@@ -3,7 +3,10 @@ import strawberry
 from phoenix.core.model_schema import ACTUAL_LABEL, PREDICTION_LABEL, Column, Model
 from phoenix.metrics import Metric
 from phoenix.metrics.mixins import EvaluationMetric
-from phoenix.server.api.types.PerformanceMetric import PerformanceMetric
+from phoenix.server.api.types.PerformanceMetric import (
+    PERFORMANCE_METRIC_FUNCTIONS,
+    PerformanceMetric,
+)
 
 
 @strawberry.input
@@ -14,5 +17,5 @@ class PerformanceMetricInput:
         return EvaluationMetric(
             actual=Column(model[ACTUAL_LABEL].name),
             predicted=Column(model[PREDICTION_LABEL].name),
-            eval=self.metric.value,
+            eval=PERFORMANCE_METRIC_FUNCTIONS[self.metric],
         )

phoenix/server/api/input_types/ProjectSessionSort.py

@@ -0,0 +1,29 @@
+from enum import Enum, auto
+
+import strawberry
+from typing_extensions import assert_never
+
+from phoenix.server.api.types.pagination import CursorSortColumnDataType
+from phoenix.server.api.types.SortDir import SortDir
+
+
+@strawberry.enum
+class ProjectSessionColumn(Enum):
+    startTime = auto()
+    endTime = auto()
+    tokenCountTotal = auto()
+    numTraces = auto()
+
+    @property
+    def data_type(self) -> CursorSortColumnDataType:
+        if self is ProjectSessionColumn.tokenCountTotal or self is ProjectSessionColumn.numTraces:
+            return CursorSortColumnDataType.INT
+        if self is ProjectSessionColumn.startTime or self is ProjectSessionColumn.endTime:
+            return CursorSortColumnDataType.DATETIME
+        assert_never(self)
+
+
+@strawberry.input(description="The sort key and direction for ProjectSession connections.")
+class ProjectSessionSort:
+    col: ProjectSessionColumn
+    dir: SortDir

phoenix/server/api/input_types/SpanAnnotationSort.py

@@ -0,0 +1,17 @@
+from enum import Enum
+
+import strawberry
+
+from phoenix.server.api.types.SortDir import SortDir
+
+
+@strawberry.enum
+class SpanAnnotationColumn(Enum):
+    createdAt = "created_at"
+    name = "name"
+
+
+@strawberry.input(description="The sort key and direction for SpanAnnotation connections")
+class SpanAnnotationSort:
+    col: SpanAnnotationColumn
+    dir: SortDir