arize 8.0.0a22__py3-none-any.whl → 8.0.0b0__py3-none-any.whl

arize/utils/proto.py

@@ -1,3 +1,5 @@
+"""Protocol buffer schema utilities for tracing data."""
+
 # type: ignore[pb2]
 from arize._generated.protocol.rec import public_pb2 as pb2
 
@@ -5,6 +7,14 @@ from arize._generated.protocol.rec import public_pb2 as pb2
 def get_pb_schema_tracing(
     project_name: str,
 ) -> pb2.Schema:
+    """Create a protobuf schema for LLM tracing data.
+
+    Args:
+        project_name: The name of the project/model.
+
+    Returns:
+        A configured pb2.Schema object for tracing environment.
+    """
     s = pb2.Schema()
     s.constants.model_id = project_name
     s.constants.environment = pb2.Schema.Environment.TRACING

arize/utils/size.py

@@ -1,13 +1,15 @@
+"""Size calculation utilities for payloads and data structures."""
+
 import sys
-from typing import Any, Dict, List
+from typing import Any
 
 import pandas as pd
 
 
-def get_payload_size_mb(payload: List[Dict[str, Any]] | pd.DataFrame) -> float:
+def get_payload_size_mb(payload: list[dict[str, Any]] | pd.DataFrame) -> float:
     """Return approximate size of payload in MB."""
     if isinstance(payload, pd.DataFrame):
-        # memory_usage(deep=True) sums all columns’ memory footprint
+        # memory_usage(deep=True) sums all columns' memory footprint
         size_bytes = payload.memory_usage(deep=True).sum()
     elif isinstance(payload, list):
         # sys.getsizeof() gives shallow size; sum all element sizes for rough total

arize/utils/types.py

@@ -0,0 +1,105 @@
+"""Common type definitions and data models used across the Arize SDK."""
+
+import json
+from collections.abc import Iterable, Sequence
+from typing import (
+    TypeVar,
+)
+
+import numpy as np
+
+
+def is_json_str(s: str) -> bool:
+    """Check if a string is valid JSON.
+
+    Args:
+        s: The string to validate.
+
+    Returns:
+        True if the string is valid JSON, False otherwise.
+    """
+    try:
+        json.loads(s)
+    except ValueError:
+        return False
+    except TypeError:
+        return False
+    return True
+
+
+T = TypeVar("T", bound=type)
+
+
+def is_array_of(arr: Sequence[object], tp: T) -> bool:
+    """Check if a value is a numpy array with all elements of a specific type.
+
+    Args:
+        arr: The sequence to check.
+        tp: The expected type for all elements.
+
+    Returns:
+        True if arr is a numpy array and all elements are of type tp.
+    """
+    return isinstance(arr, np.ndarray) and all(isinstance(x, tp) for x in arr)
+
+
+def is_list_of(lst: Sequence[object], tp: T) -> bool:
+    """Check if a value is a list with all elements of a specific type.
+
+    Args:
+        lst: The sequence to check.
+        tp: The expected type for all elements.
+
+    Returns:
+        True if lst is a list and all elements are of type tp.
+    """
+    return isinstance(lst, list) and all(isinstance(x, tp) for x in lst)
+
+
+def is_iterable_of(lst: Sequence[object], tp: T) -> bool:
+    """Check if a value is an iterable with all elements of a specific type.
+
+    Args:
+        lst: The sequence to check.
+        tp: The expected type for all elements.
+
+    Returns:
+        True if lst is an iterable and all elements are of type tp.
+    """
+    return isinstance(lst, Iterable) and all(isinstance(x, tp) for x in lst)
+
+
+def is_dict_of(
+    d: dict[object, object],
+    key_allowed_types: T,
+    value_allowed_types: T = (),
+    value_list_allowed_types: T = (),
+) -> bool:
+    """Method to check types are valid for dictionary.
+
+    Args:
+        d: Dictionary itself.
+        key_allowed_types: All allowed types for keys of dictionary.
+        value_allowed_types: All allowed types for values of dictionary.
+        value_list_allowed_types: If value is a list, these are the allowed
+            types for value list.
+
+    Returns:
+        True if the data types of dictionary match the types specified by the
+            arguments, false otherwise.
+    """
+    if value_list_allowed_types and not isinstance(
+        value_list_allowed_types, tuple
+    ):
+        value_list_allowed_types = (value_list_allowed_types,)
+
+    return (
+        isinstance(d, dict)
+        and all(isinstance(k, key_allowed_types) for k in d)
+        and all(
+            isinstance(v, value_allowed_types)
+            or any(is_list_of(v, t) for t in value_list_allowed_types)
+            for v in d.values()
+            if value_allowed_types or value_list_allowed_types
+        )
+    )

arize/version.py

@@ -1 +1,3 @@
-__version__ = "8.0.0a22"
+"""Version information for the Arize SDK."""
+
+__version__ = "8.0.0b0"

{arize-8.0.0a22.dist-info → arize-8.0.0b0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arize
-Version: 8.0.0a22
+Version: 8.0.0b0
 Summary: A helper library to interact with Arize AI APIs
 Project-URL: Homepage, https://arize.com
 Project-URL: Documentation, https://docs.arize.com/arize
@@ -9,8 +9,9 @@ Project-URL: Source, https://github.com/Arize-ai/client_python
 Project-URL: Changelog, https://github.com/Arize-ai/client_python/blob/main/CHANGELOG.md
 Author-email: Arize AI <support@arize.com>
 Maintainer-email: Arize AI <support@arize.com>
-License: BSD
-License-File: LICENSE.md
+License: Apache-2.0
+License-File: LICENSE
+License-File: NOTICE
 Keywords: Arize,Evaluations,Explainability,LLM,Monitoring,Observability,Tracing
 Classifier: Development Status :: 3 - Alpha
 Classifier: Intended Audience :: Developers
@@ -36,6 +37,12 @@ Requires-Dist: torch<3,>=1.13; extra == 'auto-embeddings'
 Requires-Dist: transformers<5,>=4.25; extra == 'auto-embeddings'
 Provides-Extra: datasets-experiments
 Requires-Dist: numpy>=2.0.0; extra == 'datasets-experiments'
+Requires-Dist: openinference-semantic-conventions<1,>=0.1.21; extra == 'datasets-experiments'
+Requires-Dist: opentelemetry-api>=1.38.0; extra == 'datasets-experiments'
+Requires-Dist: opentelemetry-exporter-otlp-proto-common>=1.38.0; extra == 'datasets-experiments'
+Requires-Dist: opentelemetry-exporter-otlp-proto-grpc>=1.38.0; extra == 'datasets-experiments'
+Requires-Dist: opentelemetry-proto>=1.38.0; extra == 'datasets-experiments'
+Requires-Dist: opentelemetry-sdk>=1.38.0; extra == 'datasets-experiments'
 Requires-Dist: pydantic; extra == 'datasets-experiments'
 Requires-Dist: wrapt<2.0.0,>=1.0.0; extra == 'datasets-experiments'
 Provides-Extra: dev
@@ -260,7 +267,7 @@ Use `arize.models` to interact with ML models: log ML data (traininv, validation
 
 ```python
 from arize import ArizeClient
-from arize.types import ModelTypes, Environments
+from arize.ml.types import ModelTypes, Environments
 
 client = ArizeClient(api_key=API_KEY)
 SPACE_ID = "<your-space-id>"
@@ -285,14 +292,14 @@ response = client.models.log_stream(
 
 ```python
 from arize import ArizeClient
-from arize.types import ModelTypes, Environments
+from arize.ml.types import ModelTypes, Environments
 
 client = ArizeClient(api_key=API_KEY)
 SPACE_ID = "<your-space-id>"
 MODEL_NAME = "<your-model-name>"
 MODEL_VERSION = "1.0"
 
-from arize.types import Schema, EmbeddingColumnNames, ObjectDetectionColumnNames, ModelTypes, Environments
+from arize.ml.types import Schema, EmbeddingColumnNames, ObjectDetectionColumnNames, ModelTypes, Environments
 
 tags = ["drift_type"]
 embedding_feature_column_names = {

arize-8.0.0b0.dist-info/RECORD

@@ -0,0 +1,175 @@
+arize/__init__.py,sha256=C2NOiNcFL9TMu5NTkJwF88kL6PKAHHjBMG8HTAIQ5TU,3551
+arize/_lazy.py,sha256=aLdgSGxdfdV9QiLllxlLlkN5k44pFf_7jdKRV9Z9sgo,2864
+arize/client.py,sha256=RwnDTmxHzPfpzonYD8u_d4V-RFyehpbUADZV8Xpki0c,13660
+arize/config.py,sha256=80iQgiaAhSqgE1Q-nKlivjQ2z3F7PifiPAsZlhQjozo,16471
+arize/logging.py,sha256=CZkgdpDKXoNNJ9fQMNwbCQEe-ipUg7fQf_CE1hRN1Qc,8574
+arize/pre_releases.py,sha256=QWZb5Kko-fHDKDKNk8HUwehMQuLrZPYnyD2c6_mnEqI,1216
+arize/regions.py,sha256=oEKJdZbW5r8X3fL3PaDVMdk-f2MoZ8OBvUvYZLMGBpw,1039
+arize/version.py,sha256=OIc4vIJor74VQDoVG5zuAmxWI0h4EEU19vIPa3y2Tjc,70
+arize/_exporter/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+arize/_exporter/client.py,sha256=DRxdREi4k0RiRI-kPvYCwX2zo6DGkdt1XMezF4ZhR08,16425
+arize/_exporter/validation.py,sha256=9gX4oHrZcChMfkwoBpe3Qp4aK-txy_zZ4azrKf-bEQw,1022
+arize/_exporter/parsers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+arize/_exporter/parsers/tracing_data_parser.py,sha256=mgpIyKhUeUeW-HMR11Kmo2HSLrUXZJLuqk27aDnytwo,5716
+arize/_flight/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+arize/_flight/client.py,sha256=HpUu22Q2r-jnh_j0RaQtJLZnaXKGBNNjZlZ0dvX98xs,20391
+arize/_flight/types.py,sha256=GB_4dQu2ElIrcDGAcqhG7oI4g-b0ZdSlbrQkf0TFzVE,194
+arize/_generated/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+arize/_generated/api_client_README.md,sha256=SfBbE1r1thnfwk3AVr6CZUP6kJ7HNy28gx1V7DjQhp8,7451
+arize/_generated/api_client/__init__.py,sha256=I_KGjOG-2Z5dokY6Ln9E5fFHUIe53WUvCaUvg4DtQss,4672
+arize/_generated/api_client/api_client.py,sha256=Cr9leKLpAJYQmL_U1hAPsMNFdaSO6Ti1X52jvGp2WsQ,27899
+arize/_generated/api_client/api_response.py,sha256=eMxw1mpmJcoGZ3gs9z6jM4oYoZ10Gjk333s9sKxGv7s,652
+arize/_generated/api_client/configuration.py,sha256=EJ6DkTKzHWqjN0BwMEHQ1D9pEDdClOsM72EcffquZ9I,19406
+arize/_generated/api_client/exceptions.py,sha256=EAN7PugiYwsF3Puj8S1iWGqptP6kheYykxvZLPEk4rw,6502
+arize/_generated/api_client/rest.py,sha256=OXfGpi__G1f4q3Edgbj4njQNBstouwHUIwrPLJBsVVE,9512
+arize/_generated/api_client/api/__init__.py,sha256=ycVTd-FBlPyFX82wbCQiuCTbj78pZbVv17kX9U3s5j0,261
+arize/_generated/api_client/api/datasets_api.py,sha256=pwS_NU7puzAdoqp49D24XjgcReptPue6K2ZI_C8DwYY,102867
+arize/_generated/api_client/api/experiments_api.py,sha256=xMMH-QA80mmXXPbGKKPIONMnxNowSGlf8wVdLKqzPss,65311
+arize/_generated/api_client/api/projects_api.py,sha256=SRMimrrXNZLlKD9zbTFs-iy2SeOHiVDdlMB0Hbp0ApE,48820
+arize/_generated/api_client/models/__init__.py,sha256=qajnlyF3Q2lCuIIeCqldVFcdgogFPd2iLr5YMYv8Alw,2261
+arize/_generated/api_client/models/dataset.py,sha256=eStUwLDPU7j3kMTdSO7RqMCScsGU9rOZtU41xnSXUUs,4332
+arize/_generated/api_client/models/dataset_example.py,sha256=avmtSuitMruY7Yphw5gDpPbfkx_GKDhePBSoG6S_sc0,3825
+arize/_generated/api_client/models/dataset_example_update.py,sha256=DMV1_IkN74tQJrnWFPsJ-7Yj9d-tBaF2rVZZlktprpo,3290
+arize/_generated/api_client/models/dataset_version.py,sha256=OlkhTlcQuNvxJdNxxU5CpHZ9a2OCro5FRPAUMCYLBdE,3709
+arize/_generated/api_client/models/datasets_create_request.py,sha256=av3O6kxoeLL1JACsDL_SrHnNKTNVMHlUyrcBhrsiOac,3673
+arize/_generated/api_client/models/datasets_examples_insert_request.py,sha256=u_9QqAgcEUVUyZoZJCWnd9Wuz00-vyRrDKL-82hXO4U,3456
+arize/_generated/api_client/models/datasets_examples_list200_response.py,sha256=fIY3yg5cbIcRBdawxBRnWqDt5dxOLN1Pxju_g6BDU30,3869
+arize/_generated/api_client/models/datasets_examples_update_request.py,sha256=awa70zUxyMPLiEddQmcigVCR4tPhqTjVL6ClV-CeqK4,3797
+arize/_generated/api_client/models/datasets_list200_response.py,sha256=KsDfQU44q7gg7TwR5MDmJfkp69-SNTo_c87d9uz_8fk,3777
+arize/_generated/api_client/models/experiment.py,sha256=hmolU5fJAJbZ8HQaApPgmODAiqTJ-mudkc525_uwhrI,4416
+arize/_generated/api_client/models/experiment_run.py,sha256=YB2I6rjBCVYdg42iuqLDIV4APssawXGkPJsBdF_Mv9U,3692
+arize/_generated/api_client/models/experiment_run_create.py,sha256=nkMc9ojyFG77wf2jaY0iipiDIg4FmgsNr_inp_XK8rw,3444
+arize/_generated/api_client/models/experiments_create_request.py,sha256=L0DllTYtkIuS7sEX7tnxULTkvrhCBYwIDTICvSkwcvs,3773
+arize/_generated/api_client/models/experiments_list200_response.py,sha256=AKot7N4isrGFHNr6-mXsSr8ikeo5veneFFsuGYRKSRc,3843
+arize/_generated/api_client/models/experiments_runs_list200_response.py,sha256=vkaYsUKdAnChSBKCEzxc1DrE5vs_0tG1Ya-yXhjuPBo,3979
+arize/_generated/api_client/models/pagination_metadata.py,sha256=2NgzlV140yhFqS8IURPfMrVtW2p-0Xr3DaA9GxFfInw,3262
+arize/_generated/api_client/models/primitive_value.py,sha256=iDHyARo1ILnJymsS9EaVFE5V37X0BvUB6uTX_aVhKbg,6400
+arize/_generated/api_client/models/problem.py,sha256=mbAAHZAC8mKYC0r902SGE4QcLCUMfSzFJOLvjyjO8Pw,3584
+arize/_generated/api_client/models/project.py,sha256=w4OT222655N-dd8IHauiDobgUyMnol4xzZ80Rniv3ts,3636
+arize/_generated/api_client/models/projects_create_request.py,sha256=hnnerivgBpJ4BNcdE1hfZ8EoIawtmxsx1iv8NVoHkw4,3024
+arize/_generated/api_client/models/projects_list200_response.py,sha256=ibz4CJNTDX5Y6eiEY2_j2R-L_rTWsTTzaRVCN3iWRDI,3777
+arize/_generated/api_client/test/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+arize/_generated/api_client/test/test_dataset.py,sha256=ZVNd_Nt8qkcelrpCuX3mYx8JP5q23EBbudWR8m3OR5U,2391
+arize/_generated/api_client/test/test_dataset_example.py,sha256=musVzDrorF1yf5ozlAnntPyMad9QQPgxoldJjxb7lAc,1876
+arize/_generated/api_client/test/test_dataset_example_update.py,sha256=Z59KydV8_su5QlPdFx7Vs4WAOh-NEYvNVMsIe_5bz0k,1521
+arize/_generated/api_client/test/test_dataset_version.py,sha256=rPczEI--Htxd5tRYzCgNMYEisu7ub87EOxRE-ZIEbp0,1996
+arize/_generated/api_client/test/test_datasets_api.py,sha256=2WYqfqygypzUEB61iMzqWZRAZIyYhrZQ3LaNZDAOIhc,1707
+arize/_generated/api_client/test/test_datasets_create_request.py,sha256=wjDWgl-NUlaQZTM0eDpVN15fvJ0tH9c3RegBwOKAYCk,1873
+arize/_generated/api_client/test/test_datasets_examples_insert_request.py,sha256=N3mAB0BjNAXGdEyN6l3BmRiUeXG4WUoRxgjtvjmWX7w,1854
+arize/_generated/api_client/test/test_datasets_examples_list200_response.py,sha256=gfm4cmvhpOr0bXLgoRN1G2I9wc155F4IPDjvVySLWcw,2244
+arize/_generated/api_client/test/test_datasets_examples_update_request.py,sha256=a1m-IaI0RGUd3dd6ktQG0nADEORX_r0KLbY-9IqbwBw,1888
+arize/_generated/api_client/test/test_datasets_list200_response.py,sha256=jRWD_HBWE3doypQuaexSp39F_fFdp-aJVXsosxM36M8,3905
+arize/_generated/api_client/test/test_experiment.py,sha256=sOwd8Rep47jAf7vhq1hm3rK0P7I0VpGS9cZJLOb1RSE,2080
+arize/_generated/api_client/test/test_experiment_run.py,sha256=FPmaQpEPqs6_jR-o7S-Ejaj2OKeJjO7pLOr-r51fetk,1560
+arize/_generated/api_client/test/test_experiment_run_create.py,sha256=oZorTWEiI6fTmSFw35vZe-1Npw92Ne01nTIbRmfPnAc,1583
+arize/_generated/api_client/test/test_experiments_api.py,sha256=Ez6hw0XNW9y2IDLR6_7YlgjbTXgfZr9JxBoaVxEU8N8,1414
+arize/_generated/api_client/test/test_experiments_create_request.py,sha256=7bNXzkDE3kL5VZCi1ogey9pWmWWqGi_rDG60_93bQoA,1927
+arize/_generated/api_client/test/test_experiments_list200_response.py,sha256=NlLuUOjLRXlUqFsgqSI04B2wCl_4OEZUMtGJrVGUwWA,3085
+arize/_generated/api_client/test/test_experiments_runs_list200_response.py,sha256=BtOi_RMEYJciMCazYfmCkR3BbXi8VDdYm1qs_8razkE,2246
+arize/_generated/api_client/test/test_pagination_metadata.py,sha256=6I0fz9C-PeS0xGlqfnpCb95jieoCp6wzC_77pFefiI4,1546
+arize/_generated/api_client/test/test_primitive_value.py,sha256=e3Hj8RATS82D1QpAtxysqRrgccFaYXHJHsX0hkLNC90,1399
+arize/_generated/api_client/test/test_problem.py,sha256=cP3YV_YYLjkZuh7RCKUoqJy1FuxukuRj1swYKzFmz9g,1514
+arize/_generated/api_client/test/test_project.py,sha256=89LB5u8gUqrNBGWt9UfC47A9pBFpLZlJ1DqwjUGKsfU,1693
+arize/_generated/api_client/test/test_projects_api.py,sha256=oFZh9KZlPpc-Jaf5IQ1NhLOHJf0sJBHAJIrvPACYmXI,1207
+arize/_generated/api_client/test/test_projects_create_request.py,sha256=fhmr9HTlcn9khjeiLGm5zMfCtPvXIiRu1Wb3bbcWOj4,1599
+arize/_generated/api_client/test/test_projects_list200_response.py,sha256=l0yfAWNkYNRQn7d3ziANxtz-OlqGGWVlyqaxlyEBJeE,2575
+arize/_generated/protocol/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+arize/_generated/protocol/flight/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+arize/_generated/protocol/flight/flight_pb2.py,sha256=buKnlVPQSFpynK8AYJTfHRjfIFsfwf7khjOhyTjv4gk,27790
+arize/_generated/protocol/rec/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+arize/_generated/protocol/rec/public_pb2.py,sha256=vgP-yTSZLeomVwfIzcOo6t3i1mPCCNJGgd41ZkfLNng,79898
+arize/constants/__init__.py,sha256=QYJ_twp380ASZP1eJzEHru-utxKwxL6wTyiRM575RXM,68
+arize/constants/config.py,sha256=GzyD4z5pw6ZXc1VbUcEPAsxKGmPsFMYbbJFVltPtAP8,1731
+arize/constants/ml.py,sha256=LzkvxpFk7VdEZxOxjJq19OkDwo1wywCgRAVpsl5ZcAk,2457
+arize/constants/model_mapping.json,sha256=OPE54rBATzmwRhx0tycsxnGae1jBhtqEmQqQvzleTSc,5725
+arize/constants/openinference.py,sha256=mnjtbelFbJYJDZ65HQ3lZ6Xr0YyYudBIyTw5cLKA_CI,610
+arize/constants/pyarrow.py,sha256=4OXNqSYxWxkbmjJTwGxHst8Ch0cmCXsBjUuy1T7rSf4,78
+arize/constants/spans.py,sha256=fQpzvQBDGYDHg_48wymVwGSvAnMnUUopHBLwNAskP3Q,2597
+arize/datasets/__init__.py,sha256=t-nswFsscEAy7ixz_g2gEdLN3TCNVdeD9JiOcPdBvwI,69
+arize/datasets/client.py,sha256=Lt0UU2QSMzTk7dREoaOsCv8MMztx20Ynw_w6Kq4WRzQ,18175
+arize/datasets/errors.py,sha256=Vx54dsl1TEXNch8dht49NXr1dZFDp6AGX-Yhw-Qxh9w,3078
+arize/datasets/validation.py,sha256=vnfSs3PrhtVD4nAeNL-Q2yk_6lHXGVfGekGPo9wHcVI,1684
+arize/embeddings/__init__.py,sha256=Dsh7Ncap6Q47M5_5n0vWzwKkPJtVhZBGke3ygXx02KE,226
+arize/embeddings/auto_generator.py,sha256=iFbU1pentuQCzqmb0PMEuTS7p2PyXzbE38OpmahfKH8,4518
+arize/embeddings/base_generators.py,sha256=MdQOMcSHhNUs-VygIPpOAcmsruaQzFfBe_dzzrHCXXQ,11452
+arize/embeddings/constants.py,sha256=oS7rM8Rqn03B76EKpg6uNtfP5dsx0ptCA34SI0X2RAA,1177
+arize/embeddings/cv_generators.py,sha256=U_MA7YVh4s1bDqCOHjIJkYE-y3SDRJ68Nw_HusRDKX8,1639
+arize/embeddings/errors.py,sha256=CcXjRUsQUq_1nQUVv9Qj8fLgQdAo9bzFq3ZEiEWFf9s,2366
+arize/embeddings/nlp_generators.py,sha256=_DE4qaTGPnHMPqlGn0HuBxCw-x5dzMFvBLKMJfZYCKU,4573
+arize/embeddings/tabular_generators.py,sha256=QPfVKfbvDghYEey7cfaC0D9MBkFgD2-7ODDlmflCXFc,6539
+arize/embeddings/usecases.py,sha256=BAcEHTVF1abfe_Izf4eqCj5l-jPgLGs2phTfYVZ1nB0,850
+arize/exceptions/__init__.py,sha256=-16kxctvh4CU_PAekPti-UI7ZTVcCkUrttwoXBa-Eq4,59
+arize/exceptions/auth.py,sha256=ZCkJ15dW_3h7Fm4AawQaGJejrOEWn1wAXVoxBGA3XhA,749
+arize/exceptions/base.py,sha256=0D5fqNul3SBPh2tvVAwocZwiBNspeTh7umI1CTqISw8,4400
+arize/exceptions/models.py,sha256=Pe9QzdPl7hAWkJwHVm0QXBzmG2WFi2Xa_GNQOqGAJeM,1259
+arize/exceptions/parameters.py,sha256=xbF38TNZI14PruYMor6DYVMB7uVhyusJfRPUn_YPHzU,15175
+arize/exceptions/spaces.py,sha256=RZdXCnxkc-t5_jY2je7gVUDxaXhBQvKpXcUGXptO8mA,631
+arize/exceptions/types.py,sha256=ks-4B2YpiVxHVopWUh4Gv_iM2h-LhdJ_zZDXtN3Ctc0,9468
+arize/exceptions/values.py,sha256=rzT8sZbY2m2Vxc5vt5V-ZjsQyiGbxM0pxPi-bp9d-pU,27920
+arize/experiments/__init__.py,sha256=8ESMbJbtGc3VGDo4efQxxCeXmuZlT-0tPv-EBS48Xpo,358
+arize/experiments/client.py,sha256=ScUgT44GkNJ6SdwctHeLPNefev3hLCjW55PVkVf_-xI,26233
+arize/experiments/functions.py,sha256=b51Dv1qk4oiDiJv6VZgpBhfdKlo4ff8FLTtH5ruEPOA,34675
+arize/experiments/tracing.py,sha256=P3Q2DTUGNsRhvxRNAx_NXY30Id8y7cw9MV9piduSbB0,10207
+arize/experiments/types.py,sha256=qwv-LV9DqfdlKwXc4SdL9p2QsI8pHoEy2_HCldzsdZs,14093
+arize/experiments/evaluators/__init__.py,sha256=FUTfWRir8jaY8PC3pRHUy5m4RqrKRBPD1EJ4GYkIMLY,76
+arize/experiments/evaluators/base.py,sha256=m9Gk2efWyssSYnZIdSJouB6CNx1KkqHjHUrRWft1L54,10981
+arize/experiments/evaluators/exceptions.py,sha256=iqg5-OcGfRWY354OCcdmVC39cb3jEb-RO1LnVn7BD7M,376
+arize/experiments/evaluators/executors.py,sha256=G6StVzZDd5MbXWENiCgFd1eDgNgMs6IWRc-0OtRyKvk,20722
+arize/experiments/evaluators/rate_limiters.py,sha256=9PextZkLHljvY9t1yR4Lx_ryFW5Eht00irB87dxnWeA,13567
+arize/experiments/evaluators/types.py,sha256=q_p5gaGiKvV14OgfDAI_Kn44Ribou3UfxizAJ9Bl5sM,4509
+arize/experiments/evaluators/utils.py,sha256=nzl7DtucZoIQTwDpcLJUSlTbCsnyLNLFmwvovWB6GOg,7884
+arize/ml/__init__.py,sha256=jWGnNLghA0TkmR033NewriFUleH46zCTC4MCoGADE5c,68
+arize/ml/bounded_executor.py,sha256=BZF19icwV8UaRm8Otw8leiAgjDYultDDmvuSPXw2sx8,1669
+arize/ml/casting.py,sha256=lNEDTremFv5R5IR2F8mSAD4ufDgawQOjFIWo5AyAGi0,14016
+arize/ml/client.py,sha256=SVLVjB4C3bk__LNlFQyfb4nb5v_7jiZudWR-kweI8lo,58403
+arize/ml/proto.py,sha256=_zzE2ESKU2NmRHjCU0lTUR55m9LKSi3s0DKTGhJFYcI,19077
+arize/ml/stream_validation.py,sha256=5P-OFUiXCvdEoHTTbr1VsI_PRmeMaWtrLpG_Fagf-CE,8287
+arize/ml/types.py,sha256=oSGmfIKxRZxX5FXsN0-sWRPHVjZUjkh58GV_zGwdeE8,71033
+arize/ml/batch_validation/__init__.py,sha256=mvn_TLQvgL6lwLvaTedzH0GGABrJViDkh77M38aFXQ4,49
+arize/ml/batch_validation/errors.py,sha256=FXiQKRrWh-a_9siEOO6dc9xEn_onS7jtDvNT-eFWnp4,61681
+arize/ml/batch_validation/validator.py,sha256=Hz9ctp_iUHoS9kViPq5qBz5uTHIaHm4w4WILQngen1g,147757
+arize/ml/surrogate_explainer/__init__.py,sha256=jxFXugcjhip4Lg5M1ikrnq5xODwS1Z5Tt30NeVkCpH0,70
+arize/ml/surrogate_explainer/mimic.py,sha256=Zx5idIHUyln0AxsU4K0NJoQ1HFOtKk0GoA6bZ4ctTLg,6214
+arize/projects/__init__.py,sha256=hx29_muM9ZJxH5j518QJ88-_VVRAFymF5WLRvQVpDH8,64
+arize/projects/client.py,sha256=hjyqg8yfINEIyorPIEwndurnaIfQimjxHQG6AQ_lt-U,4341
+arize/spans/__init__.py,sha256=rXRx4DshOoP9WmbVRUyYdM_QHtUmUTbfO2pZPkztKL0,57
+arize/spans/client.py,sha256=45wkvUmdxtYQKNp75RQYSkuM_Hwx_6Uf-V8osF0h8S8,48785
+arize/spans/columns.py,sha256=Sap8UrOeeFebxsI1LbelustuirAPmKrtjW8z3vHJVFk,13378
+arize/spans/conversion.py,sha256=9GrNd9puZtJJZma2C6UMmNEcTfkb4maG1BxkRiE7zZU,4937
+arize/spans/validation/__init__.py,sha256=QSb0EFPBCI2riIrgtSaFXPDGDPQ_aYK4N1Dej499cu8,55
+arize/spans/validation/annotations/__init__.py,sha256=AW7uOEA-uBbM39cdy5SEyi6dC_oYUP_zzW72qkYoFlI,51
+arize/spans/validation/annotations/annotations_validation.py,sha256=WDChwwqiwlI-lPT7jGA7THllL3a93foZWnyjIkrUjUg,2763
+arize/spans/validation/annotations/dataframe_form_validation.py,sha256=V050mg0IW_rc6Iimv9E3ZEdfi7A14ZuacE3mfuDR45A,7079
+arize/spans/validation/annotations/value_validation.py,sha256=mZq9iQIfUl7YE-ctwYrt8vQvgDeKsgMhb-uPF1_PFZA,8650
+arize/spans/validation/common/__init__.py,sha256=oJ2KdLO6gpl0AUuA8zCuyU5kB_TFW0dPyVpBQtFk_EU,66
+arize/spans/validation/common/argument_validation.py,sha256=Xa0hzsC0YIkE7dF62_Bij4Km8QR41zFJ6fUTUlOPnag,2316
+arize/spans/validation/common/dataframe_form_validation.py,sha256=GmN0Tx_4Btsf1ChenTI0fEHT8-IhnzOEEQ4ek3T4Bmk,2023
+arize/spans/validation/common/errors.py,sha256=pWkQlNuQiUcSNy1scQhbVpL7OiXo_bjYH7Su_R-9AP0,22388
+arize/spans/validation/common/value_validation.py,sha256=593mcfW73y9vvMZvFyhfqhs4TXHoKDxj_Aksu3IQzGg,9314
+arize/spans/validation/evals/__init__.py,sha256=RMMOzvDKk1c-X1OmFeMoezDom4LnhVELDfU5EmojqIY,56
+arize/spans/validation/evals/dataframe_form_validation.py,sha256=LBqIRUBDAGeDU313bb89Ng8J75U7eBQ3muVtInO2TqY,3957
+arize/spans/validation/evals/evals_validation.py,sha256=-FA2uyrpe9gKeZX9_AQayOMq9Am5ib8mZjDXwgf4MFI,3285
+arize/spans/validation/evals/value_validation.py,sha256=Gn0V_xhY0uk-qWtQLLV-1N4mi-3amOmbu7e2DvRHVGs,4080
+arize/spans/validation/metadata/__init__.py,sha256=gfKaohBZWKsZV2u0NImP-u8mziLHputqSRDwtdVgfnI,49
+arize/spans/validation/metadata/argument_validation.py,sha256=HY5j8S0SHCz4AzVy8bl2A5HeDIJESxwStnA39bACMX8,1953
+arize/spans/validation/metadata/dataframe_form_validation.py,sha256=CVoO0-8_0cGoFhsfYD2IpTuucAdpy13CbaTZcti0Ngs,4011
+arize/spans/validation/metadata/value_validation.py,sha256=bUWpDPPiw42iMQuBfLw81ToauMOHduk8H7U_LK4nioc,10251
+arize/spans/validation/spans/__init__.py,sha256=hAgCnolxy6tyb6q1NBzGziezviJswtcnTsChwbkLIvA,44
+arize/spans/validation/spans/dataframe_form_validation.py,sha256=HdM5JUPef4uxyxZeTzjb5FDG7ntSAOashdiXfFxomFM,6377
+arize/spans/validation/spans/spans_validation.py,sha256=sULXzYMJ_INTiRlddJ7Z-iqTAszhIuMEmvaSPk-zrus,3485
+arize/spans/validation/spans/value_validation.py,sha256=nM2tHyt6nIf5yCjAUfDrzbDnXUBWPlAJ0Yht6MObQ2A,21430
+arize/utils/__init__.py,sha256=ScWPiN7pTi2-i6gqRryH9Nlp78zz6X_tbgu_RxLJTEM,62
+arize/utils/arrow.py,sha256=mX6aLjHYNPZrMKJD0Nc-rki_n10Mpq7w6CsSHnVIcLI,5955
+arize/utils/cache.py,sha256=a8jNEZx7R8t3UvmyyyPh2sbjiMaMQlmDRK2-1eLkUeU,3018
+arize/utils/dataframe.py,sha256=qSjT2b9K1gIBBOY7o5FpRMi-qTL1NrGIPp43buw3ito,1758
+arize/utils/openinference_conversion.py,sha256=cUUaWSu9FR99IF0hoIvijcBPrKcHVUuPu4tFoW0obrY,2912
+arize/utils/proto.py,sha256=cXxZr4UwZpEOlxZdZA8B-YY5herdFbZCnxLngGhtrZc,643
+arize/utils/size.py,sha256=qIjUYYg-NttqK25ObX17_32MzkAepLJDDv8N7YxJwAo,805
+arize/utils/types.py,sha256=BxAQwrqRMeA5BaDZcmEczcKkd-T33mi25FL9kBroNXA,2892
+arize/utils/online_tasks/__init__.py,sha256=T7778_bgutP0U7h8e9NRPYfwBQSmHOi859M1K2fjpjw,213
+arize/utils/online_tasks/dataframe_preprocessor.py,sha256=-VcQSetJjZJamWyZryvkA1LhPOT7BdqDGOL4hP-wtik,9399
+arize-8.0.0b0.dist-info/METADATA,sha256=wRxcpzxQ06IxGjITW8n-kRBjEr_DlocmQXRAIW74oA0,29211
+arize-8.0.0b0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+arize-8.0.0b0.dist-info/licenses/LICENSE,sha256=Yseh419WQGiW16p8pS0MwNJyrAIrXSeW59aQXbijY2o,9723
+arize-8.0.0b0.dist-info/licenses/NOTICE,sha256=V9Mdpy_w2tgo5GxgjpsZFTu2WcltLQIh6wBVFwBnJuY,550
+arize-8.0.0b0.dist-info/RECORD,,

{arize-8.0.0a22.dist-info → arize-8.0.0b0.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.27.0
+Generator: hatchling 1.28.0
 Root-Is-Purelib: true
 Tag: py3-none-any

arize-8.0.0b0.dist-info/licenses/LICENSE

@@ -0,0 +1,176 @@
+                              Apache License
+                        Version 2.0, January 2004
+                     http://www.apache.org/licenses/
+
+TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+1. Definitions.
+
+   "License" shall mean the terms and conditions for use, reproduction,
+   and distribution as defined by Sections 1 through 9 of this document.
+
+   "Licensor" shall mean the copyright owner or entity authorized by
+   the copyright owner that is granting the License.
+
+   "Legal Entity" shall mean the union of the acting entity and all
+   other entities that control, are controlled by, or are under common
+   control with that entity. For the purposes of this definition,
+   "control" means (i) the power, direct or indirect, to cause the
+   direction or management of such entity, whether by contract or
+   otherwise, or (ii) ownership of fifty percent (50%) or more of the
+   outstanding shares, or (iii) beneficial ownership of such entity.
+
+   "You" (or "Your") shall mean an individual or Legal Entity
+   exercising permissions granted by this License.
+
+   "Source" form shall mean the preferred form for making modifications,
+   including but not limited to software source code, documentation
+   source, and configuration files.
+
+   "Object" form shall mean any form resulting from mechanical
+   transformation or translation of a Source form, including but
+   not limited to compiled object code, generated documentation,
+   and conversions to other media types.
+
+   "Work" shall mean the work of authorship, whether in Source or
+   Object form, made available under the License, as indicated by a
+   copyright notice that is included in or attached to the work
+   (an example is provided in the Appendix below).
+
+   "Derivative Works" shall mean any work, whether in Source or Object
+   form, that is based on (or derived from) the Work and for which the
+   editorial revisions, annotations, elaborations, or other modifications
+   represent, as a whole, an original work of authorship. For the purposes
+   of this License, Derivative Works shall not include works that remain
+   separable from, or merely link (or bind by name) to the interfaces of,
+   the Work and Derivative Works thereof.
+
+   "Contribution" shall mean any work of authorship, including
+   the original version of the Work and any modifications or additions
+   to that Work or Derivative Works thereof, that is intentionally
+   submitted to Licensor for inclusion in the Work by the copyright owner
+   or by an individual or Legal Entity authorized to submit on behalf of
+   the copyright owner. For the purposes of this definition, "submitted"
+   means any form of electronic, verbal, or written communication sent
+   to the Licensor or its representatives, including but not limited to
+   communication on electronic mailing lists, source code control systems,
+   and issue tracking systems that are managed by, or on behalf of, the
+   Licensor for the purpose of discussing and improving the Work, but
+   excluding communication that is conspicuously marked or otherwise
+   designated in writing by the copyright owner as "Not a Contribution."
+
+   "Contributor" shall mean Licensor and any individual or Legal Entity
+   on behalf of whom a Contribution has been received by Licensor and
+   subsequently incorporated within the Work.
+
+2. Grant of Copyright License. Subject to the terms and conditions of
+   this License, each Contributor hereby grants to You a perpetual,
+   worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+   copyright license to reproduce, prepare Derivative Works of,
+   publicly display, publicly perform, sublicense, and distribute the
+   Work and such Derivative Works in Source or Object form.
+
+3. Grant of Patent License. Subject to the terms and conditions of
+   this License, each Contributor hereby grants to You a perpetual,
+   worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+   (except as stated in this section) patent license to make, have made,
+   use, offer to sell, sell, import, and otherwise transfer the Work,
+   where such license applies only to those patent claims licensable
+   by such Contributor that are necessarily infringed by their
+   Contribution(s) alone or by combination of their Contribution(s)
+   with the Work to which such Contribution(s) was submitted. If You
+   institute patent litigation against any entity (including a
+   cross-claim or counterclaim in a lawsuit) alleging that the Work
+   or a Contribution incorporated within the Work constitutes direct
+   or contributory patent infringement, then any patent licenses
+   granted to You under this License for that Work shall terminate
+   as of the date such litigation is filed.
+
+4. Redistribution. You may reproduce and distribute copies of the
+   Work or Derivative Works thereof in any medium, with or without
+   modifications, and in Source or Object form, provided that You
+   meet the following conditions:
+
+   (a) You must give any other recipients of the Work or
+       Derivative Works a copy of this License; and
+
+   (b) You must cause any modified files to carry prominent notices
+       stating that You changed the files; and
+
+   (c) You must retain, in the Source form of any Derivative Works
+       that You distribute, all copyright, patent, trademark, and
+       attribution notices from the Source form of the Work,
+       excluding those notices that do not pertain to any part of
+       the Derivative Works; and
+
+   (d) If the Work includes a "NOTICE" text file as part of its
+       distribution, then any Derivative Works that You distribute must
+       include a readable copy of the attribution notices contained
+       within such NOTICE file, excluding those notices that do not
+       pertain to any part of the Derivative Works, in at least one
+       of the following places: within a NOTICE text file distributed
+       as part of the Derivative Works; within the Source form or
+       documentation, if provided along with the Derivative Works; or,
+       within a display generated by the Derivative Works, if and
+       wherever such third-party notices normally appear. The contents
+       of the NOTICE file are for informational purposes only and
+       do not modify the License. You may add Your own attribution
+       notices within Derivative Works that You distribute, alongside
+       or as an addendum to the NOTICE text from the Work, provided
+       that such additional attribution notices cannot be construed
+       as modifying the License.
+
+   You may add Your own copyright statement to Your modifications and
+   may provide additional or different license terms and conditions
+   for use, reproduction, or distribution of Your modifications, or
+   for any such Derivative Works as a whole, provided Your use,
+   reproduction, and distribution of the Work otherwise complies with
+   the conditions stated in this License.
+
+5. Submission of Contributions. Unless You explicitly state otherwise,
+   any Contribution intentionally submitted for inclusion in the Work
+   by You to the Licensor shall be under the terms and conditions of
+   this License, without any additional terms or conditions.
+   Notwithstanding the above, nothing herein shall supersede or modify
+   the terms of any separate license agreement you may have executed
+   with Licensor regarding such Contributions.
+
+6. Trademarks. This License does not grant permission to use the trade
+   names, trademarks, service marks, or product names of the Licensor,
+   except as required for reasonable and customary use in describing the
+   origin of the Work and reproducing the content of the NOTICE file.
+
+7. Disclaimer of Warranty. Unless required by applicable law or
+   agreed to in writing, Licensor provides the Work (and each
+   Contributor provides its Contributions) on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+   implied, including, without limitation, any warranties or conditions
+   of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+   PARTICULAR PURPOSE. You are solely responsible for determining the
+   appropriateness of using or redistributing the Work and assume any
+   risks associated with Your exercise of permissions under this License.
+
+8. Limitation of Liability. In no event and under no legal theory,
+   whether in tort (including negligence), contract, or otherwise,
+   unless required by applicable law (such as deliberate and grossly
+   negligent acts) or agreed to in writing, shall any Contributor be
+   liable to You for damages, including any direct, indirect, special,
+   incidental, or consequential damages of any character arising as a
+   result of this License or out of the use or inability to use the
+   Work (including but not limited to damages for loss of goodwill,
+   work stoppage, computer failure or malfunction, or any and all
+   other commercial damages or losses), even if such Contributor
+   has been advised of the possibility of such damages.
+
+9. Accepting Warranty or Additional Liability. While redistributing
+   the Work or Derivative Works thereof, You may choose to offer,
+   and charge a fee for, acceptance of support, warranty, indemnity,
+   or other liability obligations and/or rights consistent with this
+   License. However, in accepting such obligations, You may act only
+   on Your own behalf and on Your sole responsibility, not on behalf
+   of any other Contributor, and only if You agree to indemnify,
+   defend, and hold each Contributor harmless for any liability
+   incurred by, or claims asserted against, such Contributor by reason
+   of your accepting any such warranty or additional liability.
+
+END OF TERMS AND CONDITIONS

arize-8.0.0b0.dist-info/licenses/NOTICE

@@ -0,0 +1,13 @@
+Copyright 2020, Arize AI
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.

arize/_generated/protocol/flight/export_pb2.py

@@ -1,61 +0,0 @@
-# -*- coding: utf-8 -*-
-# Generated by the protocol buffer compiler.  DO NOT EDIT!
-# source: export.proto
-"""Generated protocol buffer code."""
-from google.protobuf import descriptor as _descriptor
-from google.protobuf import descriptor_pool as _descriptor_pool
-from google.protobuf import message as _message
-from google.protobuf import reflection as _reflection
-from google.protobuf import symbol_database as _symbol_database
-# @@protoc_insertion_point(imports)
-
-_sym_db = _symbol_database.Default()
-
-
-from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
-from google.protobuf import wrappers_pb2 as google_dot_protobuf_dot_wrappers__pb2
-
-
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0c\x65xport.proto\x12\x06public\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x1egoogle/protobuf/wrappers.proto\"\xdb\x04\n\x15RecordQueryDescriptor\x12\x10\n\x08space_id\x18\x01 \x01(\t\x12\x10\n\x08model_id\x18\x02 \x01(\t\x12>\n\x0b\x65nvironment\x18\x03 \x01(\x0e\x32).public.RecordQueryDescriptor.Environment\x12\x15\n\rmodel_version\x18\x04 \x01(\t\x12\x10\n\x08\x62\x61tch_id\x18\x05 \x01(\t\x12.\n\nstart_time\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12,\n\x08\x65nd_time\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x17\n\x0finclude_actuals\x18\x08 \x01(\x08\x12\x19\n\x11\x66ilter_expression\x18\t \x01(\t\x12@\n\x18similarity_search_params\x18\n \x01(\x0b\x32\x1e.public.SimilaritySearchParams\x12\x19\n\x11projected_columns\x18\x0b \x03(\t\x12\x36\n\x11stream_chunk_size\x18\x0c \x01(\x0b\x32\x1b.google.protobuf.Int64Value\x12\x37\n\x13parallelize_exports\x18\r \x01(\x0b\x32\x1a.google.protobuf.BoolValue\"U\n\x0b\x45nvironment\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x0c\n\x08TRAINING\x10\x01\x12\x0e\n\nVALIDATION\x10\x02\x12\x0e\n\nPRODUCTION\x10\x03\x12\x0b\n\x07TRACING\x10\x04\"\x93\x02\n\x16SimilaritySearchParams\x12<\n\nreferences\x18\x01 \x03(\x0b\x32(.public.SimilaritySearchParams.Reference\x12\x1a\n\x12search_column_name\x18\x02 \x01(\t\x12\x11\n\tthreshold\x18\x03 \x01(\x01\x1a\x8b\x01\n\tReference\x12\x15\n\rprediction_id\x18\x01 \x01(\t\x12\x1d\n\x15reference_column_name\x18\x02 \x01(\t\x12\x38\n\x14prediction_timestamp\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x0e\n\x06vector\x18\x04 \x03(\x01\x42?Z=github.com/Arize-ai/arize/go/pkg/flightserver/protocol/publicb\x06proto3')
-
-
-
-_RECORDQUERYDESCRIPTOR = DESCRIPTOR.message_types_by_name['RecordQueryDescriptor']
-_SIMILARITYSEARCHPARAMS = DESCRIPTOR.message_types_by_name['SimilaritySearchParams']
-_SIMILARITYSEARCHPARAMS_REFERENCE = _SIMILARITYSEARCHPARAMS.nested_types_by_name['Reference']
-_RECORDQUERYDESCRIPTOR_ENVIRONMENT = _RECORDQUERYDESCRIPTOR.enum_types_by_name['Environment']
-RecordQueryDescriptor = _reflection.GeneratedProtocolMessageType('RecordQueryDescriptor', (_message.Message,), {
-  'DESCRIPTOR' : _RECORDQUERYDESCRIPTOR,
-  '__module__' : 'export_pb2'
-  # @@protoc_insertion_point(class_scope:public.RecordQueryDescriptor)
-  })
-_sym_db.RegisterMessage(RecordQueryDescriptor)
-
-SimilaritySearchParams = _reflection.GeneratedProtocolMessageType('SimilaritySearchParams', (_message.Message,), {
-
-  'Reference' : _reflection.GeneratedProtocolMessageType('Reference', (_message.Message,), {
-    'DESCRIPTOR' : _SIMILARITYSEARCHPARAMS_REFERENCE,
-    '__module__' : 'export_pb2'
-    # @@protoc_insertion_point(class_scope:public.SimilaritySearchParams.Reference)
-    })
-  ,
-  'DESCRIPTOR' : _SIMILARITYSEARCHPARAMS,
-  '__module__' : 'export_pb2'
-  # @@protoc_insertion_point(class_scope:public.SimilaritySearchParams)
-  })
-_sym_db.RegisterMessage(SimilaritySearchParams)
-_sym_db.RegisterMessage(SimilaritySearchParams.Reference)
-
-if _descriptor._USE_C_DESCRIPTORS == False:
-
-  DESCRIPTOR._options = None
-  DESCRIPTOR._serialized_options = b'Z=github.com/Arize-ai/arize/go/pkg/flightserver/protocol/public'
-  _RECORDQUERYDESCRIPTOR._serialized_start=90
-  _RECORDQUERYDESCRIPTOR._serialized_end=693
-  _RECORDQUERYDESCRIPTOR_ENVIRONMENT._serialized_start=608
-  _RECORDQUERYDESCRIPTOR_ENVIRONMENT._serialized_end=693
-  _SIMILARITYSEARCHPARAMS._serialized_start=696
-  _SIMILARITYSEARCHPARAMS._serialized_end=971
-  _SIMILARITYSEARCHPARAMS_REFERENCE._serialized_start=832
-  _SIMILARITYSEARCHPARAMS_REFERENCE._serialized_end=971
-# @@protoc_insertion_point(module_scope)