Skip to content

Commit aed1402

Browse files
Arize AI Devsfjcasti1
authored andcommitted
Project import generated by Copybara.
GitOrigin-RevId: 0530595cc4b5f97e969a3462edce40d6ec425f5a
1 parent ae4d0ff commit aed1402

37 files changed

+2860
-24
lines changed

CHANGELOG.md

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,12 @@
11
# Changelog
22

3+
## [8.4.0](https://github.com/Arize-ai/arize/compare/arize-python-sdk/v8.3.0...arize-python-sdk/v8.4.0) (2026-02-20)
4+
5+
6+
### 🎁 New Features
7+
8+
* Annotation configs ([#63369](https://github.com/Arize-ai/arize/issues/63369)) ([d6a6b00](https://github.com/Arize-ai/arize/commit/d6a6b0011dcaafd1eeaf58551b264738c95170f3))
9+
310
## [8.3.0](https://github.com/Arize-ai/arize/compare/arize-python-sdk/v8.2.1...arize-python-sdk/v8.3.0) (2026-02-18)
411

512

docs/source/_static/switcher.json

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,10 @@
44
"url": "https://arize-client-python.readthedocs.io/en/latest/",
55
"preferred": true
66
},
7+
{
8+
"version": "v8.4.0",
9+
"url": "https://arize-client-python.readthedocs.io/en/v8.4.0/"
10+
},
711
{
812
"version": "v8.3.0",
913
"url": "https://arize-client-python.readthedocs.io/en/v8.3.0/"

src/.openapi-generator/FILES

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,10 +2,12 @@ arize/_generated/__init__.py
22
arize/_generated/api_client/__init__.py
33
arize/_generated/api_client/api/__init__.py
44
arize/_generated/api_client/api/annotation_configs_api.py
5+
arize/_generated/api_client/api/annotation_queues_api.py
56
arize/_generated/api_client/api/datasets_api.py
67
arize/_generated/api_client/api/experiments_api.py
78
arize/_generated/api_client/api/projects_api.py
89
arize/_generated/api_client/api/prompts_api.py
10+
arize/_generated/api_client/api/spaces_api.py
911
arize/_generated/api_client/api/spans_api.py
1012
arize/_generated/api_client/api_client.py
1113
arize/_generated/api_client/api_response.py
@@ -15,6 +17,9 @@ arize/_generated/api_client/docs/AnnotationConfigBase.md
1517
arize/_generated/api_client/docs/AnnotationConfigCreateBase.md
1618
arize/_generated/api_client/docs/AnnotationConfigsApi.md
1719
arize/_generated/api_client/docs/AnnotationConfigsList200Response.md
20+
arize/_generated/api_client/docs/AnnotationQueue.md
21+
arize/_generated/api_client/docs/AnnotationQueuesApi.md
22+
arize/_generated/api_client/docs/AnnotationQueuesList200Response.md
1823
arize/_generated/api_client/docs/CategoricalAnnotationConfig.md
1924
arize/_generated/api_client/docs/CategoricalAnnotationConfigCreate.md
2025
arize/_generated/api_client/docs/CategoricalAnnotationValue.md
@@ -64,6 +69,9 @@ arize/_generated/api_client/docs/ProviderParamsBedrockOptions.md
6469
arize/_generated/api_client/docs/ResponseFormat.md
6570
arize/_generated/api_client/docs/ResponseFormatJsonSchema.md
6671
arize/_generated/api_client/docs/ResponseFormatType.md
72+
arize/_generated/api_client/docs/Space.md
73+
arize/_generated/api_client/docs/SpacesApi.md
74+
arize/_generated/api_client/docs/SpacesList200Response.md
6775
arize/_generated/api_client/docs/Span.md
6876
arize/_generated/api_client/docs/SpanContext.md
6977
arize/_generated/api_client/docs/SpanEvent.md
@@ -80,6 +88,8 @@ arize/_generated/api_client/models/annotation_config.py
8088
arize/_generated/api_client/models/annotation_config_base.py
8189
arize/_generated/api_client/models/annotation_config_create_base.py
8290
arize/_generated/api_client/models/annotation_configs_list200_response.py
91+
arize/_generated/api_client/models/annotation_queue.py
92+
arize/_generated/api_client/models/annotation_queues_list200_response.py
8393
arize/_generated/api_client/models/categorical_annotation_config.py
8494
arize/_generated/api_client/models/categorical_annotation_config_create.py
8595
arize/_generated/api_client/models/categorical_annotation_value.py
@@ -125,6 +135,8 @@ arize/_generated/api_client/models/provider_params_bedrock_options.py
125135
arize/_generated/api_client/models/response_format.py
126136
arize/_generated/api_client/models/response_format_json_schema.py
127137
arize/_generated/api_client/models/response_format_type.py
138+
arize/_generated/api_client/models/space.py
139+
arize/_generated/api_client/models/spaces_list200_response.py
128140
arize/_generated/api_client/models/span.py
129141
arize/_generated/api_client/models/span_context.py
130142
arize/_generated/api_client/models/span_event.py
@@ -141,6 +153,9 @@ arize/_generated/api_client/test/test_annotation_config_base.py
141153
arize/_generated/api_client/test/test_annotation_config_create_base.py
142154
arize/_generated/api_client/test/test_annotation_configs_api.py
143155
arize/_generated/api_client/test/test_annotation_configs_list200_response.py
156+
arize/_generated/api_client/test/test_annotation_queue.py
157+
arize/_generated/api_client/test/test_annotation_queues_api.py
158+
arize/_generated/api_client/test/test_annotation_queues_list200_response.py
144159
arize/_generated/api_client/test/test_categorical_annotation_config.py
145160
arize/_generated/api_client/test/test_categorical_annotation_config_create.py
146161
arize/_generated/api_client/test/test_categorical_annotation_value.py

src/arize/__init__.py

Lines changed: 38 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,17 @@
11
"""Arize SDK for model observability and LLM tracing."""
22

33
import logging
4-
from collections.abc import Mapping
5-
from typing import Literal, cast
4+
from collections.abc import Callable, Mapping
5+
from typing import TYPE_CHECKING, Literal, cast
66

77
from arize._generated.api_client import models
88
from arize.client import ArizeClient
99
from arize.config import SDKConfiguration
1010
from arize.regions import Region
1111

12+
if TYPE_CHECKING:
13+
import pandas as pd
14+
1215
# Attach a NullHandler by default in the top-level package
1316
# so that if no configuration is installed, nothing explodes.
1417
logging.getLogger("arize").addHandler(logging.NullHandler())
@@ -30,7 +33,7 @@
3033
]
3134

3235

33-
def make_to_df(field_name: str) -> object:
36+
def make_to_df(field_name: str) -> Callable[..., "pd.DataFrame"]:
3437
def to_df(
3538
self: object,
3639
by_alias: bool = False,
@@ -39,7 +42,7 @@ def to_df(
3942
convert_dtypes: bool = True,
4043
expand_field: str = "additional_properties",
4144
expand_prefix: str = "",
42-
) -> object:
45+
) -> "pd.DataFrame":
4346
"""Convert a list of objects to a :class:`pandas.DataFrame`.
4447
4548
Behavior:
@@ -113,11 +116,42 @@ def to_df(
113116
return to_df
114117

115118

119+
def annotation_configs_to_df(
120+
self: object,
121+
by_alias: bool = False,
122+
exclude_none: str | bool = True,
123+
json_normalize: bool = False,
124+
convert_dtypes: bool = True,
125+
expand_field: str = "actual_instance",
126+
expand_prefix: str = "",
127+
) -> "pd.DataFrame":
128+
"""Convert annotation config list response to DataFrame.
129+
130+
Defaults to expanding `actual_instance` so oneOf wrapper internals are
131+
flattened into user-facing columns.
132+
"""
133+
to_df = make_to_df("annotation_configs")
134+
df = to_df(
135+
self,
136+
by_alias=by_alias,
137+
exclude_none=exclude_none,
138+
json_normalize=json_normalize,
139+
convert_dtypes=convert_dtypes,
140+
expand_field=expand_field,
141+
expand_prefix=expand_prefix,
142+
)
143+
return df.drop(
144+
columns=["one_of_schemas", "discriminator_value_class_map"],
145+
errors="ignore",
146+
)
147+
148+
116149
# Monkey-patch convenience methods onto generated response models
117150
# Type ignore comments needed: mypy can't verify runtime attribute additions
118151
models.DatasetsExamplesList200Response.to_df = make_to_df("examples") # type: ignore[attr-defined]
119152
models.DatasetsList200Response.to_df = make_to_df("datasets") # type: ignore[attr-defined]
120153
models.ExperimentsList200Response.to_df = make_to_df("experiments") # type: ignore[attr-defined]
121154
models.ExperimentsRunsList200Response.to_df = make_to_df("experiment_runs") # type: ignore[attr-defined]
122155
models.ProjectsList200Response.to_df = make_to_df("projects") # type: ignore[attr-defined]
156+
models.AnnotationConfigsList200Response.to_df = annotation_configs_to_df # type: ignore[attr-defined]
123157
models.SpansList200Response.to_df = make_to_df("spans") # type: ignore[attr-defined]

src/arize/_generated/api_client/__init__.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,10 +19,12 @@
1919
# Define package exports
2020
__all__ = [
2121
"AnnotationConfigsApi",
22+
"AnnotationQueuesApi",
2223
"DatasetsApi",
2324
"ExperimentsApi",
2425
"ProjectsApi",
2526
"PromptsApi",
27+
"SpacesApi",
2628
"SpansApi",
2729
"ApiResponse",
2830
"ApiClient",
@@ -37,6 +39,8 @@
3739
"AnnotationConfigBase",
3840
"AnnotationConfigCreateBase",
3941
"AnnotationConfigsList200Response",
42+
"AnnotationQueue",
43+
"AnnotationQueuesList200Response",
4044
"CategoricalAnnotationConfig",
4145
"CategoricalAnnotationConfigCreate",
4246
"CategoricalAnnotationValue",
@@ -82,6 +86,8 @@
8286
"ResponseFormat",
8387
"ResponseFormatJsonSchema",
8488
"ResponseFormatType",
89+
"Space",
90+
"SpacesList200Response",
8591
"Span",
8692
"SpanContext",
8793
"SpanEvent",
@@ -95,10 +101,12 @@
95101

96102
# import apis into sdk package
97103
from arize._generated.api_client.api.annotation_configs_api import AnnotationConfigsApi as AnnotationConfigsApi
104+
from arize._generated.api_client.api.annotation_queues_api import AnnotationQueuesApi as AnnotationQueuesApi
98105
from arize._generated.api_client.api.datasets_api import DatasetsApi as DatasetsApi
99106
from arize._generated.api_client.api.experiments_api import ExperimentsApi as ExperimentsApi
100107
from arize._generated.api_client.api.projects_api import ProjectsApi as ProjectsApi
101108
from arize._generated.api_client.api.prompts_api import PromptsApi as PromptsApi
109+
from arize._generated.api_client.api.spaces_api import SpacesApi as SpacesApi
102110
from arize._generated.api_client.api.spans_api import SpansApi as SpansApi
103111

104112
# import ApiClient
@@ -117,6 +125,8 @@
117125
from arize._generated.api_client.models.annotation_config_base import AnnotationConfigBase as AnnotationConfigBase
118126
from arize._generated.api_client.models.annotation_config_create_base import AnnotationConfigCreateBase as AnnotationConfigCreateBase
119127
from arize._generated.api_client.models.annotation_configs_list200_response import AnnotationConfigsList200Response as AnnotationConfigsList200Response
128+
from arize._generated.api_client.models.annotation_queue import AnnotationQueue as AnnotationQueue
129+
from arize._generated.api_client.models.annotation_queues_list200_response import AnnotationQueuesList200Response as AnnotationQueuesList200Response
120130
from arize._generated.api_client.models.categorical_annotation_config import CategoricalAnnotationConfig as CategoricalAnnotationConfig
121131
from arize._generated.api_client.models.categorical_annotation_config_create import CategoricalAnnotationConfigCreate as CategoricalAnnotationConfigCreate
122132
from arize._generated.api_client.models.categorical_annotation_value import CategoricalAnnotationValue as CategoricalAnnotationValue
@@ -162,6 +172,8 @@
162172
from arize._generated.api_client.models.response_format import ResponseFormat as ResponseFormat
163173
from arize._generated.api_client.models.response_format_json_schema import ResponseFormatJsonSchema as ResponseFormatJsonSchema
164174
from arize._generated.api_client.models.response_format_type import ResponseFormatType as ResponseFormatType
175+
from arize._generated.api_client.models.space import Space as Space
176+
from arize._generated.api_client.models.spaces_list200_response import SpacesList200Response as SpacesList200Response
165177
from arize._generated.api_client.models.span import Span as Span
166178
from arize._generated.api_client.models.span_context import SpanContext as SpanContext
167179
from arize._generated.api_client.models.span_event import SpanEvent as SpanEvent

src/arize/_generated/api_client/api/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,9 +2,11 @@
22

33
# import apis into api package
44
from arize._generated.api_client.api.annotation_configs_api import AnnotationConfigsApi
5+
from arize._generated.api_client.api.annotation_queues_api import AnnotationQueuesApi
56
from arize._generated.api_client.api.datasets_api import DatasetsApi
67
from arize._generated.api_client.api.experiments_api import ExperimentsApi
78
from arize._generated.api_client.api.projects_api import ProjectsApi
89
from arize._generated.api_client.api.prompts_api import PromptsApi
10+
from arize._generated.api_client.api.spaces_api import SpacesApi
911
from arize._generated.api_client.api.spans_api import SpansApi
1012

0 commit comments

Comments
 (0)