antfly-sdk 0.0.2.dev2__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.
- antfly/.gitignore +23 -0
- antfly/README.md +124 -0
- antfly/__init__.py +20 -0
- antfly/client.py +260 -0
- antfly/client_generated/__init__.py +8 -0
- antfly/client_generated/api/__init__.py +1 -0
- antfly/client_generated/api/api_key/__init__.py +1 -0
- antfly/client_generated/api/api_key/create_api_key.py +209 -0
- antfly/client_generated/api/api_key/delete_api_key.py +182 -0
- antfly/client_generated/api/api_key/list_api_keys.py +182 -0
- antfly/client_generated/api/cluster_management/__init__.py +1 -0
- antfly/client_generated/api/cluster_management/backup.py +259 -0
- antfly/client_generated/api/cluster_management/delete_secret.py +177 -0
- antfly/client_generated/api/cluster_management/get_cluster.py +147 -0
- antfly/client_generated/api/cluster_management/get_status.py +145 -0
- antfly/client_generated/api/cluster_management/list_backups.py +187 -0
- antfly/client_generated/api/cluster_management/list_secrets.py +147 -0
- antfly/client_generated/api/cluster_management/put_secret.py +200 -0
- antfly/client_generated/api/cluster_management/restore.py +215 -0
- antfly/client_generated/api/data_operations/__init__.py +1 -0
- antfly/client_generated/api/data_operations/abort_transaction_session.py +176 -0
- antfly/client_generated/api/data_operations/backup_table.py +192 -0
- antfly/client_generated/api/data_operations/batch_write.py +318 -0
- antfly/client_generated/api/data_operations/begin_transaction.py +172 -0
- antfly/client_generated/api/data_operations/cleanup_transaction_sessions.py +171 -0
- antfly/client_generated/api/data_operations/commit_transaction.py +289 -0
- antfly/client_generated/api/data_operations/commit_transaction_session.py +234 -0
- antfly/client_generated/api/data_operations/create_transaction_savepoint.py +176 -0
- antfly/client_generated/api/data_operations/get_transaction_session.py +176 -0
- antfly/client_generated/api/data_operations/linear_merge.py +338 -0
- antfly/client_generated/api/data_operations/list_transaction_sessions.py +134 -0
- antfly/client_generated/api/data_operations/lookup_key.py +206 -0
- antfly/client_generated/api/data_operations/multi_batch_write.py +248 -0
- antfly/client_generated/api/data_operations/restore_table.py +187 -0
- antfly/client_generated/api/data_operations/rollback_transaction_savepoint.py +190 -0
- antfly/client_generated/api/data_operations/stage_transaction_delete.py +197 -0
- antfly/client_generated/api/data_operations/stage_transaction_read.py +198 -0
- antfly/client_generated/api/data_operations/stage_transaction_session.py +233 -0
- antfly/client_generated/api/data_operations/stage_transaction_write.py +197 -0
- antfly/client_generated/api/default/__init__.py +1 -0
- antfly/client_generated/api/default/chat_completions.py +193 -0
- antfly/client_generated/api/default/chunk_text.py +243 -0
- antfly/client_generated/api/default/create_embedding.py +204 -0
- antfly/client_generated/api/default/extract.py +201 -0
- antfly/client_generated/api/default/generate_content.py +257 -0
- antfly/client_generated/api/default/generate_embeddings.py +204 -0
- antfly/client_generated/api/default/list_models.py +327 -0
- antfly/client_generated/api/default/read_images.py +281 -0
- antfly/client_generated/api/default/rerank_multimodal_prompts.py +204 -0
- antfly/client_generated/api/default/rerank_prompts.py +289 -0
- antfly/client_generated/api/default/rewrite_text.py +241 -0
- antfly/client_generated/api/default/transcribe_audio.py +329 -0
- antfly/client_generated/api/index_management/__init__.py +1 -0
- antfly/client_generated/api/index_management/create_index.py +209 -0
- antfly/client_generated/api/index_management/drop_index.py +174 -0
- antfly/client_generated/api/index_management/get_index.py +176 -0
- antfly/client_generated/api/index_management/list_indexes.py +171 -0
- antfly/client_generated/api/permission/__init__.py +1 -0
- antfly/client_generated/api/permission/add_permission_to_user.py +198 -0
- antfly/client_generated/api/permission/get_user_permissions.py +184 -0
- antfly/client_generated/api/permission/remove_permission_from_user.py +215 -0
- antfly/client_generated/api/query_operations/__init__.py +1 -0
- antfly/client_generated/api/query_operations/evaluate.py +211 -0
- antfly/client_generated/api/query_operations/query_builder_agent.py +202 -0
- antfly/client_generated/api/role/__init__.py +1 -0
- antfly/client_generated/api/role/add_role_to_user.py +198 -0
- antfly/client_generated/api/role/list_auth_subjects.py +147 -0
- antfly/client_generated/api/role/list_user_roles.py +169 -0
- antfly/client_generated/api/role/remove_role_from_user.py +194 -0
- antfly/client_generated/api/row_filter/__init__.py +1 -0
- antfly/client_generated/api/row_filter/get_row_filter.py +186 -0
- antfly/client_generated/api/row_filter/get_subject_row_filter.py +186 -0
- antfly/client_generated/api/row_filter/list_row_filters.py +179 -0
- antfly/client_generated/api/row_filter/list_subject_row_filters.py +174 -0
- antfly/client_generated/api/row_filter/remove_row_filter.py +182 -0
- antfly/client_generated/api/row_filter/remove_subject_row_filter.py +182 -0
- antfly/client_generated/api/row_filter/set_row_filter.py +216 -0
- antfly/client_generated/api/row_filter/set_subject_row_filter.py +207 -0
- antfly/client_generated/api/table_management/__init__.py +1 -0
- antfly/client_generated/api/table_management/drop_table.py +160 -0
- antfly/client_generated/api/table_management/get_table.py +151 -0
- antfly/client_generated/api/table_management/list_tables.py +176 -0
- antfly/client_generated/api/table_management/update_schema.py +182 -0
- antfly/client_generated/api/user/__init__.py +1 -0
- antfly/client_generated/api/user/create_user.py +196 -0
- antfly/client_generated/api/user/delete_user.py +173 -0
- antfly/client_generated/api/user/get_current_user.py +147 -0
- antfly/client_generated/api/user/get_user_by_name.py +175 -0
- antfly/client_generated/api/user/list_users.py +157 -0
- antfly/client_generated/api/user/update_user_password.py +198 -0
- antfly/client_generated/client.py +268 -0
- antfly/client_generated/errors.py +16 -0
- antfly/client_generated/models/__init__.py +1087 -0
- antfly/client_generated/models/agent_decision.py +81 -0
- antfly/client_generated/models/agent_question.py +120 -0
- antfly/client_generated/models/agent_question_kind.py +12 -0
- antfly/client_generated/models/agent_status.py +12 -0
- antfly/client_generated/models/agent_step.py +155 -0
- antfly/client_generated/models/agent_step_details.py +47 -0
- antfly/client_generated/models/agent_step_kind.py +13 -0
- antfly/client_generated/models/agent_step_status.py +10 -0
- antfly/client_generated/models/aggregation_bucket.py +156 -0
- antfly/client_generated/models/aggregation_bucket_sub_aggregations.py +60 -0
- antfly/client_generated/models/aggregation_date_range.py +81 -0
- antfly/client_generated/models/aggregation_range.py +81 -0
- antfly/client_generated/models/aggregation_result.py +160 -0
- antfly/client_generated/models/aggregation_type.py +23 -0
- antfly/client_generated/models/algebraic_aggregation_join.py +96 -0
- antfly/client_generated/models/algebraic_aggregation_join_kind.py +11 -0
- antfly/client_generated/models/algebraic_index_config.py +64 -0
- antfly/client_generated/models/algebraic_index_stats.py +329 -0
- antfly/client_generated/models/algebraic_index_stats_index_type.py +8 -0
- antfly/client_generated/models/algebraic_index_stats_planner_last_decision.py +9 -0
- antfly/client_generated/models/analyses.py +70 -0
- antfly/client_generated/models/analyses_result.py +74 -0
- antfly/client_generated/models/answer_agent_result.py +159 -0
- antfly/client_generated/models/answer_agent_steps.py +136 -0
- antfly/client_generated/models/antfly_chunker_config.py +149 -0
- antfly/client_generated/models/antfly_embedder_config.py +90 -0
- antfly/client_generated/models/antfly_generator_config.py +119 -0
- antfly/client_generated/models/antfly_reranker_config.py +77 -0
- antfly/client_generated/models/antfly_type.py +19 -0
- antfly/client_generated/models/antfly_type_2.py +19 -0
- antfly/client_generated/models/anthropic_generator_config.py +119 -0
- antfly/client_generated/models/api_key.py +202 -0
- antfly/client_generated/models/api_key_row_filter_type_0.py +50 -0
- antfly/client_generated/models/api_key_with_secret.py +220 -0
- antfly/client_generated/models/audio_chunk_options.py +72 -0
- antfly/client_generated/models/auth_subject.py +72 -0
- antfly/client_generated/models/auth_subject_kind.py +11 -0
- antfly/client_generated/models/backup_info.py +115 -0
- antfly/client_generated/models/backup_info_format.py +9 -0
- antfly/client_generated/models/backup_list_response.py +75 -0
- antfly/client_generated/models/backup_request.py +101 -0
- antfly/client_generated/models/backup_request_format.py +9 -0
- antfly/client_generated/models/backup_table_response_201.py +61 -0
- antfly/client_generated/models/batch_request.py +195 -0
- antfly/client_generated/models/batch_request_inserts.py +71 -0
- antfly/client_generated/models/batch_request_inserts_additional_property.py +47 -0
- antfly/client_generated/models/batch_response.py +79 -0
- antfly/client_generated/models/bedrock_embedder_config.py +139 -0
- antfly/client_generated/models/bedrock_generator_config.py +109 -0
- antfly/client_generated/models/bing_search_config.py +159 -0
- antfly/client_generated/models/bing_search_config_freshness.py +10 -0
- antfly/client_generated/models/bool_field_query.py +93 -0
- antfly/client_generated/models/boolean_query.py +534 -0
- antfly/client_generated/models/brave_search_config.py +169 -0
- antfly/client_generated/models/brave_search_config_freshness.py +11 -0
- antfly/client_generated/models/calendar_interval.py +14 -0
- antfly/client_generated/models/chain_condition.py +11 -0
- antfly/client_generated/models/chain_link.py +110 -0
- antfly/client_generated/models/chat_message.py +208 -0
- antfly/client_generated/models/chat_message_role.py +11 -0
- antfly/client_generated/models/chat_tool_name.py +16 -0
- antfly/client_generated/models/chat_tools_config.py +157 -0
- antfly/client_generated/models/chunk_options.py +111 -0
- antfly/client_generated/models/chunker_config.py +103 -0
- antfly/client_generated/models/chunker_config_full_text_index.py +51 -0
- antfly/client_generated/models/chunker_provider.py +9 -0
- antfly/client_generated/models/classification_step_config.py +169 -0
- antfly/client_generated/models/classification_transformation_result.py +158 -0
- antfly/client_generated/models/cluster_backup_request.py +113 -0
- antfly/client_generated/models/cluster_backup_request_format.py +9 -0
- antfly/client_generated/models/cluster_backup_response.py +93 -0
- antfly/client_generated/models/cluster_backup_response_status.py +10 -0
- antfly/client_generated/models/cluster_data_group_status.py +209 -0
- antfly/client_generated/models/cluster_data_node_status.py +197 -0
- antfly/client_generated/models/cluster_data_range_status.py +200 -0
- antfly/client_generated/models/cluster_data_replica_status.py +98 -0
- antfly/client_generated/models/cluster_data_status.py +149 -0
- antfly/client_generated/models/cluster_health.py +12 -0
- antfly/client_generated/models/cluster_restore_request.py +107 -0
- antfly/client_generated/models/cluster_restore_request_restore_mode.py +10 -0
- antfly/client_generated/models/cluster_restore_response.py +85 -0
- antfly/client_generated/models/cluster_restore_response_status.py +10 -0
- antfly/client_generated/models/cluster_status.py +114 -0
- antfly/client_generated/models/cluster_topology.py +124 -0
- antfly/client_generated/models/cohere_embedder_config.py +119 -0
- antfly/client_generated/models/cohere_embedder_config_input_type.py +11 -0
- antfly/client_generated/models/cohere_embedder_config_truncate.py +10 -0
- antfly/client_generated/models/cohere_generator_config.py +127 -0
- antfly/client_generated/models/cohere_reranker_config.py +102 -0
- antfly/client_generated/models/confidence_step_config.py +119 -0
- antfly/client_generated/models/conjunction_query.py +492 -0
- antfly/client_generated/models/create_api_key_request.py +156 -0
- antfly/client_generated/models/create_api_key_request_row_filter_type_0.py +50 -0
- antfly/client_generated/models/create_user_request.py +153 -0
- antfly/client_generated/models/create_user_request_metadata_type_0.py +52 -0
- antfly/client_generated/models/credentials.py +102 -0
- antfly/client_generated/models/date_range_string_query.py +165 -0
- antfly/client_generated/models/disjunction_query.py +501 -0
- antfly/client_generated/models/distance_metric.py +10 -0
- antfly/client_generated/models/distance_range.py +81 -0
- antfly/client_generated/models/distance_unit.py +12 -0
- antfly/client_generated/models/doc_id_query.py +84 -0
- antfly/client_generated/models/document_schema.py +85 -0
- antfly/client_generated/models/document_schema_schema.py +50 -0
- antfly/client_generated/models/duck_duck_go_search_config.py +143 -0
- antfly/client_generated/models/dynamic_template.py +141 -0
- antfly/client_generated/models/dynamic_template_match_mapping_type.py +12 -0
- antfly/client_generated/models/edge.py +144 -0
- antfly/client_generated/models/edge_direction.py +10 -0
- antfly/client_generated/models/edge_metadata.py +47 -0
- antfly/client_generated/models/edge_type_config.py +133 -0
- antfly/client_generated/models/edge_type_config_topology.py +9 -0
- antfly/client_generated/models/edges_response.py +88 -0
- antfly/client_generated/models/embedder_config.py +267 -0
- antfly/client_generated/models/embedder_provider.py +16 -0
- antfly/client_generated/models/embedding_type_1.py +70 -0
- antfly/client_generated/models/embedding_type_3.py +71 -0
- antfly/client_generated/models/embeddings_index_config.py +393 -0
- antfly/client_generated/models/embeddings_index_stats.py +146 -0
- antfly/client_generated/models/embeddings_index_stats_index_type.py +8 -0
- antfly/client_generated/models/error.py +61 -0
- antfly/client_generated/models/eval_config.py +135 -0
- antfly/client_generated/models/eval_options.py +80 -0
- antfly/client_generated/models/eval_request.py +185 -0
- antfly/client_generated/models/eval_request_context_item.py +47 -0
- antfly/client_generated/models/eval_result.py +102 -0
- antfly/client_generated/models/eval_scores.py +93 -0
- antfly/client_generated/models/eval_scores_generation.py +60 -0
- antfly/client_generated/models/eval_scores_retrieval.py +60 -0
- antfly/client_generated/models/eval_summary.py +89 -0
- antfly/client_generated/models/evaluator_name.py +20 -0
- antfly/client_generated/models/evaluator_score.py +102 -0
- antfly/client_generated/models/evaluator_score_metadata.py +47 -0
- antfly/client_generated/models/extraction_classification.py +80 -0
- antfly/client_generated/models/extraction_classification_schema.py +80 -0
- antfly/client_generated/models/extraction_entity.py +98 -0
- antfly/client_generated/models/extraction_input.py +201 -0
- antfly/client_generated/models/extraction_input_metadata.py +47 -0
- antfly/client_generated/models/extraction_object.py +152 -0
- antfly/client_generated/models/extraction_object_structures.py +47 -0
- antfly/client_generated/models/extraction_options.py +110 -0
- antfly/client_generated/models/extraction_reader_options.py +88 -0
- antfly/client_generated/models/extraction_relation.py +110 -0
- antfly/client_generated/models/extraction_relation_endpoint.py +70 -0
- antfly/client_generated/models/extraction_relation_schema.py +81 -0
- antfly/client_generated/models/extraction_request.py +113 -0
- antfly/client_generated/models/extraction_response.py +112 -0
- antfly/client_generated/models/extraction_response_object.py +8 -0
- antfly/client_generated/models/extraction_response_usage.py +47 -0
- antfly/client_generated/models/extraction_schema.py +131 -0
- antfly/client_generated/models/extraction_schema_structures.py +60 -0
- antfly/client_generated/models/extraction_structure_field_type_1.py +47 -0
- antfly/client_generated/models/extraction_structure_schema.py +74 -0
- antfly/client_generated/models/extraction_structure_schema_fields.py +76 -0
- antfly/client_generated/models/extraction_token.py +74 -0
- antfly/client_generated/models/failed_operation.py +87 -0
- antfly/client_generated/models/failed_operation_operation.py +9 -0
- antfly/client_generated/models/fetch_config.py +140 -0
- antfly/client_generated/models/field_statistics.py +98 -0
- antfly/client_generated/models/filter_spec.py +88 -0
- antfly/client_generated/models/filter_spec_operator.py +17 -0
- antfly/client_generated/models/followup_step_config.py +128 -0
- antfly/client_generated/models/foreign_column.py +82 -0
- antfly/client_generated/models/foreign_source.py +114 -0
- antfly/client_generated/models/foreign_source_type.py +8 -0
- antfly/client_generated/models/full_text_index_config.py +61 -0
- antfly/client_generated/models/full_text_index_stats.py +118 -0
- antfly/client_generated/models/full_text_index_stats_index_type.py +8 -0
- antfly/client_generated/models/fuzziness_type_1.py +8 -0
- antfly/client_generated/models/fuzzy_query.py +131 -0
- antfly/client_generated/models/generation_step_config.py +128 -0
- antfly/client_generated/models/generator_config.py +67 -0
- antfly/client_generated/models/generator_provider.py +17 -0
- antfly/client_generated/models/geo_bounding_box_query.py +101 -0
- antfly/client_generated/models/geo_bounding_polygon_query.py +107 -0
- antfly/client_generated/models/geo_distance_query.py +101 -0
- antfly/client_generated/models/geo_point.py +70 -0
- antfly/client_generated/models/geo_shape_geometry_relation.py +10 -0
- antfly/client_generated/models/get_current_user_response_200.py +122 -0
- antfly/client_generated/models/get_current_user_response_200_metadata_type_0.py +47 -0
- antfly/client_generated/models/google_embedder_config.py +120 -0
- antfly/client_generated/models/google_generator_config.py +136 -0
- antfly/client_generated/models/google_search_config.py +173 -0
- antfly/client_generated/models/google_search_config_search_type.py +9 -0
- antfly/client_generated/models/graph_index_config.py +121 -0
- antfly/client_generated/models/graph_index_stats.py +151 -0
- antfly/client_generated/models/graph_index_stats_algebraic_graph.py +75 -0
- antfly/client_generated/models/graph_index_stats_algebraic_graph_traversal.py +97 -0
- antfly/client_generated/models/graph_index_stats_edge_types.py +47 -0
- antfly/client_generated/models/graph_index_stats_index_type.py +8 -0
- antfly/client_generated/models/graph_node_selector.py +106 -0
- antfly/client_generated/models/graph_query.py +192 -0
- antfly/client_generated/models/graph_query_params.py +216 -0
- antfly/client_generated/models/graph_query_params_algorithm_params.py +47 -0
- antfly/client_generated/models/graph_query_result.py +155 -0
- antfly/client_generated/models/graph_query_type.py +12 -0
- antfly/client_generated/models/graph_result_node.py +173 -0
- antfly/client_generated/models/graph_result_node_document.py +47 -0
- antfly/client_generated/models/ground_truth.py +74 -0
- antfly/client_generated/models/image_url.py +62 -0
- antfly/client_generated/models/image_url_content_part.py +78 -0
- antfly/client_generated/models/image_url_content_part_type.py +8 -0
- antfly/client_generated/models/incomplete_details.py +68 -0
- antfly/client_generated/models/incomplete_details_reason.py +11 -0
- antfly/client_generated/models/index_status.py +190 -0
- antfly/client_generated/models/index_status_shard_status.py +116 -0
- antfly/client_generated/models/index_type.py +11 -0
- antfly/client_generated/models/inference_audio_chunk_config.py +95 -0
- antfly/client_generated/models/inference_backend_runtimes.py +116 -0
- antfly/client_generated/models/inference_binary_content.py +98 -0
- antfly/client_generated/models/inference_chat_message.py +198 -0
- antfly/client_generated/models/inference_chunk.py +70 -0
- antfly/client_generated/models/inference_chunk_config.py +123 -0
- antfly/client_generated/models/inference_chunk_object.py +88 -0
- antfly/client_generated/models/inference_chunk_object_object.py +8 -0
- antfly/client_generated/models/inference_chunk_request.py +148 -0
- antfly/client_generated/models/inference_chunk_response.py +117 -0
- antfly/client_generated/models/inference_chunk_response_object.py +8 -0
- antfly/client_generated/models/inference_classify_object.py +93 -0
- antfly/client_generated/models/inference_classify_object_object.py +8 -0
- antfly/client_generated/models/inference_classify_request.py +105 -0
- antfly/client_generated/models/inference_classify_response.py +103 -0
- antfly/client_generated/models/inference_classify_response_object.py +8 -0
- antfly/client_generated/models/inference_classify_result.py +69 -0
- antfly/client_generated/models/inference_config.py +327 -0
- antfly/client_generated/models/inference_config_model_strategies.py +74 -0
- antfly/client_generated/models/inference_config_model_strategies_additional_property.py +10 -0
- antfly/client_generated/models/inference_content_security_config.py +128 -0
- antfly/client_generated/models/inference_credentials.py +102 -0
- antfly/client_generated/models/inference_document_classification_features.py +141 -0
- antfly/client_generated/models/inference_document_classification_object.py +162 -0
- antfly/client_generated/models/inference_document_classification_object_input.py +69 -0
- antfly/client_generated/models/inference_document_classification_object_object.py +8 -0
- antfly/client_generated/models/inference_document_classification_request.py +99 -0
- antfly/client_generated/models/inference_document_classification_response.py +103 -0
- antfly/client_generated/models/inference_document_classification_response_object.py +8 -0
- antfly/client_generated/models/inference_document_classification_result.py +69 -0
- antfly/client_generated/models/inference_document_token_box.py +70 -0
- antfly/client_generated/models/inference_document_token_classification_features.py +101 -0
- antfly/client_generated/models/inference_document_token_classification_object.py +124 -0
- antfly/client_generated/models/inference_document_token_classification_object_object.py +8 -0
- antfly/client_generated/models/inference_document_token_classification_prediction.py +145 -0
- antfly/client_generated/models/inference_document_token_classification_request.py +105 -0
- antfly/client_generated/models/inference_document_token_classification_response.py +105 -0
- antfly/client_generated/models/inference_document_token_classification_response_object.py +8 -0
- antfly/client_generated/models/inference_document_token_classification_result.py +69 -0
- antfly/client_generated/models/inference_embed_request.py +226 -0
- antfly/client_generated/models/inference_embed_request_encoding_format.py +8 -0
- antfly/client_generated/models/inference_embed_request_input_type.py +13 -0
- antfly/client_generated/models/inference_embed_request_task_type.py +15 -0
- antfly/client_generated/models/inference_embed_response.py +104 -0
- antfly/client_generated/models/inference_embed_response_object.py +8 -0
- antfly/client_generated/models/inference_embedding_object.py +113 -0
- antfly/client_generated/models/inference_embedding_object_object.py +8 -0
- antfly/client_generated/models/inference_embedding_usage.py +70 -0
- antfly/client_generated/models/inference_error.py +61 -0
- antfly/client_generated/models/inference_extract_field_value.py +90 -0
- antfly/client_generated/models/inference_extract_object.py +88 -0
- antfly/client_generated/models/inference_extract_object_object.py +8 -0
- antfly/client_generated/models/inference_extract_object_results.py +80 -0
- antfly/client_generated/models/inference_extract_object_results_additional_property_item.py +47 -0
- antfly/client_generated/models/inference_extract_request.py +179 -0
- antfly/client_generated/models/inference_extract_request_schema.py +62 -0
- antfly/client_generated/models/inference_extract_response.py +103 -0
- antfly/client_generated/models/inference_extract_response_object.py +8 -0
- antfly/client_generated/models/inference_finish_reason.py +12 -0
- antfly/client_generated/models/inference_function_definition.py +107 -0
- antfly/client_generated/models/inference_function_definition_parameters.py +53 -0
- antfly/client_generated/models/inference_generate_choice.py +121 -0
- antfly/client_generated/models/inference_generate_choice_logprobs_type_0.py +47 -0
- antfly/client_generated/models/inference_generate_chunk.py +110 -0
- antfly/client_generated/models/inference_generate_chunk_choice.py +94 -0
- antfly/client_generated/models/inference_generate_chunk_object.py +8 -0
- antfly/client_generated/models/inference_generate_delta.py +117 -0
- antfly/client_generated/models/inference_generate_json_schema_config.py +92 -0
- antfly/client_generated/models/inference_generate_json_schema_config_schema.py +47 -0
- antfly/client_generated/models/inference_generate_message.py +112 -0
- antfly/client_generated/models/inference_generate_request.py +380 -0
- antfly/client_generated/models/inference_generate_request_backend.py +14 -0
- antfly/client_generated/models/inference_generate_request_cache_dtype.py +12 -0
- antfly/client_generated/models/inference_generate_request_compiled_target.py +9 -0
- antfly/client_generated/models/inference_generate_request_mode.py +9 -0
- antfly/client_generated/models/inference_generate_response.py +120 -0
- antfly/client_generated/models/inference_generate_response_format.py +86 -0
- antfly/client_generated/models/inference_generate_response_format_type.py +10 -0
- antfly/client_generated/models/inference_generate_response_object.py +8 -0
- antfly/client_generated/models/inference_generate_usage.py +77 -0
- antfly/client_generated/models/inference_image_url.py +62 -0
- antfly/client_generated/models/inference_image_url_content_part.py +78 -0
- antfly/client_generated/models/inference_image_url_content_part_type.py +8 -0
- antfly/client_generated/models/inference_level.py +11 -0
- antfly/client_generated/models/inference_media_content_part.py +80 -0
- antfly/client_generated/models/inference_media_content_part_type.py +8 -0
- antfly/client_generated/models/inference_model_info.py +76 -0
- antfly/client_generated/models/inference_models_response.py +204 -0
- antfly/client_generated/models/inference_models_response_chunkers.py +60 -0
- antfly/client_generated/models/inference_models_response_classifiers.py +60 -0
- antfly/client_generated/models/inference_models_response_data_item.py +47 -0
- antfly/client_generated/models/inference_models_response_embedders.py +60 -0
- antfly/client_generated/models/inference_models_response_extractors.py +60 -0
- antfly/client_generated/models/inference_models_response_generators.py +60 -0
- antfly/client_generated/models/inference_models_response_object.py +8 -0
- antfly/client_generated/models/inference_models_response_readers.py +60 -0
- antfly/client_generated/models/inference_models_response_recognizers.py +60 -0
- antfly/client_generated/models/inference_models_response_rerankers.py +60 -0
- antfly/client_generated/models/inference_models_response_rewriters.py +60 -0
- antfly/client_generated/models/inference_models_response_transcribers.py +60 -0
- antfly/client_generated/models/inference_read_object.py +130 -0
- antfly/client_generated/models/inference_read_object_object.py +8 -0
- antfly/client_generated/models/inference_read_request.py +113 -0
- antfly/client_generated/models/inference_read_response.py +103 -0
- antfly/client_generated/models/inference_read_response_object.py +8 -0
- antfly/client_generated/models/inference_read_result.py +113 -0
- antfly/client_generated/models/inference_read_result_fields.py +54 -0
- antfly/client_generated/models/inference_recognize_entity.py +93 -0
- antfly/client_generated/models/inference_recognize_object.py +118 -0
- antfly/client_generated/models/inference_recognize_object_object.py +8 -0
- antfly/client_generated/models/inference_recognize_request.py +129 -0
- antfly/client_generated/models/inference_recognize_response.py +103 -0
- antfly/client_generated/models/inference_recognize_response_object.py +8 -0
- antfly/client_generated/models/inference_relation.py +91 -0
- antfly/client_generated/models/inference_rerank_multimodal_document.py +160 -0
- antfly/client_generated/models/inference_rerank_multimodal_request.py +91 -0
- antfly/client_generated/models/inference_rerank_object.py +79 -0
- antfly/client_generated/models/inference_rerank_object_object.py +8 -0
- antfly/client_generated/models/inference_rerank_request.py +79 -0
- antfly/client_generated/models/inference_rerank_response.py +103 -0
- antfly/client_generated/models/inference_rerank_response_object.py +8 -0
- antfly/client_generated/models/inference_resolver_config.py +110 -0
- antfly/client_generated/models/inference_rewrite_object.py +79 -0
- antfly/client_generated/models/inference_rewrite_object_object.py +8 -0
- antfly/client_generated/models/inference_rewrite_request.py +69 -0
- antfly/client_generated/models/inference_rewrite_response.py +103 -0
- antfly/client_generated/models/inference_rewrite_response_object.py +8 -0
- antfly/client_generated/models/inference_role.py +11 -0
- antfly/client_generated/models/inference_sparse_vector.py +70 -0
- antfly/client_generated/models/inference_style.py +11 -0
- antfly/client_generated/models/inference_text_chunk_options.py +82 -0
- antfly/client_generated/models/inference_text_content.py +78 -0
- antfly/client_generated/models/inference_text_content_part.py +72 -0
- antfly/client_generated/models/inference_text_content_part_type.py +8 -0
- antfly/client_generated/models/inference_text_region.py +89 -0
- antfly/client_generated/models/inference_tool.py +78 -0
- antfly/client_generated/models/inference_tool_call_delta.py +110 -0
- antfly/client_generated/models/inference_tool_call_delta_type.py +8 -0
- antfly/client_generated/models/inference_tool_call_function_delta.py +71 -0
- antfly/client_generated/models/inference_tool_choice_type_0.py +10 -0
- antfly/client_generated/models/inference_tool_choice_type_1_function.py +61 -0
- antfly/client_generated/models/inference_tool_choice_type_1_type.py +8 -0
- antfly/client_generated/models/inference_tool_type.py +8 -0
- antfly/client_generated/models/inference_transcribe_object.py +89 -0
- antfly/client_generated/models/inference_transcribe_object_object.py +8 -0
- antfly/client_generated/models/inference_transcribe_request.py +81 -0
- antfly/client_generated/models/inference_transcribe_response.py +103 -0
- antfly/client_generated/models/inference_transcribe_response_object.py +8 -0
- antfly/client_generated/models/inference_vad_options.py +92 -0
- antfly/client_generated/models/inferenceschemas_config.py +88 -0
- antfly/client_generated/models/ip_range_query.py +93 -0
- antfly/client_generated/models/join_condition.py +95 -0
- antfly/client_generated/models/join_operator.py +13 -0
- antfly/client_generated/models/join_profile.py +130 -0
- antfly/client_generated/models/join_strategy.py +10 -0
- antfly/client_generated/models/join_type.py +10 -0
- antfly/client_generated/models/key_range.py +71 -0
- antfly/client_generated/models/linear_merge_page_status.py +10 -0
- antfly/client_generated/models/linear_merge_request.py +150 -0
- antfly/client_generated/models/linear_merge_request_records.py +59 -0
- antfly/client_generated/models/linear_merge_result.py +183 -0
- antfly/client_generated/models/list_users_response_200_item.py +61 -0
- antfly/client_generated/models/lookup_key_response_200.py +47 -0
- antfly/client_generated/models/lsm_storage_status.py +116 -0
- antfly/client_generated/models/match_all_query.py +90 -0
- antfly/client_generated/models/match_all_query_match_all.py +47 -0
- antfly/client_generated/models/match_none_query.py +90 -0
- antfly/client_generated/models/match_none_query_match_none.py +47 -0
- antfly/client_generated/models/match_phrase_query.py +131 -0
- antfly/client_generated/models/match_query.py +157 -0
- antfly/client_generated/models/match_query_operator.py +9 -0
- antfly/client_generated/models/media_content_part.py +80 -0
- antfly/client_generated/models/media_content_part_type.py +8 -0
- antfly/client_generated/models/merge_config.py +118 -0
- antfly/client_generated/models/merge_config_weights.py +54 -0
- antfly/client_generated/models/merge_profile.py +101 -0
- antfly/client_generated/models/merge_strategy.py +10 -0
- antfly/client_generated/models/multi_batch_request.py +104 -0
- antfly/client_generated/models/multi_batch_request_tables.py +67 -0
- antfly/client_generated/models/multi_batch_response.py +75 -0
- antfly/client_generated/models/multi_batch_response_tables.py +60 -0
- antfly/client_generated/models/multi_match_body.py +101 -0
- antfly/client_generated/models/multi_match_body_type.py +8 -0
- antfly/client_generated/models/multi_match_query.py +67 -0
- antfly/client_generated/models/multi_phrase_query.py +131 -0
- antfly/client_generated/models/node_filter.py +84 -0
- antfly/client_generated/models/node_filter_filter_query.py +47 -0
- antfly/client_generated/models/numeric_range_query.py +163 -0
- antfly/client_generated/models/ollama_embedder_config.py +84 -0
- antfly/client_generated/models/ollama_generator_config.py +118 -0
- antfly/client_generated/models/ollama_reranker_config.py +73 -0
- antfly/client_generated/models/open_ai_embedder_config.py +104 -0
- antfly/client_generated/models/open_ai_generator_config.py +127 -0
- antfly/client_generated/models/open_router_embedder_config.py +94 -0
- antfly/client_generated/models/open_router_generator_config.py +128 -0
- antfly/client_generated/models/path.py +108 -0
- antfly/client_generated/models/path_edge.py +110 -0
- antfly/client_generated/models/path_edge_metadata.py +47 -0
- antfly/client_generated/models/path_find_request.py +158 -0
- antfly/client_generated/models/path_find_result.py +135 -0
- antfly/client_generated/models/path_find_weight_mode.py +10 -0
- antfly/client_generated/models/path_weight_mode.py +10 -0
- antfly/client_generated/models/pattern_edge_step.py +120 -0
- antfly/client_generated/models/pattern_match.py +98 -0
- antfly/client_generated/models/pattern_match_bindings.py +60 -0
- antfly/client_generated/models/pattern_step.py +102 -0
- antfly/client_generated/models/permission.py +80 -0
- antfly/client_generated/models/permission_type.py +10 -0
- antfly/client_generated/models/phrase_query.py +122 -0
- antfly/client_generated/models/prefix_query.py +93 -0
- antfly/client_generated/models/prune_stats.py +89 -0
- antfly/client_generated/models/pruner.py +115 -0
- antfly/client_generated/models/query_builder_request.py +250 -0
- antfly/client_generated/models/query_builder_request_constraints.py +53 -0
- antfly/client_generated/models/query_builder_request_example_documents_item.py +47 -0
- antfly/client_generated/models/query_hit.py +124 -0
- antfly/client_generated/models/query_hit_index_scores.py +47 -0
- antfly/client_generated/models/query_hit_source.py +47 -0
- antfly/client_generated/models/query_hits.py +98 -0
- antfly/client_generated/models/query_profile.py +130 -0
- antfly/client_generated/models/query_responses.py +80 -0
- antfly/client_generated/models/query_result.py +187 -0
- antfly/client_generated/models/query_result_aggregations.py +63 -0
- antfly/client_generated/models/query_result_analyses.py +60 -0
- antfly/client_generated/models/query_result_graph_results.py +60 -0
- antfly/client_generated/models/query_strategy.py +11 -0
- antfly/client_generated/models/query_string_query.py +84 -0
- antfly/client_generated/models/regexp_query.py +93 -0
- antfly/client_generated/models/replication_source_action_hint.py +81 -0
- antfly/client_generated/models/replication_source_status.py +225 -0
- antfly/client_generated/models/replication_transform_op.py +87 -0
- antfly/client_generated/models/reranker_config.py +86 -0
- antfly/client_generated/models/reranker_profile.py +80 -0
- antfly/client_generated/models/reranker_provider.py +11 -0
- antfly/client_generated/models/resource_type.py +10 -0
- antfly/client_generated/models/restore_table_response_202.py +61 -0
- antfly/client_generated/models/retrieval_agent_result.py +397 -0
- antfly/client_generated/models/retrieval_agent_steps.py +179 -0
- antfly/client_generated/models/retrieval_agent_usage.py +129 -0
- antfly/client_generated/models/retrieval_strategy.py +13 -0
- antfly/client_generated/models/retry_config.py +89 -0
- antfly/client_generated/models/role_assignment.py +61 -0
- antfly/client_generated/models/route_type.py +9 -0
- antfly/client_generated/models/row_filter_entry.py +77 -0
- antfly/client_generated/models/row_filter_entry_filter.py +52 -0
- antfly/client_generated/models/secret_entry.py +115 -0
- antfly/client_generated/models/secret_list.py +75 -0
- antfly/client_generated/models/secret_status.py +10 -0
- antfly/client_generated/models/secret_store_status.py +62 -0
- antfly/client_generated/models/secret_write_request.py +61 -0
- antfly/client_generated/models/semantic_query_mode.py +9 -0
- antfly/client_generated/models/serper_search_config.py +170 -0
- antfly/client_generated/models/serper_search_config_search_type.py +12 -0
- antfly/client_generated/models/serper_search_config_time_period.py +11 -0
- antfly/client_generated/models/set_row_filter_body.py +47 -0
- antfly/client_generated/models/set_subject_row_filter_body.py +47 -0
- antfly/client_generated/models/shard_config.py +61 -0
- antfly/client_generated/models/shards_profile.py +80 -0
- antfly/client_generated/models/significance_algorithm.py +11 -0
- antfly/client_generated/models/sort_field.py +73 -0
- antfly/client_generated/models/sse_error.py +62 -0
- antfly/client_generated/models/sse_event.py +19 -0
- antfly/client_generated/models/sse_step_completed.py +155 -0
- antfly/client_generated/models/sse_step_completed_details.py +47 -0
- antfly/client_generated/models/sse_step_progress.py +64 -0
- antfly/client_generated/models/sse_step_started.py +98 -0
- antfly/client_generated/models/sse_tool_mode.py +74 -0
- antfly/client_generated/models/sse_tool_mode_mode.py +9 -0
- antfly/client_generated/models/storage_status.py +93 -0
- antfly/client_generated/models/success_message.py +61 -0
- antfly/client_generated/models/sync_level.py +13 -0
- antfly/client_generated/models/table_backup_status.py +81 -0
- antfly/client_generated/models/table_backup_status_status.py +10 -0
- antfly/client_generated/models/table_migration.py +79 -0
- antfly/client_generated/models/table_migration_state.py +8 -0
- antfly/client_generated/models/table_restore_status.py +81 -0
- antfly/client_generated/models/table_restore_status_status.py +10 -0
- antfly/client_generated/models/table_schema.py +150 -0
- antfly/client_generated/models/table_schema_document_schemas.py +60 -0
- antfly/client_generated/models/table_statistics.py +111 -0
- antfly/client_generated/models/table_statistics_field_stats.py +60 -0
- antfly/client_generated/models/tavily_search_config.py +195 -0
- antfly/client_generated/models/tavily_search_config_search_depth.py +9 -0
- antfly/client_generated/models/template_field_mapping.py +116 -0
- antfly/client_generated/models/term_query.py +93 -0
- antfly/client_generated/models/term_range_query.py +163 -0
- antfly/client_generated/models/text_chunk_options.py +82 -0
- antfly/client_generated/models/text_content_part.py +72 -0
- antfly/client_generated/models/text_content_part_type.py +8 -0
- antfly/client_generated/models/tool_call.py +86 -0
- antfly/client_generated/models/tool_call_function.py +70 -0
- antfly/client_generated/models/tool_call_type.py +8 -0
- antfly/client_generated/models/transaction_begin_request.py +76 -0
- antfly/client_generated/models/transaction_begin_response.py +77 -0
- antfly/client_generated/models/transaction_commit_request.py +123 -0
- antfly/client_generated/models/transaction_commit_request_tables.py +60 -0
- antfly/client_generated/models/transaction_commit_response.py +107 -0
- antfly/client_generated/models/transaction_commit_response_conflict.py +80 -0
- antfly/client_generated/models/transaction_commit_response_status.py +9 -0
- antfly/client_generated/models/transaction_commit_response_tables.py +60 -0
- antfly/client_generated/models/transaction_read_item.py +80 -0
- antfly/client_generated/models/transaction_savepoint_response.py +77 -0
- antfly/client_generated/models/transaction_session_cleanup_response.py +70 -0
- antfly/client_generated/models/transaction_session_commit_response.py +114 -0
- antfly/client_generated/models/transaction_session_details_response.py +268 -0
- antfly/client_generated/models/transaction_session_list_response.py +106 -0
- antfly/client_generated/models/transaction_session_read_snapshot.py +88 -0
- antfly/client_generated/models/transaction_session_status.py +207 -0
- antfly/client_generated/models/transaction_session_table_detail.py +97 -0
- antfly/client_generated/models/transaction_stage_delete_request.py +69 -0
- antfly/client_generated/models/transaction_stage_read_request.py +77 -0
- antfly/client_generated/models/transaction_stage_read_response.py +83 -0
- antfly/client_generated/models/transaction_stage_read_snapshot.py +85 -0
- antfly/client_generated/models/transaction_stage_write_request.py +83 -0
- antfly/client_generated/models/transaction_stage_write_request_document.py +47 -0
- antfly/client_generated/models/transaction_status_response.py +69 -0
- antfly/client_generated/models/transform.py +104 -0
- antfly/client_generated/models/transform_op.py +82 -0
- antfly/client_generated/models/transform_op_type.py +19 -0
- antfly/client_generated/models/traversal_result.py +137 -0
- antfly/client_generated/models/traversal_result_document.py +47 -0
- antfly/client_generated/models/traversal_rules.py +138 -0
- antfly/client_generated/models/traverse_response.py +88 -0
- antfly/client_generated/models/tree_search_config.py +97 -0
- antfly/client_generated/models/update_password_request.py +61 -0
- antfly/client_generated/models/user.py +110 -0
- antfly/client_generated/models/user_metadata_type_0.py +52 -0
- antfly/client_generated/models/vertex_embedder_config.py +115 -0
- antfly/client_generated/models/vertex_generator_config.py +127 -0
- antfly/client_generated/models/vertex_reranker_config.py +108 -0
- antfly/client_generated/models/web_search_config.py +127 -0
- antfly/client_generated/models/web_search_provider.py +13 -0
- antfly/client_generated/models/wildcard_query.py +93 -0
- antfly/client_generated/py.typed +1 -0
- antfly/client_generated/types.py +54 -0
- antfly/exceptions.py +19 -0
- antfly/pyproject.toml +26 -0
- antfly_sdk-0.0.2.dev2.dist-info/METADATA +153 -0
- antfly_sdk-0.0.2.dev2.dist-info/RECORD +640 -0
- antfly_sdk-0.0.2.dev2.dist-info/WHEEL +4 -0
|
@@ -0,0 +1,329 @@
|
|
|
1
|
+
from __future__ import annotations
|
|
2
|
+
|
|
3
|
+
from collections.abc import Mapping
|
|
4
|
+
from typing import Any, TypeVar
|
|
5
|
+
|
|
6
|
+
from attrs import define as _attrs_define
|
|
7
|
+
from attrs import field as _attrs_field
|
|
8
|
+
|
|
9
|
+
from ..models.algebraic_index_stats_index_type import AlgebraicIndexStatsIndexType
|
|
10
|
+
from ..models.algebraic_index_stats_planner_last_decision import AlgebraicIndexStatsPlannerLastDecision
|
|
11
|
+
from ..types import UNSET, Unset
|
|
12
|
+
|
|
13
|
+
T = TypeVar("T", bound="AlgebraicIndexStats")
|
|
14
|
+
|
|
15
|
+
|
|
16
|
+
@_attrs_define
|
|
17
|
+
class AlgebraicIndexStats:
|
|
18
|
+
"""Compact public statistics for an algebraic sidecar index. Detailed runtime, adaptive, and materialization records
|
|
19
|
+
remain internal diagnostics.
|
|
20
|
+
|
|
21
|
+
Attributes:
|
|
22
|
+
index_type (AlgebraicIndexStatsIndexType): Discriminator for the index stats variant.
|
|
23
|
+
error (str | Unset): Error message if stats could not be retrieved
|
|
24
|
+
total_indexed (int | Unset): Number of documents reflected in the algebraic sidecar
|
|
25
|
+
disk_usage (int | Unset): Size of the index in bytes
|
|
26
|
+
rebuilding (bool | Unset): Whether the sidecar is currently rebuilding
|
|
27
|
+
backfill_progress (float | Unset): Backfill progress as a ratio from 0.0 to 1.0
|
|
28
|
+
backfill_items_processed (int | Unset): Number of documents processed during current backfill
|
|
29
|
+
healthy (bool | Unset):
|
|
30
|
+
parse_error_count (int | Unset):
|
|
31
|
+
schema_version (int | Unset):
|
|
32
|
+
capability_lifecycle_status (str | Unset): Schema-derived algebraic capability lifecycle, for example current,
|
|
33
|
+
stale, or rebuild_required.
|
|
34
|
+
planner_selected (int | Unset):
|
|
35
|
+
planner_fallback_count (int | Unset):
|
|
36
|
+
planner_last_decision (AlgebraicIndexStatsPlannerLastDecision | Unset):
|
|
37
|
+
planner_last_fallback_reason (str | Unset):
|
|
38
|
+
planner_last_estimated_scan_rows (int | Unset): Latest algebraic planner scan-row estimate for the last selected
|
|
39
|
+
or fallback decision.
|
|
40
|
+
planner_last_estimated_result_buckets (int | Unset): Latest algebraic planner result-bucket estimate for the
|
|
41
|
+
last selected or fallback decision.
|
|
42
|
+
planner_lifecycle_ready (bool | Unset):
|
|
43
|
+
planner_lifecycle_blocking_reason (str | Unset):
|
|
44
|
+
adaptive_progress_count (int | Unset):
|
|
45
|
+
recommendation_count (int | Unset): Number of currently recommended algebraic shapes.
|
|
46
|
+
adaptive_backfilling_count (int | Unset):
|
|
47
|
+
adaptive_ready_count (int | Unset):
|
|
48
|
+
adaptive_stale_count (int | Unset):
|
|
49
|
+
adaptive_cleanup_recommended_count (int | Unset):
|
|
50
|
+
last_error_reason (str | Unset):
|
|
51
|
+
active_progress_lifecycle (str | Unset):
|
|
52
|
+
active_progress_rows_processed (int | Unset):
|
|
53
|
+
active_progress_target_rows (int | Unset):
|
|
54
|
+
"""
|
|
55
|
+
|
|
56
|
+
index_type: AlgebraicIndexStatsIndexType
|
|
57
|
+
error: str | Unset = UNSET
|
|
58
|
+
total_indexed: int | Unset = UNSET
|
|
59
|
+
disk_usage: int | Unset = UNSET
|
|
60
|
+
rebuilding: bool | Unset = UNSET
|
|
61
|
+
backfill_progress: float | Unset = UNSET
|
|
62
|
+
backfill_items_processed: int | Unset = UNSET
|
|
63
|
+
healthy: bool | Unset = UNSET
|
|
64
|
+
parse_error_count: int | Unset = UNSET
|
|
65
|
+
schema_version: int | Unset = UNSET
|
|
66
|
+
capability_lifecycle_status: str | Unset = UNSET
|
|
67
|
+
planner_selected: int | Unset = UNSET
|
|
68
|
+
planner_fallback_count: int | Unset = UNSET
|
|
69
|
+
planner_last_decision: AlgebraicIndexStatsPlannerLastDecision | Unset = UNSET
|
|
70
|
+
planner_last_fallback_reason: str | Unset = UNSET
|
|
71
|
+
planner_last_estimated_scan_rows: int | Unset = UNSET
|
|
72
|
+
planner_last_estimated_result_buckets: int | Unset = UNSET
|
|
73
|
+
planner_lifecycle_ready: bool | Unset = UNSET
|
|
74
|
+
planner_lifecycle_blocking_reason: str | Unset = UNSET
|
|
75
|
+
adaptive_progress_count: int | Unset = UNSET
|
|
76
|
+
recommendation_count: int | Unset = UNSET
|
|
77
|
+
adaptive_backfilling_count: int | Unset = UNSET
|
|
78
|
+
adaptive_ready_count: int | Unset = UNSET
|
|
79
|
+
adaptive_stale_count: int | Unset = UNSET
|
|
80
|
+
adaptive_cleanup_recommended_count: int | Unset = UNSET
|
|
81
|
+
last_error_reason: str | Unset = UNSET
|
|
82
|
+
active_progress_lifecycle: str | Unset = UNSET
|
|
83
|
+
active_progress_rows_processed: int | Unset = UNSET
|
|
84
|
+
active_progress_target_rows: int | Unset = UNSET
|
|
85
|
+
additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
|
|
86
|
+
|
|
87
|
+
def to_dict(self) -> dict[str, Any]:
|
|
88
|
+
index_type = self.index_type.value
|
|
89
|
+
|
|
90
|
+
error = self.error
|
|
91
|
+
|
|
92
|
+
total_indexed = self.total_indexed
|
|
93
|
+
|
|
94
|
+
disk_usage = self.disk_usage
|
|
95
|
+
|
|
96
|
+
rebuilding = self.rebuilding
|
|
97
|
+
|
|
98
|
+
backfill_progress = self.backfill_progress
|
|
99
|
+
|
|
100
|
+
backfill_items_processed = self.backfill_items_processed
|
|
101
|
+
|
|
102
|
+
healthy = self.healthy
|
|
103
|
+
|
|
104
|
+
parse_error_count = self.parse_error_count
|
|
105
|
+
|
|
106
|
+
schema_version = self.schema_version
|
|
107
|
+
|
|
108
|
+
capability_lifecycle_status = self.capability_lifecycle_status
|
|
109
|
+
|
|
110
|
+
planner_selected = self.planner_selected
|
|
111
|
+
|
|
112
|
+
planner_fallback_count = self.planner_fallback_count
|
|
113
|
+
|
|
114
|
+
planner_last_decision: str | Unset = UNSET
|
|
115
|
+
if not isinstance(self.planner_last_decision, Unset):
|
|
116
|
+
planner_last_decision = self.planner_last_decision.value
|
|
117
|
+
|
|
118
|
+
planner_last_fallback_reason = self.planner_last_fallback_reason
|
|
119
|
+
|
|
120
|
+
planner_last_estimated_scan_rows = self.planner_last_estimated_scan_rows
|
|
121
|
+
|
|
122
|
+
planner_last_estimated_result_buckets = self.planner_last_estimated_result_buckets
|
|
123
|
+
|
|
124
|
+
planner_lifecycle_ready = self.planner_lifecycle_ready
|
|
125
|
+
|
|
126
|
+
planner_lifecycle_blocking_reason = self.planner_lifecycle_blocking_reason
|
|
127
|
+
|
|
128
|
+
adaptive_progress_count = self.adaptive_progress_count
|
|
129
|
+
|
|
130
|
+
recommendation_count = self.recommendation_count
|
|
131
|
+
|
|
132
|
+
adaptive_backfilling_count = self.adaptive_backfilling_count
|
|
133
|
+
|
|
134
|
+
adaptive_ready_count = self.adaptive_ready_count
|
|
135
|
+
|
|
136
|
+
adaptive_stale_count = self.adaptive_stale_count
|
|
137
|
+
|
|
138
|
+
adaptive_cleanup_recommended_count = self.adaptive_cleanup_recommended_count
|
|
139
|
+
|
|
140
|
+
last_error_reason = self.last_error_reason
|
|
141
|
+
|
|
142
|
+
active_progress_lifecycle = self.active_progress_lifecycle
|
|
143
|
+
|
|
144
|
+
active_progress_rows_processed = self.active_progress_rows_processed
|
|
145
|
+
|
|
146
|
+
active_progress_target_rows = self.active_progress_target_rows
|
|
147
|
+
|
|
148
|
+
field_dict: dict[str, Any] = {}
|
|
149
|
+
field_dict.update(self.additional_properties)
|
|
150
|
+
field_dict.update(
|
|
151
|
+
{
|
|
152
|
+
"index_type": index_type,
|
|
153
|
+
}
|
|
154
|
+
)
|
|
155
|
+
if error is not UNSET:
|
|
156
|
+
field_dict["error"] = error
|
|
157
|
+
if total_indexed is not UNSET:
|
|
158
|
+
field_dict["total_indexed"] = total_indexed
|
|
159
|
+
if disk_usage is not UNSET:
|
|
160
|
+
field_dict["disk_usage"] = disk_usage
|
|
161
|
+
if rebuilding is not UNSET:
|
|
162
|
+
field_dict["rebuilding"] = rebuilding
|
|
163
|
+
if backfill_progress is not UNSET:
|
|
164
|
+
field_dict["backfill_progress"] = backfill_progress
|
|
165
|
+
if backfill_items_processed is not UNSET:
|
|
166
|
+
field_dict["backfill_items_processed"] = backfill_items_processed
|
|
167
|
+
if healthy is not UNSET:
|
|
168
|
+
field_dict["healthy"] = healthy
|
|
169
|
+
if parse_error_count is not UNSET:
|
|
170
|
+
field_dict["parse_error_count"] = parse_error_count
|
|
171
|
+
if schema_version is not UNSET:
|
|
172
|
+
field_dict["schema_version"] = schema_version
|
|
173
|
+
if capability_lifecycle_status is not UNSET:
|
|
174
|
+
field_dict["capability_lifecycle_status"] = capability_lifecycle_status
|
|
175
|
+
if planner_selected is not UNSET:
|
|
176
|
+
field_dict["planner_selected"] = planner_selected
|
|
177
|
+
if planner_fallback_count is not UNSET:
|
|
178
|
+
field_dict["planner_fallback_count"] = planner_fallback_count
|
|
179
|
+
if planner_last_decision is not UNSET:
|
|
180
|
+
field_dict["planner_last_decision"] = planner_last_decision
|
|
181
|
+
if planner_last_fallback_reason is not UNSET:
|
|
182
|
+
field_dict["planner_last_fallback_reason"] = planner_last_fallback_reason
|
|
183
|
+
if planner_last_estimated_scan_rows is not UNSET:
|
|
184
|
+
field_dict["planner_last_estimated_scan_rows"] = planner_last_estimated_scan_rows
|
|
185
|
+
if planner_last_estimated_result_buckets is not UNSET:
|
|
186
|
+
field_dict["planner_last_estimated_result_buckets"] = planner_last_estimated_result_buckets
|
|
187
|
+
if planner_lifecycle_ready is not UNSET:
|
|
188
|
+
field_dict["planner_lifecycle_ready"] = planner_lifecycle_ready
|
|
189
|
+
if planner_lifecycle_blocking_reason is not UNSET:
|
|
190
|
+
field_dict["planner_lifecycle_blocking_reason"] = planner_lifecycle_blocking_reason
|
|
191
|
+
if adaptive_progress_count is not UNSET:
|
|
192
|
+
field_dict["adaptive_progress_count"] = adaptive_progress_count
|
|
193
|
+
if recommendation_count is not UNSET:
|
|
194
|
+
field_dict["recommendation_count"] = recommendation_count
|
|
195
|
+
if adaptive_backfilling_count is not UNSET:
|
|
196
|
+
field_dict["adaptive_backfilling_count"] = adaptive_backfilling_count
|
|
197
|
+
if adaptive_ready_count is not UNSET:
|
|
198
|
+
field_dict["adaptive_ready_count"] = adaptive_ready_count
|
|
199
|
+
if adaptive_stale_count is not UNSET:
|
|
200
|
+
field_dict["adaptive_stale_count"] = adaptive_stale_count
|
|
201
|
+
if adaptive_cleanup_recommended_count is not UNSET:
|
|
202
|
+
field_dict["adaptive_cleanup_recommended_count"] = adaptive_cleanup_recommended_count
|
|
203
|
+
if last_error_reason is not UNSET:
|
|
204
|
+
field_dict["last_error_reason"] = last_error_reason
|
|
205
|
+
if active_progress_lifecycle is not UNSET:
|
|
206
|
+
field_dict["active_progress_lifecycle"] = active_progress_lifecycle
|
|
207
|
+
if active_progress_rows_processed is not UNSET:
|
|
208
|
+
field_dict["active_progress_rows_processed"] = active_progress_rows_processed
|
|
209
|
+
if active_progress_target_rows is not UNSET:
|
|
210
|
+
field_dict["active_progress_target_rows"] = active_progress_target_rows
|
|
211
|
+
|
|
212
|
+
return field_dict
|
|
213
|
+
|
|
214
|
+
@classmethod
|
|
215
|
+
def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
|
|
216
|
+
d = dict(src_dict)
|
|
217
|
+
index_type = AlgebraicIndexStatsIndexType(d.pop("index_type"))
|
|
218
|
+
|
|
219
|
+
error = d.pop("error", UNSET)
|
|
220
|
+
|
|
221
|
+
total_indexed = d.pop("total_indexed", UNSET)
|
|
222
|
+
|
|
223
|
+
disk_usage = d.pop("disk_usage", UNSET)
|
|
224
|
+
|
|
225
|
+
rebuilding = d.pop("rebuilding", UNSET)
|
|
226
|
+
|
|
227
|
+
backfill_progress = d.pop("backfill_progress", UNSET)
|
|
228
|
+
|
|
229
|
+
backfill_items_processed = d.pop("backfill_items_processed", UNSET)
|
|
230
|
+
|
|
231
|
+
healthy = d.pop("healthy", UNSET)
|
|
232
|
+
|
|
233
|
+
parse_error_count = d.pop("parse_error_count", UNSET)
|
|
234
|
+
|
|
235
|
+
schema_version = d.pop("schema_version", UNSET)
|
|
236
|
+
|
|
237
|
+
capability_lifecycle_status = d.pop("capability_lifecycle_status", UNSET)
|
|
238
|
+
|
|
239
|
+
planner_selected = d.pop("planner_selected", UNSET)
|
|
240
|
+
|
|
241
|
+
planner_fallback_count = d.pop("planner_fallback_count", UNSET)
|
|
242
|
+
|
|
243
|
+
_planner_last_decision = d.pop("planner_last_decision", UNSET)
|
|
244
|
+
planner_last_decision: AlgebraicIndexStatsPlannerLastDecision | Unset
|
|
245
|
+
if isinstance(_planner_last_decision, Unset):
|
|
246
|
+
planner_last_decision = UNSET
|
|
247
|
+
else:
|
|
248
|
+
planner_last_decision = AlgebraicIndexStatsPlannerLastDecision(_planner_last_decision)
|
|
249
|
+
|
|
250
|
+
planner_last_fallback_reason = d.pop("planner_last_fallback_reason", UNSET)
|
|
251
|
+
|
|
252
|
+
planner_last_estimated_scan_rows = d.pop("planner_last_estimated_scan_rows", UNSET)
|
|
253
|
+
|
|
254
|
+
planner_last_estimated_result_buckets = d.pop("planner_last_estimated_result_buckets", UNSET)
|
|
255
|
+
|
|
256
|
+
planner_lifecycle_ready = d.pop("planner_lifecycle_ready", UNSET)
|
|
257
|
+
|
|
258
|
+
planner_lifecycle_blocking_reason = d.pop("planner_lifecycle_blocking_reason", UNSET)
|
|
259
|
+
|
|
260
|
+
adaptive_progress_count = d.pop("adaptive_progress_count", UNSET)
|
|
261
|
+
|
|
262
|
+
recommendation_count = d.pop("recommendation_count", UNSET)
|
|
263
|
+
|
|
264
|
+
adaptive_backfilling_count = d.pop("adaptive_backfilling_count", UNSET)
|
|
265
|
+
|
|
266
|
+
adaptive_ready_count = d.pop("adaptive_ready_count", UNSET)
|
|
267
|
+
|
|
268
|
+
adaptive_stale_count = d.pop("adaptive_stale_count", UNSET)
|
|
269
|
+
|
|
270
|
+
adaptive_cleanup_recommended_count = d.pop("adaptive_cleanup_recommended_count", UNSET)
|
|
271
|
+
|
|
272
|
+
last_error_reason = d.pop("last_error_reason", UNSET)
|
|
273
|
+
|
|
274
|
+
active_progress_lifecycle = d.pop("active_progress_lifecycle", UNSET)
|
|
275
|
+
|
|
276
|
+
active_progress_rows_processed = d.pop("active_progress_rows_processed", UNSET)
|
|
277
|
+
|
|
278
|
+
active_progress_target_rows = d.pop("active_progress_target_rows", UNSET)
|
|
279
|
+
|
|
280
|
+
algebraic_index_stats = cls(
|
|
281
|
+
index_type=index_type,
|
|
282
|
+
error=error,
|
|
283
|
+
total_indexed=total_indexed,
|
|
284
|
+
disk_usage=disk_usage,
|
|
285
|
+
rebuilding=rebuilding,
|
|
286
|
+
backfill_progress=backfill_progress,
|
|
287
|
+
backfill_items_processed=backfill_items_processed,
|
|
288
|
+
healthy=healthy,
|
|
289
|
+
parse_error_count=parse_error_count,
|
|
290
|
+
schema_version=schema_version,
|
|
291
|
+
capability_lifecycle_status=capability_lifecycle_status,
|
|
292
|
+
planner_selected=planner_selected,
|
|
293
|
+
planner_fallback_count=planner_fallback_count,
|
|
294
|
+
planner_last_decision=planner_last_decision,
|
|
295
|
+
planner_last_fallback_reason=planner_last_fallback_reason,
|
|
296
|
+
planner_last_estimated_scan_rows=planner_last_estimated_scan_rows,
|
|
297
|
+
planner_last_estimated_result_buckets=planner_last_estimated_result_buckets,
|
|
298
|
+
planner_lifecycle_ready=planner_lifecycle_ready,
|
|
299
|
+
planner_lifecycle_blocking_reason=planner_lifecycle_blocking_reason,
|
|
300
|
+
adaptive_progress_count=adaptive_progress_count,
|
|
301
|
+
recommendation_count=recommendation_count,
|
|
302
|
+
adaptive_backfilling_count=adaptive_backfilling_count,
|
|
303
|
+
adaptive_ready_count=adaptive_ready_count,
|
|
304
|
+
adaptive_stale_count=adaptive_stale_count,
|
|
305
|
+
adaptive_cleanup_recommended_count=adaptive_cleanup_recommended_count,
|
|
306
|
+
last_error_reason=last_error_reason,
|
|
307
|
+
active_progress_lifecycle=active_progress_lifecycle,
|
|
308
|
+
active_progress_rows_processed=active_progress_rows_processed,
|
|
309
|
+
active_progress_target_rows=active_progress_target_rows,
|
|
310
|
+
)
|
|
311
|
+
|
|
312
|
+
algebraic_index_stats.additional_properties = d
|
|
313
|
+
return algebraic_index_stats
|
|
314
|
+
|
|
315
|
+
@property
|
|
316
|
+
def additional_keys(self) -> list[str]:
|
|
317
|
+
return list(self.additional_properties.keys())
|
|
318
|
+
|
|
319
|
+
def __getitem__(self, key: str) -> Any:
|
|
320
|
+
return self.additional_properties[key]
|
|
321
|
+
|
|
322
|
+
def __setitem__(self, key: str, value: Any) -> None:
|
|
323
|
+
self.additional_properties[key] = value
|
|
324
|
+
|
|
325
|
+
def __delitem__(self, key: str) -> None:
|
|
326
|
+
del self.additional_properties[key]
|
|
327
|
+
|
|
328
|
+
def __contains__(self, key: str) -> bool:
|
|
329
|
+
return key in self.additional_properties
|
|
@@ -0,0 +1,70 @@
|
|
|
1
|
+
from __future__ import annotations
|
|
2
|
+
|
|
3
|
+
from collections.abc import Mapping
|
|
4
|
+
from typing import Any, TypeVar
|
|
5
|
+
|
|
6
|
+
from attrs import define as _attrs_define
|
|
7
|
+
from attrs import field as _attrs_field
|
|
8
|
+
|
|
9
|
+
from ..types import UNSET, Unset
|
|
10
|
+
|
|
11
|
+
T = TypeVar("T", bound="Analyses")
|
|
12
|
+
|
|
13
|
+
|
|
14
|
+
@_attrs_define
|
|
15
|
+
class Analyses:
|
|
16
|
+
"""
|
|
17
|
+
Attributes:
|
|
18
|
+
pca (bool | Unset):
|
|
19
|
+
tsne (bool | Unset):
|
|
20
|
+
"""
|
|
21
|
+
|
|
22
|
+
pca: bool | Unset = UNSET
|
|
23
|
+
tsne: bool | Unset = UNSET
|
|
24
|
+
additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
|
|
25
|
+
|
|
26
|
+
def to_dict(self) -> dict[str, Any]:
|
|
27
|
+
pca = self.pca
|
|
28
|
+
|
|
29
|
+
tsne = self.tsne
|
|
30
|
+
|
|
31
|
+
field_dict: dict[str, Any] = {}
|
|
32
|
+
field_dict.update(self.additional_properties)
|
|
33
|
+
field_dict.update({})
|
|
34
|
+
if pca is not UNSET:
|
|
35
|
+
field_dict["pca"] = pca
|
|
36
|
+
if tsne is not UNSET:
|
|
37
|
+
field_dict["tsne"] = tsne
|
|
38
|
+
|
|
39
|
+
return field_dict
|
|
40
|
+
|
|
41
|
+
@classmethod
|
|
42
|
+
def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
|
|
43
|
+
d = dict(src_dict)
|
|
44
|
+
pca = d.pop("pca", UNSET)
|
|
45
|
+
|
|
46
|
+
tsne = d.pop("tsne", UNSET)
|
|
47
|
+
|
|
48
|
+
analyses = cls(
|
|
49
|
+
pca=pca,
|
|
50
|
+
tsne=tsne,
|
|
51
|
+
)
|
|
52
|
+
|
|
53
|
+
analyses.additional_properties = d
|
|
54
|
+
return analyses
|
|
55
|
+
|
|
56
|
+
@property
|
|
57
|
+
def additional_keys(self) -> list[str]:
|
|
58
|
+
return list(self.additional_properties.keys())
|
|
59
|
+
|
|
60
|
+
def __getitem__(self, key: str) -> Any:
|
|
61
|
+
return self.additional_properties[key]
|
|
62
|
+
|
|
63
|
+
def __setitem__(self, key: str, value: Any) -> None:
|
|
64
|
+
self.additional_properties[key] = value
|
|
65
|
+
|
|
66
|
+
def __delitem__(self, key: str) -> None:
|
|
67
|
+
del self.additional_properties[key]
|
|
68
|
+
|
|
69
|
+
def __contains__(self, key: str) -> bool:
|
|
70
|
+
return key in self.additional_properties
|
|
@@ -0,0 +1,74 @@
|
|
|
1
|
+
from __future__ import annotations
|
|
2
|
+
|
|
3
|
+
from collections.abc import Mapping
|
|
4
|
+
from typing import Any, TypeVar, cast
|
|
5
|
+
|
|
6
|
+
from attrs import define as _attrs_define
|
|
7
|
+
from attrs import field as _attrs_field
|
|
8
|
+
|
|
9
|
+
from ..types import UNSET, Unset
|
|
10
|
+
|
|
11
|
+
T = TypeVar("T", bound="AnalysesResult")
|
|
12
|
+
|
|
13
|
+
|
|
14
|
+
@_attrs_define
|
|
15
|
+
class AnalysesResult:
|
|
16
|
+
"""
|
|
17
|
+
Attributes:
|
|
18
|
+
pca (list[float] | Unset):
|
|
19
|
+
tsne (list[float] | Unset):
|
|
20
|
+
"""
|
|
21
|
+
|
|
22
|
+
pca: list[float] | Unset = UNSET
|
|
23
|
+
tsne: list[float] | Unset = UNSET
|
|
24
|
+
additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
|
|
25
|
+
|
|
26
|
+
def to_dict(self) -> dict[str, Any]:
|
|
27
|
+
pca: list[float] | Unset = UNSET
|
|
28
|
+
if not isinstance(self.pca, Unset):
|
|
29
|
+
pca = self.pca
|
|
30
|
+
|
|
31
|
+
tsne: list[float] | Unset = UNSET
|
|
32
|
+
if not isinstance(self.tsne, Unset):
|
|
33
|
+
tsne = self.tsne
|
|
34
|
+
|
|
35
|
+
field_dict: dict[str, Any] = {}
|
|
36
|
+
field_dict.update(self.additional_properties)
|
|
37
|
+
field_dict.update({})
|
|
38
|
+
if pca is not UNSET:
|
|
39
|
+
field_dict["pca"] = pca
|
|
40
|
+
if tsne is not UNSET:
|
|
41
|
+
field_dict["tsne"] = tsne
|
|
42
|
+
|
|
43
|
+
return field_dict
|
|
44
|
+
|
|
45
|
+
@classmethod
|
|
46
|
+
def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
|
|
47
|
+
d = dict(src_dict)
|
|
48
|
+
pca = cast(list[float], d.pop("pca", UNSET))
|
|
49
|
+
|
|
50
|
+
tsne = cast(list[float], d.pop("tsne", UNSET))
|
|
51
|
+
|
|
52
|
+
analyses_result = cls(
|
|
53
|
+
pca=pca,
|
|
54
|
+
tsne=tsne,
|
|
55
|
+
)
|
|
56
|
+
|
|
57
|
+
analyses_result.additional_properties = d
|
|
58
|
+
return analyses_result
|
|
59
|
+
|
|
60
|
+
@property
|
|
61
|
+
def additional_keys(self) -> list[str]:
|
|
62
|
+
return list(self.additional_properties.keys())
|
|
63
|
+
|
|
64
|
+
def __getitem__(self, key: str) -> Any:
|
|
65
|
+
return self.additional_properties[key]
|
|
66
|
+
|
|
67
|
+
def __setitem__(self, key: str, value: Any) -> None:
|
|
68
|
+
self.additional_properties[key] = value
|
|
69
|
+
|
|
70
|
+
def __delitem__(self, key: str) -> None:
|
|
71
|
+
del self.additional_properties[key]
|
|
72
|
+
|
|
73
|
+
def __contains__(self, key: str) -> bool:
|
|
74
|
+
return key in self.additional_properties
|
|
@@ -0,0 +1,159 @@
|
|
|
1
|
+
from __future__ import annotations
|
|
2
|
+
|
|
3
|
+
from collections.abc import Mapping
|
|
4
|
+
from typing import TYPE_CHECKING, Any, TypeVar, cast
|
|
5
|
+
|
|
6
|
+
from attrs import define as _attrs_define
|
|
7
|
+
from attrs import field as _attrs_field
|
|
8
|
+
|
|
9
|
+
from ..types import UNSET, Unset
|
|
10
|
+
|
|
11
|
+
if TYPE_CHECKING:
|
|
12
|
+
from ..models.classification_transformation_result import ClassificationTransformationResult
|
|
13
|
+
from ..models.eval_result import EvalResult
|
|
14
|
+
from ..models.query_result import QueryResult
|
|
15
|
+
|
|
16
|
+
|
|
17
|
+
T = TypeVar("T", bound="AnswerAgentResult")
|
|
18
|
+
|
|
19
|
+
|
|
20
|
+
@_attrs_define
|
|
21
|
+
class AnswerAgentResult:
|
|
22
|
+
"""DEPRECATED: Use RetrievalAgentResult instead.
|
|
23
|
+
Result from the answer agent.
|
|
24
|
+
|
|
25
|
+
Attributes:
|
|
26
|
+
answer (str | Unset): DEPRECATED: Use generation on RetrievalAgentResult instead.
|
|
27
|
+
Generated answer in markdown format.
|
|
28
|
+
answer_confidence (float | Unset): DEPRECATED: Use generation_confidence on RetrievalAgentResult instead.
|
|
29
|
+
Confidence in the generated answer.
|
|
30
|
+
context_relevance (float | Unset): Relevance of retrieved documents to the query
|
|
31
|
+
classification_transformation (ClassificationTransformationResult | Unset): Query classification and
|
|
32
|
+
transformation result combining all query enhancements including strategy selection and semantic optimization
|
|
33
|
+
query_results (list[QueryResult] | Unset): DEPRECATED: Use hits on RetrievalAgentResult instead.
|
|
34
|
+
Query results grouped by table.
|
|
35
|
+
followup_questions (list[str] | Unset): Suggested follow-up questions
|
|
36
|
+
eval_result (EvalResult | Unset): Complete evaluation result
|
|
37
|
+
"""
|
|
38
|
+
|
|
39
|
+
answer: str | Unset = UNSET
|
|
40
|
+
answer_confidence: float | Unset = UNSET
|
|
41
|
+
context_relevance: float | Unset = UNSET
|
|
42
|
+
classification_transformation: ClassificationTransformationResult | Unset = UNSET
|
|
43
|
+
query_results: list[QueryResult] | Unset = UNSET
|
|
44
|
+
followup_questions: list[str] | Unset = UNSET
|
|
45
|
+
eval_result: EvalResult | Unset = UNSET
|
|
46
|
+
additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
|
|
47
|
+
|
|
48
|
+
def to_dict(self) -> dict[str, Any]:
|
|
49
|
+
answer = self.answer
|
|
50
|
+
|
|
51
|
+
answer_confidence = self.answer_confidence
|
|
52
|
+
|
|
53
|
+
context_relevance = self.context_relevance
|
|
54
|
+
|
|
55
|
+
classification_transformation: dict[str, Any] | Unset = UNSET
|
|
56
|
+
if not isinstance(self.classification_transformation, Unset):
|
|
57
|
+
classification_transformation = self.classification_transformation.to_dict()
|
|
58
|
+
|
|
59
|
+
query_results: list[dict[str, Any]] | Unset = UNSET
|
|
60
|
+
if not isinstance(self.query_results, Unset):
|
|
61
|
+
query_results = []
|
|
62
|
+
for query_results_item_data in self.query_results:
|
|
63
|
+
query_results_item = query_results_item_data.to_dict()
|
|
64
|
+
query_results.append(query_results_item)
|
|
65
|
+
|
|
66
|
+
followup_questions: list[str] | Unset = UNSET
|
|
67
|
+
if not isinstance(self.followup_questions, Unset):
|
|
68
|
+
followup_questions = self.followup_questions
|
|
69
|
+
|
|
70
|
+
eval_result: dict[str, Any] | Unset = UNSET
|
|
71
|
+
if not isinstance(self.eval_result, Unset):
|
|
72
|
+
eval_result = self.eval_result.to_dict()
|
|
73
|
+
|
|
74
|
+
field_dict: dict[str, Any] = {}
|
|
75
|
+
field_dict.update(self.additional_properties)
|
|
76
|
+
field_dict.update({})
|
|
77
|
+
if answer is not UNSET:
|
|
78
|
+
field_dict["answer"] = answer
|
|
79
|
+
if answer_confidence is not UNSET:
|
|
80
|
+
field_dict["answer_confidence"] = answer_confidence
|
|
81
|
+
if context_relevance is not UNSET:
|
|
82
|
+
field_dict["context_relevance"] = context_relevance
|
|
83
|
+
if classification_transformation is not UNSET:
|
|
84
|
+
field_dict["classification_transformation"] = classification_transformation
|
|
85
|
+
if query_results is not UNSET:
|
|
86
|
+
field_dict["query_results"] = query_results
|
|
87
|
+
if followup_questions is not UNSET:
|
|
88
|
+
field_dict["followup_questions"] = followup_questions
|
|
89
|
+
if eval_result is not UNSET:
|
|
90
|
+
field_dict["eval_result"] = eval_result
|
|
91
|
+
|
|
92
|
+
return field_dict
|
|
93
|
+
|
|
94
|
+
@classmethod
|
|
95
|
+
def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
|
|
96
|
+
from ..models.classification_transformation_result import ClassificationTransformationResult
|
|
97
|
+
from ..models.eval_result import EvalResult
|
|
98
|
+
from ..models.query_result import QueryResult
|
|
99
|
+
|
|
100
|
+
d = dict(src_dict)
|
|
101
|
+
answer = d.pop("answer", UNSET)
|
|
102
|
+
|
|
103
|
+
answer_confidence = d.pop("answer_confidence", UNSET)
|
|
104
|
+
|
|
105
|
+
context_relevance = d.pop("context_relevance", UNSET)
|
|
106
|
+
|
|
107
|
+
_classification_transformation = d.pop("classification_transformation", UNSET)
|
|
108
|
+
classification_transformation: ClassificationTransformationResult | Unset
|
|
109
|
+
if isinstance(_classification_transformation, Unset):
|
|
110
|
+
classification_transformation = UNSET
|
|
111
|
+
else:
|
|
112
|
+
classification_transformation = ClassificationTransformationResult.from_dict(_classification_transformation)
|
|
113
|
+
|
|
114
|
+
_query_results = d.pop("query_results", UNSET)
|
|
115
|
+
query_results: list[QueryResult] | Unset = UNSET
|
|
116
|
+
if _query_results is not UNSET:
|
|
117
|
+
query_results = []
|
|
118
|
+
for query_results_item_data in _query_results:
|
|
119
|
+
query_results_item = QueryResult.from_dict(query_results_item_data)
|
|
120
|
+
|
|
121
|
+
query_results.append(query_results_item)
|
|
122
|
+
|
|
123
|
+
followup_questions = cast(list[str], d.pop("followup_questions", UNSET))
|
|
124
|
+
|
|
125
|
+
_eval_result = d.pop("eval_result", UNSET)
|
|
126
|
+
eval_result: EvalResult | Unset
|
|
127
|
+
if isinstance(_eval_result, Unset):
|
|
128
|
+
eval_result = UNSET
|
|
129
|
+
else:
|
|
130
|
+
eval_result = EvalResult.from_dict(_eval_result)
|
|
131
|
+
|
|
132
|
+
answer_agent_result = cls(
|
|
133
|
+
answer=answer,
|
|
134
|
+
answer_confidence=answer_confidence,
|
|
135
|
+
context_relevance=context_relevance,
|
|
136
|
+
classification_transformation=classification_transformation,
|
|
137
|
+
query_results=query_results,
|
|
138
|
+
followup_questions=followup_questions,
|
|
139
|
+
eval_result=eval_result,
|
|
140
|
+
)
|
|
141
|
+
|
|
142
|
+
answer_agent_result.additional_properties = d
|
|
143
|
+
return answer_agent_result
|
|
144
|
+
|
|
145
|
+
@property
|
|
146
|
+
def additional_keys(self) -> list[str]:
|
|
147
|
+
return list(self.additional_properties.keys())
|
|
148
|
+
|
|
149
|
+
def __getitem__(self, key: str) -> Any:
|
|
150
|
+
return self.additional_properties[key]
|
|
151
|
+
|
|
152
|
+
def __setitem__(self, key: str, value: Any) -> None:
|
|
153
|
+
self.additional_properties[key] = value
|
|
154
|
+
|
|
155
|
+
def __delitem__(self, key: str) -> None:
|
|
156
|
+
del self.additional_properties[key]
|
|
157
|
+
|
|
158
|
+
def __contains__(self, key: str) -> bool:
|
|
159
|
+
return key in self.additional_properties
|