llama-cloud 0.1.41__py3-none-any.whl → 1.0.0b4__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.
- llama_cloud/__init__.py +101 -816
- llama_cloud/_base_client.py +2124 -0
- llama_cloud/_client.py +795 -0
- llama_cloud/_compat.py +219 -0
- llama_cloud/_constants.py +14 -0
- llama_cloud/_exceptions.py +108 -0
- llama_cloud/_files.py +127 -0
- llama_cloud/_models.py +872 -0
- llama_cloud/_polling.py +182 -0
- llama_cloud/_qs.py +150 -0
- llama_cloud/_resource.py +43 -0
- llama_cloud/_response.py +832 -0
- llama_cloud/_streaming.py +333 -0
- llama_cloud/_types.py +270 -0
- llama_cloud/_utils/__init__.py +64 -0
- llama_cloud/_utils/_compat.py +45 -0
- llama_cloud/_utils/_datetime_parse.py +136 -0
- llama_cloud/_utils/_logs.py +25 -0
- llama_cloud/_utils/_proxy.py +65 -0
- llama_cloud/_utils/_reflection.py +42 -0
- llama_cloud/_utils/_resources_proxy.py +24 -0
- llama_cloud/_utils/_streams.py +12 -0
- llama_cloud/_utils/_sync.py +58 -0
- llama_cloud/_utils/_transform.py +457 -0
- llama_cloud/_utils/_typing.py +156 -0
- llama_cloud/_utils/_utils.py +421 -0
- llama_cloud/_version.py +4 -0
- llama_cloud/lib/__init__.py +0 -0
- llama_cloud/lib/index/__init__.py +13 -0
- llama_cloud/lib/index/api_utils.py +300 -0
- llama_cloud/lib/index/base.py +1041 -0
- llama_cloud/lib/index/composite_retriever.py +272 -0
- llama_cloud/lib/index/retriever.py +233 -0
- llama_cloud/pagination.py +465 -0
- llama_cloud/py.typed +0 -0
- llama_cloud/resources/__init__.py +136 -107
- llama_cloud/resources/beta/__init__.py +102 -1
- llama_cloud/resources/beta/agent_data.py +1041 -0
- llama_cloud/resources/beta/batch/__init__.py +33 -0
- llama_cloud/resources/beta/batch/batch.py +664 -0
- llama_cloud/resources/beta/batch/job_items.py +348 -0
- llama_cloud/resources/beta/beta.py +262 -0
- llama_cloud/resources/beta/directories/__init__.py +33 -0
- llama_cloud/resources/beta/directories/directories.py +719 -0
- llama_cloud/resources/beta/directories/files.py +913 -0
- llama_cloud/resources/beta/parse_configurations.py +743 -0
- llama_cloud/resources/beta/sheets.py +1130 -0
- llama_cloud/resources/beta/split.py +917 -0
- llama_cloud/resources/classifier/__init__.py +32 -1
- llama_cloud/resources/classifier/classifier.py +588 -0
- llama_cloud/resources/classifier/jobs.py +563 -0
- llama_cloud/resources/data_sinks.py +579 -0
- llama_cloud/resources/data_sources.py +651 -0
- llama_cloud/resources/extraction/__init__.py +61 -0
- llama_cloud/resources/extraction/extraction.py +609 -0
- llama_cloud/resources/extraction/extraction_agents/__init__.py +33 -0
- llama_cloud/resources/extraction/extraction_agents/extraction_agents.py +633 -0
- llama_cloud/resources/extraction/extraction_agents/schema.py +308 -0
- llama_cloud/resources/extraction/jobs.py +1106 -0
- llama_cloud/resources/extraction/runs.py +498 -0
- llama_cloud/resources/files.py +784 -0
- llama_cloud/resources/parsing.py +1296 -0
- llama_cloud/resources/pipelines/__init__.py +98 -24
- llama_cloud/resources/pipelines/data_sources.py +529 -0
- llama_cloud/resources/pipelines/documents.py +810 -0
- llama_cloud/resources/pipelines/files.py +682 -0
- llama_cloud/resources/pipelines/images.py +513 -0
- llama_cloud/resources/pipelines/metadata.py +265 -0
- llama_cloud/resources/pipelines/pipelines.py +1525 -0
- llama_cloud/resources/pipelines/sync.py +243 -0
- llama_cloud/resources/projects.py +276 -0
- llama_cloud/resources/retrievers/__init__.py +32 -1
- llama_cloud/resources/retrievers/retriever.py +238 -0
- llama_cloud/resources/retrievers/retrievers.py +920 -0
- llama_cloud/types/__init__.py +171 -721
- llama_cloud/types/advanced_mode_transform_config.py +102 -38
- llama_cloud/types/advanced_mode_transform_config_param.py +102 -0
- llama_cloud/types/auto_transform_config.py +11 -25
- llama_cloud/types/auto_transform_config_param.py +17 -0
- llama_cloud/types/azure_openai_embedding.py +62 -0
- llama_cloud/types/azure_openai_embedding_config.py +17 -0
- llama_cloud/types/azure_openai_embedding_config_param.py +17 -0
- llama_cloud/types/azure_openai_embedding_param.py +61 -0
- llama_cloud/types/b_box.py +37 -0
- llama_cloud/types/bedrock_embedding.py +49 -46
- llama_cloud/types/bedrock_embedding_config.py +10 -27
- llama_cloud/types/bedrock_embedding_config_param.py +17 -0
- llama_cloud/types/bedrock_embedding_param.py +48 -0
- llama_cloud/types/beta/__init__.py +59 -0
- llama_cloud/types/beta/agent_data.py +26 -0
- llama_cloud/types/beta/agent_data_agent_data_params.py +20 -0
- llama_cloud/types/beta/agent_data_aggregate_params.py +79 -0
- llama_cloud/types/beta/agent_data_aggregate_response.py +17 -0
- llama_cloud/types/beta/agent_data_delete_by_query_params.py +43 -0
- llama_cloud/types/beta/agent_data_delete_by_query_response.py +11 -0
- llama_cloud/types/beta/agent_data_delete_params.py +14 -0
- llama_cloud/types/beta/agent_data_delete_response.py +8 -0
- llama_cloud/types/beta/agent_data_get_params.py +14 -0
- llama_cloud/types/beta/agent_data_search_params.py +69 -0
- llama_cloud/types/beta/agent_data_update_params.py +16 -0
- llama_cloud/types/beta/batch/__init__.py +12 -0
- llama_cloud/types/beta/batch/job_item_get_processing_results_params.py +17 -0
- llama_cloud/types/beta/batch/job_item_get_processing_results_response.py +409 -0
- llama_cloud/types/beta/batch/job_item_list_params.py +23 -0
- llama_cloud/types/beta/batch/job_item_list_response.py +42 -0
- llama_cloud/types/beta/batch_cancel_params.py +21 -0
- llama_cloud/types/beta/batch_cancel_response.py +23 -0
- llama_cloud/types/beta/batch_create_params.py +399 -0
- llama_cloud/types/beta/batch_create_response.py +63 -0
- llama_cloud/types/beta/batch_get_status_params.py +14 -0
- llama_cloud/types/beta/batch_get_status_response.py +73 -0
- llama_cloud/types/beta/batch_list_params.py +29 -0
- llama_cloud/types/beta/batch_list_response.py +63 -0
- llama_cloud/types/beta/directories/__init__.py +15 -0
- llama_cloud/types/beta/directories/file_add_params.py +26 -0
- llama_cloud/types/beta/directories/file_add_response.py +42 -0
- llama_cloud/types/beta/directories/file_delete_params.py +16 -0
- llama_cloud/types/beta/directories/file_get_params.py +16 -0
- llama_cloud/types/beta/directories/file_get_response.py +42 -0
- llama_cloud/types/beta/directories/file_list_params.py +28 -0
- llama_cloud/types/beta/directories/file_list_response.py +42 -0
- llama_cloud/types/beta/directories/file_update_params.py +27 -0
- llama_cloud/types/beta/directories/file_update_response.py +42 -0
- llama_cloud/types/beta/directories/file_upload_params.py +24 -0
- llama_cloud/types/beta/directories/file_upload_response.py +42 -0
- llama_cloud/types/beta/directory_create_params.py +23 -0
- llama_cloud/types/beta/directory_create_response.py +36 -0
- llama_cloud/types/beta/directory_delete_params.py +14 -0
- llama_cloud/types/beta/directory_get_params.py +14 -0
- llama_cloud/types/beta/directory_get_response.py +36 -0
- llama_cloud/types/beta/directory_list_params.py +24 -0
- llama_cloud/types/beta/directory_list_response.py +36 -0
- llama_cloud/types/beta/directory_update_params.py +20 -0
- llama_cloud/types/beta/directory_update_response.py +36 -0
- llama_cloud/types/beta/parse_configuration.py +40 -0
- llama_cloud/types/beta/parse_configuration_create_params.py +34 -0
- llama_cloud/types/beta/parse_configuration_delete_params.py +14 -0
- llama_cloud/types/beta/parse_configuration_get_params.py +14 -0
- llama_cloud/types/beta/parse_configuration_list_params.py +24 -0
- llama_cloud/types/beta/parse_configuration_query_response.py +28 -0
- llama_cloud/types/beta/parse_configuration_update_params.py +22 -0
- llama_cloud/types/beta/sheet_create_params.py +22 -0
- llama_cloud/types/beta/sheet_delete_job_params.py +14 -0
- llama_cloud/types/beta/sheet_get_params.py +16 -0
- llama_cloud/types/beta/sheet_get_result_table_params.py +20 -0
- llama_cloud/types/beta/sheet_list_params.py +20 -0
- llama_cloud/types/beta/sheets_job.py +88 -0
- llama_cloud/types/beta/sheets_parsing_config.py +49 -0
- llama_cloud/types/beta/sheets_parsing_config_param.py +51 -0
- llama_cloud/types/beta/split_category.py +17 -0
- llama_cloud/types/beta/split_category_param.py +18 -0
- llama_cloud/types/beta/split_create_params.py +36 -0
- llama_cloud/types/beta/split_create_response.py +48 -0
- llama_cloud/types/beta/split_document_input.py +15 -0
- llama_cloud/types/beta/split_document_input_param.py +17 -0
- llama_cloud/types/beta/split_get_params.py +14 -0
- llama_cloud/types/beta/split_get_response.py +48 -0
- llama_cloud/types/beta/split_list_params.py +18 -0
- llama_cloud/types/beta/split_list_response.py +48 -0
- llama_cloud/types/beta/split_result_response.py +15 -0
- llama_cloud/types/beta/split_segment_response.py +20 -0
- llama_cloud/types/classifier/__init__.py +15 -0
- llama_cloud/types/classifier/classifier_rule.py +25 -0
- llama_cloud/types/classifier/classifier_rule_param.py +27 -0
- llama_cloud/types/classifier/classify_job.py +51 -0
- llama_cloud/types/classifier/classify_job_param.py +53 -0
- llama_cloud/types/classifier/classify_parsing_configuration.py +21 -0
- llama_cloud/types/classifier/classify_parsing_configuration_param.py +23 -0
- llama_cloud/types/classifier/job_create_params.py +30 -0
- llama_cloud/types/classifier/job_get_params.py +14 -0
- llama_cloud/types/classifier/job_get_results_params.py +14 -0
- llama_cloud/types/classifier/job_get_results_response.py +66 -0
- llama_cloud/types/classifier/job_list_params.py +18 -0
- llama_cloud/types/cohere_embedding.py +37 -40
- llama_cloud/types/cohere_embedding_config.py +10 -27
- llama_cloud/types/cohere_embedding_config_param.py +17 -0
- llama_cloud/types/cohere_embedding_param.py +36 -0
- llama_cloud/types/composite_retrieval_mode.py +4 -18
- llama_cloud/types/composite_retrieval_result.py +52 -37
- llama_cloud/types/data_sink.py +46 -39
- llama_cloud/types/data_sink_create_param.py +41 -0
- llama_cloud/types/data_sink_create_params.py +44 -0
- llama_cloud/types/data_sink_list_params.py +14 -0
- llama_cloud/types/data_sink_list_response.py +10 -0
- llama_cloud/types/data_sink_update_params.py +40 -0
- llama_cloud/types/data_source.py +67 -39
- llama_cloud/types/data_source_create_params.py +65 -0
- llama_cloud/types/data_source_list_params.py +14 -0
- llama_cloud/types/data_source_list_response.py +10 -0
- llama_cloud/types/data_source_reader_version_metadata.py +8 -27
- llama_cloud/types/data_source_update_params.py +61 -0
- llama_cloud/types/extraction/__init__.py +25 -0
- llama_cloud/types/extraction/extract_agent.py +41 -0
- llama_cloud/types/extraction/extract_config.py +118 -0
- llama_cloud/types/extraction/extract_config_param.py +118 -0
- llama_cloud/types/extraction/extract_job.py +32 -0
- llama_cloud/types/extraction/extract_run.py +64 -0
- llama_cloud/types/extraction/extraction_agent_create_params.py +25 -0
- llama_cloud/types/extraction/extraction_agent_list_params.py +17 -0
- llama_cloud/types/extraction/extraction_agent_list_response.py +10 -0
- llama_cloud/types/extraction/extraction_agent_update_params.py +18 -0
- llama_cloud/types/extraction/extraction_agents/__init__.py +8 -0
- llama_cloud/types/extraction/extraction_agents/schema_generate_schema_params.py +23 -0
- llama_cloud/types/extraction/extraction_agents/schema_generate_schema_response.py +14 -0
- llama_cloud/types/extraction/extraction_agents/schema_validate_schema_params.py +12 -0
- llama_cloud/types/extraction/extraction_agents/schema_validate_schema_response.py +13 -0
- llama_cloud/types/extraction/job_create_params.py +38 -0
- llama_cloud/types/extraction/job_file_params.py +29 -0
- llama_cloud/types/extraction/job_get_result_params.py +14 -0
- llama_cloud/types/extraction/job_get_result_response.py +27 -0
- llama_cloud/types/extraction/job_list_params.py +11 -0
- llama_cloud/types/extraction/job_list_response.py +10 -0
- llama_cloud/types/extraction/run_delete_params.py +14 -0
- llama_cloud/types/extraction/run_get_by_job_params.py +14 -0
- llama_cloud/types/extraction/run_get_params.py +14 -0
- llama_cloud/types/extraction/run_list_params.py +15 -0
- llama_cloud/types/extraction/webhook_configuration.py +43 -0
- llama_cloud/types/extraction/webhook_configuration_param.py +43 -0
- llama_cloud/types/extraction_run_params.py +45 -0
- llama_cloud/types/fail_page_mode.py +4 -26
- llama_cloud/types/file.py +48 -40
- llama_cloud/types/file_create_params.py +28 -0
- llama_cloud/types/file_create_response.py +38 -0
- llama_cloud/types/file_delete_params.py +14 -0
- llama_cloud/types/file_get_params.py +16 -0
- llama_cloud/types/file_list_params.py +40 -0
- llama_cloud/types/file_list_response.py +38 -0
- llama_cloud/types/file_query_params.py +61 -0
- llama_cloud/types/file_query_response.py +47 -27
- llama_cloud/types/gemini_embedding.py +40 -39
- llama_cloud/types/gemini_embedding_config.py +10 -27
- llama_cloud/types/gemini_embedding_config_param.py +17 -0
- llama_cloud/types/gemini_embedding_param.py +39 -0
- llama_cloud/types/hugging_face_inference_api_embedding.py +62 -46
- llama_cloud/types/hugging_face_inference_api_embedding_config.py +11 -28
- llama_cloud/types/hugging_face_inference_api_embedding_config_param.py +17 -0
- llama_cloud/types/hugging_face_inference_api_embedding_param.py +60 -0
- llama_cloud/types/list_item.py +48 -0
- llama_cloud/types/llama_parse_parameters.py +251 -130
- llama_cloud/types/llama_parse_parameters_param.py +261 -0
- llama_cloud/types/llama_parse_supported_file_extensions.py +84 -310
- llama_cloud/types/managed_ingestion_status_response.py +39 -37
- llama_cloud/types/message_role.py +4 -46
- llama_cloud/types/metadata_filters.py +45 -29
- llama_cloud/types/metadata_filters_param.py +58 -0
- llama_cloud/types/openai_embedding.py +56 -0
- llama_cloud/types/openai_embedding_config.py +17 -0
- llama_cloud/types/openai_embedding_config_param.py +17 -0
- llama_cloud/types/openai_embedding_param.py +55 -0
- llama_cloud/types/page_figure_node_with_score.py +32 -29
- llama_cloud/types/page_screenshot_node_with_score.py +23 -29
- llama_cloud/types/parsing_create_params.py +586 -0
- llama_cloud/types/parsing_create_response.py +33 -0
- llama_cloud/types/parsing_get_params.py +27 -0
- llama_cloud/types/parsing_get_response.py +364 -0
- llama_cloud/types/parsing_languages.py +94 -0
- llama_cloud/types/parsing_list_params.py +23 -0
- llama_cloud/types/parsing_list_response.py +33 -0
- llama_cloud/types/parsing_mode.py +13 -46
- llama_cloud/types/parsing_upload_file_params.py +14 -0
- llama_cloud/types/parsing_upload_file_response.py +33 -0
- llama_cloud/types/pipeline.py +180 -62
- llama_cloud/types/pipeline_create_params.py +95 -0
- llama_cloud/types/pipeline_get_status_params.py +12 -0
- llama_cloud/types/pipeline_list_params.py +23 -0
- llama_cloud/types/pipeline_list_response.py +12 -0
- llama_cloud/types/pipeline_metadata_config.py +9 -30
- llama_cloud/types/pipeline_metadata_config_param.py +17 -0
- llama_cloud/types/pipeline_retrieve_params.py +74 -0
- llama_cloud/types/pipeline_retrieve_response.py +63 -0
- llama_cloud/types/pipeline_type.py +4 -18
- llama_cloud/types/pipeline_update_params.py +90 -0
- llama_cloud/types/pipeline_upsert_params.py +95 -0
- llama_cloud/types/pipelines/__init__.py +38 -0
- llama_cloud/types/pipelines/cloud_document.py +29 -0
- llama_cloud/types/pipelines/cloud_document_create_param.py +30 -0
- llama_cloud/types/pipelines/data_source_get_data_sources_response.py +10 -0
- llama_cloud/types/pipelines/data_source_sync_params.py +16 -0
- llama_cloud/types/pipelines/data_source_update_data_sources_params.py +25 -0
- llama_cloud/types/pipelines/data_source_update_data_sources_response.py +10 -0
- llama_cloud/types/pipelines/data_source_update_params.py +15 -0
- llama_cloud/types/pipelines/document_create_params.py +14 -0
- llama_cloud/types/pipelines/document_create_response.py +10 -0
- llama_cloud/types/pipelines/document_get_chunks_response.py +10 -0
- llama_cloud/types/pipelines/document_list_params.py +22 -0
- llama_cloud/types/pipelines/document_upsert_params.py +14 -0
- llama_cloud/types/pipelines/document_upsert_response.py +10 -0
- llama_cloud/types/pipelines/file_create_params.py +22 -0
- llama_cloud/types/pipelines/file_create_response.py +10 -0
- llama_cloud/types/pipelines/file_get_status_counts_params.py +14 -0
- llama_cloud/types/pipelines/file_get_status_counts_response.py +24 -0
- llama_cloud/types/pipelines/file_list_params.py +22 -0
- llama_cloud/types/pipelines/file_update_params.py +15 -0
- llama_cloud/types/pipelines/image_get_page_figure_params.py +18 -0
- llama_cloud/types/pipelines/image_get_page_screenshot_params.py +16 -0
- llama_cloud/types/pipelines/image_list_page_figures_params.py +14 -0
- llama_cloud/types/pipelines/image_list_page_figures_response.py +34 -0
- llama_cloud/types/pipelines/image_list_page_screenshots_params.py +14 -0
- llama_cloud/types/pipelines/image_list_page_screenshots_response.py +25 -0
- llama_cloud/types/pipelines/metadata_create_params.py +13 -0
- llama_cloud/types/pipelines/metadata_create_response.py +8 -0
- llama_cloud/types/pipelines/pipeline_data_source.py +96 -0
- llama_cloud/types/pipelines/pipeline_file.py +70 -0
- llama_cloud/types/pipelines/text_node.py +89 -0
- llama_cloud/types/preset_retrieval_params.py +61 -49
- llama_cloud/types/preset_retrieval_params_param.py +71 -0
- llama_cloud/types/presigned_url.py +13 -29
- llama_cloud/types/project.py +24 -36
- llama_cloud/types/project_get_params.py +12 -0
- llama_cloud/types/project_list_params.py +14 -0
- llama_cloud/types/project_list_response.py +10 -0
- llama_cloud/types/re_rank_config_param.py +18 -0
- llama_cloud/types/retrieval_mode.py +4 -26
- llama_cloud/types/retriever.py +31 -38
- llama_cloud/types/retriever_create_params.py +26 -0
- llama_cloud/types/retriever_get_params.py +14 -0
- llama_cloud/types/retriever_list_params.py +16 -0
- llama_cloud/types/retriever_list_response.py +12 -0
- llama_cloud/types/retriever_pipeline.py +26 -34
- llama_cloud/types/retriever_pipeline_param.py +28 -0
- llama_cloud/types/retriever_search_params.py +38 -0
- llama_cloud/types/retriever_update_params.py +19 -0
- llama_cloud/types/retriever_upsert_params.py +26 -0
- llama_cloud/types/retrievers/__init__.py +5 -0
- llama_cloud/types/retrievers/retriever_search_params.py +32 -0
- llama_cloud/types/shared/__init__.py +21 -0
- llama_cloud/types/shared/cloud_astra_db_vector_store.py +39 -0
- llama_cloud/types/shared/cloud_az_storage_blob_data_source.py +34 -0
- llama_cloud/types/shared/cloud_azure_ai_search_vector_store.py +30 -0
- llama_cloud/types/shared/cloud_box_data_source.py +31 -0
- llama_cloud/types/shared/cloud_confluence_data_source.py +53 -0
- llama_cloud/types/shared/cloud_jira_data_source.py +30 -0
- llama_cloud/types/shared/cloud_jira_data_source_v2.py +49 -0
- llama_cloud/types/shared/cloud_milvus_vector_store.py +21 -0
- llama_cloud/types/shared/cloud_mongodb_atlas_vector_search.py +36 -0
- llama_cloud/types/shared/cloud_notion_page_data_source.py +19 -0
- llama_cloud/types/shared/cloud_one_drive_data_source.py +32 -0
- llama_cloud/types/shared/cloud_pinecone_vector_store.py +32 -0
- llama_cloud/types/shared/cloud_postgres_vector_store.py +35 -0
- llama_cloud/types/shared/cloud_qdrant_vector_store.py +35 -0
- llama_cloud/types/shared/cloud_s3_data_source.py +28 -0
- llama_cloud/types/shared/cloud_sharepoint_data_source.py +55 -0
- llama_cloud/types/shared/cloud_slack_data_source.py +31 -0
- llama_cloud/types/shared/failure_handling_config.py +16 -0
- llama_cloud/types/shared/pg_vector_hnsw_settings.py +27 -0
- llama_cloud/types/shared_params/__init__.py +21 -0
- llama_cloud/types/shared_params/cloud_astra_db_vector_store.py +42 -0
- llama_cloud/types/shared_params/cloud_az_storage_blob_data_source.py +41 -0
- llama_cloud/types/shared_params/cloud_azure_ai_search_vector_store.py +34 -0
- llama_cloud/types/shared_params/cloud_box_data_source.py +40 -0
- llama_cloud/types/shared_params/cloud_confluence_data_source.py +58 -0
- llama_cloud/types/shared_params/cloud_jira_data_source.py +34 -0
- llama_cloud/types/shared_params/cloud_jira_data_source_v2.py +54 -0
- llama_cloud/types/shared_params/cloud_milvus_vector_store.py +24 -0
- llama_cloud/types/shared_params/cloud_mongodb_atlas_vector_search.py +39 -0
- llama_cloud/types/shared_params/cloud_notion_page_data_source.py +23 -0
- llama_cloud/types/shared_params/cloud_one_drive_data_source.py +37 -0
- llama_cloud/types/shared_params/cloud_pinecone_vector_store.py +35 -0
- llama_cloud/types/shared_params/cloud_postgres_vector_store.py +39 -0
- llama_cloud/types/shared_params/cloud_qdrant_vector_store.py +37 -0
- llama_cloud/types/shared_params/cloud_s3_data_source.py +32 -0
- llama_cloud/types/shared_params/cloud_sharepoint_data_source.py +60 -0
- llama_cloud/types/shared_params/cloud_slack_data_source.py +35 -0
- llama_cloud/types/shared_params/failure_handling_config.py +16 -0
- llama_cloud/types/shared_params/pg_vector_hnsw_settings.py +26 -0
- llama_cloud/types/sparse_model_config.py +16 -30
- llama_cloud/types/sparse_model_config_param.py +25 -0
- llama_cloud/types/status_enum.py +4 -34
- llama_cloud/types/vertex_ai_embedding_config.py +10 -27
- llama_cloud/types/vertex_ai_embedding_config_param.py +17 -0
- llama_cloud/types/vertex_text_embedding.py +47 -45
- llama_cloud/types/vertex_text_embedding_param.py +45 -0
- llama_cloud-1.0.0b4.dist-info/METADATA +546 -0
- llama_cloud-1.0.0b4.dist-info/RECORD +376 -0
- {llama_cloud-0.1.41.dist-info → llama_cloud-1.0.0b4.dist-info}/WHEEL +1 -1
- llama_cloud-1.0.0b4.dist-info/licenses/LICENSE +7 -0
- llama_cloud/client.py +0 -108
- llama_cloud/core/__init__.py +0 -17
- llama_cloud/core/api_error.py +0 -15
- llama_cloud/core/client_wrapper.py +0 -51
- llama_cloud/core/datetime_utils.py +0 -28
- llama_cloud/core/jsonable_encoder.py +0 -106
- llama_cloud/core/remove_none_from_dict.py +0 -11
- llama_cloud/environment.py +0 -7
- llama_cloud/errors/__init__.py +0 -5
- llama_cloud/errors/unprocessable_entity_error.py +0 -9
- llama_cloud/resources/admin/__init__.py +0 -2
- llama_cloud/resources/admin/client.py +0 -196
- llama_cloud/resources/agent_deployments/__init__.py +0 -2
- llama_cloud/resources/agent_deployments/client.py +0 -160
- llama_cloud/resources/alpha/__init__.py +0 -2
- llama_cloud/resources/alpha/client.py +0 -112
- llama_cloud/resources/beta/client.py +0 -2664
- llama_cloud/resources/chat_apps/__init__.py +0 -2
- llama_cloud/resources/chat_apps/client.py +0 -616
- llama_cloud/resources/classifier/client.py +0 -444
- llama_cloud/resources/data_sinks/__init__.py +0 -5
- llama_cloud/resources/data_sinks/client.py +0 -535
- llama_cloud/resources/data_sinks/types/__init__.py +0 -5
- llama_cloud/resources/data_sinks/types/data_sink_update_component.py +0 -22
- llama_cloud/resources/data_sources/__init__.py +0 -5
- llama_cloud/resources/data_sources/client.py +0 -548
- llama_cloud/resources/data_sources/types/__init__.py +0 -6
- llama_cloud/resources/data_sources/types/data_source_update_component.py +0 -28
- llama_cloud/resources/data_sources/types/data_source_update_custom_metadata_value.py +0 -7
- llama_cloud/resources/embedding_model_configs/__init__.py +0 -23
- llama_cloud/resources/embedding_model_configs/client.py +0 -420
- llama_cloud/resources/embedding_model_configs/types/__init__.py +0 -23
- llama_cloud/resources/embedding_model_configs/types/embedding_model_config_create_embedding_config.py +0 -89
- llama_cloud/resources/evals/__init__.py +0 -2
- llama_cloud/resources/evals/client.py +0 -85
- llama_cloud/resources/files/__init__.py +0 -5
- llama_cloud/resources/files/client.py +0 -1454
- llama_cloud/resources/files/types/__init__.py +0 -5
- llama_cloud/resources/files/types/file_create_from_url_resource_info_value.py +0 -7
- llama_cloud/resources/jobs/__init__.py +0 -2
- llama_cloud/resources/jobs/client.py +0 -164
- llama_cloud/resources/llama_extract/__init__.py +0 -27
- llama_cloud/resources/llama_extract/client.py +0 -2082
- llama_cloud/resources/llama_extract/types/__init__.py +0 -25
- llama_cloud/resources/llama_extract/types/extract_agent_create_data_schema.py +0 -9
- llama_cloud/resources/llama_extract/types/extract_agent_create_data_schema_zero_value.py +0 -7
- llama_cloud/resources/llama_extract/types/extract_agent_update_data_schema.py +0 -9
- llama_cloud/resources/llama_extract/types/extract_agent_update_data_schema_zero_value.py +0 -7
- llama_cloud/resources/llama_extract/types/extract_job_create_batch_data_schema_override.py +0 -9
- llama_cloud/resources/llama_extract/types/extract_job_create_batch_data_schema_override_zero_value.py +0 -7
- llama_cloud/resources/llama_extract/types/extract_schema_validate_request_data_schema.py +0 -9
- llama_cloud/resources/llama_extract/types/extract_schema_validate_request_data_schema_zero_value.py +0 -7
- llama_cloud/resources/llama_extract/types/extract_stateless_request_data_schema.py +0 -9
- llama_cloud/resources/llama_extract/types/extract_stateless_request_data_schema_zero_value.py +0 -7
- llama_cloud/resources/organizations/__init__.py +0 -2
- llama_cloud/resources/organizations/client.py +0 -1448
- llama_cloud/resources/parsing/__init__.py +0 -2
- llama_cloud/resources/parsing/client.py +0 -2392
- llama_cloud/resources/pipelines/client.py +0 -3436
- llama_cloud/resources/pipelines/types/__init__.py +0 -29
- llama_cloud/resources/pipelines/types/pipeline_file_update_custom_metadata_value.py +0 -7
- llama_cloud/resources/pipelines/types/pipeline_update_embedding_config.py +0 -89
- llama_cloud/resources/pipelines/types/pipeline_update_transform_config.py +0 -8
- llama_cloud/resources/pipelines/types/retrieval_params_search_filters_inference_schema_value.py +0 -7
- llama_cloud/resources/projects/__init__.py +0 -2
- llama_cloud/resources/projects/client.py +0 -636
- llama_cloud/resources/retrievers/client.py +0 -837
- llama_cloud/resources/users/__init__.py +0 -2
- llama_cloud/resources/users/client.py +0 -155
- llama_cloud/types/advanced_mode_transform_config_chunking_config.py +0 -67
- llama_cloud/types/advanced_mode_transform_config_segmentation_config.py +0 -45
- llama_cloud/types/agent_data.py +0 -40
- llama_cloud/types/agent_deployment_list.py +0 -32
- llama_cloud/types/agent_deployment_summary.py +0 -39
- llama_cloud/types/aggregate_group.py +0 -37
- llama_cloud/types/azure_open_ai_embedding.py +0 -49
- llama_cloud/types/azure_open_ai_embedding_config.py +0 -34
- llama_cloud/types/base_plan.py +0 -53
- llama_cloud/types/base_plan_metronome_plan_type.py +0 -17
- llama_cloud/types/base_plan_name.py +0 -57
- llama_cloud/types/base_plan_plan_frequency.py +0 -25
- llama_cloud/types/batch.py +0 -47
- llama_cloud/types/batch_item.py +0 -40
- llama_cloud/types/batch_paginated_list.py +0 -35
- llama_cloud/types/batch_public_output.py +0 -36
- llama_cloud/types/billing_period.py +0 -32
- llama_cloud/types/box_auth_mechanism.py +0 -17
- llama_cloud/types/character_chunking_config.py +0 -32
- llama_cloud/types/chat_app.py +0 -46
- llama_cloud/types/chat_app_response.py +0 -43
- llama_cloud/types/chat_data.py +0 -35
- llama_cloud/types/chat_message.py +0 -43
- llama_cloud/types/chunk_mode.py +0 -29
- llama_cloud/types/classification_result.py +0 -39
- llama_cloud/types/classifier_rule.py +0 -43
- llama_cloud/types/classify_job.py +0 -47
- llama_cloud/types/classify_job_results.py +0 -38
- llama_cloud/types/classify_parsing_configuration.py +0 -38
- llama_cloud/types/cloud_astra_db_vector_store.py +0 -51
- llama_cloud/types/cloud_az_storage_blob_data_source.py +0 -41
- llama_cloud/types/cloud_azure_ai_search_vector_store.py +0 -45
- llama_cloud/types/cloud_box_data_source.py +0 -42
- llama_cloud/types/cloud_confluence_data_source.py +0 -59
- llama_cloud/types/cloud_document.py +0 -40
- llama_cloud/types/cloud_document_create.py +0 -40
- llama_cloud/types/cloud_jira_data_source.py +0 -42
- llama_cloud/types/cloud_jira_data_source_v_2.py +0 -52
- llama_cloud/types/cloud_jira_data_source_v_2_api_version.py +0 -21
- llama_cloud/types/cloud_milvus_vector_store.py +0 -40
- llama_cloud/types/cloud_mongo_db_atlas_vector_search.py +0 -52
- llama_cloud/types/cloud_notion_page_data_source.py +0 -35
- llama_cloud/types/cloud_one_drive_data_source.py +0 -39
- llama_cloud/types/cloud_pinecone_vector_store.py +0 -49
- llama_cloud/types/cloud_postgres_vector_store.py +0 -44
- llama_cloud/types/cloud_qdrant_vector_store.py +0 -51
- llama_cloud/types/cloud_s_3_data_source.py +0 -39
- llama_cloud/types/cloud_sharepoint_data_source.py +0 -42
- llama_cloud/types/cloud_slack_data_source.py +0 -39
- llama_cloud/types/composite_retrieved_text_node.py +0 -42
- llama_cloud/types/composite_retrieved_text_node_with_score.py +0 -34
- llama_cloud/types/configurable_data_sink_names.py +0 -41
- llama_cloud/types/configurable_data_source_names.py +0 -57
- llama_cloud/types/credit_type.py +0 -32
- llama_cloud/types/data_sink_component.py +0 -22
- llama_cloud/types/data_sink_create.py +0 -39
- llama_cloud/types/data_sink_create_component.py +0 -22
- llama_cloud/types/data_source_component.py +0 -28
- llama_cloud/types/data_source_create.py +0 -41
- llama_cloud/types/data_source_create_component.py +0 -28
- llama_cloud/types/data_source_create_custom_metadata_value.py +0 -7
- llama_cloud/types/data_source_custom_metadata_value.py +0 -7
- llama_cloud/types/data_source_reader_version_metadata_reader_version.py +0 -25
- llama_cloud/types/data_source_update_dispatcher_config.py +0 -38
- llama_cloud/types/delete_params.py +0 -39
- llama_cloud/types/document_chunk_mode.py +0 -17
- llama_cloud/types/document_ingestion_job_params.py +0 -43
- llama_cloud/types/element_segmentation_config.py +0 -29
- llama_cloud/types/embedding_model_config.py +0 -43
- llama_cloud/types/embedding_model_config_embedding_config.py +0 -89
- llama_cloud/types/embedding_model_config_update.py +0 -33
- llama_cloud/types/embedding_model_config_update_embedding_config.py +0 -89
- llama_cloud/types/eval_execution_params.py +0 -41
- llama_cloud/types/extract_agent.py +0 -48
- llama_cloud/types/extract_agent_data_schema_value.py +0 -5
- llama_cloud/types/extract_config.py +0 -66
- llama_cloud/types/extract_config_priority.py +0 -29
- llama_cloud/types/extract_job.py +0 -38
- llama_cloud/types/extract_job_create.py +0 -46
- llama_cloud/types/extract_job_create_data_schema_override.py +0 -9
- llama_cloud/types/extract_job_create_data_schema_override_zero_value.py +0 -7
- llama_cloud/types/extract_job_create_priority.py +0 -29
- llama_cloud/types/extract_mode.py +0 -29
- llama_cloud/types/extract_models.py +0 -53
- llama_cloud/types/extract_resultset.py +0 -42
- llama_cloud/types/extract_resultset_data.py +0 -11
- llama_cloud/types/extract_resultset_data_item_value.py +0 -7
- llama_cloud/types/extract_resultset_data_zero_value.py +0 -7
- llama_cloud/types/extract_resultset_extraction_metadata_value.py +0 -7
- llama_cloud/types/extract_run.py +0 -55
- llama_cloud/types/extract_run_data.py +0 -11
- llama_cloud/types/extract_run_data_item_value.py +0 -5
- llama_cloud/types/extract_run_data_schema_value.py +0 -5
- llama_cloud/types/extract_run_data_zero_value.py +0 -5
- llama_cloud/types/extract_run_extraction_metadata_value.py +0 -7
- llama_cloud/types/extract_schema_generate_response.py +0 -38
- llama_cloud/types/extract_schema_generate_response_data_schema_value.py +0 -7
- llama_cloud/types/extract_schema_validate_response.py +0 -32
- llama_cloud/types/extract_schema_validate_response_data_schema_value.py +0 -7
- llama_cloud/types/extract_state.py +0 -29
- llama_cloud/types/extract_target.py +0 -17
- llama_cloud/types/failure_handling_config.py +0 -37
- llama_cloud/types/file_classification.py +0 -41
- llama_cloud/types/file_count_by_status_response.py +0 -37
- llama_cloud/types/file_create.py +0 -41
- llama_cloud/types/file_create_permission_info_value.py +0 -7
- llama_cloud/types/file_create_resource_info_value.py +0 -5
- llama_cloud/types/file_data.py +0 -36
- llama_cloud/types/file_filter.py +0 -40
- llama_cloud/types/file_id_presigned_url.py +0 -38
- llama_cloud/types/file_parse_public.py +0 -36
- llama_cloud/types/file_permission_info_value.py +0 -5
- llama_cloud/types/file_resource_info_value.py +0 -5
- llama_cloud/types/file_store_info_response.py +0 -34
- llama_cloud/types/file_store_info_response_status.py +0 -25
- llama_cloud/types/filter_condition.py +0 -29
- llama_cloud/types/filter_operation.py +0 -46
- llama_cloud/types/filter_operation_eq.py +0 -6
- llama_cloud/types/filter_operation_gt.py +0 -6
- llama_cloud/types/filter_operation_gte.py +0 -6
- llama_cloud/types/filter_operation_includes_item.py +0 -6
- llama_cloud/types/filter_operation_lt.py +0 -6
- llama_cloud/types/filter_operation_lte.py +0 -6
- llama_cloud/types/filter_operator.py +0 -73
- llama_cloud/types/free_credits_usage.py +0 -34
- llama_cloud/types/http_validation_error.py +0 -32
- llama_cloud/types/hugging_face_inference_api_embedding_token.py +0 -5
- llama_cloud/types/ingestion_error_response.py +0 -34
- llama_cloud/types/input_message.py +0 -40
- llama_cloud/types/job_name_mapping.py +0 -49
- llama_cloud/types/job_names.py +0 -81
- llama_cloud/types/job_record.py +0 -58
- llama_cloud/types/job_record_parameters.py +0 -111
- llama_cloud/types/job_record_with_usage_metrics.py +0 -36
- llama_cloud/types/l_lama_parse_transform_config.py +0 -37
- llama_cloud/types/legacy_parse_job_config.py +0 -207
- llama_cloud/types/license_info_response.py +0 -34
- llama_cloud/types/llama_extract_feature_availability.py +0 -34
- llama_cloud/types/llama_extract_mode_availability.py +0 -38
- llama_cloud/types/llama_extract_mode_availability_status.py +0 -17
- llama_cloud/types/llama_extract_settings.py +0 -67
- llama_cloud/types/llama_parse_parameters_priority.py +0 -29
- llama_cloud/types/llm_model_data.py +0 -38
- llama_cloud/types/llm_parameters.py +0 -39
- llama_cloud/types/load_files_job_config.py +0 -35
- llama_cloud/types/managed_ingestion_status.py +0 -41
- llama_cloud/types/managed_open_ai_embedding.py +0 -36
- llama_cloud/types/managed_open_ai_embedding_config.py +0 -34
- llama_cloud/types/message_annotation.py +0 -33
- llama_cloud/types/metadata_filter.py +0 -44
- llama_cloud/types/metadata_filter_value.py +0 -5
- llama_cloud/types/metadata_filters_filters_item.py +0 -8
- llama_cloud/types/multimodal_parse_resolution.py +0 -17
- llama_cloud/types/node_relationship.py +0 -44
- llama_cloud/types/none_chunking_config.py +0 -29
- llama_cloud/types/none_segmentation_config.py +0 -29
- llama_cloud/types/object_type.py +0 -33
- llama_cloud/types/open_ai_embedding.py +0 -47
- llama_cloud/types/open_ai_embedding_config.py +0 -34
- llama_cloud/types/organization.py +0 -43
- llama_cloud/types/organization_create.py +0 -35
- llama_cloud/types/page_figure_metadata.py +0 -37
- llama_cloud/types/page_screenshot_metadata.py +0 -34
- llama_cloud/types/page_segmentation_config.py +0 -31
- llama_cloud/types/paginated_extract_runs_response.py +0 -39
- llama_cloud/types/paginated_jobs_history_with_metrics.py +0 -35
- llama_cloud/types/paginated_list_cloud_documents_response.py +0 -35
- llama_cloud/types/paginated_list_pipeline_files_response.py +0 -35
- llama_cloud/types/paginated_response_agent_data.py +0 -34
- llama_cloud/types/paginated_response_aggregate_group.py +0 -34
- llama_cloud/types/paginated_response_classify_job.py +0 -34
- llama_cloud/types/paginated_response_quota_configuration.py +0 -36
- llama_cloud/types/parse_configuration.py +0 -44
- llama_cloud/types/parse_configuration_create.py +0 -41
- llama_cloud/types/parse_configuration_filter.py +0 -40
- llama_cloud/types/parse_configuration_query_response.py +0 -38
- llama_cloud/types/parse_job_config.py +0 -149
- llama_cloud/types/parse_job_config_priority.py +0 -29
- llama_cloud/types/parse_plan_level.py +0 -21
- llama_cloud/types/parser_languages.py +0 -361
- llama_cloud/types/parsing_history_item.py +0 -39
- llama_cloud/types/parsing_job.py +0 -35
- llama_cloud/types/parsing_job_json_result.py +0 -32
- llama_cloud/types/parsing_job_markdown_result.py +0 -32
- llama_cloud/types/parsing_job_structured_result.py +0 -32
- llama_cloud/types/parsing_job_text_result.py +0 -32
- llama_cloud/types/partition_names.py +0 -45
- llama_cloud/types/permission.py +0 -40
- llama_cloud/types/pg_vector_distance_method.py +0 -43
- llama_cloud/types/pg_vector_hnsw_settings.py +0 -45
- llama_cloud/types/pg_vector_vector_type.py +0 -35
- llama_cloud/types/pipeline_configuration_hashes.py +0 -37
- llama_cloud/types/pipeline_create.py +0 -65
- llama_cloud/types/pipeline_create_embedding_config.py +0 -89
- llama_cloud/types/pipeline_create_transform_config.py +0 -8
- llama_cloud/types/pipeline_data_source.py +0 -55
- llama_cloud/types/pipeline_data_source_component.py +0 -28
- llama_cloud/types/pipeline_data_source_create.py +0 -36
- llama_cloud/types/pipeline_data_source_custom_metadata_value.py +0 -7
- llama_cloud/types/pipeline_data_source_status.py +0 -33
- llama_cloud/types/pipeline_deployment.py +0 -37
- llama_cloud/types/pipeline_embedding_config.py +0 -100
- llama_cloud/types/pipeline_file.py +0 -58
- llama_cloud/types/pipeline_file_config_hash_value.py +0 -5
- llama_cloud/types/pipeline_file_create.py +0 -37
- llama_cloud/types/pipeline_file_create_custom_metadata_value.py +0 -7
- llama_cloud/types/pipeline_file_custom_metadata_value.py +0 -7
- llama_cloud/types/pipeline_file_permission_info_value.py +0 -7
- llama_cloud/types/pipeline_file_resource_info_value.py +0 -7
- llama_cloud/types/pipeline_file_status.py +0 -33
- llama_cloud/types/pipeline_file_update_dispatcher_config.py +0 -38
- llama_cloud/types/pipeline_file_updater_config.py +0 -44
- llama_cloud/types/pipeline_managed_ingestion_job_params.py +0 -37
- llama_cloud/types/pipeline_status.py +0 -17
- llama_cloud/types/pipeline_transform_config.py +0 -31
- llama_cloud/types/plan_limits.py +0 -53
- llama_cloud/types/playground_session.py +0 -51
- llama_cloud/types/pooling.py +0 -29
- llama_cloud/types/preset_composite_retrieval_params.py +0 -37
- llama_cloud/types/preset_retrieval_params_search_filters_inference_schema_value.py +0 -7
- llama_cloud/types/project_create.py +0 -35
- llama_cloud/types/prompt_conf.py +0 -38
- llama_cloud/types/public_model_name.py +0 -97
- llama_cloud/types/quota_configuration.py +0 -53
- llama_cloud/types/quota_configuration_configuration_type.py +0 -33
- llama_cloud/types/quota_configuration_status.py +0 -21
- llama_cloud/types/quota_rate_limit_configuration_value.py +0 -38
- llama_cloud/types/quota_rate_limit_configuration_value_denominator_units.py +0 -29
- llama_cloud/types/re_rank_config.py +0 -35
- llama_cloud/types/re_ranker_type.py +0 -41
- llama_cloud/types/recurring_credit_grant.py +0 -44
- llama_cloud/types/related_node_info.py +0 -36
- llama_cloud/types/related_node_info_node_type.py +0 -7
- llama_cloud/types/retrieve_results.py +0 -56
- llama_cloud/types/retriever_create.py +0 -37
- llama_cloud/types/role.py +0 -40
- llama_cloud/types/schema_generation_availability.py +0 -33
- llama_cloud/types/schema_generation_availability_status.py +0 -17
- llama_cloud/types/schema_relax_mode.py +0 -25
- llama_cloud/types/semantic_chunking_config.py +0 -32
- llama_cloud/types/sentence_chunking_config.py +0 -34
- llama_cloud/types/sparse_model_type.py +0 -33
- llama_cloud/types/struct_mode.py +0 -33
- llama_cloud/types/struct_parse_conf.py +0 -63
- llama_cloud/types/supported_llm_model.py +0 -40
- llama_cloud/types/supported_llm_model_names.py +0 -69
- llama_cloud/types/text_node.py +0 -67
- llama_cloud/types/text_node_relationships_value.py +0 -7
- llama_cloud/types/text_node_with_score.py +0 -39
- llama_cloud/types/token_chunking_config.py +0 -33
- llama_cloud/types/update_user_response.py +0 -33
- llama_cloud/types/usage_and_plan.py +0 -34
- llama_cloud/types/usage_metric_response.py +0 -34
- llama_cloud/types/usage_response.py +0 -43
- llama_cloud/types/usage_response_active_alerts_item.py +0 -37
- llama_cloud/types/user_job_record.py +0 -32
- llama_cloud/types/user_organization.py +0 -47
- llama_cloud/types/user_organization_create.py +0 -38
- llama_cloud/types/user_organization_delete.py +0 -37
- llama_cloud/types/user_organization_role.py +0 -42
- llama_cloud/types/user_summary.py +0 -38
- llama_cloud/types/validation_error.py +0 -34
- llama_cloud/types/validation_error_loc_item.py +0 -5
- llama_cloud/types/vertex_embedding_mode.py +0 -38
- llama_cloud/types/webhook_configuration.py +0 -39
- llama_cloud/types/webhook_configuration_webhook_events_item.py +0 -57
- llama_cloud-0.1.41.dist-info/LICENSE +0 -21
- llama_cloud-0.1.41.dist-info/METADATA +0 -106
- llama_cloud-0.1.41.dist-info/RECORD +0 -385
|
@@ -0,0 +1,1525 @@
|
|
|
1
|
+
# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
|
|
2
|
+
|
|
3
|
+
from __future__ import annotations
|
|
4
|
+
|
|
5
|
+
from typing import Dict, Union, Iterable, Optional
|
|
6
|
+
|
|
7
|
+
import httpx
|
|
8
|
+
|
|
9
|
+
from .sync import (
|
|
10
|
+
SyncResource,
|
|
11
|
+
AsyncSyncResource,
|
|
12
|
+
SyncResourceWithRawResponse,
|
|
13
|
+
AsyncSyncResourceWithRawResponse,
|
|
14
|
+
SyncResourceWithStreamingResponse,
|
|
15
|
+
AsyncSyncResourceWithStreamingResponse,
|
|
16
|
+
)
|
|
17
|
+
from .files import (
|
|
18
|
+
FilesResource,
|
|
19
|
+
AsyncFilesResource,
|
|
20
|
+
FilesResourceWithRawResponse,
|
|
21
|
+
AsyncFilesResourceWithRawResponse,
|
|
22
|
+
FilesResourceWithStreamingResponse,
|
|
23
|
+
AsyncFilesResourceWithStreamingResponse,
|
|
24
|
+
)
|
|
25
|
+
from .images import (
|
|
26
|
+
ImagesResource,
|
|
27
|
+
AsyncImagesResource,
|
|
28
|
+
ImagesResourceWithRawResponse,
|
|
29
|
+
AsyncImagesResourceWithRawResponse,
|
|
30
|
+
ImagesResourceWithStreamingResponse,
|
|
31
|
+
AsyncImagesResourceWithStreamingResponse,
|
|
32
|
+
)
|
|
33
|
+
from ...types import (
|
|
34
|
+
PipelineType,
|
|
35
|
+
RetrievalMode,
|
|
36
|
+
pipeline_list_params,
|
|
37
|
+
pipeline_create_params,
|
|
38
|
+
pipeline_update_params,
|
|
39
|
+
pipeline_upsert_params,
|
|
40
|
+
pipeline_retrieve_params,
|
|
41
|
+
pipeline_get_status_params,
|
|
42
|
+
)
|
|
43
|
+
from ..._types import Body, Omit, Query, Headers, NoneType, NotGiven, omit, not_given
|
|
44
|
+
from ..._utils import maybe_transform, async_maybe_transform
|
|
45
|
+
from .metadata import (
|
|
46
|
+
MetadataResource,
|
|
47
|
+
AsyncMetadataResource,
|
|
48
|
+
MetadataResourceWithRawResponse,
|
|
49
|
+
AsyncMetadataResourceWithRawResponse,
|
|
50
|
+
MetadataResourceWithStreamingResponse,
|
|
51
|
+
AsyncMetadataResourceWithStreamingResponse,
|
|
52
|
+
)
|
|
53
|
+
from ..._compat import cached_property
|
|
54
|
+
from .documents import (
|
|
55
|
+
DocumentsResource,
|
|
56
|
+
AsyncDocumentsResource,
|
|
57
|
+
DocumentsResourceWithRawResponse,
|
|
58
|
+
AsyncDocumentsResourceWithRawResponse,
|
|
59
|
+
DocumentsResourceWithStreamingResponse,
|
|
60
|
+
AsyncDocumentsResourceWithStreamingResponse,
|
|
61
|
+
)
|
|
62
|
+
from ..._resource import SyncAPIResource, AsyncAPIResource
|
|
63
|
+
from ..._response import (
|
|
64
|
+
to_raw_response_wrapper,
|
|
65
|
+
to_streamed_response_wrapper,
|
|
66
|
+
async_to_raw_response_wrapper,
|
|
67
|
+
async_to_streamed_response_wrapper,
|
|
68
|
+
)
|
|
69
|
+
from .data_sources import (
|
|
70
|
+
DataSourcesResource,
|
|
71
|
+
AsyncDataSourcesResource,
|
|
72
|
+
DataSourcesResourceWithRawResponse,
|
|
73
|
+
AsyncDataSourcesResourceWithRawResponse,
|
|
74
|
+
DataSourcesResourceWithStreamingResponse,
|
|
75
|
+
AsyncDataSourcesResourceWithStreamingResponse,
|
|
76
|
+
)
|
|
77
|
+
from ..._base_client import make_request_options
|
|
78
|
+
from ...types.pipeline import Pipeline
|
|
79
|
+
from ...types.pipeline_type import PipelineType
|
|
80
|
+
from ...types.retrieval_mode import RetrievalMode
|
|
81
|
+
from ...types.data_sink_create_param import DataSinkCreateParam
|
|
82
|
+
from ...types.metadata_filters_param import MetadataFiltersParam
|
|
83
|
+
from ...types.pipeline_list_response import PipelineListResponse
|
|
84
|
+
from ...types.sparse_model_config_param import SparseModelConfigParam
|
|
85
|
+
from ...types.pipeline_retrieve_response import PipelineRetrieveResponse
|
|
86
|
+
from ...types.llama_parse_parameters_param import LlamaParseParametersParam
|
|
87
|
+
from ...types.preset_retrieval_params_param import PresetRetrievalParamsParam
|
|
88
|
+
from ...types.pipeline_metadata_config_param import PipelineMetadataConfigParam
|
|
89
|
+
from ...types.managed_ingestion_status_response import ManagedIngestionStatusResponse
|
|
90
|
+
|
|
91
|
+
__all__ = ["PipelinesResource", "AsyncPipelinesResource"]
|
|
92
|
+
|
|
93
|
+
|
|
94
|
+
class PipelinesResource(SyncAPIResource):
|
|
95
|
+
@cached_property
|
|
96
|
+
def sync(self) -> SyncResource:
|
|
97
|
+
return SyncResource(self._client)
|
|
98
|
+
|
|
99
|
+
@cached_property
|
|
100
|
+
def data_sources(self) -> DataSourcesResource:
|
|
101
|
+
return DataSourcesResource(self._client)
|
|
102
|
+
|
|
103
|
+
@cached_property
|
|
104
|
+
def images(self) -> ImagesResource:
|
|
105
|
+
return ImagesResource(self._client)
|
|
106
|
+
|
|
107
|
+
@cached_property
|
|
108
|
+
def files(self) -> FilesResource:
|
|
109
|
+
return FilesResource(self._client)
|
|
110
|
+
|
|
111
|
+
@cached_property
|
|
112
|
+
def metadata(self) -> MetadataResource:
|
|
113
|
+
return MetadataResource(self._client)
|
|
114
|
+
|
|
115
|
+
@cached_property
|
|
116
|
+
def documents(self) -> DocumentsResource:
|
|
117
|
+
return DocumentsResource(self._client)
|
|
118
|
+
|
|
119
|
+
@cached_property
|
|
120
|
+
def with_raw_response(self) -> PipelinesResourceWithRawResponse:
|
|
121
|
+
"""
|
|
122
|
+
This property can be used as a prefix for any HTTP method call to return
|
|
123
|
+
the raw response object instead of the parsed content.
|
|
124
|
+
|
|
125
|
+
For more information, see https://www.github.com/run-llama/llama-cloud-py#accessing-raw-response-data-eg-headers
|
|
126
|
+
"""
|
|
127
|
+
return PipelinesResourceWithRawResponse(self)
|
|
128
|
+
|
|
129
|
+
@cached_property
|
|
130
|
+
def with_streaming_response(self) -> PipelinesResourceWithStreamingResponse:
|
|
131
|
+
"""
|
|
132
|
+
An alternative to `.with_raw_response` that doesn't eagerly read the response body.
|
|
133
|
+
|
|
134
|
+
For more information, see https://www.github.com/run-llama/llama-cloud-py#with_streaming_response
|
|
135
|
+
"""
|
|
136
|
+
return PipelinesResourceWithStreamingResponse(self)
|
|
137
|
+
|
|
138
|
+
def create(
|
|
139
|
+
self,
|
|
140
|
+
*,
|
|
141
|
+
name: str,
|
|
142
|
+
organization_id: Optional[str] | Omit = omit,
|
|
143
|
+
project_id: Optional[str] | Omit = omit,
|
|
144
|
+
data_sink: Optional[DataSinkCreateParam] | Omit = omit,
|
|
145
|
+
data_sink_id: Optional[str] | Omit = omit,
|
|
146
|
+
embedding_config: Optional[pipeline_create_params.EmbeddingConfig] | Omit = omit,
|
|
147
|
+
embedding_model_config_id: Optional[str] | Omit = omit,
|
|
148
|
+
llama_parse_parameters: LlamaParseParametersParam | Omit = omit,
|
|
149
|
+
managed_pipeline_id: Optional[str] | Omit = omit,
|
|
150
|
+
metadata_config: Optional[PipelineMetadataConfigParam] | Omit = omit,
|
|
151
|
+
pipeline_type: PipelineType | Omit = omit,
|
|
152
|
+
preset_retrieval_parameters: PresetRetrievalParamsParam | Omit = omit,
|
|
153
|
+
sparse_model_config: Optional[SparseModelConfigParam] | Omit = omit,
|
|
154
|
+
status: Optional[str] | Omit = omit,
|
|
155
|
+
transform_config: Optional[pipeline_create_params.TransformConfig] | Omit = omit,
|
|
156
|
+
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
|
|
157
|
+
# The extra values given here take precedence over values defined on the client or passed to this method.
|
|
158
|
+
extra_headers: Headers | None = None,
|
|
159
|
+
extra_query: Query | None = None,
|
|
160
|
+
extra_body: Body | None = None,
|
|
161
|
+
timeout: float | httpx.Timeout | None | NotGiven = not_given,
|
|
162
|
+
) -> Pipeline:
|
|
163
|
+
"""
|
|
164
|
+
Create a new pipeline for a project.
|
|
165
|
+
|
|
166
|
+
Args:
|
|
167
|
+
data_sink: Schema for creating a data sink.
|
|
168
|
+
|
|
169
|
+
data_sink_id: Data sink ID. When provided instead of data_sink, the data sink will be looked
|
|
170
|
+
up by ID.
|
|
171
|
+
|
|
172
|
+
embedding_model_config_id: Embedding model config ID. When provided instead of embedding_config, the
|
|
173
|
+
embedding model config will be looked up by ID.
|
|
174
|
+
|
|
175
|
+
llama_parse_parameters: Settings that can be configured for how to use LlamaParse to parse files within
|
|
176
|
+
a LlamaCloud pipeline.
|
|
177
|
+
|
|
178
|
+
managed_pipeline_id: The ID of the ManagedPipeline this playground pipeline is linked to.
|
|
179
|
+
|
|
180
|
+
metadata_config: Metadata configuration for the pipeline.
|
|
181
|
+
|
|
182
|
+
pipeline_type: Type of pipeline. Either PLAYGROUND or MANAGED.
|
|
183
|
+
|
|
184
|
+
preset_retrieval_parameters: Preset retrieval parameters for the pipeline.
|
|
185
|
+
|
|
186
|
+
sparse_model_config: Configuration for sparse embedding models used in hybrid search.
|
|
187
|
+
|
|
188
|
+
This allows users to choose between Splade and BM25 models for sparse retrieval
|
|
189
|
+
in managed data sinks.
|
|
190
|
+
|
|
191
|
+
status: Status of the pipeline deployment.
|
|
192
|
+
|
|
193
|
+
transform_config: Configuration for the transformation.
|
|
194
|
+
|
|
195
|
+
extra_headers: Send extra headers
|
|
196
|
+
|
|
197
|
+
extra_query: Add additional query parameters to the request
|
|
198
|
+
|
|
199
|
+
extra_body: Add additional JSON properties to the request
|
|
200
|
+
|
|
201
|
+
timeout: Override the client-level default timeout for this request, in seconds
|
|
202
|
+
"""
|
|
203
|
+
return self._post(
|
|
204
|
+
"/api/v1/pipelines",
|
|
205
|
+
body=maybe_transform(
|
|
206
|
+
{
|
|
207
|
+
"name": name,
|
|
208
|
+
"data_sink": data_sink,
|
|
209
|
+
"data_sink_id": data_sink_id,
|
|
210
|
+
"embedding_config": embedding_config,
|
|
211
|
+
"embedding_model_config_id": embedding_model_config_id,
|
|
212
|
+
"llama_parse_parameters": llama_parse_parameters,
|
|
213
|
+
"managed_pipeline_id": managed_pipeline_id,
|
|
214
|
+
"metadata_config": metadata_config,
|
|
215
|
+
"pipeline_type": pipeline_type,
|
|
216
|
+
"preset_retrieval_parameters": preset_retrieval_parameters,
|
|
217
|
+
"sparse_model_config": sparse_model_config,
|
|
218
|
+
"status": status,
|
|
219
|
+
"transform_config": transform_config,
|
|
220
|
+
},
|
|
221
|
+
pipeline_create_params.PipelineCreateParams,
|
|
222
|
+
),
|
|
223
|
+
options=make_request_options(
|
|
224
|
+
extra_headers=extra_headers,
|
|
225
|
+
extra_query=extra_query,
|
|
226
|
+
extra_body=extra_body,
|
|
227
|
+
timeout=timeout,
|
|
228
|
+
query=maybe_transform(
|
|
229
|
+
{
|
|
230
|
+
"organization_id": organization_id,
|
|
231
|
+
"project_id": project_id,
|
|
232
|
+
},
|
|
233
|
+
pipeline_create_params.PipelineCreateParams,
|
|
234
|
+
),
|
|
235
|
+
),
|
|
236
|
+
cast_to=Pipeline,
|
|
237
|
+
)
|
|
238
|
+
|
|
239
|
+
def retrieve(
|
|
240
|
+
self,
|
|
241
|
+
pipeline_id: str,
|
|
242
|
+
*,
|
|
243
|
+
query: str,
|
|
244
|
+
organization_id: Optional[str] | Omit = omit,
|
|
245
|
+
project_id: Optional[str] | Omit = omit,
|
|
246
|
+
alpha: Optional[float] | Omit = omit,
|
|
247
|
+
class_name: str | Omit = omit,
|
|
248
|
+
dense_similarity_cutoff: Optional[float] | Omit = omit,
|
|
249
|
+
dense_similarity_top_k: Optional[int] | Omit = omit,
|
|
250
|
+
enable_reranking: Optional[bool] | Omit = omit,
|
|
251
|
+
files_top_k: Optional[int] | Omit = omit,
|
|
252
|
+
rerank_top_n: Optional[int] | Omit = omit,
|
|
253
|
+
retrieval_mode: RetrievalMode | Omit = omit,
|
|
254
|
+
retrieve_image_nodes: bool | Omit = omit,
|
|
255
|
+
retrieve_page_figure_nodes: bool | Omit = omit,
|
|
256
|
+
retrieve_page_screenshot_nodes: bool | Omit = omit,
|
|
257
|
+
search_filters: Optional[MetadataFiltersParam] | Omit = omit,
|
|
258
|
+
search_filters_inference_schema: Optional[
|
|
259
|
+
Dict[str, Union[Dict[str, object], Iterable[object], str, float, bool, None]]
|
|
260
|
+
]
|
|
261
|
+
| Omit = omit,
|
|
262
|
+
sparse_similarity_top_k: Optional[int] | Omit = omit,
|
|
263
|
+
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
|
|
264
|
+
# The extra values given here take precedence over values defined on the client or passed to this method.
|
|
265
|
+
extra_headers: Headers | None = None,
|
|
266
|
+
extra_query: Query | None = None,
|
|
267
|
+
extra_body: Body | None = None,
|
|
268
|
+
timeout: float | httpx.Timeout | None | NotGiven = not_given,
|
|
269
|
+
) -> PipelineRetrieveResponse:
|
|
270
|
+
"""
|
|
271
|
+
Get retrieval results for a managed pipeline and a query
|
|
272
|
+
|
|
273
|
+
Args:
|
|
274
|
+
query: The query to retrieve against.
|
|
275
|
+
|
|
276
|
+
alpha: Alpha value for hybrid retrieval to determine the weights between dense and
|
|
277
|
+
sparse retrieval. 0 is sparse retrieval and 1 is dense retrieval.
|
|
278
|
+
|
|
279
|
+
dense_similarity_cutoff: Minimum similarity score wrt query for retrieval
|
|
280
|
+
|
|
281
|
+
dense_similarity_top_k: Number of nodes for dense retrieval.
|
|
282
|
+
|
|
283
|
+
enable_reranking: Enable reranking for retrieval
|
|
284
|
+
|
|
285
|
+
files_top_k: Number of files to retrieve (only for retrieval mode files_via_metadata and
|
|
286
|
+
files_via_content).
|
|
287
|
+
|
|
288
|
+
rerank_top_n: Number of reranked nodes for returning.
|
|
289
|
+
|
|
290
|
+
retrieval_mode: The retrieval mode for the query.
|
|
291
|
+
|
|
292
|
+
retrieve_image_nodes: Whether to retrieve image nodes.
|
|
293
|
+
|
|
294
|
+
retrieve_page_figure_nodes: Whether to retrieve page figure nodes.
|
|
295
|
+
|
|
296
|
+
retrieve_page_screenshot_nodes: Whether to retrieve page screenshot nodes.
|
|
297
|
+
|
|
298
|
+
search_filters: Metadata filters for vector stores.
|
|
299
|
+
|
|
300
|
+
search_filters_inference_schema: JSON Schema that will be used to infer search_filters. Omit or leave as null to
|
|
301
|
+
skip inference.
|
|
302
|
+
|
|
303
|
+
sparse_similarity_top_k: Number of nodes for sparse retrieval.
|
|
304
|
+
|
|
305
|
+
extra_headers: Send extra headers
|
|
306
|
+
|
|
307
|
+
extra_query: Add additional query parameters to the request
|
|
308
|
+
|
|
309
|
+
extra_body: Add additional JSON properties to the request
|
|
310
|
+
|
|
311
|
+
timeout: Override the client-level default timeout for this request, in seconds
|
|
312
|
+
"""
|
|
313
|
+
if not pipeline_id:
|
|
314
|
+
raise ValueError(f"Expected a non-empty value for `pipeline_id` but received {pipeline_id!r}")
|
|
315
|
+
return self._post(
|
|
316
|
+
f"/api/v1/pipelines/{pipeline_id}/retrieve",
|
|
317
|
+
body=maybe_transform(
|
|
318
|
+
{
|
|
319
|
+
"query": query,
|
|
320
|
+
"alpha": alpha,
|
|
321
|
+
"class_name": class_name,
|
|
322
|
+
"dense_similarity_cutoff": dense_similarity_cutoff,
|
|
323
|
+
"dense_similarity_top_k": dense_similarity_top_k,
|
|
324
|
+
"enable_reranking": enable_reranking,
|
|
325
|
+
"files_top_k": files_top_k,
|
|
326
|
+
"rerank_top_n": rerank_top_n,
|
|
327
|
+
"retrieval_mode": retrieval_mode,
|
|
328
|
+
"retrieve_image_nodes": retrieve_image_nodes,
|
|
329
|
+
"retrieve_page_figure_nodes": retrieve_page_figure_nodes,
|
|
330
|
+
"retrieve_page_screenshot_nodes": retrieve_page_screenshot_nodes,
|
|
331
|
+
"search_filters": search_filters,
|
|
332
|
+
"search_filters_inference_schema": search_filters_inference_schema,
|
|
333
|
+
"sparse_similarity_top_k": sparse_similarity_top_k,
|
|
334
|
+
},
|
|
335
|
+
pipeline_retrieve_params.PipelineRetrieveParams,
|
|
336
|
+
),
|
|
337
|
+
options=make_request_options(
|
|
338
|
+
extra_headers=extra_headers,
|
|
339
|
+
extra_query=extra_query,
|
|
340
|
+
extra_body=extra_body,
|
|
341
|
+
timeout=timeout,
|
|
342
|
+
query=maybe_transform(
|
|
343
|
+
{
|
|
344
|
+
"organization_id": organization_id,
|
|
345
|
+
"project_id": project_id,
|
|
346
|
+
},
|
|
347
|
+
pipeline_retrieve_params.PipelineRetrieveParams,
|
|
348
|
+
),
|
|
349
|
+
),
|
|
350
|
+
cast_to=PipelineRetrieveResponse,
|
|
351
|
+
)
|
|
352
|
+
|
|
353
|
+
def update(
|
|
354
|
+
self,
|
|
355
|
+
pipeline_id: str,
|
|
356
|
+
*,
|
|
357
|
+
data_sink: Optional[DataSinkCreateParam] | Omit = omit,
|
|
358
|
+
data_sink_id: Optional[str] | Omit = omit,
|
|
359
|
+
embedding_config: Optional[pipeline_update_params.EmbeddingConfig] | Omit = omit,
|
|
360
|
+
embedding_model_config_id: Optional[str] | Omit = omit,
|
|
361
|
+
llama_parse_parameters: Optional[LlamaParseParametersParam] | Omit = omit,
|
|
362
|
+
managed_pipeline_id: Optional[str] | Omit = omit,
|
|
363
|
+
metadata_config: Optional[PipelineMetadataConfigParam] | Omit = omit,
|
|
364
|
+
name: Optional[str] | Omit = omit,
|
|
365
|
+
preset_retrieval_parameters: Optional[PresetRetrievalParamsParam] | Omit = omit,
|
|
366
|
+
sparse_model_config: Optional[SparseModelConfigParam] | Omit = omit,
|
|
367
|
+
status: Optional[str] | Omit = omit,
|
|
368
|
+
transform_config: Optional[pipeline_update_params.TransformConfig] | Omit = omit,
|
|
369
|
+
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
|
|
370
|
+
# The extra values given here take precedence over values defined on the client or passed to this method.
|
|
371
|
+
extra_headers: Headers | None = None,
|
|
372
|
+
extra_query: Query | None = None,
|
|
373
|
+
extra_body: Body | None = None,
|
|
374
|
+
timeout: float | httpx.Timeout | None | NotGiven = not_given,
|
|
375
|
+
) -> Pipeline:
|
|
376
|
+
"""
|
|
377
|
+
Update an existing pipeline for a project.
|
|
378
|
+
|
|
379
|
+
Args:
|
|
380
|
+
data_sink: Schema for creating a data sink.
|
|
381
|
+
|
|
382
|
+
data_sink_id: Data sink ID. When provided instead of data_sink, the data sink will be looked
|
|
383
|
+
up by ID.
|
|
384
|
+
|
|
385
|
+
embedding_model_config_id: Embedding model config ID. When provided instead of embedding_config, the
|
|
386
|
+
embedding model config will be looked up by ID.
|
|
387
|
+
|
|
388
|
+
llama_parse_parameters: Settings that can be configured for how to use LlamaParse to parse files within
|
|
389
|
+
a LlamaCloud pipeline.
|
|
390
|
+
|
|
391
|
+
managed_pipeline_id: The ID of the ManagedPipeline this playground pipeline is linked to.
|
|
392
|
+
|
|
393
|
+
metadata_config: Metadata configuration for the pipeline.
|
|
394
|
+
|
|
395
|
+
preset_retrieval_parameters: Schema for the search params for an retrieval execution that can be preset for a
|
|
396
|
+
pipeline.
|
|
397
|
+
|
|
398
|
+
sparse_model_config: Configuration for sparse embedding models used in hybrid search.
|
|
399
|
+
|
|
400
|
+
This allows users to choose between Splade and BM25 models for sparse retrieval
|
|
401
|
+
in managed data sinks.
|
|
402
|
+
|
|
403
|
+
status: Status of the pipeline deployment.
|
|
404
|
+
|
|
405
|
+
transform_config: Configuration for the transformation.
|
|
406
|
+
|
|
407
|
+
extra_headers: Send extra headers
|
|
408
|
+
|
|
409
|
+
extra_query: Add additional query parameters to the request
|
|
410
|
+
|
|
411
|
+
extra_body: Add additional JSON properties to the request
|
|
412
|
+
|
|
413
|
+
timeout: Override the client-level default timeout for this request, in seconds
|
|
414
|
+
"""
|
|
415
|
+
if not pipeline_id:
|
|
416
|
+
raise ValueError(f"Expected a non-empty value for `pipeline_id` but received {pipeline_id!r}")
|
|
417
|
+
return self._put(
|
|
418
|
+
f"/api/v1/pipelines/{pipeline_id}",
|
|
419
|
+
body=maybe_transform(
|
|
420
|
+
{
|
|
421
|
+
"data_sink": data_sink,
|
|
422
|
+
"data_sink_id": data_sink_id,
|
|
423
|
+
"embedding_config": embedding_config,
|
|
424
|
+
"embedding_model_config_id": embedding_model_config_id,
|
|
425
|
+
"llama_parse_parameters": llama_parse_parameters,
|
|
426
|
+
"managed_pipeline_id": managed_pipeline_id,
|
|
427
|
+
"metadata_config": metadata_config,
|
|
428
|
+
"name": name,
|
|
429
|
+
"preset_retrieval_parameters": preset_retrieval_parameters,
|
|
430
|
+
"sparse_model_config": sparse_model_config,
|
|
431
|
+
"status": status,
|
|
432
|
+
"transform_config": transform_config,
|
|
433
|
+
},
|
|
434
|
+
pipeline_update_params.PipelineUpdateParams,
|
|
435
|
+
),
|
|
436
|
+
options=make_request_options(
|
|
437
|
+
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
|
|
438
|
+
),
|
|
439
|
+
cast_to=Pipeline,
|
|
440
|
+
)
|
|
441
|
+
|
|
442
|
+
def list(
|
|
443
|
+
self,
|
|
444
|
+
*,
|
|
445
|
+
organization_id: Optional[str] | Omit = omit,
|
|
446
|
+
pipeline_name: Optional[str] | Omit = omit,
|
|
447
|
+
pipeline_type: Optional[PipelineType] | Omit = omit,
|
|
448
|
+
project_id: Optional[str] | Omit = omit,
|
|
449
|
+
project_name: Optional[str] | Omit = omit,
|
|
450
|
+
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
|
|
451
|
+
# The extra values given here take precedence over values defined on the client or passed to this method.
|
|
452
|
+
extra_headers: Headers | None = None,
|
|
453
|
+
extra_query: Query | None = None,
|
|
454
|
+
extra_body: Body | None = None,
|
|
455
|
+
timeout: float | httpx.Timeout | None | NotGiven = not_given,
|
|
456
|
+
) -> PipelineListResponse:
|
|
457
|
+
"""
|
|
458
|
+
Search for pipelines by various parameters.
|
|
459
|
+
|
|
460
|
+
Args:
|
|
461
|
+
pipeline_type: Enum for representing the type of a pipeline
|
|
462
|
+
|
|
463
|
+
extra_headers: Send extra headers
|
|
464
|
+
|
|
465
|
+
extra_query: Add additional query parameters to the request
|
|
466
|
+
|
|
467
|
+
extra_body: Add additional JSON properties to the request
|
|
468
|
+
|
|
469
|
+
timeout: Override the client-level default timeout for this request, in seconds
|
|
470
|
+
"""
|
|
471
|
+
return self._get(
|
|
472
|
+
"/api/v1/pipelines",
|
|
473
|
+
options=make_request_options(
|
|
474
|
+
extra_headers=extra_headers,
|
|
475
|
+
extra_query=extra_query,
|
|
476
|
+
extra_body=extra_body,
|
|
477
|
+
timeout=timeout,
|
|
478
|
+
query=maybe_transform(
|
|
479
|
+
{
|
|
480
|
+
"organization_id": organization_id,
|
|
481
|
+
"pipeline_name": pipeline_name,
|
|
482
|
+
"pipeline_type": pipeline_type,
|
|
483
|
+
"project_id": project_id,
|
|
484
|
+
"project_name": project_name,
|
|
485
|
+
},
|
|
486
|
+
pipeline_list_params.PipelineListParams,
|
|
487
|
+
),
|
|
488
|
+
),
|
|
489
|
+
cast_to=PipelineListResponse,
|
|
490
|
+
)
|
|
491
|
+
|
|
492
|
+
def delete(
|
|
493
|
+
self,
|
|
494
|
+
pipeline_id: str,
|
|
495
|
+
*,
|
|
496
|
+
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
|
|
497
|
+
# The extra values given here take precedence over values defined on the client or passed to this method.
|
|
498
|
+
extra_headers: Headers | None = None,
|
|
499
|
+
extra_query: Query | None = None,
|
|
500
|
+
extra_body: Body | None = None,
|
|
501
|
+
timeout: float | httpx.Timeout | None | NotGiven = not_given,
|
|
502
|
+
) -> None:
|
|
503
|
+
"""
|
|
504
|
+
Delete a pipeline by ID.
|
|
505
|
+
|
|
506
|
+
Args:
|
|
507
|
+
extra_headers: Send extra headers
|
|
508
|
+
|
|
509
|
+
extra_query: Add additional query parameters to the request
|
|
510
|
+
|
|
511
|
+
extra_body: Add additional JSON properties to the request
|
|
512
|
+
|
|
513
|
+
timeout: Override the client-level default timeout for this request, in seconds
|
|
514
|
+
"""
|
|
515
|
+
if not pipeline_id:
|
|
516
|
+
raise ValueError(f"Expected a non-empty value for `pipeline_id` but received {pipeline_id!r}")
|
|
517
|
+
extra_headers = {"Accept": "*/*", **(extra_headers or {})}
|
|
518
|
+
return self._delete(
|
|
519
|
+
f"/api/v1/pipelines/{pipeline_id}",
|
|
520
|
+
options=make_request_options(
|
|
521
|
+
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
|
|
522
|
+
),
|
|
523
|
+
cast_to=NoneType,
|
|
524
|
+
)
|
|
525
|
+
|
|
526
|
+
def get(
|
|
527
|
+
self,
|
|
528
|
+
pipeline_id: str,
|
|
529
|
+
*,
|
|
530
|
+
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
|
|
531
|
+
# The extra values given here take precedence over values defined on the client or passed to this method.
|
|
532
|
+
extra_headers: Headers | None = None,
|
|
533
|
+
extra_query: Query | None = None,
|
|
534
|
+
extra_body: Body | None = None,
|
|
535
|
+
timeout: float | httpx.Timeout | None | NotGiven = not_given,
|
|
536
|
+
) -> Pipeline:
|
|
537
|
+
"""
|
|
538
|
+
Get a pipeline by ID for a given project.
|
|
539
|
+
|
|
540
|
+
Args:
|
|
541
|
+
extra_headers: Send extra headers
|
|
542
|
+
|
|
543
|
+
extra_query: Add additional query parameters to the request
|
|
544
|
+
|
|
545
|
+
extra_body: Add additional JSON properties to the request
|
|
546
|
+
|
|
547
|
+
timeout: Override the client-level default timeout for this request, in seconds
|
|
548
|
+
"""
|
|
549
|
+
if not pipeline_id:
|
|
550
|
+
raise ValueError(f"Expected a non-empty value for `pipeline_id` but received {pipeline_id!r}")
|
|
551
|
+
return self._get(
|
|
552
|
+
f"/api/v1/pipelines/{pipeline_id}",
|
|
553
|
+
options=make_request_options(
|
|
554
|
+
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
|
|
555
|
+
),
|
|
556
|
+
cast_to=Pipeline,
|
|
557
|
+
)
|
|
558
|
+
|
|
559
|
+
def get_status(
|
|
560
|
+
self,
|
|
561
|
+
pipeline_id: str,
|
|
562
|
+
*,
|
|
563
|
+
full_details: Optional[bool] | Omit = omit,
|
|
564
|
+
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
|
|
565
|
+
# The extra values given here take precedence over values defined on the client or passed to this method.
|
|
566
|
+
extra_headers: Headers | None = None,
|
|
567
|
+
extra_query: Query | None = None,
|
|
568
|
+
extra_body: Body | None = None,
|
|
569
|
+
timeout: float | httpx.Timeout | None | NotGiven = not_given,
|
|
570
|
+
) -> ManagedIngestionStatusResponse:
|
|
571
|
+
"""
|
|
572
|
+
Get the status of a pipeline by ID.
|
|
573
|
+
|
|
574
|
+
Args:
|
|
575
|
+
extra_headers: Send extra headers
|
|
576
|
+
|
|
577
|
+
extra_query: Add additional query parameters to the request
|
|
578
|
+
|
|
579
|
+
extra_body: Add additional JSON properties to the request
|
|
580
|
+
|
|
581
|
+
timeout: Override the client-level default timeout for this request, in seconds
|
|
582
|
+
"""
|
|
583
|
+
if not pipeline_id:
|
|
584
|
+
raise ValueError(f"Expected a non-empty value for `pipeline_id` but received {pipeline_id!r}")
|
|
585
|
+
return self._get(
|
|
586
|
+
f"/api/v1/pipelines/{pipeline_id}/status",
|
|
587
|
+
options=make_request_options(
|
|
588
|
+
extra_headers=extra_headers,
|
|
589
|
+
extra_query=extra_query,
|
|
590
|
+
extra_body=extra_body,
|
|
591
|
+
timeout=timeout,
|
|
592
|
+
query=maybe_transform(
|
|
593
|
+
{"full_details": full_details}, pipeline_get_status_params.PipelineGetStatusParams
|
|
594
|
+
),
|
|
595
|
+
),
|
|
596
|
+
cast_to=ManagedIngestionStatusResponse,
|
|
597
|
+
)
|
|
598
|
+
|
|
599
|
+
def upsert(
|
|
600
|
+
self,
|
|
601
|
+
*,
|
|
602
|
+
name: str,
|
|
603
|
+
organization_id: Optional[str] | Omit = omit,
|
|
604
|
+
project_id: Optional[str] | Omit = omit,
|
|
605
|
+
data_sink: Optional[DataSinkCreateParam] | Omit = omit,
|
|
606
|
+
data_sink_id: Optional[str] | Omit = omit,
|
|
607
|
+
embedding_config: Optional[pipeline_upsert_params.EmbeddingConfig] | Omit = omit,
|
|
608
|
+
embedding_model_config_id: Optional[str] | Omit = omit,
|
|
609
|
+
llama_parse_parameters: LlamaParseParametersParam | Omit = omit,
|
|
610
|
+
managed_pipeline_id: Optional[str] | Omit = omit,
|
|
611
|
+
metadata_config: Optional[PipelineMetadataConfigParam] | Omit = omit,
|
|
612
|
+
pipeline_type: PipelineType | Omit = omit,
|
|
613
|
+
preset_retrieval_parameters: PresetRetrievalParamsParam | Omit = omit,
|
|
614
|
+
sparse_model_config: Optional[SparseModelConfigParam] | Omit = omit,
|
|
615
|
+
status: Optional[str] | Omit = omit,
|
|
616
|
+
transform_config: Optional[pipeline_upsert_params.TransformConfig] | Omit = omit,
|
|
617
|
+
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
|
|
618
|
+
# The extra values given here take precedence over values defined on the client or passed to this method.
|
|
619
|
+
extra_headers: Headers | None = None,
|
|
620
|
+
extra_query: Query | None = None,
|
|
621
|
+
extra_body: Body | None = None,
|
|
622
|
+
timeout: float | httpx.Timeout | None | NotGiven = not_given,
|
|
623
|
+
) -> Pipeline:
|
|
624
|
+
"""Upsert a pipeline for a project.
|
|
625
|
+
|
|
626
|
+
Updates if a pipeline with the same name and
|
|
627
|
+
project_id already exists. Otherwise, creates a new pipeline.
|
|
628
|
+
|
|
629
|
+
Args:
|
|
630
|
+
data_sink: Schema for creating a data sink.
|
|
631
|
+
|
|
632
|
+
data_sink_id: Data sink ID. When provided instead of data_sink, the data sink will be looked
|
|
633
|
+
up by ID.
|
|
634
|
+
|
|
635
|
+
embedding_model_config_id: Embedding model config ID. When provided instead of embedding_config, the
|
|
636
|
+
embedding model config will be looked up by ID.
|
|
637
|
+
|
|
638
|
+
llama_parse_parameters: Settings that can be configured for how to use LlamaParse to parse files within
|
|
639
|
+
a LlamaCloud pipeline.
|
|
640
|
+
|
|
641
|
+
managed_pipeline_id: The ID of the ManagedPipeline this playground pipeline is linked to.
|
|
642
|
+
|
|
643
|
+
metadata_config: Metadata configuration for the pipeline.
|
|
644
|
+
|
|
645
|
+
pipeline_type: Type of pipeline. Either PLAYGROUND or MANAGED.
|
|
646
|
+
|
|
647
|
+
preset_retrieval_parameters: Preset retrieval parameters for the pipeline.
|
|
648
|
+
|
|
649
|
+
sparse_model_config: Configuration for sparse embedding models used in hybrid search.
|
|
650
|
+
|
|
651
|
+
This allows users to choose between Splade and BM25 models for sparse retrieval
|
|
652
|
+
in managed data sinks.
|
|
653
|
+
|
|
654
|
+
status: Status of the pipeline deployment.
|
|
655
|
+
|
|
656
|
+
transform_config: Configuration for the transformation.
|
|
657
|
+
|
|
658
|
+
extra_headers: Send extra headers
|
|
659
|
+
|
|
660
|
+
extra_query: Add additional query parameters to the request
|
|
661
|
+
|
|
662
|
+
extra_body: Add additional JSON properties to the request
|
|
663
|
+
|
|
664
|
+
timeout: Override the client-level default timeout for this request, in seconds
|
|
665
|
+
"""
|
|
666
|
+
return self._put(
|
|
667
|
+
"/api/v1/pipelines",
|
|
668
|
+
body=maybe_transform(
|
|
669
|
+
{
|
|
670
|
+
"name": name,
|
|
671
|
+
"data_sink": data_sink,
|
|
672
|
+
"data_sink_id": data_sink_id,
|
|
673
|
+
"embedding_config": embedding_config,
|
|
674
|
+
"embedding_model_config_id": embedding_model_config_id,
|
|
675
|
+
"llama_parse_parameters": llama_parse_parameters,
|
|
676
|
+
"managed_pipeline_id": managed_pipeline_id,
|
|
677
|
+
"metadata_config": metadata_config,
|
|
678
|
+
"pipeline_type": pipeline_type,
|
|
679
|
+
"preset_retrieval_parameters": preset_retrieval_parameters,
|
|
680
|
+
"sparse_model_config": sparse_model_config,
|
|
681
|
+
"status": status,
|
|
682
|
+
"transform_config": transform_config,
|
|
683
|
+
},
|
|
684
|
+
pipeline_upsert_params.PipelineUpsertParams,
|
|
685
|
+
),
|
|
686
|
+
options=make_request_options(
|
|
687
|
+
extra_headers=extra_headers,
|
|
688
|
+
extra_query=extra_query,
|
|
689
|
+
extra_body=extra_body,
|
|
690
|
+
timeout=timeout,
|
|
691
|
+
query=maybe_transform(
|
|
692
|
+
{
|
|
693
|
+
"organization_id": organization_id,
|
|
694
|
+
"project_id": project_id,
|
|
695
|
+
},
|
|
696
|
+
pipeline_upsert_params.PipelineUpsertParams,
|
|
697
|
+
),
|
|
698
|
+
),
|
|
699
|
+
cast_to=Pipeline,
|
|
700
|
+
)
|
|
701
|
+
|
|
702
|
+
|
|
703
|
+
class AsyncPipelinesResource(AsyncAPIResource):
|
|
704
|
+
@cached_property
|
|
705
|
+
def sync(self) -> AsyncSyncResource:
|
|
706
|
+
return AsyncSyncResource(self._client)
|
|
707
|
+
|
|
708
|
+
@cached_property
|
|
709
|
+
def data_sources(self) -> AsyncDataSourcesResource:
|
|
710
|
+
return AsyncDataSourcesResource(self._client)
|
|
711
|
+
|
|
712
|
+
@cached_property
|
|
713
|
+
def images(self) -> AsyncImagesResource:
|
|
714
|
+
return AsyncImagesResource(self._client)
|
|
715
|
+
|
|
716
|
+
@cached_property
|
|
717
|
+
def files(self) -> AsyncFilesResource:
|
|
718
|
+
return AsyncFilesResource(self._client)
|
|
719
|
+
|
|
720
|
+
@cached_property
|
|
721
|
+
def metadata(self) -> AsyncMetadataResource:
|
|
722
|
+
return AsyncMetadataResource(self._client)
|
|
723
|
+
|
|
724
|
+
@cached_property
|
|
725
|
+
def documents(self) -> AsyncDocumentsResource:
|
|
726
|
+
return AsyncDocumentsResource(self._client)
|
|
727
|
+
|
|
728
|
+
@cached_property
|
|
729
|
+
def with_raw_response(self) -> AsyncPipelinesResourceWithRawResponse:
|
|
730
|
+
"""
|
|
731
|
+
This property can be used as a prefix for any HTTP method call to return
|
|
732
|
+
the raw response object instead of the parsed content.
|
|
733
|
+
|
|
734
|
+
For more information, see https://www.github.com/run-llama/llama-cloud-py#accessing-raw-response-data-eg-headers
|
|
735
|
+
"""
|
|
736
|
+
return AsyncPipelinesResourceWithRawResponse(self)
|
|
737
|
+
|
|
738
|
+
@cached_property
|
|
739
|
+
def with_streaming_response(self) -> AsyncPipelinesResourceWithStreamingResponse:
|
|
740
|
+
"""
|
|
741
|
+
An alternative to `.with_raw_response` that doesn't eagerly read the response body.
|
|
742
|
+
|
|
743
|
+
For more information, see https://www.github.com/run-llama/llama-cloud-py#with_streaming_response
|
|
744
|
+
"""
|
|
745
|
+
return AsyncPipelinesResourceWithStreamingResponse(self)
|
|
746
|
+
|
|
747
|
+
async def create(
|
|
748
|
+
self,
|
|
749
|
+
*,
|
|
750
|
+
name: str,
|
|
751
|
+
organization_id: Optional[str] | Omit = omit,
|
|
752
|
+
project_id: Optional[str] | Omit = omit,
|
|
753
|
+
data_sink: Optional[DataSinkCreateParam] | Omit = omit,
|
|
754
|
+
data_sink_id: Optional[str] | Omit = omit,
|
|
755
|
+
embedding_config: Optional[pipeline_create_params.EmbeddingConfig] | Omit = omit,
|
|
756
|
+
embedding_model_config_id: Optional[str] | Omit = omit,
|
|
757
|
+
llama_parse_parameters: LlamaParseParametersParam | Omit = omit,
|
|
758
|
+
managed_pipeline_id: Optional[str] | Omit = omit,
|
|
759
|
+
metadata_config: Optional[PipelineMetadataConfigParam] | Omit = omit,
|
|
760
|
+
pipeline_type: PipelineType | Omit = omit,
|
|
761
|
+
preset_retrieval_parameters: PresetRetrievalParamsParam | Omit = omit,
|
|
762
|
+
sparse_model_config: Optional[SparseModelConfigParam] | Omit = omit,
|
|
763
|
+
status: Optional[str] | Omit = omit,
|
|
764
|
+
transform_config: Optional[pipeline_create_params.TransformConfig] | Omit = omit,
|
|
765
|
+
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
|
|
766
|
+
# The extra values given here take precedence over values defined on the client or passed to this method.
|
|
767
|
+
extra_headers: Headers | None = None,
|
|
768
|
+
extra_query: Query | None = None,
|
|
769
|
+
extra_body: Body | None = None,
|
|
770
|
+
timeout: float | httpx.Timeout | None | NotGiven = not_given,
|
|
771
|
+
) -> Pipeline:
|
|
772
|
+
"""
|
|
773
|
+
Create a new pipeline for a project.
|
|
774
|
+
|
|
775
|
+
Args:
|
|
776
|
+
data_sink: Schema for creating a data sink.
|
|
777
|
+
|
|
778
|
+
data_sink_id: Data sink ID. When provided instead of data_sink, the data sink will be looked
|
|
779
|
+
up by ID.
|
|
780
|
+
|
|
781
|
+
embedding_model_config_id: Embedding model config ID. When provided instead of embedding_config, the
|
|
782
|
+
embedding model config will be looked up by ID.
|
|
783
|
+
|
|
784
|
+
llama_parse_parameters: Settings that can be configured for how to use LlamaParse to parse files within
|
|
785
|
+
a LlamaCloud pipeline.
|
|
786
|
+
|
|
787
|
+
managed_pipeline_id: The ID of the ManagedPipeline this playground pipeline is linked to.
|
|
788
|
+
|
|
789
|
+
metadata_config: Metadata configuration for the pipeline.
|
|
790
|
+
|
|
791
|
+
pipeline_type: Type of pipeline. Either PLAYGROUND or MANAGED.
|
|
792
|
+
|
|
793
|
+
preset_retrieval_parameters: Preset retrieval parameters for the pipeline.
|
|
794
|
+
|
|
795
|
+
sparse_model_config: Configuration for sparse embedding models used in hybrid search.
|
|
796
|
+
|
|
797
|
+
This allows users to choose between Splade and BM25 models for sparse retrieval
|
|
798
|
+
in managed data sinks.
|
|
799
|
+
|
|
800
|
+
status: Status of the pipeline deployment.
|
|
801
|
+
|
|
802
|
+
transform_config: Configuration for the transformation.
|
|
803
|
+
|
|
804
|
+
extra_headers: Send extra headers
|
|
805
|
+
|
|
806
|
+
extra_query: Add additional query parameters to the request
|
|
807
|
+
|
|
808
|
+
extra_body: Add additional JSON properties to the request
|
|
809
|
+
|
|
810
|
+
timeout: Override the client-level default timeout for this request, in seconds
|
|
811
|
+
"""
|
|
812
|
+
return await self._post(
|
|
813
|
+
"/api/v1/pipelines",
|
|
814
|
+
body=await async_maybe_transform(
|
|
815
|
+
{
|
|
816
|
+
"name": name,
|
|
817
|
+
"data_sink": data_sink,
|
|
818
|
+
"data_sink_id": data_sink_id,
|
|
819
|
+
"embedding_config": embedding_config,
|
|
820
|
+
"embedding_model_config_id": embedding_model_config_id,
|
|
821
|
+
"llama_parse_parameters": llama_parse_parameters,
|
|
822
|
+
"managed_pipeline_id": managed_pipeline_id,
|
|
823
|
+
"metadata_config": metadata_config,
|
|
824
|
+
"pipeline_type": pipeline_type,
|
|
825
|
+
"preset_retrieval_parameters": preset_retrieval_parameters,
|
|
826
|
+
"sparse_model_config": sparse_model_config,
|
|
827
|
+
"status": status,
|
|
828
|
+
"transform_config": transform_config,
|
|
829
|
+
},
|
|
830
|
+
pipeline_create_params.PipelineCreateParams,
|
|
831
|
+
),
|
|
832
|
+
options=make_request_options(
|
|
833
|
+
extra_headers=extra_headers,
|
|
834
|
+
extra_query=extra_query,
|
|
835
|
+
extra_body=extra_body,
|
|
836
|
+
timeout=timeout,
|
|
837
|
+
query=await async_maybe_transform(
|
|
838
|
+
{
|
|
839
|
+
"organization_id": organization_id,
|
|
840
|
+
"project_id": project_id,
|
|
841
|
+
},
|
|
842
|
+
pipeline_create_params.PipelineCreateParams,
|
|
843
|
+
),
|
|
844
|
+
),
|
|
845
|
+
cast_to=Pipeline,
|
|
846
|
+
)
|
|
847
|
+
|
|
848
|
+
async def retrieve(
|
|
849
|
+
self,
|
|
850
|
+
pipeline_id: str,
|
|
851
|
+
*,
|
|
852
|
+
query: str,
|
|
853
|
+
organization_id: Optional[str] | Omit = omit,
|
|
854
|
+
project_id: Optional[str] | Omit = omit,
|
|
855
|
+
alpha: Optional[float] | Omit = omit,
|
|
856
|
+
class_name: str | Omit = omit,
|
|
857
|
+
dense_similarity_cutoff: Optional[float] | Omit = omit,
|
|
858
|
+
dense_similarity_top_k: Optional[int] | Omit = omit,
|
|
859
|
+
enable_reranking: Optional[bool] | Omit = omit,
|
|
860
|
+
files_top_k: Optional[int] | Omit = omit,
|
|
861
|
+
rerank_top_n: Optional[int] | Omit = omit,
|
|
862
|
+
retrieval_mode: RetrievalMode | Omit = omit,
|
|
863
|
+
retrieve_image_nodes: bool | Omit = omit,
|
|
864
|
+
retrieve_page_figure_nodes: bool | Omit = omit,
|
|
865
|
+
retrieve_page_screenshot_nodes: bool | Omit = omit,
|
|
866
|
+
search_filters: Optional[MetadataFiltersParam] | Omit = omit,
|
|
867
|
+
search_filters_inference_schema: Optional[
|
|
868
|
+
Dict[str, Union[Dict[str, object], Iterable[object], str, float, bool, None]]
|
|
869
|
+
]
|
|
870
|
+
| Omit = omit,
|
|
871
|
+
sparse_similarity_top_k: Optional[int] | Omit = omit,
|
|
872
|
+
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
|
|
873
|
+
# The extra values given here take precedence over values defined on the client or passed to this method.
|
|
874
|
+
extra_headers: Headers | None = None,
|
|
875
|
+
extra_query: Query | None = None,
|
|
876
|
+
extra_body: Body | None = None,
|
|
877
|
+
timeout: float | httpx.Timeout | None | NotGiven = not_given,
|
|
878
|
+
) -> PipelineRetrieveResponse:
|
|
879
|
+
"""
|
|
880
|
+
Get retrieval results for a managed pipeline and a query
|
|
881
|
+
|
|
882
|
+
Args:
|
|
883
|
+
query: The query to retrieve against.
|
|
884
|
+
|
|
885
|
+
alpha: Alpha value for hybrid retrieval to determine the weights between dense and
|
|
886
|
+
sparse retrieval. 0 is sparse retrieval and 1 is dense retrieval.
|
|
887
|
+
|
|
888
|
+
dense_similarity_cutoff: Minimum similarity score wrt query for retrieval
|
|
889
|
+
|
|
890
|
+
dense_similarity_top_k: Number of nodes for dense retrieval.
|
|
891
|
+
|
|
892
|
+
enable_reranking: Enable reranking for retrieval
|
|
893
|
+
|
|
894
|
+
files_top_k: Number of files to retrieve (only for retrieval mode files_via_metadata and
|
|
895
|
+
files_via_content).
|
|
896
|
+
|
|
897
|
+
rerank_top_n: Number of reranked nodes for returning.
|
|
898
|
+
|
|
899
|
+
retrieval_mode: The retrieval mode for the query.
|
|
900
|
+
|
|
901
|
+
retrieve_image_nodes: Whether to retrieve image nodes.
|
|
902
|
+
|
|
903
|
+
retrieve_page_figure_nodes: Whether to retrieve page figure nodes.
|
|
904
|
+
|
|
905
|
+
retrieve_page_screenshot_nodes: Whether to retrieve page screenshot nodes.
|
|
906
|
+
|
|
907
|
+
search_filters: Metadata filters for vector stores.
|
|
908
|
+
|
|
909
|
+
search_filters_inference_schema: JSON Schema that will be used to infer search_filters. Omit or leave as null to
|
|
910
|
+
skip inference.
|
|
911
|
+
|
|
912
|
+
sparse_similarity_top_k: Number of nodes for sparse retrieval.
|
|
913
|
+
|
|
914
|
+
extra_headers: Send extra headers
|
|
915
|
+
|
|
916
|
+
extra_query: Add additional query parameters to the request
|
|
917
|
+
|
|
918
|
+
extra_body: Add additional JSON properties to the request
|
|
919
|
+
|
|
920
|
+
timeout: Override the client-level default timeout for this request, in seconds
|
|
921
|
+
"""
|
|
922
|
+
if not pipeline_id:
|
|
923
|
+
raise ValueError(f"Expected a non-empty value for `pipeline_id` but received {pipeline_id!r}")
|
|
924
|
+
return await self._post(
|
|
925
|
+
f"/api/v1/pipelines/{pipeline_id}/retrieve",
|
|
926
|
+
body=await async_maybe_transform(
|
|
927
|
+
{
|
|
928
|
+
"query": query,
|
|
929
|
+
"alpha": alpha,
|
|
930
|
+
"class_name": class_name,
|
|
931
|
+
"dense_similarity_cutoff": dense_similarity_cutoff,
|
|
932
|
+
"dense_similarity_top_k": dense_similarity_top_k,
|
|
933
|
+
"enable_reranking": enable_reranking,
|
|
934
|
+
"files_top_k": files_top_k,
|
|
935
|
+
"rerank_top_n": rerank_top_n,
|
|
936
|
+
"retrieval_mode": retrieval_mode,
|
|
937
|
+
"retrieve_image_nodes": retrieve_image_nodes,
|
|
938
|
+
"retrieve_page_figure_nodes": retrieve_page_figure_nodes,
|
|
939
|
+
"retrieve_page_screenshot_nodes": retrieve_page_screenshot_nodes,
|
|
940
|
+
"search_filters": search_filters,
|
|
941
|
+
"search_filters_inference_schema": search_filters_inference_schema,
|
|
942
|
+
"sparse_similarity_top_k": sparse_similarity_top_k,
|
|
943
|
+
},
|
|
944
|
+
pipeline_retrieve_params.PipelineRetrieveParams,
|
|
945
|
+
),
|
|
946
|
+
options=make_request_options(
|
|
947
|
+
extra_headers=extra_headers,
|
|
948
|
+
extra_query=extra_query,
|
|
949
|
+
extra_body=extra_body,
|
|
950
|
+
timeout=timeout,
|
|
951
|
+
query=await async_maybe_transform(
|
|
952
|
+
{
|
|
953
|
+
"organization_id": organization_id,
|
|
954
|
+
"project_id": project_id,
|
|
955
|
+
},
|
|
956
|
+
pipeline_retrieve_params.PipelineRetrieveParams,
|
|
957
|
+
),
|
|
958
|
+
),
|
|
959
|
+
cast_to=PipelineRetrieveResponse,
|
|
960
|
+
)
|
|
961
|
+
|
|
962
|
+
async def update(
|
|
963
|
+
self,
|
|
964
|
+
pipeline_id: str,
|
|
965
|
+
*,
|
|
966
|
+
data_sink: Optional[DataSinkCreateParam] | Omit = omit,
|
|
967
|
+
data_sink_id: Optional[str] | Omit = omit,
|
|
968
|
+
embedding_config: Optional[pipeline_update_params.EmbeddingConfig] | Omit = omit,
|
|
969
|
+
embedding_model_config_id: Optional[str] | Omit = omit,
|
|
970
|
+
llama_parse_parameters: Optional[LlamaParseParametersParam] | Omit = omit,
|
|
971
|
+
managed_pipeline_id: Optional[str] | Omit = omit,
|
|
972
|
+
metadata_config: Optional[PipelineMetadataConfigParam] | Omit = omit,
|
|
973
|
+
name: Optional[str] | Omit = omit,
|
|
974
|
+
preset_retrieval_parameters: Optional[PresetRetrievalParamsParam] | Omit = omit,
|
|
975
|
+
sparse_model_config: Optional[SparseModelConfigParam] | Omit = omit,
|
|
976
|
+
status: Optional[str] | Omit = omit,
|
|
977
|
+
transform_config: Optional[pipeline_update_params.TransformConfig] | Omit = omit,
|
|
978
|
+
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
|
|
979
|
+
# The extra values given here take precedence over values defined on the client or passed to this method.
|
|
980
|
+
extra_headers: Headers | None = None,
|
|
981
|
+
extra_query: Query | None = None,
|
|
982
|
+
extra_body: Body | None = None,
|
|
983
|
+
timeout: float | httpx.Timeout | None | NotGiven = not_given,
|
|
984
|
+
) -> Pipeline:
|
|
985
|
+
"""
|
|
986
|
+
Update an existing pipeline for a project.
|
|
987
|
+
|
|
988
|
+
Args:
|
|
989
|
+
data_sink: Schema for creating a data sink.
|
|
990
|
+
|
|
991
|
+
data_sink_id: Data sink ID. When provided instead of data_sink, the data sink will be looked
|
|
992
|
+
up by ID.
|
|
993
|
+
|
|
994
|
+
embedding_model_config_id: Embedding model config ID. When provided instead of embedding_config, the
|
|
995
|
+
embedding model config will be looked up by ID.
|
|
996
|
+
|
|
997
|
+
llama_parse_parameters: Settings that can be configured for how to use LlamaParse to parse files within
|
|
998
|
+
a LlamaCloud pipeline.
|
|
999
|
+
|
|
1000
|
+
managed_pipeline_id: The ID of the ManagedPipeline this playground pipeline is linked to.
|
|
1001
|
+
|
|
1002
|
+
metadata_config: Metadata configuration for the pipeline.
|
|
1003
|
+
|
|
1004
|
+
preset_retrieval_parameters: Schema for the search params for an retrieval execution that can be preset for a
|
|
1005
|
+
pipeline.
|
|
1006
|
+
|
|
1007
|
+
sparse_model_config: Configuration for sparse embedding models used in hybrid search.
|
|
1008
|
+
|
|
1009
|
+
This allows users to choose between Splade and BM25 models for sparse retrieval
|
|
1010
|
+
in managed data sinks.
|
|
1011
|
+
|
|
1012
|
+
status: Status of the pipeline deployment.
|
|
1013
|
+
|
|
1014
|
+
transform_config: Configuration for the transformation.
|
|
1015
|
+
|
|
1016
|
+
extra_headers: Send extra headers
|
|
1017
|
+
|
|
1018
|
+
extra_query: Add additional query parameters to the request
|
|
1019
|
+
|
|
1020
|
+
extra_body: Add additional JSON properties to the request
|
|
1021
|
+
|
|
1022
|
+
timeout: Override the client-level default timeout for this request, in seconds
|
|
1023
|
+
"""
|
|
1024
|
+
if not pipeline_id:
|
|
1025
|
+
raise ValueError(f"Expected a non-empty value for `pipeline_id` but received {pipeline_id!r}")
|
|
1026
|
+
return await self._put(
|
|
1027
|
+
f"/api/v1/pipelines/{pipeline_id}",
|
|
1028
|
+
body=await async_maybe_transform(
|
|
1029
|
+
{
|
|
1030
|
+
"data_sink": data_sink,
|
|
1031
|
+
"data_sink_id": data_sink_id,
|
|
1032
|
+
"embedding_config": embedding_config,
|
|
1033
|
+
"embedding_model_config_id": embedding_model_config_id,
|
|
1034
|
+
"llama_parse_parameters": llama_parse_parameters,
|
|
1035
|
+
"managed_pipeline_id": managed_pipeline_id,
|
|
1036
|
+
"metadata_config": metadata_config,
|
|
1037
|
+
"name": name,
|
|
1038
|
+
"preset_retrieval_parameters": preset_retrieval_parameters,
|
|
1039
|
+
"sparse_model_config": sparse_model_config,
|
|
1040
|
+
"status": status,
|
|
1041
|
+
"transform_config": transform_config,
|
|
1042
|
+
},
|
|
1043
|
+
pipeline_update_params.PipelineUpdateParams,
|
|
1044
|
+
),
|
|
1045
|
+
options=make_request_options(
|
|
1046
|
+
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
|
|
1047
|
+
),
|
|
1048
|
+
cast_to=Pipeline,
|
|
1049
|
+
)
|
|
1050
|
+
|
|
1051
|
+
async def list(
|
|
1052
|
+
self,
|
|
1053
|
+
*,
|
|
1054
|
+
organization_id: Optional[str] | Omit = omit,
|
|
1055
|
+
pipeline_name: Optional[str] | Omit = omit,
|
|
1056
|
+
pipeline_type: Optional[PipelineType] | Omit = omit,
|
|
1057
|
+
project_id: Optional[str] | Omit = omit,
|
|
1058
|
+
project_name: Optional[str] | Omit = omit,
|
|
1059
|
+
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
|
|
1060
|
+
# The extra values given here take precedence over values defined on the client or passed to this method.
|
|
1061
|
+
extra_headers: Headers | None = None,
|
|
1062
|
+
extra_query: Query | None = None,
|
|
1063
|
+
extra_body: Body | None = None,
|
|
1064
|
+
timeout: float | httpx.Timeout | None | NotGiven = not_given,
|
|
1065
|
+
) -> PipelineListResponse:
|
|
1066
|
+
"""
|
|
1067
|
+
Search for pipelines by various parameters.
|
|
1068
|
+
|
|
1069
|
+
Args:
|
|
1070
|
+
pipeline_type: Enum for representing the type of a pipeline
|
|
1071
|
+
|
|
1072
|
+
extra_headers: Send extra headers
|
|
1073
|
+
|
|
1074
|
+
extra_query: Add additional query parameters to the request
|
|
1075
|
+
|
|
1076
|
+
extra_body: Add additional JSON properties to the request
|
|
1077
|
+
|
|
1078
|
+
timeout: Override the client-level default timeout for this request, in seconds
|
|
1079
|
+
"""
|
|
1080
|
+
return await self._get(
|
|
1081
|
+
"/api/v1/pipelines",
|
|
1082
|
+
options=make_request_options(
|
|
1083
|
+
extra_headers=extra_headers,
|
|
1084
|
+
extra_query=extra_query,
|
|
1085
|
+
extra_body=extra_body,
|
|
1086
|
+
timeout=timeout,
|
|
1087
|
+
query=await async_maybe_transform(
|
|
1088
|
+
{
|
|
1089
|
+
"organization_id": organization_id,
|
|
1090
|
+
"pipeline_name": pipeline_name,
|
|
1091
|
+
"pipeline_type": pipeline_type,
|
|
1092
|
+
"project_id": project_id,
|
|
1093
|
+
"project_name": project_name,
|
|
1094
|
+
},
|
|
1095
|
+
pipeline_list_params.PipelineListParams,
|
|
1096
|
+
),
|
|
1097
|
+
),
|
|
1098
|
+
cast_to=PipelineListResponse,
|
|
1099
|
+
)
|
|
1100
|
+
|
|
1101
|
+
async def delete(
|
|
1102
|
+
self,
|
|
1103
|
+
pipeline_id: str,
|
|
1104
|
+
*,
|
|
1105
|
+
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
|
|
1106
|
+
# The extra values given here take precedence over values defined on the client or passed to this method.
|
|
1107
|
+
extra_headers: Headers | None = None,
|
|
1108
|
+
extra_query: Query | None = None,
|
|
1109
|
+
extra_body: Body | None = None,
|
|
1110
|
+
timeout: float | httpx.Timeout | None | NotGiven = not_given,
|
|
1111
|
+
) -> None:
|
|
1112
|
+
"""
|
|
1113
|
+
Delete a pipeline by ID.
|
|
1114
|
+
|
|
1115
|
+
Args:
|
|
1116
|
+
extra_headers: Send extra headers
|
|
1117
|
+
|
|
1118
|
+
extra_query: Add additional query parameters to the request
|
|
1119
|
+
|
|
1120
|
+
extra_body: Add additional JSON properties to the request
|
|
1121
|
+
|
|
1122
|
+
timeout: Override the client-level default timeout for this request, in seconds
|
|
1123
|
+
"""
|
|
1124
|
+
if not pipeline_id:
|
|
1125
|
+
raise ValueError(f"Expected a non-empty value for `pipeline_id` but received {pipeline_id!r}")
|
|
1126
|
+
extra_headers = {"Accept": "*/*", **(extra_headers or {})}
|
|
1127
|
+
return await self._delete(
|
|
1128
|
+
f"/api/v1/pipelines/{pipeline_id}",
|
|
1129
|
+
options=make_request_options(
|
|
1130
|
+
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
|
|
1131
|
+
),
|
|
1132
|
+
cast_to=NoneType,
|
|
1133
|
+
)
|
|
1134
|
+
|
|
1135
|
+
async def get(
|
|
1136
|
+
self,
|
|
1137
|
+
pipeline_id: str,
|
|
1138
|
+
*,
|
|
1139
|
+
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
|
|
1140
|
+
# The extra values given here take precedence over values defined on the client or passed to this method.
|
|
1141
|
+
extra_headers: Headers | None = None,
|
|
1142
|
+
extra_query: Query | None = None,
|
|
1143
|
+
extra_body: Body | None = None,
|
|
1144
|
+
timeout: float | httpx.Timeout | None | NotGiven = not_given,
|
|
1145
|
+
) -> Pipeline:
|
|
1146
|
+
"""
|
|
1147
|
+
Get a pipeline by ID for a given project.
|
|
1148
|
+
|
|
1149
|
+
Args:
|
|
1150
|
+
extra_headers: Send extra headers
|
|
1151
|
+
|
|
1152
|
+
extra_query: Add additional query parameters to the request
|
|
1153
|
+
|
|
1154
|
+
extra_body: Add additional JSON properties to the request
|
|
1155
|
+
|
|
1156
|
+
timeout: Override the client-level default timeout for this request, in seconds
|
|
1157
|
+
"""
|
|
1158
|
+
if not pipeline_id:
|
|
1159
|
+
raise ValueError(f"Expected a non-empty value for `pipeline_id` but received {pipeline_id!r}")
|
|
1160
|
+
return await self._get(
|
|
1161
|
+
f"/api/v1/pipelines/{pipeline_id}",
|
|
1162
|
+
options=make_request_options(
|
|
1163
|
+
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
|
|
1164
|
+
),
|
|
1165
|
+
cast_to=Pipeline,
|
|
1166
|
+
)
|
|
1167
|
+
|
|
1168
|
+
async def get_status(
|
|
1169
|
+
self,
|
|
1170
|
+
pipeline_id: str,
|
|
1171
|
+
*,
|
|
1172
|
+
full_details: Optional[bool] | Omit = omit,
|
|
1173
|
+
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
|
|
1174
|
+
# The extra values given here take precedence over values defined on the client or passed to this method.
|
|
1175
|
+
extra_headers: Headers | None = None,
|
|
1176
|
+
extra_query: Query | None = None,
|
|
1177
|
+
extra_body: Body | None = None,
|
|
1178
|
+
timeout: float | httpx.Timeout | None | NotGiven = not_given,
|
|
1179
|
+
) -> ManagedIngestionStatusResponse:
|
|
1180
|
+
"""
|
|
1181
|
+
Get the status of a pipeline by ID.
|
|
1182
|
+
|
|
1183
|
+
Args:
|
|
1184
|
+
extra_headers: Send extra headers
|
|
1185
|
+
|
|
1186
|
+
extra_query: Add additional query parameters to the request
|
|
1187
|
+
|
|
1188
|
+
extra_body: Add additional JSON properties to the request
|
|
1189
|
+
|
|
1190
|
+
timeout: Override the client-level default timeout for this request, in seconds
|
|
1191
|
+
"""
|
|
1192
|
+
if not pipeline_id:
|
|
1193
|
+
raise ValueError(f"Expected a non-empty value for `pipeline_id` but received {pipeline_id!r}")
|
|
1194
|
+
return await self._get(
|
|
1195
|
+
f"/api/v1/pipelines/{pipeline_id}/status",
|
|
1196
|
+
options=make_request_options(
|
|
1197
|
+
extra_headers=extra_headers,
|
|
1198
|
+
extra_query=extra_query,
|
|
1199
|
+
extra_body=extra_body,
|
|
1200
|
+
timeout=timeout,
|
|
1201
|
+
query=await async_maybe_transform(
|
|
1202
|
+
{"full_details": full_details}, pipeline_get_status_params.PipelineGetStatusParams
|
|
1203
|
+
),
|
|
1204
|
+
),
|
|
1205
|
+
cast_to=ManagedIngestionStatusResponse,
|
|
1206
|
+
)
|
|
1207
|
+
|
|
1208
|
+
async def upsert(
|
|
1209
|
+
self,
|
|
1210
|
+
*,
|
|
1211
|
+
name: str,
|
|
1212
|
+
organization_id: Optional[str] | Omit = omit,
|
|
1213
|
+
project_id: Optional[str] | Omit = omit,
|
|
1214
|
+
data_sink: Optional[DataSinkCreateParam] | Omit = omit,
|
|
1215
|
+
data_sink_id: Optional[str] | Omit = omit,
|
|
1216
|
+
embedding_config: Optional[pipeline_upsert_params.EmbeddingConfig] | Omit = omit,
|
|
1217
|
+
embedding_model_config_id: Optional[str] | Omit = omit,
|
|
1218
|
+
llama_parse_parameters: LlamaParseParametersParam | Omit = omit,
|
|
1219
|
+
managed_pipeline_id: Optional[str] | Omit = omit,
|
|
1220
|
+
metadata_config: Optional[PipelineMetadataConfigParam] | Omit = omit,
|
|
1221
|
+
pipeline_type: PipelineType | Omit = omit,
|
|
1222
|
+
preset_retrieval_parameters: PresetRetrievalParamsParam | Omit = omit,
|
|
1223
|
+
sparse_model_config: Optional[SparseModelConfigParam] | Omit = omit,
|
|
1224
|
+
status: Optional[str] | Omit = omit,
|
|
1225
|
+
transform_config: Optional[pipeline_upsert_params.TransformConfig] | Omit = omit,
|
|
1226
|
+
# Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
|
|
1227
|
+
# The extra values given here take precedence over values defined on the client or passed to this method.
|
|
1228
|
+
extra_headers: Headers | None = None,
|
|
1229
|
+
extra_query: Query | None = None,
|
|
1230
|
+
extra_body: Body | None = None,
|
|
1231
|
+
timeout: float | httpx.Timeout | None | NotGiven = not_given,
|
|
1232
|
+
) -> Pipeline:
|
|
1233
|
+
"""Upsert a pipeline for a project.
|
|
1234
|
+
|
|
1235
|
+
Updates if a pipeline with the same name and
|
|
1236
|
+
project_id already exists. Otherwise, creates a new pipeline.
|
|
1237
|
+
|
|
1238
|
+
Args:
|
|
1239
|
+
data_sink: Schema for creating a data sink.
|
|
1240
|
+
|
|
1241
|
+
data_sink_id: Data sink ID. When provided instead of data_sink, the data sink will be looked
|
|
1242
|
+
up by ID.
|
|
1243
|
+
|
|
1244
|
+
embedding_model_config_id: Embedding model config ID. When provided instead of embedding_config, the
|
|
1245
|
+
embedding model config will be looked up by ID.
|
|
1246
|
+
|
|
1247
|
+
llama_parse_parameters: Settings that can be configured for how to use LlamaParse to parse files within
|
|
1248
|
+
a LlamaCloud pipeline.
|
|
1249
|
+
|
|
1250
|
+
managed_pipeline_id: The ID of the ManagedPipeline this playground pipeline is linked to.
|
|
1251
|
+
|
|
1252
|
+
metadata_config: Metadata configuration for the pipeline.
|
|
1253
|
+
|
|
1254
|
+
pipeline_type: Type of pipeline. Either PLAYGROUND or MANAGED.
|
|
1255
|
+
|
|
1256
|
+
preset_retrieval_parameters: Preset retrieval parameters for the pipeline.
|
|
1257
|
+
|
|
1258
|
+
sparse_model_config: Configuration for sparse embedding models used in hybrid search.
|
|
1259
|
+
|
|
1260
|
+
This allows users to choose between Splade and BM25 models for sparse retrieval
|
|
1261
|
+
in managed data sinks.
|
|
1262
|
+
|
|
1263
|
+
status: Status of the pipeline deployment.
|
|
1264
|
+
|
|
1265
|
+
transform_config: Configuration for the transformation.
|
|
1266
|
+
|
|
1267
|
+
extra_headers: Send extra headers
|
|
1268
|
+
|
|
1269
|
+
extra_query: Add additional query parameters to the request
|
|
1270
|
+
|
|
1271
|
+
extra_body: Add additional JSON properties to the request
|
|
1272
|
+
|
|
1273
|
+
timeout: Override the client-level default timeout for this request, in seconds
|
|
1274
|
+
"""
|
|
1275
|
+
return await self._put(
|
|
1276
|
+
"/api/v1/pipelines",
|
|
1277
|
+
body=await async_maybe_transform(
|
|
1278
|
+
{
|
|
1279
|
+
"name": name,
|
|
1280
|
+
"data_sink": data_sink,
|
|
1281
|
+
"data_sink_id": data_sink_id,
|
|
1282
|
+
"embedding_config": embedding_config,
|
|
1283
|
+
"embedding_model_config_id": embedding_model_config_id,
|
|
1284
|
+
"llama_parse_parameters": llama_parse_parameters,
|
|
1285
|
+
"managed_pipeline_id": managed_pipeline_id,
|
|
1286
|
+
"metadata_config": metadata_config,
|
|
1287
|
+
"pipeline_type": pipeline_type,
|
|
1288
|
+
"preset_retrieval_parameters": preset_retrieval_parameters,
|
|
1289
|
+
"sparse_model_config": sparse_model_config,
|
|
1290
|
+
"status": status,
|
|
1291
|
+
"transform_config": transform_config,
|
|
1292
|
+
},
|
|
1293
|
+
pipeline_upsert_params.PipelineUpsertParams,
|
|
1294
|
+
),
|
|
1295
|
+
options=make_request_options(
|
|
1296
|
+
extra_headers=extra_headers,
|
|
1297
|
+
extra_query=extra_query,
|
|
1298
|
+
extra_body=extra_body,
|
|
1299
|
+
timeout=timeout,
|
|
1300
|
+
query=await async_maybe_transform(
|
|
1301
|
+
{
|
|
1302
|
+
"organization_id": organization_id,
|
|
1303
|
+
"project_id": project_id,
|
|
1304
|
+
},
|
|
1305
|
+
pipeline_upsert_params.PipelineUpsertParams,
|
|
1306
|
+
),
|
|
1307
|
+
),
|
|
1308
|
+
cast_to=Pipeline,
|
|
1309
|
+
)
|
|
1310
|
+
|
|
1311
|
+
|
|
1312
|
+
class PipelinesResourceWithRawResponse:
|
|
1313
|
+
def __init__(self, pipelines: PipelinesResource) -> None:
|
|
1314
|
+
self._pipelines = pipelines
|
|
1315
|
+
|
|
1316
|
+
self.create = to_raw_response_wrapper(
|
|
1317
|
+
pipelines.create,
|
|
1318
|
+
)
|
|
1319
|
+
self.retrieve = to_raw_response_wrapper(
|
|
1320
|
+
pipelines.retrieve,
|
|
1321
|
+
)
|
|
1322
|
+
self.update = to_raw_response_wrapper(
|
|
1323
|
+
pipelines.update,
|
|
1324
|
+
)
|
|
1325
|
+
self.list = to_raw_response_wrapper(
|
|
1326
|
+
pipelines.list,
|
|
1327
|
+
)
|
|
1328
|
+
self.delete = to_raw_response_wrapper(
|
|
1329
|
+
pipelines.delete,
|
|
1330
|
+
)
|
|
1331
|
+
self.get = to_raw_response_wrapper(
|
|
1332
|
+
pipelines.get,
|
|
1333
|
+
)
|
|
1334
|
+
self.get_status = to_raw_response_wrapper(
|
|
1335
|
+
pipelines.get_status,
|
|
1336
|
+
)
|
|
1337
|
+
self.upsert = to_raw_response_wrapper(
|
|
1338
|
+
pipelines.upsert,
|
|
1339
|
+
)
|
|
1340
|
+
|
|
1341
|
+
@cached_property
|
|
1342
|
+
def sync(self) -> SyncResourceWithRawResponse:
|
|
1343
|
+
return SyncResourceWithRawResponse(self._pipelines.sync)
|
|
1344
|
+
|
|
1345
|
+
@cached_property
|
|
1346
|
+
def data_sources(self) -> DataSourcesResourceWithRawResponse:
|
|
1347
|
+
return DataSourcesResourceWithRawResponse(self._pipelines.data_sources)
|
|
1348
|
+
|
|
1349
|
+
@cached_property
|
|
1350
|
+
def images(self) -> ImagesResourceWithRawResponse:
|
|
1351
|
+
return ImagesResourceWithRawResponse(self._pipelines.images)
|
|
1352
|
+
|
|
1353
|
+
@cached_property
|
|
1354
|
+
def files(self) -> FilesResourceWithRawResponse:
|
|
1355
|
+
return FilesResourceWithRawResponse(self._pipelines.files)
|
|
1356
|
+
|
|
1357
|
+
@cached_property
|
|
1358
|
+
def metadata(self) -> MetadataResourceWithRawResponse:
|
|
1359
|
+
return MetadataResourceWithRawResponse(self._pipelines.metadata)
|
|
1360
|
+
|
|
1361
|
+
@cached_property
|
|
1362
|
+
def documents(self) -> DocumentsResourceWithRawResponse:
|
|
1363
|
+
return DocumentsResourceWithRawResponse(self._pipelines.documents)
|
|
1364
|
+
|
|
1365
|
+
|
|
1366
|
+
class AsyncPipelinesResourceWithRawResponse:
|
|
1367
|
+
def __init__(self, pipelines: AsyncPipelinesResource) -> None:
|
|
1368
|
+
self._pipelines = pipelines
|
|
1369
|
+
|
|
1370
|
+
self.create = async_to_raw_response_wrapper(
|
|
1371
|
+
pipelines.create,
|
|
1372
|
+
)
|
|
1373
|
+
self.retrieve = async_to_raw_response_wrapper(
|
|
1374
|
+
pipelines.retrieve,
|
|
1375
|
+
)
|
|
1376
|
+
self.update = async_to_raw_response_wrapper(
|
|
1377
|
+
pipelines.update,
|
|
1378
|
+
)
|
|
1379
|
+
self.list = async_to_raw_response_wrapper(
|
|
1380
|
+
pipelines.list,
|
|
1381
|
+
)
|
|
1382
|
+
self.delete = async_to_raw_response_wrapper(
|
|
1383
|
+
pipelines.delete,
|
|
1384
|
+
)
|
|
1385
|
+
self.get = async_to_raw_response_wrapper(
|
|
1386
|
+
pipelines.get,
|
|
1387
|
+
)
|
|
1388
|
+
self.get_status = async_to_raw_response_wrapper(
|
|
1389
|
+
pipelines.get_status,
|
|
1390
|
+
)
|
|
1391
|
+
self.upsert = async_to_raw_response_wrapper(
|
|
1392
|
+
pipelines.upsert,
|
|
1393
|
+
)
|
|
1394
|
+
|
|
1395
|
+
@cached_property
|
|
1396
|
+
def sync(self) -> AsyncSyncResourceWithRawResponse:
|
|
1397
|
+
return AsyncSyncResourceWithRawResponse(self._pipelines.sync)
|
|
1398
|
+
|
|
1399
|
+
@cached_property
|
|
1400
|
+
def data_sources(self) -> AsyncDataSourcesResourceWithRawResponse:
|
|
1401
|
+
return AsyncDataSourcesResourceWithRawResponse(self._pipelines.data_sources)
|
|
1402
|
+
|
|
1403
|
+
@cached_property
|
|
1404
|
+
def images(self) -> AsyncImagesResourceWithRawResponse:
|
|
1405
|
+
return AsyncImagesResourceWithRawResponse(self._pipelines.images)
|
|
1406
|
+
|
|
1407
|
+
@cached_property
|
|
1408
|
+
def files(self) -> AsyncFilesResourceWithRawResponse:
|
|
1409
|
+
return AsyncFilesResourceWithRawResponse(self._pipelines.files)
|
|
1410
|
+
|
|
1411
|
+
@cached_property
|
|
1412
|
+
def metadata(self) -> AsyncMetadataResourceWithRawResponse:
|
|
1413
|
+
return AsyncMetadataResourceWithRawResponse(self._pipelines.metadata)
|
|
1414
|
+
|
|
1415
|
+
@cached_property
|
|
1416
|
+
def documents(self) -> AsyncDocumentsResourceWithRawResponse:
|
|
1417
|
+
return AsyncDocumentsResourceWithRawResponse(self._pipelines.documents)
|
|
1418
|
+
|
|
1419
|
+
|
|
1420
|
+
class PipelinesResourceWithStreamingResponse:
|
|
1421
|
+
def __init__(self, pipelines: PipelinesResource) -> None:
|
|
1422
|
+
self._pipelines = pipelines
|
|
1423
|
+
|
|
1424
|
+
self.create = to_streamed_response_wrapper(
|
|
1425
|
+
pipelines.create,
|
|
1426
|
+
)
|
|
1427
|
+
self.retrieve = to_streamed_response_wrapper(
|
|
1428
|
+
pipelines.retrieve,
|
|
1429
|
+
)
|
|
1430
|
+
self.update = to_streamed_response_wrapper(
|
|
1431
|
+
pipelines.update,
|
|
1432
|
+
)
|
|
1433
|
+
self.list = to_streamed_response_wrapper(
|
|
1434
|
+
pipelines.list,
|
|
1435
|
+
)
|
|
1436
|
+
self.delete = to_streamed_response_wrapper(
|
|
1437
|
+
pipelines.delete,
|
|
1438
|
+
)
|
|
1439
|
+
self.get = to_streamed_response_wrapper(
|
|
1440
|
+
pipelines.get,
|
|
1441
|
+
)
|
|
1442
|
+
self.get_status = to_streamed_response_wrapper(
|
|
1443
|
+
pipelines.get_status,
|
|
1444
|
+
)
|
|
1445
|
+
self.upsert = to_streamed_response_wrapper(
|
|
1446
|
+
pipelines.upsert,
|
|
1447
|
+
)
|
|
1448
|
+
|
|
1449
|
+
@cached_property
|
|
1450
|
+
def sync(self) -> SyncResourceWithStreamingResponse:
|
|
1451
|
+
return SyncResourceWithStreamingResponse(self._pipelines.sync)
|
|
1452
|
+
|
|
1453
|
+
@cached_property
|
|
1454
|
+
def data_sources(self) -> DataSourcesResourceWithStreamingResponse:
|
|
1455
|
+
return DataSourcesResourceWithStreamingResponse(self._pipelines.data_sources)
|
|
1456
|
+
|
|
1457
|
+
@cached_property
|
|
1458
|
+
def images(self) -> ImagesResourceWithStreamingResponse:
|
|
1459
|
+
return ImagesResourceWithStreamingResponse(self._pipelines.images)
|
|
1460
|
+
|
|
1461
|
+
@cached_property
|
|
1462
|
+
def files(self) -> FilesResourceWithStreamingResponse:
|
|
1463
|
+
return FilesResourceWithStreamingResponse(self._pipelines.files)
|
|
1464
|
+
|
|
1465
|
+
@cached_property
|
|
1466
|
+
def metadata(self) -> MetadataResourceWithStreamingResponse:
|
|
1467
|
+
return MetadataResourceWithStreamingResponse(self._pipelines.metadata)
|
|
1468
|
+
|
|
1469
|
+
@cached_property
|
|
1470
|
+
def documents(self) -> DocumentsResourceWithStreamingResponse:
|
|
1471
|
+
return DocumentsResourceWithStreamingResponse(self._pipelines.documents)
|
|
1472
|
+
|
|
1473
|
+
|
|
1474
|
+
class AsyncPipelinesResourceWithStreamingResponse:
|
|
1475
|
+
def __init__(self, pipelines: AsyncPipelinesResource) -> None:
|
|
1476
|
+
self._pipelines = pipelines
|
|
1477
|
+
|
|
1478
|
+
self.create = async_to_streamed_response_wrapper(
|
|
1479
|
+
pipelines.create,
|
|
1480
|
+
)
|
|
1481
|
+
self.retrieve = async_to_streamed_response_wrapper(
|
|
1482
|
+
pipelines.retrieve,
|
|
1483
|
+
)
|
|
1484
|
+
self.update = async_to_streamed_response_wrapper(
|
|
1485
|
+
pipelines.update,
|
|
1486
|
+
)
|
|
1487
|
+
self.list = async_to_streamed_response_wrapper(
|
|
1488
|
+
pipelines.list,
|
|
1489
|
+
)
|
|
1490
|
+
self.delete = async_to_streamed_response_wrapper(
|
|
1491
|
+
pipelines.delete,
|
|
1492
|
+
)
|
|
1493
|
+
self.get = async_to_streamed_response_wrapper(
|
|
1494
|
+
pipelines.get,
|
|
1495
|
+
)
|
|
1496
|
+
self.get_status = async_to_streamed_response_wrapper(
|
|
1497
|
+
pipelines.get_status,
|
|
1498
|
+
)
|
|
1499
|
+
self.upsert = async_to_streamed_response_wrapper(
|
|
1500
|
+
pipelines.upsert,
|
|
1501
|
+
)
|
|
1502
|
+
|
|
1503
|
+
@cached_property
|
|
1504
|
+
def sync(self) -> AsyncSyncResourceWithStreamingResponse:
|
|
1505
|
+
return AsyncSyncResourceWithStreamingResponse(self._pipelines.sync)
|
|
1506
|
+
|
|
1507
|
+
@cached_property
|
|
1508
|
+
def data_sources(self) -> AsyncDataSourcesResourceWithStreamingResponse:
|
|
1509
|
+
return AsyncDataSourcesResourceWithStreamingResponse(self._pipelines.data_sources)
|
|
1510
|
+
|
|
1511
|
+
@cached_property
|
|
1512
|
+
def images(self) -> AsyncImagesResourceWithStreamingResponse:
|
|
1513
|
+
return AsyncImagesResourceWithStreamingResponse(self._pipelines.images)
|
|
1514
|
+
|
|
1515
|
+
@cached_property
|
|
1516
|
+
def files(self) -> AsyncFilesResourceWithStreamingResponse:
|
|
1517
|
+
return AsyncFilesResourceWithStreamingResponse(self._pipelines.files)
|
|
1518
|
+
|
|
1519
|
+
@cached_property
|
|
1520
|
+
def metadata(self) -> AsyncMetadataResourceWithStreamingResponse:
|
|
1521
|
+
return AsyncMetadataResourceWithStreamingResponse(self._pipelines.metadata)
|
|
1522
|
+
|
|
1523
|
+
@cached_property
|
|
1524
|
+
def documents(self) -> AsyncDocumentsResourceWithStreamingResponse:
|
|
1525
|
+
return AsyncDocumentsResourceWithStreamingResponse(self._pipelines.documents)
|