aeri-python 4.0.0__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 (391) hide show
  1. aeri/__init__.py +72 -0
  2. aeri/_client/_validation.py +204 -0
  3. aeri/_client/attributes.py +188 -0
  4. aeri/_client/client.py +3761 -0
  5. aeri/_client/constants.py +65 -0
  6. aeri/_client/datasets.py +302 -0
  7. aeri/_client/environment_variables.py +158 -0
  8. aeri/_client/get_client.py +149 -0
  9. aeri/_client/observe.py +661 -0
  10. aeri/_client/propagation.py +475 -0
  11. aeri/_client/resource_manager.py +510 -0
  12. aeri/_client/span.py +1519 -0
  13. aeri/_client/span_filter.py +76 -0
  14. aeri/_client/span_processor.py +206 -0
  15. aeri/_client/utils.py +132 -0
  16. aeri/_task_manager/media_manager.py +331 -0
  17. aeri/_task_manager/media_upload_consumer.py +44 -0
  18. aeri/_task_manager/media_upload_queue.py +12 -0
  19. aeri/_task_manager/score_ingestion_consumer.py +208 -0
  20. aeri/_task_manager/task_manager.py +475 -0
  21. aeri/_utils/__init__.py +19 -0
  22. aeri/_utils/environment.py +34 -0
  23. aeri/_utils/error_logging.py +47 -0
  24. aeri/_utils/parse_error.py +99 -0
  25. aeri/_utils/prompt_cache.py +188 -0
  26. aeri/_utils/request.py +137 -0
  27. aeri/_utils/serializer.py +205 -0
  28. aeri/api/.fern/metadata.json +14 -0
  29. aeri/api/__init__.py +836 -0
  30. aeri/api/annotation_queues/__init__.py +82 -0
  31. aeri/api/annotation_queues/client.py +1111 -0
  32. aeri/api/annotation_queues/raw_client.py +2288 -0
  33. aeri/api/annotation_queues/types/__init__.py +84 -0
  34. aeri/api/annotation_queues/types/annotation_queue.py +28 -0
  35. aeri/api/annotation_queues/types/annotation_queue_assignment_request.py +16 -0
  36. aeri/api/annotation_queues/types/annotation_queue_item.py +34 -0
  37. aeri/api/annotation_queues/types/annotation_queue_object_type.py +26 -0
  38. aeri/api/annotation_queues/types/annotation_queue_status.py +22 -0
  39. aeri/api/annotation_queues/types/create_annotation_queue_assignment_response.py +18 -0
  40. aeri/api/annotation_queues/types/create_annotation_queue_item_request.py +25 -0
  41. aeri/api/annotation_queues/types/create_annotation_queue_request.py +20 -0
  42. aeri/api/annotation_queues/types/delete_annotation_queue_assignment_response.py +14 -0
  43. aeri/api/annotation_queues/types/delete_annotation_queue_item_response.py +15 -0
  44. aeri/api/annotation_queues/types/paginated_annotation_queue_items.py +17 -0
  45. aeri/api/annotation_queues/types/paginated_annotation_queues.py +17 -0
  46. aeri/api/annotation_queues/types/update_annotation_queue_item_request.py +15 -0
  47. aeri/api/blob_storage_integrations/__init__.py +73 -0
  48. aeri/api/blob_storage_integrations/client.py +550 -0
  49. aeri/api/blob_storage_integrations/raw_client.py +976 -0
  50. aeri/api/blob_storage_integrations/types/__init__.py +77 -0
  51. aeri/api/blob_storage_integrations/types/blob_storage_export_frequency.py +26 -0
  52. aeri/api/blob_storage_integrations/types/blob_storage_export_mode.py +26 -0
  53. aeri/api/blob_storage_integrations/types/blob_storage_integration_deletion_response.py +14 -0
  54. aeri/api/blob_storage_integrations/types/blob_storage_integration_file_type.py +26 -0
  55. aeri/api/blob_storage_integrations/types/blob_storage_integration_response.py +64 -0
  56. aeri/api/blob_storage_integrations/types/blob_storage_integration_status_response.py +50 -0
  57. aeri/api/blob_storage_integrations/types/blob_storage_integration_type.py +26 -0
  58. aeri/api/blob_storage_integrations/types/blob_storage_integrations_response.py +15 -0
  59. aeri/api/blob_storage_integrations/types/blob_storage_sync_status.py +47 -0
  60. aeri/api/blob_storage_integrations/types/create_blob_storage_integration_request.py +91 -0
  61. aeri/api/client.py +679 -0
  62. aeri/api/comments/__init__.py +44 -0
  63. aeri/api/comments/client.py +407 -0
  64. aeri/api/comments/raw_client.py +750 -0
  65. aeri/api/comments/types/__init__.py +46 -0
  66. aeri/api/comments/types/create_comment_request.py +47 -0
  67. aeri/api/comments/types/create_comment_response.py +17 -0
  68. aeri/api/comments/types/get_comments_response.py +17 -0
  69. aeri/api/commons/__init__.py +210 -0
  70. aeri/api/commons/errors/__init__.py +56 -0
  71. aeri/api/commons/errors/access_denied_error.py +12 -0
  72. aeri/api/commons/errors/error.py +12 -0
  73. aeri/api/commons/errors/method_not_allowed_error.py +12 -0
  74. aeri/api/commons/errors/not_found_error.py +12 -0
  75. aeri/api/commons/errors/unauthorized_error.py +12 -0
  76. aeri/api/commons/types/__init__.py +190 -0
  77. aeri/api/commons/types/base_score.py +90 -0
  78. aeri/api/commons/types/base_score_v1.py +70 -0
  79. aeri/api/commons/types/boolean_score.py +26 -0
  80. aeri/api/commons/types/boolean_score_v1.py +26 -0
  81. aeri/api/commons/types/categorical_score.py +26 -0
  82. aeri/api/commons/types/categorical_score_v1.py +26 -0
  83. aeri/api/commons/types/comment.py +36 -0
  84. aeri/api/commons/types/comment_object_type.py +30 -0
  85. aeri/api/commons/types/config_category.py +15 -0
  86. aeri/api/commons/types/correction_score.py +26 -0
  87. aeri/api/commons/types/create_score_value.py +5 -0
  88. aeri/api/commons/types/dataset.py +49 -0
  89. aeri/api/commons/types/dataset_item.py +58 -0
  90. aeri/api/commons/types/dataset_run.py +63 -0
  91. aeri/api/commons/types/dataset_run_item.py +40 -0
  92. aeri/api/commons/types/dataset_run_with_items.py +19 -0
  93. aeri/api/commons/types/dataset_status.py +22 -0
  94. aeri/api/commons/types/map_value.py +11 -0
  95. aeri/api/commons/types/model.py +125 -0
  96. aeri/api/commons/types/model_price.py +14 -0
  97. aeri/api/commons/types/model_usage_unit.py +42 -0
  98. aeri/api/commons/types/numeric_score.py +17 -0
  99. aeri/api/commons/types/numeric_score_v1.py +17 -0
  100. aeri/api/commons/types/observation.py +142 -0
  101. aeri/api/commons/types/observation_level.py +30 -0
  102. aeri/api/commons/types/observation_v2.py +235 -0
  103. aeri/api/commons/types/observations_view.py +89 -0
  104. aeri/api/commons/types/pricing_tier.py +91 -0
  105. aeri/api/commons/types/pricing_tier_condition.py +68 -0
  106. aeri/api/commons/types/pricing_tier_input.py +76 -0
  107. aeri/api/commons/types/pricing_tier_operator.py +42 -0
  108. aeri/api/commons/types/score.py +201 -0
  109. aeri/api/commons/types/score_config.py +66 -0
  110. aeri/api/commons/types/score_config_data_type.py +26 -0
  111. aeri/api/commons/types/score_data_type.py +30 -0
  112. aeri/api/commons/types/score_source.py +26 -0
  113. aeri/api/commons/types/score_v1.py +131 -0
  114. aeri/api/commons/types/session.py +25 -0
  115. aeri/api/commons/types/session_with_traces.py +15 -0
  116. aeri/api/commons/types/trace.py +84 -0
  117. aeri/api/commons/types/trace_with_details.py +43 -0
  118. aeri/api/commons/types/trace_with_full_details.py +45 -0
  119. aeri/api/commons/types/usage.py +59 -0
  120. aeri/api/core/__init__.py +111 -0
  121. aeri/api/core/api_error.py +23 -0
  122. aeri/api/core/client_wrapper.py +141 -0
  123. aeri/api/core/datetime_utils.py +30 -0
  124. aeri/api/core/enum.py +20 -0
  125. aeri/api/core/file.py +70 -0
  126. aeri/api/core/force_multipart.py +18 -0
  127. aeri/api/core/http_client.py +711 -0
  128. aeri/api/core/http_response.py +55 -0
  129. aeri/api/core/http_sse/__init__.py +48 -0
  130. aeri/api/core/http_sse/_api.py +114 -0
  131. aeri/api/core/http_sse/_decoders.py +66 -0
  132. aeri/api/core/http_sse/_exceptions.py +7 -0
  133. aeri/api/core/http_sse/_models.py +17 -0
  134. aeri/api/core/jsonable_encoder.py +102 -0
  135. aeri/api/core/pydantic_utilities.py +310 -0
  136. aeri/api/core/query_encoder.py +60 -0
  137. aeri/api/core/remove_none_from_dict.py +11 -0
  138. aeri/api/core/request_options.py +35 -0
  139. aeri/api/core/serialization.py +282 -0
  140. aeri/api/dataset_items/__init__.py +52 -0
  141. aeri/api/dataset_items/client.py +499 -0
  142. aeri/api/dataset_items/raw_client.py +973 -0
  143. aeri/api/dataset_items/types/__init__.py +50 -0
  144. aeri/api/dataset_items/types/create_dataset_item_request.py +37 -0
  145. aeri/api/dataset_items/types/delete_dataset_item_response.py +17 -0
  146. aeri/api/dataset_items/types/paginated_dataset_items.py +17 -0
  147. aeri/api/dataset_run_items/__init__.py +43 -0
  148. aeri/api/dataset_run_items/client.py +323 -0
  149. aeri/api/dataset_run_items/raw_client.py +547 -0
  150. aeri/api/dataset_run_items/types/__init__.py +44 -0
  151. aeri/api/dataset_run_items/types/create_dataset_run_item_request.py +51 -0
  152. aeri/api/dataset_run_items/types/paginated_dataset_run_items.py +17 -0
  153. aeri/api/datasets/__init__.py +55 -0
  154. aeri/api/datasets/client.py +661 -0
  155. aeri/api/datasets/raw_client.py +1368 -0
  156. aeri/api/datasets/types/__init__.py +53 -0
  157. aeri/api/datasets/types/create_dataset_request.py +31 -0
  158. aeri/api/datasets/types/delete_dataset_run_response.py +14 -0
  159. aeri/api/datasets/types/paginated_dataset_runs.py +17 -0
  160. aeri/api/datasets/types/paginated_datasets.py +17 -0
  161. aeri/api/health/__init__.py +44 -0
  162. aeri/api/health/client.py +112 -0
  163. aeri/api/health/errors/__init__.py +42 -0
  164. aeri/api/health/errors/service_unavailable_error.py +13 -0
  165. aeri/api/health/raw_client.py +227 -0
  166. aeri/api/health/types/__init__.py +40 -0
  167. aeri/api/health/types/health_response.py +30 -0
  168. aeri/api/ingestion/__init__.py +169 -0
  169. aeri/api/ingestion/client.py +221 -0
  170. aeri/api/ingestion/raw_client.py +293 -0
  171. aeri/api/ingestion/types/__init__.py +169 -0
  172. aeri/api/ingestion/types/base_event.py +27 -0
  173. aeri/api/ingestion/types/create_event_body.py +14 -0
  174. aeri/api/ingestion/types/create_event_event.py +15 -0
  175. aeri/api/ingestion/types/create_generation_body.py +40 -0
  176. aeri/api/ingestion/types/create_generation_event.py +15 -0
  177. aeri/api/ingestion/types/create_observation_event.py +15 -0
  178. aeri/api/ingestion/types/create_span_body.py +19 -0
  179. aeri/api/ingestion/types/create_span_event.py +15 -0
  180. aeri/api/ingestion/types/ingestion_error.py +17 -0
  181. aeri/api/ingestion/types/ingestion_event.py +155 -0
  182. aeri/api/ingestion/types/ingestion_response.py +17 -0
  183. aeri/api/ingestion/types/ingestion_success.py +15 -0
  184. aeri/api/ingestion/types/ingestion_usage.py +8 -0
  185. aeri/api/ingestion/types/observation_body.py +53 -0
  186. aeri/api/ingestion/types/observation_type.py +54 -0
  187. aeri/api/ingestion/types/open_ai_completion_usage_schema.py +26 -0
  188. aeri/api/ingestion/types/open_ai_response_usage_schema.py +24 -0
  189. aeri/api/ingestion/types/open_ai_usage.py +28 -0
  190. aeri/api/ingestion/types/optional_observation_body.py +36 -0
  191. aeri/api/ingestion/types/score_body.py +75 -0
  192. aeri/api/ingestion/types/score_event.py +15 -0
  193. aeri/api/ingestion/types/sdk_log_body.py +14 -0
  194. aeri/api/ingestion/types/sdk_log_event.py +15 -0
  195. aeri/api/ingestion/types/trace_body.py +36 -0
  196. aeri/api/ingestion/types/trace_event.py +15 -0
  197. aeri/api/ingestion/types/update_event_body.py +14 -0
  198. aeri/api/ingestion/types/update_generation_body.py +40 -0
  199. aeri/api/ingestion/types/update_generation_event.py +15 -0
  200. aeri/api/ingestion/types/update_observation_event.py +15 -0
  201. aeri/api/ingestion/types/update_span_body.py +19 -0
  202. aeri/api/ingestion/types/update_span_event.py +15 -0
  203. aeri/api/ingestion/types/usage_details.py +10 -0
  204. aeri/api/legacy/__init__.py +61 -0
  205. aeri/api/legacy/client.py +105 -0
  206. aeri/api/legacy/metrics_v1/__init__.py +40 -0
  207. aeri/api/legacy/metrics_v1/client.py +214 -0
  208. aeri/api/legacy/metrics_v1/raw_client.py +322 -0
  209. aeri/api/legacy/metrics_v1/types/__init__.py +40 -0
  210. aeri/api/legacy/metrics_v1/types/metrics_response.py +19 -0
  211. aeri/api/legacy/observations_v1/__init__.py +43 -0
  212. aeri/api/legacy/observations_v1/client.py +523 -0
  213. aeri/api/legacy/observations_v1/raw_client.py +759 -0
  214. aeri/api/legacy/observations_v1/types/__init__.py +44 -0
  215. aeri/api/legacy/observations_v1/types/observations.py +17 -0
  216. aeri/api/legacy/observations_v1/types/observations_views.py +17 -0
  217. aeri/api/legacy/raw_client.py +13 -0
  218. aeri/api/legacy/score_v1/__init__.py +43 -0
  219. aeri/api/legacy/score_v1/client.py +329 -0
  220. aeri/api/legacy/score_v1/raw_client.py +545 -0
  221. aeri/api/legacy/score_v1/types/__init__.py +44 -0
  222. aeri/api/legacy/score_v1/types/create_score_request.py +75 -0
  223. aeri/api/legacy/score_v1/types/create_score_response.py +17 -0
  224. aeri/api/llm_connections/__init__.py +55 -0
  225. aeri/api/llm_connections/client.py +311 -0
  226. aeri/api/llm_connections/raw_client.py +541 -0
  227. aeri/api/llm_connections/types/__init__.py +53 -0
  228. aeri/api/llm_connections/types/llm_adapter.py +38 -0
  229. aeri/api/llm_connections/types/llm_connection.py +77 -0
  230. aeri/api/llm_connections/types/paginated_llm_connections.py +17 -0
  231. aeri/api/llm_connections/types/upsert_llm_connection_request.py +69 -0
  232. aeri/api/media/__init__.py +58 -0
  233. aeri/api/media/client.py +427 -0
  234. aeri/api/media/raw_client.py +739 -0
  235. aeri/api/media/types/__init__.py +56 -0
  236. aeri/api/media/types/get_media_response.py +55 -0
  237. aeri/api/media/types/get_media_upload_url_request.py +51 -0
  238. aeri/api/media/types/get_media_upload_url_response.py +28 -0
  239. aeri/api/media/types/media_content_type.py +232 -0
  240. aeri/api/media/types/patch_media_body.py +43 -0
  241. aeri/api/metrics/__init__.py +40 -0
  242. aeri/api/metrics/client.py +422 -0
  243. aeri/api/metrics/raw_client.py +530 -0
  244. aeri/api/metrics/types/__init__.py +40 -0
  245. aeri/api/metrics/types/metrics_v2response.py +19 -0
  246. aeri/api/models/__init__.py +43 -0
  247. aeri/api/models/client.py +523 -0
  248. aeri/api/models/raw_client.py +993 -0
  249. aeri/api/models/types/__init__.py +44 -0
  250. aeri/api/models/types/create_model_request.py +103 -0
  251. aeri/api/models/types/paginated_models.py +17 -0
  252. aeri/api/observations/__init__.py +43 -0
  253. aeri/api/observations/client.py +522 -0
  254. aeri/api/observations/raw_client.py +641 -0
  255. aeri/api/observations/types/__init__.py +44 -0
  256. aeri/api/observations/types/observations_v2meta.py +21 -0
  257. aeri/api/observations/types/observations_v2response.py +28 -0
  258. aeri/api/opentelemetry/__init__.py +67 -0
  259. aeri/api/opentelemetry/client.py +276 -0
  260. aeri/api/opentelemetry/raw_client.py +291 -0
  261. aeri/api/opentelemetry/types/__init__.py +65 -0
  262. aeri/api/opentelemetry/types/otel_attribute.py +27 -0
  263. aeri/api/opentelemetry/types/otel_attribute_value.py +46 -0
  264. aeri/api/opentelemetry/types/otel_resource.py +24 -0
  265. aeri/api/opentelemetry/types/otel_resource_span.py +32 -0
  266. aeri/api/opentelemetry/types/otel_scope.py +34 -0
  267. aeri/api/opentelemetry/types/otel_scope_span.py +28 -0
  268. aeri/api/opentelemetry/types/otel_span.py +76 -0
  269. aeri/api/opentelemetry/types/otel_trace_response.py +16 -0
  270. aeri/api/organizations/__init__.py +73 -0
  271. aeri/api/organizations/client.py +756 -0
  272. aeri/api/organizations/raw_client.py +1707 -0
  273. aeri/api/organizations/types/__init__.py +71 -0
  274. aeri/api/organizations/types/delete_membership_request.py +16 -0
  275. aeri/api/organizations/types/membership_deletion_response.py +17 -0
  276. aeri/api/organizations/types/membership_request.py +18 -0
  277. aeri/api/organizations/types/membership_response.py +20 -0
  278. aeri/api/organizations/types/membership_role.py +30 -0
  279. aeri/api/organizations/types/memberships_response.py +15 -0
  280. aeri/api/organizations/types/organization_api_key.py +31 -0
  281. aeri/api/organizations/types/organization_api_keys_response.py +19 -0
  282. aeri/api/organizations/types/organization_project.py +25 -0
  283. aeri/api/organizations/types/organization_projects_response.py +15 -0
  284. aeri/api/projects/__init__.py +67 -0
  285. aeri/api/projects/client.py +760 -0
  286. aeri/api/projects/raw_client.py +1577 -0
  287. aeri/api/projects/types/__init__.py +65 -0
  288. aeri/api/projects/types/api_key_deletion_response.py +18 -0
  289. aeri/api/projects/types/api_key_list.py +23 -0
  290. aeri/api/projects/types/api_key_response.py +30 -0
  291. aeri/api/projects/types/api_key_summary.py +35 -0
  292. aeri/api/projects/types/organization.py +22 -0
  293. aeri/api/projects/types/project.py +34 -0
  294. aeri/api/projects/types/project_deletion_response.py +15 -0
  295. aeri/api/projects/types/projects.py +15 -0
  296. aeri/api/prompt_version/__init__.py +4 -0
  297. aeri/api/prompt_version/client.py +157 -0
  298. aeri/api/prompt_version/raw_client.py +264 -0
  299. aeri/api/prompts/__init__.py +100 -0
  300. aeri/api/prompts/client.py +550 -0
  301. aeri/api/prompts/raw_client.py +987 -0
  302. aeri/api/prompts/types/__init__.py +96 -0
  303. aeri/api/prompts/types/base_prompt.py +42 -0
  304. aeri/api/prompts/types/chat_message.py +17 -0
  305. aeri/api/prompts/types/chat_message_type.py +15 -0
  306. aeri/api/prompts/types/chat_message_with_placeholders.py +8 -0
  307. aeri/api/prompts/types/chat_prompt.py +15 -0
  308. aeri/api/prompts/types/create_chat_prompt_request.py +37 -0
  309. aeri/api/prompts/types/create_chat_prompt_type.py +15 -0
  310. aeri/api/prompts/types/create_prompt_request.py +8 -0
  311. aeri/api/prompts/types/create_text_prompt_request.py +36 -0
  312. aeri/api/prompts/types/create_text_prompt_type.py +15 -0
  313. aeri/api/prompts/types/placeholder_message.py +16 -0
  314. aeri/api/prompts/types/placeholder_message_type.py +15 -0
  315. aeri/api/prompts/types/prompt.py +58 -0
  316. aeri/api/prompts/types/prompt_meta.py +35 -0
  317. aeri/api/prompts/types/prompt_meta_list_response.py +17 -0
  318. aeri/api/prompts/types/prompt_type.py +20 -0
  319. aeri/api/prompts/types/text_prompt.py +14 -0
  320. aeri/api/scim/__init__.py +94 -0
  321. aeri/api/scim/client.py +686 -0
  322. aeri/api/scim/raw_client.py +1528 -0
  323. aeri/api/scim/types/__init__.py +92 -0
  324. aeri/api/scim/types/authentication_scheme.py +20 -0
  325. aeri/api/scim/types/bulk_config.py +22 -0
  326. aeri/api/scim/types/empty_response.py +16 -0
  327. aeri/api/scim/types/filter_config.py +17 -0
  328. aeri/api/scim/types/resource_meta.py +17 -0
  329. aeri/api/scim/types/resource_type.py +27 -0
  330. aeri/api/scim/types/resource_types_response.py +21 -0
  331. aeri/api/scim/types/schema_extension.py +17 -0
  332. aeri/api/scim/types/schema_resource.py +19 -0
  333. aeri/api/scim/types/schemas_response.py +21 -0
  334. aeri/api/scim/types/scim_email.py +16 -0
  335. aeri/api/scim/types/scim_feature_support.py +14 -0
  336. aeri/api/scim/types/scim_name.py +14 -0
  337. aeri/api/scim/types/scim_user.py +24 -0
  338. aeri/api/scim/types/scim_users_list_response.py +25 -0
  339. aeri/api/scim/types/service_provider_config.py +36 -0
  340. aeri/api/scim/types/user_meta.py +20 -0
  341. aeri/api/score_configs/__init__.py +44 -0
  342. aeri/api/score_configs/client.py +526 -0
  343. aeri/api/score_configs/raw_client.py +1012 -0
  344. aeri/api/score_configs/types/__init__.py +46 -0
  345. aeri/api/score_configs/types/create_score_config_request.py +46 -0
  346. aeri/api/score_configs/types/score_configs.py +17 -0
  347. aeri/api/score_configs/types/update_score_config_request.py +53 -0
  348. aeri/api/scores/__init__.py +76 -0
  349. aeri/api/scores/client.py +420 -0
  350. aeri/api/scores/raw_client.py +656 -0
  351. aeri/api/scores/types/__init__.py +76 -0
  352. aeri/api/scores/types/get_scores_response.py +17 -0
  353. aeri/api/scores/types/get_scores_response_data.py +211 -0
  354. aeri/api/scores/types/get_scores_response_data_boolean.py +15 -0
  355. aeri/api/scores/types/get_scores_response_data_categorical.py +15 -0
  356. aeri/api/scores/types/get_scores_response_data_correction.py +15 -0
  357. aeri/api/scores/types/get_scores_response_data_numeric.py +15 -0
  358. aeri/api/scores/types/get_scores_response_trace_data.py +38 -0
  359. aeri/api/sessions/__init__.py +40 -0
  360. aeri/api/sessions/client.py +262 -0
  361. aeri/api/sessions/raw_client.py +500 -0
  362. aeri/api/sessions/types/__init__.py +40 -0
  363. aeri/api/sessions/types/paginated_sessions.py +17 -0
  364. aeri/api/trace/__init__.py +44 -0
  365. aeri/api/trace/client.py +728 -0
  366. aeri/api/trace/raw_client.py +1208 -0
  367. aeri/api/trace/types/__init__.py +46 -0
  368. aeri/api/trace/types/delete_trace_response.py +14 -0
  369. aeri/api/trace/types/sort.py +14 -0
  370. aeri/api/trace/types/traces.py +17 -0
  371. aeri/api/utils/__init__.py +44 -0
  372. aeri/api/utils/pagination/__init__.py +40 -0
  373. aeri/api/utils/pagination/types/__init__.py +40 -0
  374. aeri/api/utils/pagination/types/meta_response.py +38 -0
  375. aeri/batch_evaluation.py +1643 -0
  376. aeri/experiment.py +1044 -0
  377. aeri/langchain/CallbackHandler.py +1377 -0
  378. aeri/langchain/__init__.py +5 -0
  379. aeri/langchain/utils.py +212 -0
  380. aeri/logger.py +28 -0
  381. aeri/media.py +352 -0
  382. aeri/model.py +477 -0
  383. aeri/openai.py +1124 -0
  384. aeri/py.typed +0 -0
  385. aeri/span_filter.py +17 -0
  386. aeri/types.py +79 -0
  387. aeri/version.py +3 -0
  388. aeri_python-4.0.0.dist-info/METADATA +51 -0
  389. aeri_python-4.0.0.dist-info/RECORD +391 -0
  390. aeri_python-4.0.0.dist-info/WHEEL +4 -0
  391. aeri_python-4.0.0.dist-info/licenses/LICENSE +21 -0
aeri/api/__init__.py ADDED
@@ -0,0 +1,836 @@
1
+ # This file was auto-generated by Fern from our API Definition.
2
+
3
+ # isort: skip_file
4
+
5
+ import typing
6
+ from importlib import import_module
7
+
8
+ if typing.TYPE_CHECKING:
9
+ from . import (
10
+ annotation_queues,
11
+ blob_storage_integrations,
12
+ comments,
13
+ commons,
14
+ dataset_items,
15
+ dataset_run_items,
16
+ datasets,
17
+ health,
18
+ ingestion,
19
+ legacy,
20
+ llm_connections,
21
+ media,
22
+ metrics,
23
+ models,
24
+ observations,
25
+ opentelemetry,
26
+ organizations,
27
+ projects,
28
+ prompt_version,
29
+ prompts,
30
+ scim,
31
+ score_configs,
32
+ scores,
33
+ sessions,
34
+ trace,
35
+ utils,
36
+ )
37
+ from .annotation_queues import (
38
+ AnnotationQueue,
39
+ AnnotationQueueAssignmentRequest,
40
+ AnnotationQueueItem,
41
+ AnnotationQueueObjectType,
42
+ AnnotationQueueStatus,
43
+ CreateAnnotationQueueAssignmentResponse,
44
+ CreateAnnotationQueueItemRequest,
45
+ CreateAnnotationQueueRequest,
46
+ DeleteAnnotationQueueAssignmentResponse,
47
+ DeleteAnnotationQueueItemResponse,
48
+ PaginatedAnnotationQueueItems,
49
+ PaginatedAnnotationQueues,
50
+ UpdateAnnotationQueueItemRequest,
51
+ )
52
+ from .blob_storage_integrations import (
53
+ BlobStorageExportFrequency,
54
+ BlobStorageExportMode,
55
+ BlobStorageIntegrationDeletionResponse,
56
+ BlobStorageIntegrationFileType,
57
+ BlobStorageIntegrationResponse,
58
+ BlobStorageIntegrationStatusResponse,
59
+ BlobStorageIntegrationType,
60
+ BlobStorageIntegrationsResponse,
61
+ BlobStorageSyncStatus,
62
+ CreateBlobStorageIntegrationRequest,
63
+ )
64
+ from .client import AsyncAeriAPI, AeriAPI
65
+ from .comments import (
66
+ CreateCommentRequest,
67
+ CreateCommentResponse,
68
+ GetCommentsResponse,
69
+ )
70
+ from .commons import (
71
+ AccessDeniedError,
72
+ BaseScore,
73
+ BaseScoreV1,
74
+ BooleanScore,
75
+ BooleanScoreV1,
76
+ CategoricalScore,
77
+ CategoricalScoreV1,
78
+ Comment,
79
+ CommentObjectType,
80
+ ConfigCategory,
81
+ CorrectionScore,
82
+ CreateScoreValue,
83
+ Dataset,
84
+ DatasetItem,
85
+ DatasetRun,
86
+ DatasetRunItem,
87
+ DatasetRunWithItems,
88
+ DatasetStatus,
89
+ Error,
90
+ MapValue,
91
+ MethodNotAllowedError,
92
+ Model,
93
+ ModelPrice,
94
+ ModelUsageUnit,
95
+ NotFoundError,
96
+ NumericScore,
97
+ NumericScoreV1,
98
+ Observation,
99
+ ObservationLevel,
100
+ ObservationV2,
101
+ ObservationsView,
102
+ PricingTier,
103
+ PricingTierCondition,
104
+ PricingTierInput,
105
+ PricingTierOperator,
106
+ Score,
107
+ ScoreConfig,
108
+ ScoreConfigDataType,
109
+ ScoreDataType,
110
+ ScoreSource,
111
+ ScoreV1,
112
+ ScoreV1_Boolean,
113
+ ScoreV1_Categorical,
114
+ ScoreV1_Numeric,
115
+ Score_Boolean,
116
+ Score_Categorical,
117
+ Score_Correction,
118
+ Score_Numeric,
119
+ Session,
120
+ SessionWithTraces,
121
+ Trace,
122
+ TraceWithDetails,
123
+ TraceWithFullDetails,
124
+ UnauthorizedError,
125
+ Usage,
126
+ )
127
+ from .dataset_items import (
128
+ CreateDatasetItemRequest,
129
+ DeleteDatasetItemResponse,
130
+ PaginatedDatasetItems,
131
+ )
132
+ from .dataset_run_items import CreateDatasetRunItemRequest, PaginatedDatasetRunItems
133
+ from .datasets import (
134
+ CreateDatasetRequest,
135
+ DeleteDatasetRunResponse,
136
+ PaginatedDatasetRuns,
137
+ PaginatedDatasets,
138
+ )
139
+ from .health import HealthResponse, ServiceUnavailableError
140
+ from .ingestion import (
141
+ BaseEvent,
142
+ CreateEventBody,
143
+ CreateEventEvent,
144
+ CreateGenerationBody,
145
+ CreateGenerationEvent,
146
+ CreateObservationEvent,
147
+ CreateSpanBody,
148
+ CreateSpanEvent,
149
+ IngestionError,
150
+ IngestionEvent,
151
+ IngestionEvent_EventCreate,
152
+ IngestionEvent_GenerationCreate,
153
+ IngestionEvent_GenerationUpdate,
154
+ IngestionEvent_ObservationCreate,
155
+ IngestionEvent_ObservationUpdate,
156
+ IngestionEvent_ScoreCreate,
157
+ IngestionEvent_SdkLog,
158
+ IngestionEvent_SpanCreate,
159
+ IngestionEvent_SpanUpdate,
160
+ IngestionEvent_TraceCreate,
161
+ IngestionResponse,
162
+ IngestionSuccess,
163
+ IngestionUsage,
164
+ ObservationBody,
165
+ ObservationType,
166
+ OpenAiCompletionUsageSchema,
167
+ OpenAiResponseUsageSchema,
168
+ OpenAiUsage,
169
+ OptionalObservationBody,
170
+ ScoreBody,
171
+ ScoreEvent,
172
+ SdkLogBody,
173
+ SdkLogEvent,
174
+ TraceBody,
175
+ TraceEvent,
176
+ UpdateEventBody,
177
+ UpdateGenerationBody,
178
+ UpdateGenerationEvent,
179
+ UpdateObservationEvent,
180
+ UpdateSpanBody,
181
+ UpdateSpanEvent,
182
+ UsageDetails,
183
+ )
184
+ from .llm_connections import (
185
+ LlmAdapter,
186
+ LlmConnection,
187
+ PaginatedLlmConnections,
188
+ UpsertLlmConnectionRequest,
189
+ )
190
+ from .media import (
191
+ GetMediaResponse,
192
+ GetMediaUploadUrlRequest,
193
+ GetMediaUploadUrlResponse,
194
+ MediaContentType,
195
+ PatchMediaBody,
196
+ )
197
+ from .metrics import MetricsV2Response
198
+ from .models import CreateModelRequest, PaginatedModels
199
+ from .observations import ObservationsV2Meta, ObservationsV2Response
200
+ from .opentelemetry import (
201
+ OtelAttribute,
202
+ OtelAttributeValue,
203
+ OtelResource,
204
+ OtelResourceSpan,
205
+ OtelScope,
206
+ OtelScopeSpan,
207
+ OtelSpan,
208
+ OtelTraceResponse,
209
+ )
210
+ from .organizations import (
211
+ DeleteMembershipRequest,
212
+ MembershipDeletionResponse,
213
+ MembershipRequest,
214
+ MembershipResponse,
215
+ MembershipRole,
216
+ MembershipsResponse,
217
+ OrganizationApiKey,
218
+ OrganizationApiKeysResponse,
219
+ OrganizationProject,
220
+ OrganizationProjectsResponse,
221
+ )
222
+ from .projects import (
223
+ ApiKeyDeletionResponse,
224
+ ApiKeyList,
225
+ ApiKeyResponse,
226
+ ApiKeySummary,
227
+ Organization,
228
+ Project,
229
+ ProjectDeletionResponse,
230
+ Projects,
231
+ )
232
+ from .prompts import (
233
+ BasePrompt,
234
+ ChatMessage,
235
+ ChatMessageType,
236
+ ChatMessageWithPlaceholders,
237
+ ChatPrompt,
238
+ CreateChatPromptRequest,
239
+ CreateChatPromptType,
240
+ CreatePromptRequest,
241
+ CreateTextPromptRequest,
242
+ CreateTextPromptType,
243
+ PlaceholderMessage,
244
+ PlaceholderMessageType,
245
+ Prompt,
246
+ PromptMeta,
247
+ PromptMetaListResponse,
248
+ PromptType,
249
+ Prompt_Chat,
250
+ Prompt_Text,
251
+ TextPrompt,
252
+ )
253
+ from .scim import (
254
+ AuthenticationScheme,
255
+ BulkConfig,
256
+ EmptyResponse,
257
+ FilterConfig,
258
+ ResourceMeta,
259
+ ResourceType,
260
+ ResourceTypesResponse,
261
+ SchemaExtension,
262
+ SchemaResource,
263
+ SchemasResponse,
264
+ ScimEmail,
265
+ ScimFeatureSupport,
266
+ ScimName,
267
+ ScimUser,
268
+ ScimUsersListResponse,
269
+ ServiceProviderConfig,
270
+ UserMeta,
271
+ )
272
+ from .score_configs import (
273
+ CreateScoreConfigRequest,
274
+ ScoreConfigs,
275
+ UpdateScoreConfigRequest,
276
+ )
277
+ from .scores import (
278
+ GetScoresResponse,
279
+ GetScoresResponseData,
280
+ GetScoresResponseDataBoolean,
281
+ GetScoresResponseDataCategorical,
282
+ GetScoresResponseDataCorrection,
283
+ GetScoresResponseDataNumeric,
284
+ GetScoresResponseData_Boolean,
285
+ GetScoresResponseData_Categorical,
286
+ GetScoresResponseData_Correction,
287
+ GetScoresResponseData_Numeric,
288
+ GetScoresResponseTraceData,
289
+ )
290
+ from .sessions import PaginatedSessions
291
+ from .trace import DeleteTraceResponse, Sort, Traces
292
+ _dynamic_imports: typing.Dict[str, str] = {
293
+ "AccessDeniedError": ".commons",
294
+ "AnnotationQueue": ".annotation_queues",
295
+ "AnnotationQueueAssignmentRequest": ".annotation_queues",
296
+ "AnnotationQueueItem": ".annotation_queues",
297
+ "AnnotationQueueObjectType": ".annotation_queues",
298
+ "AnnotationQueueStatus": ".annotation_queues",
299
+ "ApiKeyDeletionResponse": ".projects",
300
+ "ApiKeyList": ".projects",
301
+ "ApiKeyResponse": ".projects",
302
+ "ApiKeySummary": ".projects",
303
+ "AsyncAeriAPI": ".client",
304
+ "AuthenticationScheme": ".scim",
305
+ "BaseEvent": ".ingestion",
306
+ "BasePrompt": ".prompts",
307
+ "BaseScore": ".commons",
308
+ "BaseScoreV1": ".commons",
309
+ "BlobStorageExportFrequency": ".blob_storage_integrations",
310
+ "BlobStorageExportMode": ".blob_storage_integrations",
311
+ "BlobStorageIntegrationDeletionResponse": ".blob_storage_integrations",
312
+ "BlobStorageIntegrationFileType": ".blob_storage_integrations",
313
+ "BlobStorageIntegrationResponse": ".blob_storage_integrations",
314
+ "BlobStorageIntegrationStatusResponse": ".blob_storage_integrations",
315
+ "BlobStorageIntegrationType": ".blob_storage_integrations",
316
+ "BlobStorageIntegrationsResponse": ".blob_storage_integrations",
317
+ "BlobStorageSyncStatus": ".blob_storage_integrations",
318
+ "BooleanScore": ".commons",
319
+ "BooleanScoreV1": ".commons",
320
+ "BulkConfig": ".scim",
321
+ "CategoricalScore": ".commons",
322
+ "CategoricalScoreV1": ".commons",
323
+ "ChatMessage": ".prompts",
324
+ "ChatMessageType": ".prompts",
325
+ "ChatMessageWithPlaceholders": ".prompts",
326
+ "ChatPrompt": ".prompts",
327
+ "Comment": ".commons",
328
+ "CommentObjectType": ".commons",
329
+ "ConfigCategory": ".commons",
330
+ "CorrectionScore": ".commons",
331
+ "CreateAnnotationQueueAssignmentResponse": ".annotation_queues",
332
+ "CreateAnnotationQueueItemRequest": ".annotation_queues",
333
+ "CreateAnnotationQueueRequest": ".annotation_queues",
334
+ "CreateBlobStorageIntegrationRequest": ".blob_storage_integrations",
335
+ "CreateChatPromptRequest": ".prompts",
336
+ "CreateChatPromptType": ".prompts",
337
+ "CreateCommentRequest": ".comments",
338
+ "CreateCommentResponse": ".comments",
339
+ "CreateDatasetItemRequest": ".dataset_items",
340
+ "CreateDatasetRequest": ".datasets",
341
+ "CreateDatasetRunItemRequest": ".dataset_run_items",
342
+ "CreateEventBody": ".ingestion",
343
+ "CreateEventEvent": ".ingestion",
344
+ "CreateGenerationBody": ".ingestion",
345
+ "CreateGenerationEvent": ".ingestion",
346
+ "CreateModelRequest": ".models",
347
+ "CreateObservationEvent": ".ingestion",
348
+ "CreatePromptRequest": ".prompts",
349
+ "CreateScoreConfigRequest": ".score_configs",
350
+ "CreateScoreValue": ".commons",
351
+ "CreateSpanBody": ".ingestion",
352
+ "CreateSpanEvent": ".ingestion",
353
+ "CreateTextPromptRequest": ".prompts",
354
+ "CreateTextPromptType": ".prompts",
355
+ "Dataset": ".commons",
356
+ "DatasetItem": ".commons",
357
+ "DatasetRun": ".commons",
358
+ "DatasetRunItem": ".commons",
359
+ "DatasetRunWithItems": ".commons",
360
+ "DatasetStatus": ".commons",
361
+ "DeleteAnnotationQueueAssignmentResponse": ".annotation_queues",
362
+ "DeleteAnnotationQueueItemResponse": ".annotation_queues",
363
+ "DeleteDatasetItemResponse": ".dataset_items",
364
+ "DeleteDatasetRunResponse": ".datasets",
365
+ "DeleteMembershipRequest": ".organizations",
366
+ "DeleteTraceResponse": ".trace",
367
+ "EmptyResponse": ".scim",
368
+ "Error": ".commons",
369
+ "FilterConfig": ".scim",
370
+ "GetCommentsResponse": ".comments",
371
+ "GetMediaResponse": ".media",
372
+ "GetMediaUploadUrlRequest": ".media",
373
+ "GetMediaUploadUrlResponse": ".media",
374
+ "GetScoresResponse": ".scores",
375
+ "GetScoresResponseData": ".scores",
376
+ "GetScoresResponseDataBoolean": ".scores",
377
+ "GetScoresResponseDataCategorical": ".scores",
378
+ "GetScoresResponseDataCorrection": ".scores",
379
+ "GetScoresResponseDataNumeric": ".scores",
380
+ "GetScoresResponseData_Boolean": ".scores",
381
+ "GetScoresResponseData_Categorical": ".scores",
382
+ "GetScoresResponseData_Correction": ".scores",
383
+ "GetScoresResponseData_Numeric": ".scores",
384
+ "GetScoresResponseTraceData": ".scores",
385
+ "HealthResponse": ".health",
386
+ "IngestionError": ".ingestion",
387
+ "IngestionEvent": ".ingestion",
388
+ "IngestionEvent_EventCreate": ".ingestion",
389
+ "IngestionEvent_GenerationCreate": ".ingestion",
390
+ "IngestionEvent_GenerationUpdate": ".ingestion",
391
+ "IngestionEvent_ObservationCreate": ".ingestion",
392
+ "IngestionEvent_ObservationUpdate": ".ingestion",
393
+ "IngestionEvent_ScoreCreate": ".ingestion",
394
+ "IngestionEvent_SdkLog": ".ingestion",
395
+ "IngestionEvent_SpanCreate": ".ingestion",
396
+ "IngestionEvent_SpanUpdate": ".ingestion",
397
+ "IngestionEvent_TraceCreate": ".ingestion",
398
+ "IngestionResponse": ".ingestion",
399
+ "IngestionSuccess": ".ingestion",
400
+ "IngestionUsage": ".ingestion",
401
+ "AeriAPI": ".client",
402
+ "LlmAdapter": ".llm_connections",
403
+ "LlmConnection": ".llm_connections",
404
+ "MapValue": ".commons",
405
+ "MediaContentType": ".media",
406
+ "MembershipDeletionResponse": ".organizations",
407
+ "MembershipRequest": ".organizations",
408
+ "MembershipResponse": ".organizations",
409
+ "MembershipRole": ".organizations",
410
+ "MembershipsResponse": ".organizations",
411
+ "MethodNotAllowedError": ".commons",
412
+ "MetricsV2Response": ".metrics",
413
+ "Model": ".commons",
414
+ "ModelPrice": ".commons",
415
+ "ModelUsageUnit": ".commons",
416
+ "NotFoundError": ".commons",
417
+ "NumericScore": ".commons",
418
+ "NumericScoreV1": ".commons",
419
+ "Observation": ".commons",
420
+ "ObservationBody": ".ingestion",
421
+ "ObservationLevel": ".commons",
422
+ "ObservationType": ".ingestion",
423
+ "ObservationV2": ".commons",
424
+ "ObservationsV2Meta": ".observations",
425
+ "ObservationsV2Response": ".observations",
426
+ "ObservationsView": ".commons",
427
+ "OpenAiCompletionUsageSchema": ".ingestion",
428
+ "OpenAiResponseUsageSchema": ".ingestion",
429
+ "OpenAiUsage": ".ingestion",
430
+ "OptionalObservationBody": ".ingestion",
431
+ "Organization": ".projects",
432
+ "OrganizationApiKey": ".organizations",
433
+ "OrganizationApiKeysResponse": ".organizations",
434
+ "OrganizationProject": ".organizations",
435
+ "OrganizationProjectsResponse": ".organizations",
436
+ "OtelAttribute": ".opentelemetry",
437
+ "OtelAttributeValue": ".opentelemetry",
438
+ "OtelResource": ".opentelemetry",
439
+ "OtelResourceSpan": ".opentelemetry",
440
+ "OtelScope": ".opentelemetry",
441
+ "OtelScopeSpan": ".opentelemetry",
442
+ "OtelSpan": ".opentelemetry",
443
+ "OtelTraceResponse": ".opentelemetry",
444
+ "PaginatedAnnotationQueueItems": ".annotation_queues",
445
+ "PaginatedAnnotationQueues": ".annotation_queues",
446
+ "PaginatedDatasetItems": ".dataset_items",
447
+ "PaginatedDatasetRunItems": ".dataset_run_items",
448
+ "PaginatedDatasetRuns": ".datasets",
449
+ "PaginatedDatasets": ".datasets",
450
+ "PaginatedLlmConnections": ".llm_connections",
451
+ "PaginatedModels": ".models",
452
+ "PaginatedSessions": ".sessions",
453
+ "PatchMediaBody": ".media",
454
+ "PlaceholderMessage": ".prompts",
455
+ "PlaceholderMessageType": ".prompts",
456
+ "PricingTier": ".commons",
457
+ "PricingTierCondition": ".commons",
458
+ "PricingTierInput": ".commons",
459
+ "PricingTierOperator": ".commons",
460
+ "Project": ".projects",
461
+ "ProjectDeletionResponse": ".projects",
462
+ "Projects": ".projects",
463
+ "Prompt": ".prompts",
464
+ "PromptMeta": ".prompts",
465
+ "PromptMetaListResponse": ".prompts",
466
+ "PromptType": ".prompts",
467
+ "Prompt_Chat": ".prompts",
468
+ "Prompt_Text": ".prompts",
469
+ "ResourceMeta": ".scim",
470
+ "ResourceType": ".scim",
471
+ "ResourceTypesResponse": ".scim",
472
+ "SchemaExtension": ".scim",
473
+ "SchemaResource": ".scim",
474
+ "SchemasResponse": ".scim",
475
+ "ScimEmail": ".scim",
476
+ "ScimFeatureSupport": ".scim",
477
+ "ScimName": ".scim",
478
+ "ScimUser": ".scim",
479
+ "ScimUsersListResponse": ".scim",
480
+ "Score": ".commons",
481
+ "ScoreBody": ".ingestion",
482
+ "ScoreConfig": ".commons",
483
+ "ScoreConfigDataType": ".commons",
484
+ "ScoreConfigs": ".score_configs",
485
+ "ScoreDataType": ".commons",
486
+ "ScoreEvent": ".ingestion",
487
+ "ScoreSource": ".commons",
488
+ "ScoreV1": ".commons",
489
+ "ScoreV1_Boolean": ".commons",
490
+ "ScoreV1_Categorical": ".commons",
491
+ "ScoreV1_Numeric": ".commons",
492
+ "Score_Boolean": ".commons",
493
+ "Score_Categorical": ".commons",
494
+ "Score_Correction": ".commons",
495
+ "Score_Numeric": ".commons",
496
+ "SdkLogBody": ".ingestion",
497
+ "SdkLogEvent": ".ingestion",
498
+ "ServiceProviderConfig": ".scim",
499
+ "ServiceUnavailableError": ".health",
500
+ "Session": ".commons",
501
+ "SessionWithTraces": ".commons",
502
+ "Sort": ".trace",
503
+ "TextPrompt": ".prompts",
504
+ "Trace": ".commons",
505
+ "TraceBody": ".ingestion",
506
+ "TraceEvent": ".ingestion",
507
+ "TraceWithDetails": ".commons",
508
+ "TraceWithFullDetails": ".commons",
509
+ "Traces": ".trace",
510
+ "UnauthorizedError": ".commons",
511
+ "UpdateAnnotationQueueItemRequest": ".annotation_queues",
512
+ "UpdateEventBody": ".ingestion",
513
+ "UpdateGenerationBody": ".ingestion",
514
+ "UpdateGenerationEvent": ".ingestion",
515
+ "UpdateObservationEvent": ".ingestion",
516
+ "UpdateScoreConfigRequest": ".score_configs",
517
+ "UpdateSpanBody": ".ingestion",
518
+ "UpdateSpanEvent": ".ingestion",
519
+ "UpsertLlmConnectionRequest": ".llm_connections",
520
+ "Usage": ".commons",
521
+ "UsageDetails": ".ingestion",
522
+ "UserMeta": ".scim",
523
+ "annotation_queues": ".annotation_queues",
524
+ "blob_storage_integrations": ".blob_storage_integrations",
525
+ "comments": ".comments",
526
+ "commons": ".commons",
527
+ "dataset_items": ".dataset_items",
528
+ "dataset_run_items": ".dataset_run_items",
529
+ "datasets": ".datasets",
530
+ "health": ".health",
531
+ "ingestion": ".ingestion",
532
+ "legacy": ".legacy",
533
+ "llm_connections": ".llm_connections",
534
+ "media": ".media",
535
+ "metrics": ".metrics",
536
+ "models": ".models",
537
+ "observations": ".observations",
538
+ "opentelemetry": ".opentelemetry",
539
+ "organizations": ".organizations",
540
+ "projects": ".projects",
541
+ "prompt_version": ".prompt_version",
542
+ "prompts": ".prompts",
543
+ "scim": ".scim",
544
+ "score_configs": ".score_configs",
545
+ "scores": ".scores",
546
+ "sessions": ".sessions",
547
+ "trace": ".trace",
548
+ "utils": ".utils",
549
+ }
550
+
551
+
552
+ def __getattr__(attr_name: str) -> typing.Any:
553
+ module_name = _dynamic_imports.get(attr_name)
554
+ if module_name is None:
555
+ raise AttributeError(
556
+ f"No {attr_name} found in _dynamic_imports for module name -> {__name__}"
557
+ )
558
+ try:
559
+ module = import_module(module_name, __package__)
560
+ if module_name == f".{attr_name}":
561
+ return module
562
+ else:
563
+ return getattr(module, attr_name)
564
+ except ImportError as e:
565
+ raise ImportError(
566
+ f"Failed to import {attr_name} from {module_name}: {e}"
567
+ ) from e
568
+ except AttributeError as e:
569
+ raise AttributeError(
570
+ f"Failed to get {attr_name} from {module_name}: {e}"
571
+ ) from e
572
+
573
+
574
+ def __dir__():
575
+ lazy_attrs = list(_dynamic_imports.keys())
576
+ return sorted(lazy_attrs)
577
+
578
+
579
+ __all__ = [
580
+ "AccessDeniedError",
581
+ "AnnotationQueue",
582
+ "AnnotationQueueAssignmentRequest",
583
+ "AnnotationQueueItem",
584
+ "AnnotationQueueObjectType",
585
+ "AnnotationQueueStatus",
586
+ "ApiKeyDeletionResponse",
587
+ "ApiKeyList",
588
+ "ApiKeyResponse",
589
+ "ApiKeySummary",
590
+ "AsyncAeriAPI",
591
+ "AuthenticationScheme",
592
+ "BaseEvent",
593
+ "BasePrompt",
594
+ "BaseScore",
595
+ "BaseScoreV1",
596
+ "BlobStorageExportFrequency",
597
+ "BlobStorageExportMode",
598
+ "BlobStorageIntegrationDeletionResponse",
599
+ "BlobStorageIntegrationFileType",
600
+ "BlobStorageIntegrationResponse",
601
+ "BlobStorageIntegrationStatusResponse",
602
+ "BlobStorageIntegrationType",
603
+ "BlobStorageIntegrationsResponse",
604
+ "BlobStorageSyncStatus",
605
+ "BooleanScore",
606
+ "BooleanScoreV1",
607
+ "BulkConfig",
608
+ "CategoricalScore",
609
+ "CategoricalScoreV1",
610
+ "ChatMessage",
611
+ "ChatMessageType",
612
+ "ChatMessageWithPlaceholders",
613
+ "ChatPrompt",
614
+ "Comment",
615
+ "CommentObjectType",
616
+ "ConfigCategory",
617
+ "CorrectionScore",
618
+ "CreateAnnotationQueueAssignmentResponse",
619
+ "CreateAnnotationQueueItemRequest",
620
+ "CreateAnnotationQueueRequest",
621
+ "CreateBlobStorageIntegrationRequest",
622
+ "CreateChatPromptRequest",
623
+ "CreateChatPromptType",
624
+ "CreateCommentRequest",
625
+ "CreateCommentResponse",
626
+ "CreateDatasetItemRequest",
627
+ "CreateDatasetRequest",
628
+ "CreateDatasetRunItemRequest",
629
+ "CreateEventBody",
630
+ "CreateEventEvent",
631
+ "CreateGenerationBody",
632
+ "CreateGenerationEvent",
633
+ "CreateModelRequest",
634
+ "CreateObservationEvent",
635
+ "CreatePromptRequest",
636
+ "CreateScoreConfigRequest",
637
+ "CreateScoreValue",
638
+ "CreateSpanBody",
639
+ "CreateSpanEvent",
640
+ "CreateTextPromptRequest",
641
+ "CreateTextPromptType",
642
+ "Dataset",
643
+ "DatasetItem",
644
+ "DatasetRun",
645
+ "DatasetRunItem",
646
+ "DatasetRunWithItems",
647
+ "DatasetStatus",
648
+ "DeleteAnnotationQueueAssignmentResponse",
649
+ "DeleteAnnotationQueueItemResponse",
650
+ "DeleteDatasetItemResponse",
651
+ "DeleteDatasetRunResponse",
652
+ "DeleteMembershipRequest",
653
+ "DeleteTraceResponse",
654
+ "EmptyResponse",
655
+ "Error",
656
+ "FilterConfig",
657
+ "GetCommentsResponse",
658
+ "GetMediaResponse",
659
+ "GetMediaUploadUrlRequest",
660
+ "GetMediaUploadUrlResponse",
661
+ "GetScoresResponse",
662
+ "GetScoresResponseData",
663
+ "GetScoresResponseDataBoolean",
664
+ "GetScoresResponseDataCategorical",
665
+ "GetScoresResponseDataCorrection",
666
+ "GetScoresResponseDataNumeric",
667
+ "GetScoresResponseData_Boolean",
668
+ "GetScoresResponseData_Categorical",
669
+ "GetScoresResponseData_Correction",
670
+ "GetScoresResponseData_Numeric",
671
+ "GetScoresResponseTraceData",
672
+ "HealthResponse",
673
+ "IngestionError",
674
+ "IngestionEvent",
675
+ "IngestionEvent_EventCreate",
676
+ "IngestionEvent_GenerationCreate",
677
+ "IngestionEvent_GenerationUpdate",
678
+ "IngestionEvent_ObservationCreate",
679
+ "IngestionEvent_ObservationUpdate",
680
+ "IngestionEvent_ScoreCreate",
681
+ "IngestionEvent_SdkLog",
682
+ "IngestionEvent_SpanCreate",
683
+ "IngestionEvent_SpanUpdate",
684
+ "IngestionEvent_TraceCreate",
685
+ "IngestionResponse",
686
+ "IngestionSuccess",
687
+ "IngestionUsage",
688
+ "AeriAPI",
689
+ "LlmAdapter",
690
+ "LlmConnection",
691
+ "MapValue",
692
+ "MediaContentType",
693
+ "MembershipDeletionResponse",
694
+ "MembershipRequest",
695
+ "MembershipResponse",
696
+ "MembershipRole",
697
+ "MembershipsResponse",
698
+ "MethodNotAllowedError",
699
+ "MetricsV2Response",
700
+ "Model",
701
+ "ModelPrice",
702
+ "ModelUsageUnit",
703
+ "NotFoundError",
704
+ "NumericScore",
705
+ "NumericScoreV1",
706
+ "Observation",
707
+ "ObservationBody",
708
+ "ObservationLevel",
709
+ "ObservationType",
710
+ "ObservationV2",
711
+ "ObservationsV2Meta",
712
+ "ObservationsV2Response",
713
+ "ObservationsView",
714
+ "OpenAiCompletionUsageSchema",
715
+ "OpenAiResponseUsageSchema",
716
+ "OpenAiUsage",
717
+ "OptionalObservationBody",
718
+ "Organization",
719
+ "OrganizationApiKey",
720
+ "OrganizationApiKeysResponse",
721
+ "OrganizationProject",
722
+ "OrganizationProjectsResponse",
723
+ "OtelAttribute",
724
+ "OtelAttributeValue",
725
+ "OtelResource",
726
+ "OtelResourceSpan",
727
+ "OtelScope",
728
+ "OtelScopeSpan",
729
+ "OtelSpan",
730
+ "OtelTraceResponse",
731
+ "PaginatedAnnotationQueueItems",
732
+ "PaginatedAnnotationQueues",
733
+ "PaginatedDatasetItems",
734
+ "PaginatedDatasetRunItems",
735
+ "PaginatedDatasetRuns",
736
+ "PaginatedDatasets",
737
+ "PaginatedLlmConnections",
738
+ "PaginatedModels",
739
+ "PaginatedSessions",
740
+ "PatchMediaBody",
741
+ "PlaceholderMessage",
742
+ "PlaceholderMessageType",
743
+ "PricingTier",
744
+ "PricingTierCondition",
745
+ "PricingTierInput",
746
+ "PricingTierOperator",
747
+ "Project",
748
+ "ProjectDeletionResponse",
749
+ "Projects",
750
+ "Prompt",
751
+ "PromptMeta",
752
+ "PromptMetaListResponse",
753
+ "PromptType",
754
+ "Prompt_Chat",
755
+ "Prompt_Text",
756
+ "ResourceMeta",
757
+ "ResourceType",
758
+ "ResourceTypesResponse",
759
+ "SchemaExtension",
760
+ "SchemaResource",
761
+ "SchemasResponse",
762
+ "ScimEmail",
763
+ "ScimFeatureSupport",
764
+ "ScimName",
765
+ "ScimUser",
766
+ "ScimUsersListResponse",
767
+ "Score",
768
+ "ScoreBody",
769
+ "ScoreConfig",
770
+ "ScoreConfigDataType",
771
+ "ScoreConfigs",
772
+ "ScoreDataType",
773
+ "ScoreEvent",
774
+ "ScoreSource",
775
+ "ScoreV1",
776
+ "ScoreV1_Boolean",
777
+ "ScoreV1_Categorical",
778
+ "ScoreV1_Numeric",
779
+ "Score_Boolean",
780
+ "Score_Categorical",
781
+ "Score_Correction",
782
+ "Score_Numeric",
783
+ "SdkLogBody",
784
+ "SdkLogEvent",
785
+ "ServiceProviderConfig",
786
+ "ServiceUnavailableError",
787
+ "Session",
788
+ "SessionWithTraces",
789
+ "Sort",
790
+ "TextPrompt",
791
+ "Trace",
792
+ "TraceBody",
793
+ "TraceEvent",
794
+ "TraceWithDetails",
795
+ "TraceWithFullDetails",
796
+ "Traces",
797
+ "UnauthorizedError",
798
+ "UpdateAnnotationQueueItemRequest",
799
+ "UpdateEventBody",
800
+ "UpdateGenerationBody",
801
+ "UpdateGenerationEvent",
802
+ "UpdateObservationEvent",
803
+ "UpdateScoreConfigRequest",
804
+ "UpdateSpanBody",
805
+ "UpdateSpanEvent",
806
+ "UpsertLlmConnectionRequest",
807
+ "Usage",
808
+ "UsageDetails",
809
+ "UserMeta",
810
+ "annotation_queues",
811
+ "blob_storage_integrations",
812
+ "comments",
813
+ "commons",
814
+ "dataset_items",
815
+ "dataset_run_items",
816
+ "datasets",
817
+ "health",
818
+ "ingestion",
819
+ "legacy",
820
+ "llm_connections",
821
+ "media",
822
+ "metrics",
823
+ "models",
824
+ "observations",
825
+ "opentelemetry",
826
+ "organizations",
827
+ "projects",
828
+ "prompt_version",
829
+ "prompts",
830
+ "scim",
831
+ "score_configs",
832
+ "scores",
833
+ "sessions",
834
+ "trace",
835
+ "utils",
836
+ ]