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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (171) hide show
  1. arize/__init__.py +28 -19
  2. arize/_exporter/client.py +56 -37
  3. arize/_exporter/parsers/tracing_data_parser.py +41 -30
  4. arize/_exporter/validation.py +3 -3
  5. arize/_flight/client.py +207 -76
  6. arize/_generated/api_client/__init__.py +30 -6
  7. arize/_generated/api_client/api/__init__.py +1 -0
  8. arize/_generated/api_client/api/datasets_api.py +864 -190
  9. arize/_generated/api_client/api/experiments_api.py +167 -131
  10. arize/_generated/api_client/api/projects_api.py +1197 -0
  11. arize/_generated/api_client/api_client.py +2 -2
  12. arize/_generated/api_client/configuration.py +42 -34
  13. arize/_generated/api_client/exceptions.py +2 -2
  14. arize/_generated/api_client/models/__init__.py +15 -4
  15. arize/_generated/api_client/models/dataset.py +10 -10
  16. arize/_generated/api_client/models/dataset_example.py +111 -0
  17. arize/_generated/api_client/models/dataset_example_update.py +100 -0
  18. arize/_generated/api_client/models/dataset_version.py +13 -13
  19. arize/_generated/api_client/models/datasets_create_request.py +16 -8
  20. arize/_generated/api_client/models/datasets_examples_insert_request.py +100 -0
  21. arize/_generated/api_client/models/datasets_examples_list200_response.py +106 -0
  22. arize/_generated/api_client/models/datasets_examples_update_request.py +102 -0
  23. arize/_generated/api_client/models/datasets_list200_response.py +10 -4
  24. arize/_generated/api_client/models/experiment.py +14 -16
  25. arize/_generated/api_client/models/experiment_run.py +108 -0
  26. arize/_generated/api_client/models/experiment_run_create.py +102 -0
  27. arize/_generated/api_client/models/experiments_create_request.py +16 -10
  28. arize/_generated/api_client/models/experiments_list200_response.py +10 -4
  29. arize/_generated/api_client/models/experiments_runs_list200_response.py +19 -5
  30. arize/_generated/api_client/models/{error.py → pagination_metadata.py} +13 -11
  31. arize/_generated/api_client/models/primitive_value.py +172 -0
  32. arize/_generated/api_client/models/problem.py +100 -0
  33. arize/_generated/api_client/models/project.py +99 -0
  34. arize/_generated/api_client/models/{datasets_list_examples200_response.py → projects_create_request.py} +13 -11
  35. arize/_generated/api_client/models/projects_list200_response.py +106 -0
  36. arize/_generated/api_client/rest.py +2 -2
  37. arize/_generated/api_client/test/test_dataset.py +4 -2
  38. arize/_generated/api_client/test/test_dataset_example.py +56 -0
  39. arize/_generated/api_client/test/test_dataset_example_update.py +52 -0
  40. arize/_generated/api_client/test/test_dataset_version.py +7 -2
  41. arize/_generated/api_client/test/test_datasets_api.py +27 -13
  42. arize/_generated/api_client/test/test_datasets_create_request.py +8 -4
  43. arize/_generated/api_client/test/{test_datasets_list_examples200_response.py → test_datasets_examples_insert_request.py} +19 -15
  44. arize/_generated/api_client/test/test_datasets_examples_list200_response.py +66 -0
  45. arize/_generated/api_client/test/test_datasets_examples_update_request.py +61 -0
  46. arize/_generated/api_client/test/test_datasets_list200_response.py +9 -3
  47. arize/_generated/api_client/test/test_experiment.py +2 -4
  48. arize/_generated/api_client/test/test_experiment_run.py +56 -0
  49. arize/_generated/api_client/test/test_experiment_run_create.py +54 -0
  50. arize/_generated/api_client/test/test_experiments_api.py +6 -6
  51. arize/_generated/api_client/test/test_experiments_create_request.py +9 -6
  52. arize/_generated/api_client/test/test_experiments_list200_response.py +9 -5
  53. arize/_generated/api_client/test/test_experiments_runs_list200_response.py +15 -5
  54. arize/_generated/api_client/test/test_pagination_metadata.py +53 -0
  55. arize/_generated/api_client/test/{test_error.py → test_primitive_value.py} +13 -14
  56. arize/_generated/api_client/test/test_problem.py +57 -0
  57. arize/_generated/api_client/test/test_project.py +58 -0
  58. arize/_generated/api_client/test/test_projects_api.py +59 -0
  59. arize/_generated/api_client/test/test_projects_create_request.py +54 -0
  60. arize/_generated/api_client/test/test_projects_list200_response.py +70 -0
  61. arize/_generated/api_client_README.md +43 -29
  62. arize/_generated/protocol/flight/flight_pb2.py +400 -0
  63. arize/_lazy.py +27 -19
  64. arize/client.py +181 -58
  65. arize/config.py +324 -116
  66. arize/constants/__init__.py +1 -0
  67. arize/constants/config.py +11 -4
  68. arize/constants/ml.py +6 -4
  69. arize/constants/openinference.py +2 -0
  70. arize/constants/pyarrow.py +2 -0
  71. arize/constants/spans.py +3 -1
  72. arize/datasets/__init__.py +1 -0
  73. arize/datasets/client.py +304 -84
  74. arize/datasets/errors.py +32 -2
  75. arize/datasets/validation.py +18 -8
  76. arize/embeddings/__init__.py +2 -0
  77. arize/embeddings/auto_generator.py +23 -19
  78. arize/embeddings/base_generators.py +89 -36
  79. arize/embeddings/constants.py +2 -0
  80. arize/embeddings/cv_generators.py +26 -4
  81. arize/embeddings/errors.py +27 -5
  82. arize/embeddings/nlp_generators.py +43 -18
  83. arize/embeddings/tabular_generators.py +46 -31
  84. arize/embeddings/usecases.py +12 -2
  85. arize/exceptions/__init__.py +1 -0
  86. arize/exceptions/auth.py +11 -1
  87. arize/exceptions/base.py +29 -4
  88. arize/exceptions/models.py +21 -2
  89. arize/exceptions/parameters.py +31 -0
  90. arize/exceptions/spaces.py +12 -1
  91. arize/exceptions/types.py +86 -7
  92. arize/exceptions/values.py +220 -20
  93. arize/experiments/__init__.py +13 -0
  94. arize/experiments/client.py +394 -285
  95. arize/experiments/evaluators/__init__.py +1 -0
  96. arize/experiments/evaluators/base.py +74 -41
  97. arize/experiments/evaluators/exceptions.py +6 -3
  98. arize/experiments/evaluators/executors.py +121 -73
  99. arize/experiments/evaluators/rate_limiters.py +106 -57
  100. arize/experiments/evaluators/types.py +34 -7
  101. arize/experiments/evaluators/utils.py +65 -27
  102. arize/experiments/functions.py +103 -101
  103. arize/experiments/tracing.py +52 -44
  104. arize/experiments/types.py +56 -31
  105. arize/logging.py +54 -22
  106. arize/ml/__init__.py +1 -0
  107. arize/ml/batch_validation/__init__.py +1 -0
  108. arize/{models → ml}/batch_validation/errors.py +545 -67
  109. arize/{models → ml}/batch_validation/validator.py +344 -303
  110. arize/ml/bounded_executor.py +47 -0
  111. arize/{models → ml}/casting.py +118 -108
  112. arize/{models → ml}/client.py +339 -118
  113. arize/{models → ml}/proto.py +97 -42
  114. arize/{models → ml}/stream_validation.py +43 -15
  115. arize/ml/surrogate_explainer/__init__.py +1 -0
  116. arize/{models → ml}/surrogate_explainer/mimic.py +25 -10
  117. arize/{types.py → ml/types.py} +355 -354
  118. arize/pre_releases.py +44 -0
  119. arize/projects/__init__.py +1 -0
  120. arize/projects/client.py +134 -0
  121. arize/regions.py +40 -0
  122. arize/spans/__init__.py +1 -0
  123. arize/spans/client.py +204 -175
  124. arize/spans/columns.py +13 -0
  125. arize/spans/conversion.py +60 -37
  126. arize/spans/validation/__init__.py +1 -0
  127. arize/spans/validation/annotations/__init__.py +1 -0
  128. arize/spans/validation/annotations/annotations_validation.py +6 -4
  129. arize/spans/validation/annotations/dataframe_form_validation.py +13 -11
  130. arize/spans/validation/annotations/value_validation.py +35 -11
  131. arize/spans/validation/common/__init__.py +1 -0
  132. arize/spans/validation/common/argument_validation.py +33 -8
  133. arize/spans/validation/common/dataframe_form_validation.py +35 -9
  134. arize/spans/validation/common/errors.py +211 -11
  135. arize/spans/validation/common/value_validation.py +81 -14
  136. arize/spans/validation/evals/__init__.py +1 -0
  137. arize/spans/validation/evals/dataframe_form_validation.py +28 -8
  138. arize/spans/validation/evals/evals_validation.py +34 -4
  139. arize/spans/validation/evals/value_validation.py +26 -3
  140. arize/spans/validation/metadata/__init__.py +1 -1
  141. arize/spans/validation/metadata/argument_validation.py +14 -5
  142. arize/spans/validation/metadata/dataframe_form_validation.py +26 -10
  143. arize/spans/validation/metadata/value_validation.py +24 -10
  144. arize/spans/validation/spans/__init__.py +1 -0
  145. arize/spans/validation/spans/dataframe_form_validation.py +35 -14
  146. arize/spans/validation/spans/spans_validation.py +35 -4
  147. arize/spans/validation/spans/value_validation.py +78 -8
  148. arize/utils/__init__.py +1 -0
  149. arize/utils/arrow.py +31 -15
  150. arize/utils/cache.py +34 -6
  151. arize/utils/dataframe.py +20 -3
  152. arize/utils/online_tasks/__init__.py +2 -0
  153. arize/utils/online_tasks/dataframe_preprocessor.py +58 -47
  154. arize/utils/openinference_conversion.py +44 -5
  155. arize/utils/proto.py +10 -0
  156. arize/utils/size.py +5 -3
  157. arize/utils/types.py +105 -0
  158. arize/version.py +3 -1
  159. {arize-8.0.0a22.dist-info → arize-8.0.0b0.dist-info}/METADATA +13 -6
  160. arize-8.0.0b0.dist-info/RECORD +175 -0
  161. {arize-8.0.0a22.dist-info → arize-8.0.0b0.dist-info}/WHEEL +1 -1
  162. arize-8.0.0b0.dist-info/licenses/LICENSE +176 -0
  163. arize-8.0.0b0.dist-info/licenses/NOTICE +13 -0
  164. arize/_generated/protocol/flight/export_pb2.py +0 -61
  165. arize/_generated/protocol/flight/ingest_pb2.py +0 -365
  166. arize/models/__init__.py +0 -0
  167. arize/models/batch_validation/__init__.py +0 -0
  168. arize/models/bounded_executor.py +0 -34
  169. arize/models/surrogate_explainer/__init__.py +0 -0
  170. arize-8.0.0a22.dist-info/RECORD +0 -146
  171. arize-8.0.0a22.dist-info/licenses/LICENSE.md +0 -12
@@ -0,0 +1,70 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ Arize REST API
5
+
6
+ API specification for the backend data server. The API is hosted globally at https://api.arize.com/v2 or in your own environment.
7
+
8
+ The version of the OpenAPI document: 2.0.0
9
+ Generated by OpenAPI Generator (https://openapi-generator.tech)
10
+
11
+ Do not edit the class manually.
12
+ """ # noqa: E501
13
+
14
+
15
+ import unittest
16
+
17
+ from arize._generated.api_client.models.projects_list200_response import ProjectsList200Response
18
+
19
+ class TestProjectsList200Response(unittest.TestCase):
20
+ """ProjectsList200Response unit test stubs"""
21
+
22
+ def setUp(self):
23
+ pass
24
+
25
+ def tearDown(self):
26
+ pass
27
+
28
+ def make_instance(self, include_optional) -> ProjectsList200Response:
29
+ """Test ProjectsList200Response
30
+ include_optional is a boolean, when False only required
31
+ params are included, when True both required and
32
+ optional params are included """
33
+ # uncomment below to create an instance of `ProjectsList200Response`
34
+ """
35
+ model = ProjectsList200Response()
36
+ if include_optional:
37
+ return ProjectsList200Response(
38
+ projects = [
39
+ arize._generated.api_client.models.project.Project(
40
+ id = '',
41
+ name = '',
42
+ space_id = '',
43
+ created_at = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'), )
44
+ ],
45
+ pagination = arize._generated.api_client.models.pagination_metadata.PaginationMetadata(
46
+ next_cursor = '',
47
+ has_more = True, )
48
+ )
49
+ else:
50
+ return ProjectsList200Response(
51
+ projects = [
52
+ arize._generated.api_client.models.project.Project(
53
+ id = '',
54
+ name = '',
55
+ space_id = '',
56
+ created_at = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'), )
57
+ ],
58
+ pagination = arize._generated.api_client.models.pagination_metadata.PaginationMetadata(
59
+ next_cursor = '',
60
+ has_more = True, ),
61
+ )
62
+ """
63
+
64
+ def testProjectsList200Response(self):
65
+ """Test ProjectsList200Response"""
66
+ # inst_req_only = self.make_instance(include_optional=False)
67
+ # inst_req_and_optional = self.make_instance(include_optional=True)
68
+
69
+ if __name__ == '__main__':
70
+ unittest.main()
@@ -1,11 +1,11 @@
1
1
  # Arize
2
- API specification for the backend data server. The API is hosted globally at https://api.arize.com/v2 or in your own environment. You can access the OpenAPI spec for this API at https://api.arize.com/v2/spec.yaml
2
+ API specification for the backend data server. The API is hosted globally at https://api.arize.com/v2 or in your own environment.
3
3
 
4
4
  The `arize._generated.api_client` package is automatically generated by the [OpenAPI Generator](https://openapi-generator.tech) project:
5
5
 
6
- - API version: 0.0.1
6
+ - API version: 2.0.0
7
7
  - Package version: 1.0.0
8
- - Generator version: 7.16.0
8
+ - Generator version: 7.17.0
9
9
  - Build package: org.openapitools.codegen.languages.PythonClientCodegen
10
10
 
11
11
  ## Requirements.
@@ -34,10 +34,10 @@ import arize._generated.api_client
34
34
  from arize._generated.api_client.rest import ApiException
35
35
  from pprint import pprint
36
36
 
37
- # Defining the host is optional and defaults to http://localhost
37
+ # Defining the host is optional and defaults to https://api.arize.com
38
38
  # See configuration.py for a list of all supported configuration parameters.
39
39
  configuration = arize._generated.api_client.Configuration(
40
- host = "http://localhost"
40
+ host = "https://api.arize.com"
41
41
  )
42
42
 
43
43
  # The client must configure the authentication and authorization parameters
@@ -45,21 +45,20 @@ configuration = arize._generated.api_client.Configuration(
45
45
  # Examples for each auth method are provided below, use the example that
46
46
  # satisfies your auth use case.
47
47
 
48
- # Configure API key authorization: ApiKeyAuth
49
- configuration.api_key['ApiKeyAuth'] = os.environ["API_KEY"]
50
-
51
- # Uncomment below to setup prefix (e.g. Bearer) for API key, if needed
52
- # configuration.api_key_prefix['ApiKeyAuth'] = 'Bearer'
48
+ # Configure Bearer authorization (<api-key>): bearerAuth
49
+ configuration = arize._generated.api_client.Configuration(
50
+ access_token = os.environ["BEARER_TOKEN"]
51
+ )
53
52
 
54
53
 
55
54
  # Enter a context with an instance of the API client
56
55
  with arize._generated.api_client.ApiClient(configuration) as api_client:
57
56
  # Create an instance of the API class
58
57
  api_instance = arize._generated.api_client.DatasetsApi(api_client)
59
- datasets_create_request = arize._generated.api_client.DatasetsCreateRequest() # DatasetsCreateRequest | Body containing dataset creation parameters
58
+ datasets_create_request = {"name":"Math Questions Dataset","space_id":"space_12345","examples":[{"question":"What is 2 + 2?","answer":"4","topic":"arithmetic"},{"question":"What is the square root of 16?","answer":"4","topic":"geometry"},{"question":"If 3x = 12, what is x?","answer":"4","topic":"algebra"}]} # DatasetsCreateRequest | Body containing dataset creation parameters
60
59
 
61
60
  try:
62
- # Create a new dataset with JSON examples
61
+ # Create a dataset
63
62
  api_response = api_instance.datasets_create(datasets_create_request)
64
63
  print("The response of DatasetsApi->datasets_create:\n")
65
64
  pprint(api_response)
@@ -70,34 +69,51 @@ with arize._generated.api_client.ApiClient(configuration) as api_client:
70
69
 
71
70
  ## Documentation for API Endpoints
72
71
 
73
- All URIs are relative to *http://localhost*
72
+ All URIs are relative to *https://api.arize.com*
74
73
 
75
74
  Class | Method | HTTP request | Description
76
75
  ------------ | ------------- | ------------- | -------------
77
- *DatasetsApi* | [**datasets_create**](arize/_generated/api_client/docs/DatasetsApi.md#datasets_create) | **POST** /v2/datasets | Create a new dataset with JSON examples
78
- *DatasetsApi* | [**datasets_delete**](arize/_generated/api_client/docs/DatasetsApi.md#datasets_delete) | **DELETE** /v2/datasets/{datasetId} | Delete a dataset by ID
79
- *DatasetsApi* | [**datasets_get**](arize/_generated/api_client/docs/DatasetsApi.md#datasets_get) | **GET** /v2/datasets/{datasetId} | Get dataset by ID
80
- *DatasetsApi* | [**datasets_list**](arize/_generated/api_client/docs/DatasetsApi.md#datasets_list) | **GET** /v2/datasets | List datasets the user has access to
81
- *DatasetsApi* | [**datasets_list_examples**](arize/_generated/api_client/docs/DatasetsApi.md#datasets_list_examples) | **GET** /v2/datasets/{datasetId}/examples | List examples for a dataset
82
- *ExperimentsApi* | [**experiments_create**](arize/_generated/api_client/docs/ExperimentsApi.md#experiments_create) | **POST** /v2/experiments | Create a new experiment with run data
83
- *ExperimentsApi* | [**experiments_delete**](arize/_generated/api_client/docs/ExperimentsApi.md#experiments_delete) | **DELETE** /v2/experiments/{experimentId} | Delete an experiment by ID
84
- *ExperimentsApi* | [**experiments_get**](arize/_generated/api_client/docs/ExperimentsApi.md#experiments_get) | **GET** /v2/experiments/{experimentId} | Get experiment by ID
76
+ *DatasetsApi* | [**datasets_create**](arize/_generated/api_client/docs/DatasetsApi.md#datasets_create) | **POST** /v2/datasets | Create a dataset
77
+ *DatasetsApi* | [**datasets_delete**](arize/_generated/api_client/docs/DatasetsApi.md#datasets_delete) | **DELETE** /v2/datasets/{dataset_id} | Delete a dataset
78
+ *DatasetsApi* | [**datasets_examples_insert**](arize/_generated/api_client/docs/DatasetsApi.md#datasets_examples_insert) | **POST** /v2/datasets/{dataset_id}/examples | Add new examples to a dataset
79
+ *DatasetsApi* | [**datasets_examples_list**](arize/_generated/api_client/docs/DatasetsApi.md#datasets_examples_list) | **GET** /v2/datasets/{dataset_id}/examples | List dataset examples
80
+ *DatasetsApi* | [**datasets_examples_update**](arize/_generated/api_client/docs/DatasetsApi.md#datasets_examples_update) | **PATCH** /v2/datasets/{dataset_id}/examples | Update existing examples in a dataset
81
+ *DatasetsApi* | [**datasets_get**](arize/_generated/api_client/docs/DatasetsApi.md#datasets_get) | **GET** /v2/datasets/{dataset_id} | Get a dataset
82
+ *DatasetsApi* | [**datasets_list**](arize/_generated/api_client/docs/DatasetsApi.md#datasets_list) | **GET** /v2/datasets | List datasets
83
+ *ExperimentsApi* | [**experiments_create**](arize/_generated/api_client/docs/ExperimentsApi.md#experiments_create) | **POST** /v2/experiments | Create an experiment
84
+ *ExperimentsApi* | [**experiments_delete**](arize/_generated/api_client/docs/ExperimentsApi.md#experiments_delete) | **DELETE** /v2/experiments/{experiment_id} | Delete an experiment
85
+ *ExperimentsApi* | [**experiments_get**](arize/_generated/api_client/docs/ExperimentsApi.md#experiments_get) | **GET** /v2/experiments/{experiment_id} | Get an experiment
85
86
  *ExperimentsApi* | [**experiments_list**](arize/_generated/api_client/docs/ExperimentsApi.md#experiments_list) | **GET** /v2/experiments | List experiments
86
- *ExperimentsApi* | [**experiments_runs_list**](arize/_generated/api_client/docs/ExperimentsApi.md#experiments_runs_list) | **GET** /v2/experiments/{experimentId}/runs | List experiment runs for a specific experiment
87
+ *ExperimentsApi* | [**experiments_runs_list**](arize/_generated/api_client/docs/ExperimentsApi.md#experiments_runs_list) | **GET** /v2/experiments/{experiment_id}/runs | List experiment runs
88
+ *ProjectsApi* | [**projects_create**](arize/_generated/api_client/docs/ProjectsApi.md#projects_create) | **POST** /v2/projects | Create a project
89
+ *ProjectsApi* | [**projects_delete**](arize/_generated/api_client/docs/ProjectsApi.md#projects_delete) | **DELETE** /v2/projects/{project_id} | Delete a project
90
+ *ProjectsApi* | [**projects_get**](arize/_generated/api_client/docs/ProjectsApi.md#projects_get) | **GET** /v2/projects/{project_id} | Get a project
91
+ *ProjectsApi* | [**projects_list**](arize/_generated/api_client/docs/ProjectsApi.md#projects_list) | **GET** /v2/projects | List projects
87
92
 
88
93
 
89
94
  ## Documentation For Models
90
95
 
91
96
  - [Dataset](arize/_generated/api_client/docs/Dataset.md)
97
+ - [DatasetExample](arize/_generated/api_client/docs/DatasetExample.md)
98
+ - [DatasetExampleUpdate](arize/_generated/api_client/docs/DatasetExampleUpdate.md)
92
99
  - [DatasetVersion](arize/_generated/api_client/docs/DatasetVersion.md)
93
100
  - [DatasetsCreateRequest](arize/_generated/api_client/docs/DatasetsCreateRequest.md)
101
+ - [DatasetsExamplesInsertRequest](arize/_generated/api_client/docs/DatasetsExamplesInsertRequest.md)
102
+ - [DatasetsExamplesList200Response](arize/_generated/api_client/docs/DatasetsExamplesList200Response.md)
103
+ - [DatasetsExamplesUpdateRequest](arize/_generated/api_client/docs/DatasetsExamplesUpdateRequest.md)
94
104
  - [DatasetsList200Response](arize/_generated/api_client/docs/DatasetsList200Response.md)
95
- - [DatasetsListExamples200Response](arize/_generated/api_client/docs/DatasetsListExamples200Response.md)
96
- - [Error](arize/_generated/api_client/docs/Error.md)
97
105
  - [Experiment](arize/_generated/api_client/docs/Experiment.md)
106
+ - [ExperimentRun](arize/_generated/api_client/docs/ExperimentRun.md)
107
+ - [ExperimentRunCreate](arize/_generated/api_client/docs/ExperimentRunCreate.md)
98
108
  - [ExperimentsCreateRequest](arize/_generated/api_client/docs/ExperimentsCreateRequest.md)
99
109
  - [ExperimentsList200Response](arize/_generated/api_client/docs/ExperimentsList200Response.md)
100
110
  - [ExperimentsRunsList200Response](arize/_generated/api_client/docs/ExperimentsRunsList200Response.md)
111
+ - [PaginationMetadata](arize/_generated/api_client/docs/PaginationMetadata.md)
112
+ - [PrimitiveValue](arize/_generated/api_client/docs/PrimitiveValue.md)
113
+ - [Problem](arize/_generated/api_client/docs/Problem.md)
114
+ - [Project](arize/_generated/api_client/docs/Project.md)
115
+ - [ProjectsCreateRequest](arize/_generated/api_client/docs/ProjectsCreateRequest.md)
116
+ - [ProjectsList200Response](arize/_generated/api_client/docs/ProjectsList200Response.md)
101
117
 
102
118
 
103
119
  <a id="documentation-for-authorization"></a>
@@ -105,12 +121,10 @@ Class | Method | HTTP request | Description
105
121
 
106
122
 
107
123
  Authentication schemes defined for the API:
108
- <a id="ApiKeyAuth"></a>
109
- ### ApiKeyAuth
124
+ <a id="bearerAuth"></a>
125
+ ### bearerAuth
110
126
 
111
- - **Type**: API key
112
- - **API key parameter name**: Authorization
113
- - **Location**: HTTP header
127
+ - **Type**: Bearer authentication (<api-key>)
114
128
 
115
129
 
116
130
  ## Author
@@ -0,0 +1,400 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
3
+ # source: flight.proto
4
+ """Generated protocol buffer code."""
5
+ from google.protobuf.internal import enum_type_wrapper
6
+ from google.protobuf import descriptor as _descriptor
7
+ from google.protobuf import descriptor_pool as _descriptor_pool
8
+ from google.protobuf import message as _message
9
+ from google.protobuf import reflection as _reflection
10
+ from google.protobuf import symbol_database as _symbol_database
11
+ # @@protoc_insertion_point(imports)
12
+
13
+ _sym_db = _symbol_database.Default()
14
+
15
+
16
+ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
17
+ from google.protobuf import wrappers_pb2 as google_dot_protobuf_dot_wrappers__pb2
18
+
19
+
20
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0c\x66light.proto\x12\x06public\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x1egoogle/protobuf/wrappers.proto\"\x83\x01\n\x0c\x44oGetRequest\x12\x30\n\x0bget_dataset\x18\x01 \x01(\x0b\x32\x19.public.GetDatasetRequestH\x00\x12\x36\n\x0eget_experiment\x18\x02 \x01(\x0b\x32\x1c.public.GetExperimentRequestH\x00\x42\t\n\x07request\"\x82\x04\n\x0c\x44oPutRequest\x12\x36\n\x0e\x63reate_dataset\x18\x01 \x01(\x0b\x32\x1c.public.CreateDatasetRequestH\x00\x12\x36\n\x0eupdate_dataset\x18\x02 \x01(\x0b\x32\x1c.public.UpdateDatasetRequestH\x00\x12<\n\x11\x63reate_experiment\x18\x03 \x01(\x0b\x32\x1f.public.CreateExperimentRequestH\x00\x12\x41\n\x14post_experiment_data\x18\x04 \x01(\x0b\x32!.public.PostExperimentDataRequestH\x00\x12K\n\x1dwrite_span_evaluation_request\x18\x05 \x01(\x0b\x32\".public.WriteSpanEvaluationRequestH\x00\x12K\n\x1dwrite_span_annotation_request\x18\x06 \x01(\x0b\x32\".public.WriteSpanAnnotationRequestH\x00\x12\\\n&write_span_attributes_metadata_request\x18\x07 \x01(\x0b\x32*.public.WriteSpanAttributesMetadataRequestH\x00\x42\t\n\x07request\"\xd9\x02\n\x0f\x44oActionRequest\x12\x34\n\rlist_datasets\x18\x01 \x01(\x0b\x32\x1b.public.ListDatasetsRequestH\x00\x12\x36\n\x0e\x64\x65lete_dataset\x18\x02 \x01(\x0b\x32\x1c.public.DeleteDatasetRequestH\x00\x12\x41\n\x14get_dataset_versions\x18\x03 \x01(\x0b\x32!.public.GetDatasetVersionsRequestH\x00\x12L\n\x1a\x63reate_experiment_db_entry\x18\x04 \x01(\x0b\x32&.public.CreateExperimentDBEntryRequestH\x00\x12<\n\x11\x64\x65lete_experiment\x18\x05 \x01(\x0b\x32\x1f.public.DeleteExperimentRequestH\x00\x42\t\n\x07request\"\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\"|\n\x1aWriteSpanEvaluationRequest\x12\x19\n\x11\x65xternal_model_id\x18\x01 \x01(\t\x12\x15\n\rmodel_version\x18\x02 \x01(\t\x12\x10\n\x08space_id\x18\x03 \x01(\t\x12\x1a\n\x12row_id_column_name\x18\x04 \x01(\t\"M\n\x1bWriteSpanEvaluationResponse\x12\x17\n\x0frecords_updated\x18\x01 \x01(\x03\x12\x15\n\runmatched_ids\x18\x02 \x03(\t\"|\n\x1aWriteSpanAnnotationRequest\x12\x19\n\x11\x65xternal_model_id\x18\x01 \x01(\t\x12\x15\n\rmodel_version\x18\x02 \x01(\t\x12\x10\n\x08space_id\x18\x03 \x01(\t\x12\x1a\n\x12row_id_column_name\x18\x04 \x01(\t\"M\n\x1bWriteSpanAnnotationResponse\x12\x17\n\x0frecords_updated\x18\x01 \x01(\x03\x12\x15\n\runmatched_ids\x18\x02 \x03(\t\"Q\n\"WriteSpanAttributesMetadataRequest\x12\x10\n\x08space_id\x18\x01 \x01(\t\x12\x19\n\x11\x65xternal_model_id\x18\x02 \x01(\t\"\xe7\x01\n#WriteSpanAttributesMetadataResponse\x12\x17\n\x0fspans_processed\x18\x01 \x01(\x03\x12\x15\n\rspans_updated\x18\x02 \x01(\x03\x12\x14\n\x0cspans_failed\x18\x03 \x01(\x03\x12\x45\n\x06\x65rrors\x18\x04 \x03(\x0b\x32\x35.public.WriteSpanAttributesMetadataResponse.SpanError\x1a\x33\n\tSpanError\x12\x0f\n\x07span_id\x18\x01 \x01(\t\x12\x15\n\rerror_message\x18\x02 \x01(\t\"\'\n\x13ListDatasetsRequest\x12\x10\n\x08space_id\x18\x01 \x01(\t\"=\n\x14ListDatasetsResponse\x12%\n\x08\x64\x61tasets\x18\x01 \x03(\x0b\x32\x13.public.DatasetInfo\"i\n\x14\x43reateDatasetRequest\x12\x10\n\x08space_id\x18\x01 \x01(\t\x12\x14\n\x0c\x64\x61taset_name\x18\x02 \x01(\t\x12)\n\x0c\x64\x61taset_type\x18\x03 \x01(\x0e\x32\x13.public.DatasetType\"+\n\x15\x43reateDatasetResponse\x12\x12\n\ndataset_id\x18\x01 \x01(\t\"\x82\x01\n\x11GetDatasetRequest\x12\x10\n\x08space_id\x18\x01 \x01(\t\x12\x17\n\x0f\x64\x61taset_version\x18\x03 \x01(\t\x12\x14\n\ndataset_id\x18\x02 \x01(\tH\x00\x12\x16\n\x0c\x64\x61taset_name\x18\x04 \x01(\tH\x00\x42\x14\n\x12\x64\x61taset_identifier\"l\n\x14UpdateDatasetRequest\x12\x10\n\x08space_id\x18\x01 \x01(\t\x12\x14\n\ndataset_id\x18\x02 \x01(\tH\x00\x12\x16\n\x0c\x64\x61taset_name\x18\x03 \x01(\tH\x00\x42\x14\n\x12\x64\x61taset_identifier\"+\n\x15UpdateDatasetResponse\x12\x12\n\ndataset_id\x18\x01 \x01(\t\"l\n\x14\x44\x65leteDatasetRequest\x12\x10\n\x08space_id\x18\x01 \x01(\t\x12\x14\n\ndataset_id\x18\x02 \x01(\tH\x00\x12\x16\n\x0c\x64\x61taset_name\x18\x03 \x01(\tH\x00\x42\x14\n\x12\x64\x61taset_identifier\"(\n\x15\x44\x65leteDatasetResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\"q\n\x19GetDatasetVersionsRequest\x12\x10\n\x08space_id\x18\x01 \x01(\t\x12\x14\n\ndataset_id\x18\x02 \x01(\tH\x00\x12\x16\n\x0c\x64\x61taset_name\x18\x03 \x01(\tH\x00\x42\x14\n\x12\x64\x61taset_identifier\"\xc2\x01\n\x0b\x44\x61tasetInfo\x12\x12\n\ndataset_id\x18\x01 \x01(\t\x12\x14\n\x0c\x64\x61taset_name\x18\x02 \x01(\t\x12)\n\x0c\x64\x61taset_type\x18\x03 \x01(\x0e\x32\x13.public.DatasetType\x12.\n\ncreated_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12.\n\nupdated_at\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\"\x8a\x01\n\x12\x44\x61tasetVersionInfo\x12\x14\n\x0cversion_name\x18\x01 \x01(\t\x12.\n\ncreated_at\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12.\n\nupdated_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\"J\n\x1aGetDatasetVersionsResponse\x12,\n\x08versions\x18\x02 \x03(\x0b\x32\x1a.public.DatasetVersionInfo\"X\n\x17\x43reateExperimentRequest\x12\x10\n\x08space_id\x18\x01 \x01(\t\x12\x12\n\ndataset_id\x18\x02 \x01(\t\x12\x17\n\x0f\x65xperiment_name\x18\x03 \x01(\t\"1\n\x18\x43reateExperimentResponse\x12\x15\n\rexperiment_id\x18\x01 \x01(\t\"n\n\x14GetExperimentRequest\x12\x10\n\x08space_id\x18\x01 \x01(\t\x12\x15\n\rexperiment_id\x18\x02 \x01(\t\x12\x17\n\x0f\x65xperiment_name\x18\x03 \x01(\t\x12\x14\n\x0c\x64\x61taset_name\x18\x04 \x01(\t\"Z\n\x19PostExperimentDataRequest\x12\x10\n\x08space_id\x18\x01 \x01(\t\x12\x12\n\ndataset_id\x18\x02 \x01(\t\x12\x17\n\x0f\x65xperiment_name\x18\x03 \x01(\t\"G\n\x1aPostExperimentDataResponse\x12\x15\n\rexperiment_id\x18\x01 \x01(\t\x12\x12\n\ndataset_id\x18\x02 \x01(\t\"\xa5\x01\n\x17\x44\x65leteExperimentRequest\x12\x10\n\x08space_id\x18\x01 \x01(\t\x12\x17\n\rexperiment_id\x18\x02 \x01(\tH\x00\x12\x46\n\x1a\x65xperiment_name_identifier\x18\x03 \x01(\x0b\x32 .public.ExperimentNameIdentifierH\x00\x42\x17\n\x15\x65xperiment_identifier\"I\n\x18\x45xperimentNameIdentifier\x12\x17\n\x0f\x65xperiment_name\x18\x01 \x01(\t\x12\x14\n\x0c\x64\x61taset_name\x18\x02 \x01(\t\"1\n\x18\x44\x65leteExperimentResponse\x12\x15\n\rexperiment_id\x18\x01 \x01(\t\"\xa9\x01\n\x1e\x43reateExperimentDBEntryRequest\x12\x10\n\x08space_id\x18\x01 \x01(\t\x12\x14\n\ndataset_id\x18\x02 \x01(\tH\x00\x12\x16\n\x0c\x64\x61taset_name\x18\x03 \x01(\tH\x00\x12\x17\n\x0f\x65xperiment_name\x18\x04 \x01(\t\x12\x18\n\x10trace_model_name\x18\x05 \x01(\tB\x14\n\x12\x64\x61taset_identifier\"f\n\x1f\x43reateExperimentDBEntryResponse\x12\x15\n\rexperiment_id\x18\x01 \x01(\t\x12\x12\n\ndataset_id\x18\x02 \x01(\t\x12\x18\n\x10trace_model_name\x18\x03 \x01(\t*:\n\x0b\x44\x61tasetType\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x0e\n\nGENERATIVE\x10\x01\x12\x0e\n\nINFERENCES\x10\x02\x42?Z=github.com/Arize-ai/arize/go/pkg/flightserver/protocol/publicb\x06proto3')
21
+
22
+ _DATASETTYPE = DESCRIPTOR.enum_types_by_name['DatasetType']
23
+ DatasetType = enum_type_wrapper.EnumTypeWrapper(_DATASETTYPE)
24
+ UNKNOWN = 0
25
+ GENERATIVE = 1
26
+ INFERENCES = 2
27
+
28
+
29
+ _DOGETREQUEST = DESCRIPTOR.message_types_by_name['DoGetRequest']
30
+ _DOPUTREQUEST = DESCRIPTOR.message_types_by_name['DoPutRequest']
31
+ _DOACTIONREQUEST = DESCRIPTOR.message_types_by_name['DoActionRequest']
32
+ _RECORDQUERYDESCRIPTOR = DESCRIPTOR.message_types_by_name['RecordQueryDescriptor']
33
+ _SIMILARITYSEARCHPARAMS = DESCRIPTOR.message_types_by_name['SimilaritySearchParams']
34
+ _SIMILARITYSEARCHPARAMS_REFERENCE = _SIMILARITYSEARCHPARAMS.nested_types_by_name['Reference']
35
+ _WRITESPANEVALUATIONREQUEST = DESCRIPTOR.message_types_by_name['WriteSpanEvaluationRequest']
36
+ _WRITESPANEVALUATIONRESPONSE = DESCRIPTOR.message_types_by_name['WriteSpanEvaluationResponse']
37
+ _WRITESPANANNOTATIONREQUEST = DESCRIPTOR.message_types_by_name['WriteSpanAnnotationRequest']
38
+ _WRITESPANANNOTATIONRESPONSE = DESCRIPTOR.message_types_by_name['WriteSpanAnnotationResponse']
39
+ _WRITESPANATTRIBUTESMETADATAREQUEST = DESCRIPTOR.message_types_by_name['WriteSpanAttributesMetadataRequest']
40
+ _WRITESPANATTRIBUTESMETADATARESPONSE = DESCRIPTOR.message_types_by_name['WriteSpanAttributesMetadataResponse']
41
+ _WRITESPANATTRIBUTESMETADATARESPONSE_SPANERROR = _WRITESPANATTRIBUTESMETADATARESPONSE.nested_types_by_name['SpanError']
42
+ _LISTDATASETSREQUEST = DESCRIPTOR.message_types_by_name['ListDatasetsRequest']
43
+ _LISTDATASETSRESPONSE = DESCRIPTOR.message_types_by_name['ListDatasetsResponse']
44
+ _CREATEDATASETREQUEST = DESCRIPTOR.message_types_by_name['CreateDatasetRequest']
45
+ _CREATEDATASETRESPONSE = DESCRIPTOR.message_types_by_name['CreateDatasetResponse']
46
+ _GETDATASETREQUEST = DESCRIPTOR.message_types_by_name['GetDatasetRequest']
47
+ _UPDATEDATASETREQUEST = DESCRIPTOR.message_types_by_name['UpdateDatasetRequest']
48
+ _UPDATEDATASETRESPONSE = DESCRIPTOR.message_types_by_name['UpdateDatasetResponse']
49
+ _DELETEDATASETREQUEST = DESCRIPTOR.message_types_by_name['DeleteDatasetRequest']
50
+ _DELETEDATASETRESPONSE = DESCRIPTOR.message_types_by_name['DeleteDatasetResponse']
51
+ _GETDATASETVERSIONSREQUEST = DESCRIPTOR.message_types_by_name['GetDatasetVersionsRequest']
52
+ _DATASETINFO = DESCRIPTOR.message_types_by_name['DatasetInfo']
53
+ _DATASETVERSIONINFO = DESCRIPTOR.message_types_by_name['DatasetVersionInfo']
54
+ _GETDATASETVERSIONSRESPONSE = DESCRIPTOR.message_types_by_name['GetDatasetVersionsResponse']
55
+ _CREATEEXPERIMENTREQUEST = DESCRIPTOR.message_types_by_name['CreateExperimentRequest']
56
+ _CREATEEXPERIMENTRESPONSE = DESCRIPTOR.message_types_by_name['CreateExperimentResponse']
57
+ _GETEXPERIMENTREQUEST = DESCRIPTOR.message_types_by_name['GetExperimentRequest']
58
+ _POSTEXPERIMENTDATAREQUEST = DESCRIPTOR.message_types_by_name['PostExperimentDataRequest']
59
+ _POSTEXPERIMENTDATARESPONSE = DESCRIPTOR.message_types_by_name['PostExperimentDataResponse']
60
+ _DELETEEXPERIMENTREQUEST = DESCRIPTOR.message_types_by_name['DeleteExperimentRequest']
61
+ _EXPERIMENTNAMEIDENTIFIER = DESCRIPTOR.message_types_by_name['ExperimentNameIdentifier']
62
+ _DELETEEXPERIMENTRESPONSE = DESCRIPTOR.message_types_by_name['DeleteExperimentResponse']
63
+ _CREATEEXPERIMENTDBENTRYREQUEST = DESCRIPTOR.message_types_by_name['CreateExperimentDBEntryRequest']
64
+ _CREATEEXPERIMENTDBENTRYRESPONSE = DESCRIPTOR.message_types_by_name['CreateExperimentDBEntryResponse']
65
+ _RECORDQUERYDESCRIPTOR_ENVIRONMENT = _RECORDQUERYDESCRIPTOR.enum_types_by_name['Environment']
66
+ DoGetRequest = _reflection.GeneratedProtocolMessageType('DoGetRequest', (_message.Message,), {
67
+ 'DESCRIPTOR' : _DOGETREQUEST,
68
+ '__module__' : 'flight_pb2'
69
+ # @@protoc_insertion_point(class_scope:public.DoGetRequest)
70
+ })
71
+ _sym_db.RegisterMessage(DoGetRequest)
72
+
73
+ DoPutRequest = _reflection.GeneratedProtocolMessageType('DoPutRequest', (_message.Message,), {
74
+ 'DESCRIPTOR' : _DOPUTREQUEST,
75
+ '__module__' : 'flight_pb2'
76
+ # @@protoc_insertion_point(class_scope:public.DoPutRequest)
77
+ })
78
+ _sym_db.RegisterMessage(DoPutRequest)
79
+
80
+ DoActionRequest = _reflection.GeneratedProtocolMessageType('DoActionRequest', (_message.Message,), {
81
+ 'DESCRIPTOR' : _DOACTIONREQUEST,
82
+ '__module__' : 'flight_pb2'
83
+ # @@protoc_insertion_point(class_scope:public.DoActionRequest)
84
+ })
85
+ _sym_db.RegisterMessage(DoActionRequest)
86
+
87
+ RecordQueryDescriptor = _reflection.GeneratedProtocolMessageType('RecordQueryDescriptor', (_message.Message,), {
88
+ 'DESCRIPTOR' : _RECORDQUERYDESCRIPTOR,
89
+ '__module__' : 'flight_pb2'
90
+ # @@protoc_insertion_point(class_scope:public.RecordQueryDescriptor)
91
+ })
92
+ _sym_db.RegisterMessage(RecordQueryDescriptor)
93
+
94
+ SimilaritySearchParams = _reflection.GeneratedProtocolMessageType('SimilaritySearchParams', (_message.Message,), {
95
+
96
+ 'Reference' : _reflection.GeneratedProtocolMessageType('Reference', (_message.Message,), {
97
+ 'DESCRIPTOR' : _SIMILARITYSEARCHPARAMS_REFERENCE,
98
+ '__module__' : 'flight_pb2'
99
+ # @@protoc_insertion_point(class_scope:public.SimilaritySearchParams.Reference)
100
+ })
101
+ ,
102
+ 'DESCRIPTOR' : _SIMILARITYSEARCHPARAMS,
103
+ '__module__' : 'flight_pb2'
104
+ # @@protoc_insertion_point(class_scope:public.SimilaritySearchParams)
105
+ })
106
+ _sym_db.RegisterMessage(SimilaritySearchParams)
107
+ _sym_db.RegisterMessage(SimilaritySearchParams.Reference)
108
+
109
+ WriteSpanEvaluationRequest = _reflection.GeneratedProtocolMessageType('WriteSpanEvaluationRequest', (_message.Message,), {
110
+ 'DESCRIPTOR' : _WRITESPANEVALUATIONREQUEST,
111
+ '__module__' : 'flight_pb2'
112
+ # @@protoc_insertion_point(class_scope:public.WriteSpanEvaluationRequest)
113
+ })
114
+ _sym_db.RegisterMessage(WriteSpanEvaluationRequest)
115
+
116
+ WriteSpanEvaluationResponse = _reflection.GeneratedProtocolMessageType('WriteSpanEvaluationResponse', (_message.Message,), {
117
+ 'DESCRIPTOR' : _WRITESPANEVALUATIONRESPONSE,
118
+ '__module__' : 'flight_pb2'
119
+ # @@protoc_insertion_point(class_scope:public.WriteSpanEvaluationResponse)
120
+ })
121
+ _sym_db.RegisterMessage(WriteSpanEvaluationResponse)
122
+
123
+ WriteSpanAnnotationRequest = _reflection.GeneratedProtocolMessageType('WriteSpanAnnotationRequest', (_message.Message,), {
124
+ 'DESCRIPTOR' : _WRITESPANANNOTATIONREQUEST,
125
+ '__module__' : 'flight_pb2'
126
+ # @@protoc_insertion_point(class_scope:public.WriteSpanAnnotationRequest)
127
+ })
128
+ _sym_db.RegisterMessage(WriteSpanAnnotationRequest)
129
+
130
+ WriteSpanAnnotationResponse = _reflection.GeneratedProtocolMessageType('WriteSpanAnnotationResponse', (_message.Message,), {
131
+ 'DESCRIPTOR' : _WRITESPANANNOTATIONRESPONSE,
132
+ '__module__' : 'flight_pb2'
133
+ # @@protoc_insertion_point(class_scope:public.WriteSpanAnnotationResponse)
134
+ })
135
+ _sym_db.RegisterMessage(WriteSpanAnnotationResponse)
136
+
137
+ WriteSpanAttributesMetadataRequest = _reflection.GeneratedProtocolMessageType('WriteSpanAttributesMetadataRequest', (_message.Message,), {
138
+ 'DESCRIPTOR' : _WRITESPANATTRIBUTESMETADATAREQUEST,
139
+ '__module__' : 'flight_pb2'
140
+ # @@protoc_insertion_point(class_scope:public.WriteSpanAttributesMetadataRequest)
141
+ })
142
+ _sym_db.RegisterMessage(WriteSpanAttributesMetadataRequest)
143
+
144
+ WriteSpanAttributesMetadataResponse = _reflection.GeneratedProtocolMessageType('WriteSpanAttributesMetadataResponse', (_message.Message,), {
145
+
146
+ 'SpanError' : _reflection.GeneratedProtocolMessageType('SpanError', (_message.Message,), {
147
+ 'DESCRIPTOR' : _WRITESPANATTRIBUTESMETADATARESPONSE_SPANERROR,
148
+ '__module__' : 'flight_pb2'
149
+ # @@protoc_insertion_point(class_scope:public.WriteSpanAttributesMetadataResponse.SpanError)
150
+ })
151
+ ,
152
+ 'DESCRIPTOR' : _WRITESPANATTRIBUTESMETADATARESPONSE,
153
+ '__module__' : 'flight_pb2'
154
+ # @@protoc_insertion_point(class_scope:public.WriteSpanAttributesMetadataResponse)
155
+ })
156
+ _sym_db.RegisterMessage(WriteSpanAttributesMetadataResponse)
157
+ _sym_db.RegisterMessage(WriteSpanAttributesMetadataResponse.SpanError)
158
+
159
+ ListDatasetsRequest = _reflection.GeneratedProtocolMessageType('ListDatasetsRequest', (_message.Message,), {
160
+ 'DESCRIPTOR' : _LISTDATASETSREQUEST,
161
+ '__module__' : 'flight_pb2'
162
+ # @@protoc_insertion_point(class_scope:public.ListDatasetsRequest)
163
+ })
164
+ _sym_db.RegisterMessage(ListDatasetsRequest)
165
+
166
+ ListDatasetsResponse = _reflection.GeneratedProtocolMessageType('ListDatasetsResponse', (_message.Message,), {
167
+ 'DESCRIPTOR' : _LISTDATASETSRESPONSE,
168
+ '__module__' : 'flight_pb2'
169
+ # @@protoc_insertion_point(class_scope:public.ListDatasetsResponse)
170
+ })
171
+ _sym_db.RegisterMessage(ListDatasetsResponse)
172
+
173
+ CreateDatasetRequest = _reflection.GeneratedProtocolMessageType('CreateDatasetRequest', (_message.Message,), {
174
+ 'DESCRIPTOR' : _CREATEDATASETREQUEST,
175
+ '__module__' : 'flight_pb2'
176
+ # @@protoc_insertion_point(class_scope:public.CreateDatasetRequest)
177
+ })
178
+ _sym_db.RegisterMessage(CreateDatasetRequest)
179
+
180
+ CreateDatasetResponse = _reflection.GeneratedProtocolMessageType('CreateDatasetResponse', (_message.Message,), {
181
+ 'DESCRIPTOR' : _CREATEDATASETRESPONSE,
182
+ '__module__' : 'flight_pb2'
183
+ # @@protoc_insertion_point(class_scope:public.CreateDatasetResponse)
184
+ })
185
+ _sym_db.RegisterMessage(CreateDatasetResponse)
186
+
187
+ GetDatasetRequest = _reflection.GeneratedProtocolMessageType('GetDatasetRequest', (_message.Message,), {
188
+ 'DESCRIPTOR' : _GETDATASETREQUEST,
189
+ '__module__' : 'flight_pb2'
190
+ # @@protoc_insertion_point(class_scope:public.GetDatasetRequest)
191
+ })
192
+ _sym_db.RegisterMessage(GetDatasetRequest)
193
+
194
+ UpdateDatasetRequest = _reflection.GeneratedProtocolMessageType('UpdateDatasetRequest', (_message.Message,), {
195
+ 'DESCRIPTOR' : _UPDATEDATASETREQUEST,
196
+ '__module__' : 'flight_pb2'
197
+ # @@protoc_insertion_point(class_scope:public.UpdateDatasetRequest)
198
+ })
199
+ _sym_db.RegisterMessage(UpdateDatasetRequest)
200
+
201
+ UpdateDatasetResponse = _reflection.GeneratedProtocolMessageType('UpdateDatasetResponse', (_message.Message,), {
202
+ 'DESCRIPTOR' : _UPDATEDATASETRESPONSE,
203
+ '__module__' : 'flight_pb2'
204
+ # @@protoc_insertion_point(class_scope:public.UpdateDatasetResponse)
205
+ })
206
+ _sym_db.RegisterMessage(UpdateDatasetResponse)
207
+
208
+ DeleteDatasetRequest = _reflection.GeneratedProtocolMessageType('DeleteDatasetRequest', (_message.Message,), {
209
+ 'DESCRIPTOR' : _DELETEDATASETREQUEST,
210
+ '__module__' : 'flight_pb2'
211
+ # @@protoc_insertion_point(class_scope:public.DeleteDatasetRequest)
212
+ })
213
+ _sym_db.RegisterMessage(DeleteDatasetRequest)
214
+
215
+ DeleteDatasetResponse = _reflection.GeneratedProtocolMessageType('DeleteDatasetResponse', (_message.Message,), {
216
+ 'DESCRIPTOR' : _DELETEDATASETRESPONSE,
217
+ '__module__' : 'flight_pb2'
218
+ # @@protoc_insertion_point(class_scope:public.DeleteDatasetResponse)
219
+ })
220
+ _sym_db.RegisterMessage(DeleteDatasetResponse)
221
+
222
+ GetDatasetVersionsRequest = _reflection.GeneratedProtocolMessageType('GetDatasetVersionsRequest', (_message.Message,), {
223
+ 'DESCRIPTOR' : _GETDATASETVERSIONSREQUEST,
224
+ '__module__' : 'flight_pb2'
225
+ # @@protoc_insertion_point(class_scope:public.GetDatasetVersionsRequest)
226
+ })
227
+ _sym_db.RegisterMessage(GetDatasetVersionsRequest)
228
+
229
+ DatasetInfo = _reflection.GeneratedProtocolMessageType('DatasetInfo', (_message.Message,), {
230
+ 'DESCRIPTOR' : _DATASETINFO,
231
+ '__module__' : 'flight_pb2'
232
+ # @@protoc_insertion_point(class_scope:public.DatasetInfo)
233
+ })
234
+ _sym_db.RegisterMessage(DatasetInfo)
235
+
236
+ DatasetVersionInfo = _reflection.GeneratedProtocolMessageType('DatasetVersionInfo', (_message.Message,), {
237
+ 'DESCRIPTOR' : _DATASETVERSIONINFO,
238
+ '__module__' : 'flight_pb2'
239
+ # @@protoc_insertion_point(class_scope:public.DatasetVersionInfo)
240
+ })
241
+ _sym_db.RegisterMessage(DatasetVersionInfo)
242
+
243
+ GetDatasetVersionsResponse = _reflection.GeneratedProtocolMessageType('GetDatasetVersionsResponse', (_message.Message,), {
244
+ 'DESCRIPTOR' : _GETDATASETVERSIONSRESPONSE,
245
+ '__module__' : 'flight_pb2'
246
+ # @@protoc_insertion_point(class_scope:public.GetDatasetVersionsResponse)
247
+ })
248
+ _sym_db.RegisterMessage(GetDatasetVersionsResponse)
249
+
250
+ CreateExperimentRequest = _reflection.GeneratedProtocolMessageType('CreateExperimentRequest', (_message.Message,), {
251
+ 'DESCRIPTOR' : _CREATEEXPERIMENTREQUEST,
252
+ '__module__' : 'flight_pb2'
253
+ # @@protoc_insertion_point(class_scope:public.CreateExperimentRequest)
254
+ })
255
+ _sym_db.RegisterMessage(CreateExperimentRequest)
256
+
257
+ CreateExperimentResponse = _reflection.GeneratedProtocolMessageType('CreateExperimentResponse', (_message.Message,), {
258
+ 'DESCRIPTOR' : _CREATEEXPERIMENTRESPONSE,
259
+ '__module__' : 'flight_pb2'
260
+ # @@protoc_insertion_point(class_scope:public.CreateExperimentResponse)
261
+ })
262
+ _sym_db.RegisterMessage(CreateExperimentResponse)
263
+
264
+ GetExperimentRequest = _reflection.GeneratedProtocolMessageType('GetExperimentRequest', (_message.Message,), {
265
+ 'DESCRIPTOR' : _GETEXPERIMENTREQUEST,
266
+ '__module__' : 'flight_pb2'
267
+ # @@protoc_insertion_point(class_scope:public.GetExperimentRequest)
268
+ })
269
+ _sym_db.RegisterMessage(GetExperimentRequest)
270
+
271
+ PostExperimentDataRequest = _reflection.GeneratedProtocolMessageType('PostExperimentDataRequest', (_message.Message,), {
272
+ 'DESCRIPTOR' : _POSTEXPERIMENTDATAREQUEST,
273
+ '__module__' : 'flight_pb2'
274
+ # @@protoc_insertion_point(class_scope:public.PostExperimentDataRequest)
275
+ })
276
+ _sym_db.RegisterMessage(PostExperimentDataRequest)
277
+
278
+ PostExperimentDataResponse = _reflection.GeneratedProtocolMessageType('PostExperimentDataResponse', (_message.Message,), {
279
+ 'DESCRIPTOR' : _POSTEXPERIMENTDATARESPONSE,
280
+ '__module__' : 'flight_pb2'
281
+ # @@protoc_insertion_point(class_scope:public.PostExperimentDataResponse)
282
+ })
283
+ _sym_db.RegisterMessage(PostExperimentDataResponse)
284
+
285
+ DeleteExperimentRequest = _reflection.GeneratedProtocolMessageType('DeleteExperimentRequest', (_message.Message,), {
286
+ 'DESCRIPTOR' : _DELETEEXPERIMENTREQUEST,
287
+ '__module__' : 'flight_pb2'
288
+ # @@protoc_insertion_point(class_scope:public.DeleteExperimentRequest)
289
+ })
290
+ _sym_db.RegisterMessage(DeleteExperimentRequest)
291
+
292
+ ExperimentNameIdentifier = _reflection.GeneratedProtocolMessageType('ExperimentNameIdentifier', (_message.Message,), {
293
+ 'DESCRIPTOR' : _EXPERIMENTNAMEIDENTIFIER,
294
+ '__module__' : 'flight_pb2'
295
+ # @@protoc_insertion_point(class_scope:public.ExperimentNameIdentifier)
296
+ })
297
+ _sym_db.RegisterMessage(ExperimentNameIdentifier)
298
+
299
+ DeleteExperimentResponse = _reflection.GeneratedProtocolMessageType('DeleteExperimentResponse', (_message.Message,), {
300
+ 'DESCRIPTOR' : _DELETEEXPERIMENTRESPONSE,
301
+ '__module__' : 'flight_pb2'
302
+ # @@protoc_insertion_point(class_scope:public.DeleteExperimentResponse)
303
+ })
304
+ _sym_db.RegisterMessage(DeleteExperimentResponse)
305
+
306
+ CreateExperimentDBEntryRequest = _reflection.GeneratedProtocolMessageType('CreateExperimentDBEntryRequest', (_message.Message,), {
307
+ 'DESCRIPTOR' : _CREATEEXPERIMENTDBENTRYREQUEST,
308
+ '__module__' : 'flight_pb2'
309
+ # @@protoc_insertion_point(class_scope:public.CreateExperimentDBEntryRequest)
310
+ })
311
+ _sym_db.RegisterMessage(CreateExperimentDBEntryRequest)
312
+
313
+ CreateExperimentDBEntryResponse = _reflection.GeneratedProtocolMessageType('CreateExperimentDBEntryResponse', (_message.Message,), {
314
+ 'DESCRIPTOR' : _CREATEEXPERIMENTDBENTRYRESPONSE,
315
+ '__module__' : 'flight_pb2'
316
+ # @@protoc_insertion_point(class_scope:public.CreateExperimentDBEntryResponse)
317
+ })
318
+ _sym_db.RegisterMessage(CreateExperimentDBEntryResponse)
319
+
320
+ if _descriptor._USE_C_DESCRIPTORS == False:
321
+
322
+ DESCRIPTOR._options = None
323
+ DESCRIPTOR._serialized_options = b'Z=github.com/Arize-ai/arize/go/pkg/flightserver/protocol/public'
324
+ _DATASETTYPE._serialized_start=4912
325
+ _DATASETTYPE._serialized_end=4970
326
+ _DOGETREQUEST._serialized_start=90
327
+ _DOGETREQUEST._serialized_end=221
328
+ _DOPUTREQUEST._serialized_start=224
329
+ _DOPUTREQUEST._serialized_end=738
330
+ _DOACTIONREQUEST._serialized_start=741
331
+ _DOACTIONREQUEST._serialized_end=1086
332
+ _RECORDQUERYDESCRIPTOR._serialized_start=1089
333
+ _RECORDQUERYDESCRIPTOR._serialized_end=1692
334
+ _RECORDQUERYDESCRIPTOR_ENVIRONMENT._serialized_start=1607
335
+ _RECORDQUERYDESCRIPTOR_ENVIRONMENT._serialized_end=1692
336
+ _SIMILARITYSEARCHPARAMS._serialized_start=1695
337
+ _SIMILARITYSEARCHPARAMS._serialized_end=1970
338
+ _SIMILARITYSEARCHPARAMS_REFERENCE._serialized_start=1831
339
+ _SIMILARITYSEARCHPARAMS_REFERENCE._serialized_end=1970
340
+ _WRITESPANEVALUATIONREQUEST._serialized_start=1972
341
+ _WRITESPANEVALUATIONREQUEST._serialized_end=2096
342
+ _WRITESPANEVALUATIONRESPONSE._serialized_start=2098
343
+ _WRITESPANEVALUATIONRESPONSE._serialized_end=2175
344
+ _WRITESPANANNOTATIONREQUEST._serialized_start=2177
345
+ _WRITESPANANNOTATIONREQUEST._serialized_end=2301
346
+ _WRITESPANANNOTATIONRESPONSE._serialized_start=2303
347
+ _WRITESPANANNOTATIONRESPONSE._serialized_end=2380
348
+ _WRITESPANATTRIBUTESMETADATAREQUEST._serialized_start=2382
349
+ _WRITESPANATTRIBUTESMETADATAREQUEST._serialized_end=2463
350
+ _WRITESPANATTRIBUTESMETADATARESPONSE._serialized_start=2466
351
+ _WRITESPANATTRIBUTESMETADATARESPONSE._serialized_end=2697
352
+ _WRITESPANATTRIBUTESMETADATARESPONSE_SPANERROR._serialized_start=2646
353
+ _WRITESPANATTRIBUTESMETADATARESPONSE_SPANERROR._serialized_end=2697
354
+ _LISTDATASETSREQUEST._serialized_start=2699
355
+ _LISTDATASETSREQUEST._serialized_end=2738
356
+ _LISTDATASETSRESPONSE._serialized_start=2740
357
+ _LISTDATASETSRESPONSE._serialized_end=2801
358
+ _CREATEDATASETREQUEST._serialized_start=2803
359
+ _CREATEDATASETREQUEST._serialized_end=2908
360
+ _CREATEDATASETRESPONSE._serialized_start=2910
361
+ _CREATEDATASETRESPONSE._serialized_end=2953
362
+ _GETDATASETREQUEST._serialized_start=2956
363
+ _GETDATASETREQUEST._serialized_end=3086
364
+ _UPDATEDATASETREQUEST._serialized_start=3088
365
+ _UPDATEDATASETREQUEST._serialized_end=3196
366
+ _UPDATEDATASETRESPONSE._serialized_start=3198
367
+ _UPDATEDATASETRESPONSE._serialized_end=3241
368
+ _DELETEDATASETREQUEST._serialized_start=3243
369
+ _DELETEDATASETREQUEST._serialized_end=3351
370
+ _DELETEDATASETRESPONSE._serialized_start=3353
371
+ _DELETEDATASETRESPONSE._serialized_end=3393
372
+ _GETDATASETVERSIONSREQUEST._serialized_start=3395
373
+ _GETDATASETVERSIONSREQUEST._serialized_end=3508
374
+ _DATASETINFO._serialized_start=3511
375
+ _DATASETINFO._serialized_end=3705
376
+ _DATASETVERSIONINFO._serialized_start=3708
377
+ _DATASETVERSIONINFO._serialized_end=3846
378
+ _GETDATASETVERSIONSRESPONSE._serialized_start=3848
379
+ _GETDATASETVERSIONSRESPONSE._serialized_end=3922
380
+ _CREATEEXPERIMENTREQUEST._serialized_start=3924
381
+ _CREATEEXPERIMENTREQUEST._serialized_end=4012
382
+ _CREATEEXPERIMENTRESPONSE._serialized_start=4014
383
+ _CREATEEXPERIMENTRESPONSE._serialized_end=4063
384
+ _GETEXPERIMENTREQUEST._serialized_start=4065
385
+ _GETEXPERIMENTREQUEST._serialized_end=4175
386
+ _POSTEXPERIMENTDATAREQUEST._serialized_start=4177
387
+ _POSTEXPERIMENTDATAREQUEST._serialized_end=4267
388
+ _POSTEXPERIMENTDATARESPONSE._serialized_start=4269
389
+ _POSTEXPERIMENTDATARESPONSE._serialized_end=4340
390
+ _DELETEEXPERIMENTREQUEST._serialized_start=4343
391
+ _DELETEEXPERIMENTREQUEST._serialized_end=4508
392
+ _EXPERIMENTNAMEIDENTIFIER._serialized_start=4510
393
+ _EXPERIMENTNAMEIDENTIFIER._serialized_end=4583
394
+ _DELETEEXPERIMENTRESPONSE._serialized_start=4585
395
+ _DELETEEXPERIMENTRESPONSE._serialized_end=4634
396
+ _CREATEEXPERIMENTDBENTRYREQUEST._serialized_start=4637
397
+ _CREATEEXPERIMENTDBENTRYREQUEST._serialized_end=4806
398
+ _CREATEEXPERIMENTDBENTRYRESPONSE._serialized_start=4808
399
+ _CREATEEXPERIMENTDBENTRYRESPONSE._serialized_end=4910
400
+ # @@protoc_insertion_point(module_scope)