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.
Files changed (714) hide show
  1. llama_cloud/__init__.py +101 -816
  2. llama_cloud/_base_client.py +2124 -0
  3. llama_cloud/_client.py +795 -0
  4. llama_cloud/_compat.py +219 -0
  5. llama_cloud/_constants.py +14 -0
  6. llama_cloud/_exceptions.py +108 -0
  7. llama_cloud/_files.py +127 -0
  8. llama_cloud/_models.py +872 -0
  9. llama_cloud/_polling.py +182 -0
  10. llama_cloud/_qs.py +150 -0
  11. llama_cloud/_resource.py +43 -0
  12. llama_cloud/_response.py +832 -0
  13. llama_cloud/_streaming.py +333 -0
  14. llama_cloud/_types.py +270 -0
  15. llama_cloud/_utils/__init__.py +64 -0
  16. llama_cloud/_utils/_compat.py +45 -0
  17. llama_cloud/_utils/_datetime_parse.py +136 -0
  18. llama_cloud/_utils/_logs.py +25 -0
  19. llama_cloud/_utils/_proxy.py +65 -0
  20. llama_cloud/_utils/_reflection.py +42 -0
  21. llama_cloud/_utils/_resources_proxy.py +24 -0
  22. llama_cloud/_utils/_streams.py +12 -0
  23. llama_cloud/_utils/_sync.py +58 -0
  24. llama_cloud/_utils/_transform.py +457 -0
  25. llama_cloud/_utils/_typing.py +156 -0
  26. llama_cloud/_utils/_utils.py +421 -0
  27. llama_cloud/_version.py +4 -0
  28. llama_cloud/lib/__init__.py +0 -0
  29. llama_cloud/lib/index/__init__.py +13 -0
  30. llama_cloud/lib/index/api_utils.py +300 -0
  31. llama_cloud/lib/index/base.py +1041 -0
  32. llama_cloud/lib/index/composite_retriever.py +272 -0
  33. llama_cloud/lib/index/retriever.py +233 -0
  34. llama_cloud/pagination.py +465 -0
  35. llama_cloud/py.typed +0 -0
  36. llama_cloud/resources/__init__.py +136 -107
  37. llama_cloud/resources/beta/__init__.py +102 -1
  38. llama_cloud/resources/beta/agent_data.py +1041 -0
  39. llama_cloud/resources/beta/batch/__init__.py +33 -0
  40. llama_cloud/resources/beta/batch/batch.py +664 -0
  41. llama_cloud/resources/beta/batch/job_items.py +348 -0
  42. llama_cloud/resources/beta/beta.py +262 -0
  43. llama_cloud/resources/beta/directories/__init__.py +33 -0
  44. llama_cloud/resources/beta/directories/directories.py +719 -0
  45. llama_cloud/resources/beta/directories/files.py +913 -0
  46. llama_cloud/resources/beta/parse_configurations.py +743 -0
  47. llama_cloud/resources/beta/sheets.py +1130 -0
  48. llama_cloud/resources/beta/split.py +917 -0
  49. llama_cloud/resources/classifier/__init__.py +32 -1
  50. llama_cloud/resources/classifier/classifier.py +588 -0
  51. llama_cloud/resources/classifier/jobs.py +563 -0
  52. llama_cloud/resources/data_sinks.py +579 -0
  53. llama_cloud/resources/data_sources.py +651 -0
  54. llama_cloud/resources/extraction/__init__.py +61 -0
  55. llama_cloud/resources/extraction/extraction.py +609 -0
  56. llama_cloud/resources/extraction/extraction_agents/__init__.py +33 -0
  57. llama_cloud/resources/extraction/extraction_agents/extraction_agents.py +633 -0
  58. llama_cloud/resources/extraction/extraction_agents/schema.py +308 -0
  59. llama_cloud/resources/extraction/jobs.py +1106 -0
  60. llama_cloud/resources/extraction/runs.py +498 -0
  61. llama_cloud/resources/files.py +784 -0
  62. llama_cloud/resources/parsing.py +1296 -0
  63. llama_cloud/resources/pipelines/__init__.py +98 -24
  64. llama_cloud/resources/pipelines/data_sources.py +529 -0
  65. llama_cloud/resources/pipelines/documents.py +810 -0
  66. llama_cloud/resources/pipelines/files.py +682 -0
  67. llama_cloud/resources/pipelines/images.py +513 -0
  68. llama_cloud/resources/pipelines/metadata.py +265 -0
  69. llama_cloud/resources/pipelines/pipelines.py +1525 -0
  70. llama_cloud/resources/pipelines/sync.py +243 -0
  71. llama_cloud/resources/projects.py +276 -0
  72. llama_cloud/resources/retrievers/__init__.py +32 -1
  73. llama_cloud/resources/retrievers/retriever.py +238 -0
  74. llama_cloud/resources/retrievers/retrievers.py +920 -0
  75. llama_cloud/types/__init__.py +171 -721
  76. llama_cloud/types/advanced_mode_transform_config.py +102 -38
  77. llama_cloud/types/advanced_mode_transform_config_param.py +102 -0
  78. llama_cloud/types/auto_transform_config.py +11 -25
  79. llama_cloud/types/auto_transform_config_param.py +17 -0
  80. llama_cloud/types/azure_openai_embedding.py +62 -0
  81. llama_cloud/types/azure_openai_embedding_config.py +17 -0
  82. llama_cloud/types/azure_openai_embedding_config_param.py +17 -0
  83. llama_cloud/types/azure_openai_embedding_param.py +61 -0
  84. llama_cloud/types/b_box.py +37 -0
  85. llama_cloud/types/bedrock_embedding.py +49 -46
  86. llama_cloud/types/bedrock_embedding_config.py +10 -27
  87. llama_cloud/types/bedrock_embedding_config_param.py +17 -0
  88. llama_cloud/types/bedrock_embedding_param.py +48 -0
  89. llama_cloud/types/beta/__init__.py +59 -0
  90. llama_cloud/types/beta/agent_data.py +26 -0
  91. llama_cloud/types/beta/agent_data_agent_data_params.py +20 -0
  92. llama_cloud/types/beta/agent_data_aggregate_params.py +79 -0
  93. llama_cloud/types/beta/agent_data_aggregate_response.py +17 -0
  94. llama_cloud/types/beta/agent_data_delete_by_query_params.py +43 -0
  95. llama_cloud/types/beta/agent_data_delete_by_query_response.py +11 -0
  96. llama_cloud/types/beta/agent_data_delete_params.py +14 -0
  97. llama_cloud/types/beta/agent_data_delete_response.py +8 -0
  98. llama_cloud/types/beta/agent_data_get_params.py +14 -0
  99. llama_cloud/types/beta/agent_data_search_params.py +69 -0
  100. llama_cloud/types/beta/agent_data_update_params.py +16 -0
  101. llama_cloud/types/beta/batch/__init__.py +12 -0
  102. llama_cloud/types/beta/batch/job_item_get_processing_results_params.py +17 -0
  103. llama_cloud/types/beta/batch/job_item_get_processing_results_response.py +409 -0
  104. llama_cloud/types/beta/batch/job_item_list_params.py +23 -0
  105. llama_cloud/types/beta/batch/job_item_list_response.py +42 -0
  106. llama_cloud/types/beta/batch_cancel_params.py +21 -0
  107. llama_cloud/types/beta/batch_cancel_response.py +23 -0
  108. llama_cloud/types/beta/batch_create_params.py +399 -0
  109. llama_cloud/types/beta/batch_create_response.py +63 -0
  110. llama_cloud/types/beta/batch_get_status_params.py +14 -0
  111. llama_cloud/types/beta/batch_get_status_response.py +73 -0
  112. llama_cloud/types/beta/batch_list_params.py +29 -0
  113. llama_cloud/types/beta/batch_list_response.py +63 -0
  114. llama_cloud/types/beta/directories/__init__.py +15 -0
  115. llama_cloud/types/beta/directories/file_add_params.py +26 -0
  116. llama_cloud/types/beta/directories/file_add_response.py +42 -0
  117. llama_cloud/types/beta/directories/file_delete_params.py +16 -0
  118. llama_cloud/types/beta/directories/file_get_params.py +16 -0
  119. llama_cloud/types/beta/directories/file_get_response.py +42 -0
  120. llama_cloud/types/beta/directories/file_list_params.py +28 -0
  121. llama_cloud/types/beta/directories/file_list_response.py +42 -0
  122. llama_cloud/types/beta/directories/file_update_params.py +27 -0
  123. llama_cloud/types/beta/directories/file_update_response.py +42 -0
  124. llama_cloud/types/beta/directories/file_upload_params.py +24 -0
  125. llama_cloud/types/beta/directories/file_upload_response.py +42 -0
  126. llama_cloud/types/beta/directory_create_params.py +23 -0
  127. llama_cloud/types/beta/directory_create_response.py +36 -0
  128. llama_cloud/types/beta/directory_delete_params.py +14 -0
  129. llama_cloud/types/beta/directory_get_params.py +14 -0
  130. llama_cloud/types/beta/directory_get_response.py +36 -0
  131. llama_cloud/types/beta/directory_list_params.py +24 -0
  132. llama_cloud/types/beta/directory_list_response.py +36 -0
  133. llama_cloud/types/beta/directory_update_params.py +20 -0
  134. llama_cloud/types/beta/directory_update_response.py +36 -0
  135. llama_cloud/types/beta/parse_configuration.py +40 -0
  136. llama_cloud/types/beta/parse_configuration_create_params.py +34 -0
  137. llama_cloud/types/beta/parse_configuration_delete_params.py +14 -0
  138. llama_cloud/types/beta/parse_configuration_get_params.py +14 -0
  139. llama_cloud/types/beta/parse_configuration_list_params.py +24 -0
  140. llama_cloud/types/beta/parse_configuration_query_response.py +28 -0
  141. llama_cloud/types/beta/parse_configuration_update_params.py +22 -0
  142. llama_cloud/types/beta/sheet_create_params.py +22 -0
  143. llama_cloud/types/beta/sheet_delete_job_params.py +14 -0
  144. llama_cloud/types/beta/sheet_get_params.py +16 -0
  145. llama_cloud/types/beta/sheet_get_result_table_params.py +20 -0
  146. llama_cloud/types/beta/sheet_list_params.py +20 -0
  147. llama_cloud/types/beta/sheets_job.py +88 -0
  148. llama_cloud/types/beta/sheets_parsing_config.py +49 -0
  149. llama_cloud/types/beta/sheets_parsing_config_param.py +51 -0
  150. llama_cloud/types/beta/split_category.py +17 -0
  151. llama_cloud/types/beta/split_category_param.py +18 -0
  152. llama_cloud/types/beta/split_create_params.py +36 -0
  153. llama_cloud/types/beta/split_create_response.py +48 -0
  154. llama_cloud/types/beta/split_document_input.py +15 -0
  155. llama_cloud/types/beta/split_document_input_param.py +17 -0
  156. llama_cloud/types/beta/split_get_params.py +14 -0
  157. llama_cloud/types/beta/split_get_response.py +48 -0
  158. llama_cloud/types/beta/split_list_params.py +18 -0
  159. llama_cloud/types/beta/split_list_response.py +48 -0
  160. llama_cloud/types/beta/split_result_response.py +15 -0
  161. llama_cloud/types/beta/split_segment_response.py +20 -0
  162. llama_cloud/types/classifier/__init__.py +15 -0
  163. llama_cloud/types/classifier/classifier_rule.py +25 -0
  164. llama_cloud/types/classifier/classifier_rule_param.py +27 -0
  165. llama_cloud/types/classifier/classify_job.py +51 -0
  166. llama_cloud/types/classifier/classify_job_param.py +53 -0
  167. llama_cloud/types/classifier/classify_parsing_configuration.py +21 -0
  168. llama_cloud/types/classifier/classify_parsing_configuration_param.py +23 -0
  169. llama_cloud/types/classifier/job_create_params.py +30 -0
  170. llama_cloud/types/classifier/job_get_params.py +14 -0
  171. llama_cloud/types/classifier/job_get_results_params.py +14 -0
  172. llama_cloud/types/classifier/job_get_results_response.py +66 -0
  173. llama_cloud/types/classifier/job_list_params.py +18 -0
  174. llama_cloud/types/cohere_embedding.py +37 -40
  175. llama_cloud/types/cohere_embedding_config.py +10 -27
  176. llama_cloud/types/cohere_embedding_config_param.py +17 -0
  177. llama_cloud/types/cohere_embedding_param.py +36 -0
  178. llama_cloud/types/composite_retrieval_mode.py +4 -18
  179. llama_cloud/types/composite_retrieval_result.py +52 -37
  180. llama_cloud/types/data_sink.py +46 -39
  181. llama_cloud/types/data_sink_create_param.py +41 -0
  182. llama_cloud/types/data_sink_create_params.py +44 -0
  183. llama_cloud/types/data_sink_list_params.py +14 -0
  184. llama_cloud/types/data_sink_list_response.py +10 -0
  185. llama_cloud/types/data_sink_update_params.py +40 -0
  186. llama_cloud/types/data_source.py +67 -39
  187. llama_cloud/types/data_source_create_params.py +65 -0
  188. llama_cloud/types/data_source_list_params.py +14 -0
  189. llama_cloud/types/data_source_list_response.py +10 -0
  190. llama_cloud/types/data_source_reader_version_metadata.py +8 -27
  191. llama_cloud/types/data_source_update_params.py +61 -0
  192. llama_cloud/types/extraction/__init__.py +25 -0
  193. llama_cloud/types/extraction/extract_agent.py +41 -0
  194. llama_cloud/types/extraction/extract_config.py +118 -0
  195. llama_cloud/types/extraction/extract_config_param.py +118 -0
  196. llama_cloud/types/extraction/extract_job.py +32 -0
  197. llama_cloud/types/extraction/extract_run.py +64 -0
  198. llama_cloud/types/extraction/extraction_agent_create_params.py +25 -0
  199. llama_cloud/types/extraction/extraction_agent_list_params.py +17 -0
  200. llama_cloud/types/extraction/extraction_agent_list_response.py +10 -0
  201. llama_cloud/types/extraction/extraction_agent_update_params.py +18 -0
  202. llama_cloud/types/extraction/extraction_agents/__init__.py +8 -0
  203. llama_cloud/types/extraction/extraction_agents/schema_generate_schema_params.py +23 -0
  204. llama_cloud/types/extraction/extraction_agents/schema_generate_schema_response.py +14 -0
  205. llama_cloud/types/extraction/extraction_agents/schema_validate_schema_params.py +12 -0
  206. llama_cloud/types/extraction/extraction_agents/schema_validate_schema_response.py +13 -0
  207. llama_cloud/types/extraction/job_create_params.py +38 -0
  208. llama_cloud/types/extraction/job_file_params.py +29 -0
  209. llama_cloud/types/extraction/job_get_result_params.py +14 -0
  210. llama_cloud/types/extraction/job_get_result_response.py +27 -0
  211. llama_cloud/types/extraction/job_list_params.py +11 -0
  212. llama_cloud/types/extraction/job_list_response.py +10 -0
  213. llama_cloud/types/extraction/run_delete_params.py +14 -0
  214. llama_cloud/types/extraction/run_get_by_job_params.py +14 -0
  215. llama_cloud/types/extraction/run_get_params.py +14 -0
  216. llama_cloud/types/extraction/run_list_params.py +15 -0
  217. llama_cloud/types/extraction/webhook_configuration.py +43 -0
  218. llama_cloud/types/extraction/webhook_configuration_param.py +43 -0
  219. llama_cloud/types/extraction_run_params.py +45 -0
  220. llama_cloud/types/fail_page_mode.py +4 -26
  221. llama_cloud/types/file.py +48 -40
  222. llama_cloud/types/file_create_params.py +28 -0
  223. llama_cloud/types/file_create_response.py +38 -0
  224. llama_cloud/types/file_delete_params.py +14 -0
  225. llama_cloud/types/file_get_params.py +16 -0
  226. llama_cloud/types/file_list_params.py +40 -0
  227. llama_cloud/types/file_list_response.py +38 -0
  228. llama_cloud/types/file_query_params.py +61 -0
  229. llama_cloud/types/file_query_response.py +47 -27
  230. llama_cloud/types/gemini_embedding.py +40 -39
  231. llama_cloud/types/gemini_embedding_config.py +10 -27
  232. llama_cloud/types/gemini_embedding_config_param.py +17 -0
  233. llama_cloud/types/gemini_embedding_param.py +39 -0
  234. llama_cloud/types/hugging_face_inference_api_embedding.py +62 -46
  235. llama_cloud/types/hugging_face_inference_api_embedding_config.py +11 -28
  236. llama_cloud/types/hugging_face_inference_api_embedding_config_param.py +17 -0
  237. llama_cloud/types/hugging_face_inference_api_embedding_param.py +60 -0
  238. llama_cloud/types/list_item.py +48 -0
  239. llama_cloud/types/llama_parse_parameters.py +251 -130
  240. llama_cloud/types/llama_parse_parameters_param.py +261 -0
  241. llama_cloud/types/llama_parse_supported_file_extensions.py +84 -310
  242. llama_cloud/types/managed_ingestion_status_response.py +39 -37
  243. llama_cloud/types/message_role.py +4 -46
  244. llama_cloud/types/metadata_filters.py +45 -29
  245. llama_cloud/types/metadata_filters_param.py +58 -0
  246. llama_cloud/types/openai_embedding.py +56 -0
  247. llama_cloud/types/openai_embedding_config.py +17 -0
  248. llama_cloud/types/openai_embedding_config_param.py +17 -0
  249. llama_cloud/types/openai_embedding_param.py +55 -0
  250. llama_cloud/types/page_figure_node_with_score.py +32 -29
  251. llama_cloud/types/page_screenshot_node_with_score.py +23 -29
  252. llama_cloud/types/parsing_create_params.py +586 -0
  253. llama_cloud/types/parsing_create_response.py +33 -0
  254. llama_cloud/types/parsing_get_params.py +27 -0
  255. llama_cloud/types/parsing_get_response.py +364 -0
  256. llama_cloud/types/parsing_languages.py +94 -0
  257. llama_cloud/types/parsing_list_params.py +23 -0
  258. llama_cloud/types/parsing_list_response.py +33 -0
  259. llama_cloud/types/parsing_mode.py +13 -46
  260. llama_cloud/types/parsing_upload_file_params.py +14 -0
  261. llama_cloud/types/parsing_upload_file_response.py +33 -0
  262. llama_cloud/types/pipeline.py +180 -62
  263. llama_cloud/types/pipeline_create_params.py +95 -0
  264. llama_cloud/types/pipeline_get_status_params.py +12 -0
  265. llama_cloud/types/pipeline_list_params.py +23 -0
  266. llama_cloud/types/pipeline_list_response.py +12 -0
  267. llama_cloud/types/pipeline_metadata_config.py +9 -30
  268. llama_cloud/types/pipeline_metadata_config_param.py +17 -0
  269. llama_cloud/types/pipeline_retrieve_params.py +74 -0
  270. llama_cloud/types/pipeline_retrieve_response.py +63 -0
  271. llama_cloud/types/pipeline_type.py +4 -18
  272. llama_cloud/types/pipeline_update_params.py +90 -0
  273. llama_cloud/types/pipeline_upsert_params.py +95 -0
  274. llama_cloud/types/pipelines/__init__.py +38 -0
  275. llama_cloud/types/pipelines/cloud_document.py +29 -0
  276. llama_cloud/types/pipelines/cloud_document_create_param.py +30 -0
  277. llama_cloud/types/pipelines/data_source_get_data_sources_response.py +10 -0
  278. llama_cloud/types/pipelines/data_source_sync_params.py +16 -0
  279. llama_cloud/types/pipelines/data_source_update_data_sources_params.py +25 -0
  280. llama_cloud/types/pipelines/data_source_update_data_sources_response.py +10 -0
  281. llama_cloud/types/pipelines/data_source_update_params.py +15 -0
  282. llama_cloud/types/pipelines/document_create_params.py +14 -0
  283. llama_cloud/types/pipelines/document_create_response.py +10 -0
  284. llama_cloud/types/pipelines/document_get_chunks_response.py +10 -0
  285. llama_cloud/types/pipelines/document_list_params.py +22 -0
  286. llama_cloud/types/pipelines/document_upsert_params.py +14 -0
  287. llama_cloud/types/pipelines/document_upsert_response.py +10 -0
  288. llama_cloud/types/pipelines/file_create_params.py +22 -0
  289. llama_cloud/types/pipelines/file_create_response.py +10 -0
  290. llama_cloud/types/pipelines/file_get_status_counts_params.py +14 -0
  291. llama_cloud/types/pipelines/file_get_status_counts_response.py +24 -0
  292. llama_cloud/types/pipelines/file_list_params.py +22 -0
  293. llama_cloud/types/pipelines/file_update_params.py +15 -0
  294. llama_cloud/types/pipelines/image_get_page_figure_params.py +18 -0
  295. llama_cloud/types/pipelines/image_get_page_screenshot_params.py +16 -0
  296. llama_cloud/types/pipelines/image_list_page_figures_params.py +14 -0
  297. llama_cloud/types/pipelines/image_list_page_figures_response.py +34 -0
  298. llama_cloud/types/pipelines/image_list_page_screenshots_params.py +14 -0
  299. llama_cloud/types/pipelines/image_list_page_screenshots_response.py +25 -0
  300. llama_cloud/types/pipelines/metadata_create_params.py +13 -0
  301. llama_cloud/types/pipelines/metadata_create_response.py +8 -0
  302. llama_cloud/types/pipelines/pipeline_data_source.py +96 -0
  303. llama_cloud/types/pipelines/pipeline_file.py +70 -0
  304. llama_cloud/types/pipelines/text_node.py +89 -0
  305. llama_cloud/types/preset_retrieval_params.py +61 -49
  306. llama_cloud/types/preset_retrieval_params_param.py +71 -0
  307. llama_cloud/types/presigned_url.py +13 -29
  308. llama_cloud/types/project.py +24 -36
  309. llama_cloud/types/project_get_params.py +12 -0
  310. llama_cloud/types/project_list_params.py +14 -0
  311. llama_cloud/types/project_list_response.py +10 -0
  312. llama_cloud/types/re_rank_config_param.py +18 -0
  313. llama_cloud/types/retrieval_mode.py +4 -26
  314. llama_cloud/types/retriever.py +31 -38
  315. llama_cloud/types/retriever_create_params.py +26 -0
  316. llama_cloud/types/retriever_get_params.py +14 -0
  317. llama_cloud/types/retriever_list_params.py +16 -0
  318. llama_cloud/types/retriever_list_response.py +12 -0
  319. llama_cloud/types/retriever_pipeline.py +26 -34
  320. llama_cloud/types/retriever_pipeline_param.py +28 -0
  321. llama_cloud/types/retriever_search_params.py +38 -0
  322. llama_cloud/types/retriever_update_params.py +19 -0
  323. llama_cloud/types/retriever_upsert_params.py +26 -0
  324. llama_cloud/types/retrievers/__init__.py +5 -0
  325. llama_cloud/types/retrievers/retriever_search_params.py +32 -0
  326. llama_cloud/types/shared/__init__.py +21 -0
  327. llama_cloud/types/shared/cloud_astra_db_vector_store.py +39 -0
  328. llama_cloud/types/shared/cloud_az_storage_blob_data_source.py +34 -0
  329. llama_cloud/types/shared/cloud_azure_ai_search_vector_store.py +30 -0
  330. llama_cloud/types/shared/cloud_box_data_source.py +31 -0
  331. llama_cloud/types/shared/cloud_confluence_data_source.py +53 -0
  332. llama_cloud/types/shared/cloud_jira_data_source.py +30 -0
  333. llama_cloud/types/shared/cloud_jira_data_source_v2.py +49 -0
  334. llama_cloud/types/shared/cloud_milvus_vector_store.py +21 -0
  335. llama_cloud/types/shared/cloud_mongodb_atlas_vector_search.py +36 -0
  336. llama_cloud/types/shared/cloud_notion_page_data_source.py +19 -0
  337. llama_cloud/types/shared/cloud_one_drive_data_source.py +32 -0
  338. llama_cloud/types/shared/cloud_pinecone_vector_store.py +32 -0
  339. llama_cloud/types/shared/cloud_postgres_vector_store.py +35 -0
  340. llama_cloud/types/shared/cloud_qdrant_vector_store.py +35 -0
  341. llama_cloud/types/shared/cloud_s3_data_source.py +28 -0
  342. llama_cloud/types/shared/cloud_sharepoint_data_source.py +55 -0
  343. llama_cloud/types/shared/cloud_slack_data_source.py +31 -0
  344. llama_cloud/types/shared/failure_handling_config.py +16 -0
  345. llama_cloud/types/shared/pg_vector_hnsw_settings.py +27 -0
  346. llama_cloud/types/shared_params/__init__.py +21 -0
  347. llama_cloud/types/shared_params/cloud_astra_db_vector_store.py +42 -0
  348. llama_cloud/types/shared_params/cloud_az_storage_blob_data_source.py +41 -0
  349. llama_cloud/types/shared_params/cloud_azure_ai_search_vector_store.py +34 -0
  350. llama_cloud/types/shared_params/cloud_box_data_source.py +40 -0
  351. llama_cloud/types/shared_params/cloud_confluence_data_source.py +58 -0
  352. llama_cloud/types/shared_params/cloud_jira_data_source.py +34 -0
  353. llama_cloud/types/shared_params/cloud_jira_data_source_v2.py +54 -0
  354. llama_cloud/types/shared_params/cloud_milvus_vector_store.py +24 -0
  355. llama_cloud/types/shared_params/cloud_mongodb_atlas_vector_search.py +39 -0
  356. llama_cloud/types/shared_params/cloud_notion_page_data_source.py +23 -0
  357. llama_cloud/types/shared_params/cloud_one_drive_data_source.py +37 -0
  358. llama_cloud/types/shared_params/cloud_pinecone_vector_store.py +35 -0
  359. llama_cloud/types/shared_params/cloud_postgres_vector_store.py +39 -0
  360. llama_cloud/types/shared_params/cloud_qdrant_vector_store.py +37 -0
  361. llama_cloud/types/shared_params/cloud_s3_data_source.py +32 -0
  362. llama_cloud/types/shared_params/cloud_sharepoint_data_source.py +60 -0
  363. llama_cloud/types/shared_params/cloud_slack_data_source.py +35 -0
  364. llama_cloud/types/shared_params/failure_handling_config.py +16 -0
  365. llama_cloud/types/shared_params/pg_vector_hnsw_settings.py +26 -0
  366. llama_cloud/types/sparse_model_config.py +16 -30
  367. llama_cloud/types/sparse_model_config_param.py +25 -0
  368. llama_cloud/types/status_enum.py +4 -34
  369. llama_cloud/types/vertex_ai_embedding_config.py +10 -27
  370. llama_cloud/types/vertex_ai_embedding_config_param.py +17 -0
  371. llama_cloud/types/vertex_text_embedding.py +47 -45
  372. llama_cloud/types/vertex_text_embedding_param.py +45 -0
  373. llama_cloud-1.0.0b4.dist-info/METADATA +546 -0
  374. llama_cloud-1.0.0b4.dist-info/RECORD +376 -0
  375. {llama_cloud-0.1.41.dist-info → llama_cloud-1.0.0b4.dist-info}/WHEEL +1 -1
  376. llama_cloud-1.0.0b4.dist-info/licenses/LICENSE +7 -0
  377. llama_cloud/client.py +0 -108
  378. llama_cloud/core/__init__.py +0 -17
  379. llama_cloud/core/api_error.py +0 -15
  380. llama_cloud/core/client_wrapper.py +0 -51
  381. llama_cloud/core/datetime_utils.py +0 -28
  382. llama_cloud/core/jsonable_encoder.py +0 -106
  383. llama_cloud/core/remove_none_from_dict.py +0 -11
  384. llama_cloud/environment.py +0 -7
  385. llama_cloud/errors/__init__.py +0 -5
  386. llama_cloud/errors/unprocessable_entity_error.py +0 -9
  387. llama_cloud/resources/admin/__init__.py +0 -2
  388. llama_cloud/resources/admin/client.py +0 -196
  389. llama_cloud/resources/agent_deployments/__init__.py +0 -2
  390. llama_cloud/resources/agent_deployments/client.py +0 -160
  391. llama_cloud/resources/alpha/__init__.py +0 -2
  392. llama_cloud/resources/alpha/client.py +0 -112
  393. llama_cloud/resources/beta/client.py +0 -2664
  394. llama_cloud/resources/chat_apps/__init__.py +0 -2
  395. llama_cloud/resources/chat_apps/client.py +0 -616
  396. llama_cloud/resources/classifier/client.py +0 -444
  397. llama_cloud/resources/data_sinks/__init__.py +0 -5
  398. llama_cloud/resources/data_sinks/client.py +0 -535
  399. llama_cloud/resources/data_sinks/types/__init__.py +0 -5
  400. llama_cloud/resources/data_sinks/types/data_sink_update_component.py +0 -22
  401. llama_cloud/resources/data_sources/__init__.py +0 -5
  402. llama_cloud/resources/data_sources/client.py +0 -548
  403. llama_cloud/resources/data_sources/types/__init__.py +0 -6
  404. llama_cloud/resources/data_sources/types/data_source_update_component.py +0 -28
  405. llama_cloud/resources/data_sources/types/data_source_update_custom_metadata_value.py +0 -7
  406. llama_cloud/resources/embedding_model_configs/__init__.py +0 -23
  407. llama_cloud/resources/embedding_model_configs/client.py +0 -420
  408. llama_cloud/resources/embedding_model_configs/types/__init__.py +0 -23
  409. llama_cloud/resources/embedding_model_configs/types/embedding_model_config_create_embedding_config.py +0 -89
  410. llama_cloud/resources/evals/__init__.py +0 -2
  411. llama_cloud/resources/evals/client.py +0 -85
  412. llama_cloud/resources/files/__init__.py +0 -5
  413. llama_cloud/resources/files/client.py +0 -1454
  414. llama_cloud/resources/files/types/__init__.py +0 -5
  415. llama_cloud/resources/files/types/file_create_from_url_resource_info_value.py +0 -7
  416. llama_cloud/resources/jobs/__init__.py +0 -2
  417. llama_cloud/resources/jobs/client.py +0 -164
  418. llama_cloud/resources/llama_extract/__init__.py +0 -27
  419. llama_cloud/resources/llama_extract/client.py +0 -2082
  420. llama_cloud/resources/llama_extract/types/__init__.py +0 -25
  421. llama_cloud/resources/llama_extract/types/extract_agent_create_data_schema.py +0 -9
  422. llama_cloud/resources/llama_extract/types/extract_agent_create_data_schema_zero_value.py +0 -7
  423. llama_cloud/resources/llama_extract/types/extract_agent_update_data_schema.py +0 -9
  424. llama_cloud/resources/llama_extract/types/extract_agent_update_data_schema_zero_value.py +0 -7
  425. llama_cloud/resources/llama_extract/types/extract_job_create_batch_data_schema_override.py +0 -9
  426. llama_cloud/resources/llama_extract/types/extract_job_create_batch_data_schema_override_zero_value.py +0 -7
  427. llama_cloud/resources/llama_extract/types/extract_schema_validate_request_data_schema.py +0 -9
  428. llama_cloud/resources/llama_extract/types/extract_schema_validate_request_data_schema_zero_value.py +0 -7
  429. llama_cloud/resources/llama_extract/types/extract_stateless_request_data_schema.py +0 -9
  430. llama_cloud/resources/llama_extract/types/extract_stateless_request_data_schema_zero_value.py +0 -7
  431. llama_cloud/resources/organizations/__init__.py +0 -2
  432. llama_cloud/resources/organizations/client.py +0 -1448
  433. llama_cloud/resources/parsing/__init__.py +0 -2
  434. llama_cloud/resources/parsing/client.py +0 -2392
  435. llama_cloud/resources/pipelines/client.py +0 -3436
  436. llama_cloud/resources/pipelines/types/__init__.py +0 -29
  437. llama_cloud/resources/pipelines/types/pipeline_file_update_custom_metadata_value.py +0 -7
  438. llama_cloud/resources/pipelines/types/pipeline_update_embedding_config.py +0 -89
  439. llama_cloud/resources/pipelines/types/pipeline_update_transform_config.py +0 -8
  440. llama_cloud/resources/pipelines/types/retrieval_params_search_filters_inference_schema_value.py +0 -7
  441. llama_cloud/resources/projects/__init__.py +0 -2
  442. llama_cloud/resources/projects/client.py +0 -636
  443. llama_cloud/resources/retrievers/client.py +0 -837
  444. llama_cloud/resources/users/__init__.py +0 -2
  445. llama_cloud/resources/users/client.py +0 -155
  446. llama_cloud/types/advanced_mode_transform_config_chunking_config.py +0 -67
  447. llama_cloud/types/advanced_mode_transform_config_segmentation_config.py +0 -45
  448. llama_cloud/types/agent_data.py +0 -40
  449. llama_cloud/types/agent_deployment_list.py +0 -32
  450. llama_cloud/types/agent_deployment_summary.py +0 -39
  451. llama_cloud/types/aggregate_group.py +0 -37
  452. llama_cloud/types/azure_open_ai_embedding.py +0 -49
  453. llama_cloud/types/azure_open_ai_embedding_config.py +0 -34
  454. llama_cloud/types/base_plan.py +0 -53
  455. llama_cloud/types/base_plan_metronome_plan_type.py +0 -17
  456. llama_cloud/types/base_plan_name.py +0 -57
  457. llama_cloud/types/base_plan_plan_frequency.py +0 -25
  458. llama_cloud/types/batch.py +0 -47
  459. llama_cloud/types/batch_item.py +0 -40
  460. llama_cloud/types/batch_paginated_list.py +0 -35
  461. llama_cloud/types/batch_public_output.py +0 -36
  462. llama_cloud/types/billing_period.py +0 -32
  463. llama_cloud/types/box_auth_mechanism.py +0 -17
  464. llama_cloud/types/character_chunking_config.py +0 -32
  465. llama_cloud/types/chat_app.py +0 -46
  466. llama_cloud/types/chat_app_response.py +0 -43
  467. llama_cloud/types/chat_data.py +0 -35
  468. llama_cloud/types/chat_message.py +0 -43
  469. llama_cloud/types/chunk_mode.py +0 -29
  470. llama_cloud/types/classification_result.py +0 -39
  471. llama_cloud/types/classifier_rule.py +0 -43
  472. llama_cloud/types/classify_job.py +0 -47
  473. llama_cloud/types/classify_job_results.py +0 -38
  474. llama_cloud/types/classify_parsing_configuration.py +0 -38
  475. llama_cloud/types/cloud_astra_db_vector_store.py +0 -51
  476. llama_cloud/types/cloud_az_storage_blob_data_source.py +0 -41
  477. llama_cloud/types/cloud_azure_ai_search_vector_store.py +0 -45
  478. llama_cloud/types/cloud_box_data_source.py +0 -42
  479. llama_cloud/types/cloud_confluence_data_source.py +0 -59
  480. llama_cloud/types/cloud_document.py +0 -40
  481. llama_cloud/types/cloud_document_create.py +0 -40
  482. llama_cloud/types/cloud_jira_data_source.py +0 -42
  483. llama_cloud/types/cloud_jira_data_source_v_2.py +0 -52
  484. llama_cloud/types/cloud_jira_data_source_v_2_api_version.py +0 -21
  485. llama_cloud/types/cloud_milvus_vector_store.py +0 -40
  486. llama_cloud/types/cloud_mongo_db_atlas_vector_search.py +0 -52
  487. llama_cloud/types/cloud_notion_page_data_source.py +0 -35
  488. llama_cloud/types/cloud_one_drive_data_source.py +0 -39
  489. llama_cloud/types/cloud_pinecone_vector_store.py +0 -49
  490. llama_cloud/types/cloud_postgres_vector_store.py +0 -44
  491. llama_cloud/types/cloud_qdrant_vector_store.py +0 -51
  492. llama_cloud/types/cloud_s_3_data_source.py +0 -39
  493. llama_cloud/types/cloud_sharepoint_data_source.py +0 -42
  494. llama_cloud/types/cloud_slack_data_source.py +0 -39
  495. llama_cloud/types/composite_retrieved_text_node.py +0 -42
  496. llama_cloud/types/composite_retrieved_text_node_with_score.py +0 -34
  497. llama_cloud/types/configurable_data_sink_names.py +0 -41
  498. llama_cloud/types/configurable_data_source_names.py +0 -57
  499. llama_cloud/types/credit_type.py +0 -32
  500. llama_cloud/types/data_sink_component.py +0 -22
  501. llama_cloud/types/data_sink_create.py +0 -39
  502. llama_cloud/types/data_sink_create_component.py +0 -22
  503. llama_cloud/types/data_source_component.py +0 -28
  504. llama_cloud/types/data_source_create.py +0 -41
  505. llama_cloud/types/data_source_create_component.py +0 -28
  506. llama_cloud/types/data_source_create_custom_metadata_value.py +0 -7
  507. llama_cloud/types/data_source_custom_metadata_value.py +0 -7
  508. llama_cloud/types/data_source_reader_version_metadata_reader_version.py +0 -25
  509. llama_cloud/types/data_source_update_dispatcher_config.py +0 -38
  510. llama_cloud/types/delete_params.py +0 -39
  511. llama_cloud/types/document_chunk_mode.py +0 -17
  512. llama_cloud/types/document_ingestion_job_params.py +0 -43
  513. llama_cloud/types/element_segmentation_config.py +0 -29
  514. llama_cloud/types/embedding_model_config.py +0 -43
  515. llama_cloud/types/embedding_model_config_embedding_config.py +0 -89
  516. llama_cloud/types/embedding_model_config_update.py +0 -33
  517. llama_cloud/types/embedding_model_config_update_embedding_config.py +0 -89
  518. llama_cloud/types/eval_execution_params.py +0 -41
  519. llama_cloud/types/extract_agent.py +0 -48
  520. llama_cloud/types/extract_agent_data_schema_value.py +0 -5
  521. llama_cloud/types/extract_config.py +0 -66
  522. llama_cloud/types/extract_config_priority.py +0 -29
  523. llama_cloud/types/extract_job.py +0 -38
  524. llama_cloud/types/extract_job_create.py +0 -46
  525. llama_cloud/types/extract_job_create_data_schema_override.py +0 -9
  526. llama_cloud/types/extract_job_create_data_schema_override_zero_value.py +0 -7
  527. llama_cloud/types/extract_job_create_priority.py +0 -29
  528. llama_cloud/types/extract_mode.py +0 -29
  529. llama_cloud/types/extract_models.py +0 -53
  530. llama_cloud/types/extract_resultset.py +0 -42
  531. llama_cloud/types/extract_resultset_data.py +0 -11
  532. llama_cloud/types/extract_resultset_data_item_value.py +0 -7
  533. llama_cloud/types/extract_resultset_data_zero_value.py +0 -7
  534. llama_cloud/types/extract_resultset_extraction_metadata_value.py +0 -7
  535. llama_cloud/types/extract_run.py +0 -55
  536. llama_cloud/types/extract_run_data.py +0 -11
  537. llama_cloud/types/extract_run_data_item_value.py +0 -5
  538. llama_cloud/types/extract_run_data_schema_value.py +0 -5
  539. llama_cloud/types/extract_run_data_zero_value.py +0 -5
  540. llama_cloud/types/extract_run_extraction_metadata_value.py +0 -7
  541. llama_cloud/types/extract_schema_generate_response.py +0 -38
  542. llama_cloud/types/extract_schema_generate_response_data_schema_value.py +0 -7
  543. llama_cloud/types/extract_schema_validate_response.py +0 -32
  544. llama_cloud/types/extract_schema_validate_response_data_schema_value.py +0 -7
  545. llama_cloud/types/extract_state.py +0 -29
  546. llama_cloud/types/extract_target.py +0 -17
  547. llama_cloud/types/failure_handling_config.py +0 -37
  548. llama_cloud/types/file_classification.py +0 -41
  549. llama_cloud/types/file_count_by_status_response.py +0 -37
  550. llama_cloud/types/file_create.py +0 -41
  551. llama_cloud/types/file_create_permission_info_value.py +0 -7
  552. llama_cloud/types/file_create_resource_info_value.py +0 -5
  553. llama_cloud/types/file_data.py +0 -36
  554. llama_cloud/types/file_filter.py +0 -40
  555. llama_cloud/types/file_id_presigned_url.py +0 -38
  556. llama_cloud/types/file_parse_public.py +0 -36
  557. llama_cloud/types/file_permission_info_value.py +0 -5
  558. llama_cloud/types/file_resource_info_value.py +0 -5
  559. llama_cloud/types/file_store_info_response.py +0 -34
  560. llama_cloud/types/file_store_info_response_status.py +0 -25
  561. llama_cloud/types/filter_condition.py +0 -29
  562. llama_cloud/types/filter_operation.py +0 -46
  563. llama_cloud/types/filter_operation_eq.py +0 -6
  564. llama_cloud/types/filter_operation_gt.py +0 -6
  565. llama_cloud/types/filter_operation_gte.py +0 -6
  566. llama_cloud/types/filter_operation_includes_item.py +0 -6
  567. llama_cloud/types/filter_operation_lt.py +0 -6
  568. llama_cloud/types/filter_operation_lte.py +0 -6
  569. llama_cloud/types/filter_operator.py +0 -73
  570. llama_cloud/types/free_credits_usage.py +0 -34
  571. llama_cloud/types/http_validation_error.py +0 -32
  572. llama_cloud/types/hugging_face_inference_api_embedding_token.py +0 -5
  573. llama_cloud/types/ingestion_error_response.py +0 -34
  574. llama_cloud/types/input_message.py +0 -40
  575. llama_cloud/types/job_name_mapping.py +0 -49
  576. llama_cloud/types/job_names.py +0 -81
  577. llama_cloud/types/job_record.py +0 -58
  578. llama_cloud/types/job_record_parameters.py +0 -111
  579. llama_cloud/types/job_record_with_usage_metrics.py +0 -36
  580. llama_cloud/types/l_lama_parse_transform_config.py +0 -37
  581. llama_cloud/types/legacy_parse_job_config.py +0 -207
  582. llama_cloud/types/license_info_response.py +0 -34
  583. llama_cloud/types/llama_extract_feature_availability.py +0 -34
  584. llama_cloud/types/llama_extract_mode_availability.py +0 -38
  585. llama_cloud/types/llama_extract_mode_availability_status.py +0 -17
  586. llama_cloud/types/llama_extract_settings.py +0 -67
  587. llama_cloud/types/llama_parse_parameters_priority.py +0 -29
  588. llama_cloud/types/llm_model_data.py +0 -38
  589. llama_cloud/types/llm_parameters.py +0 -39
  590. llama_cloud/types/load_files_job_config.py +0 -35
  591. llama_cloud/types/managed_ingestion_status.py +0 -41
  592. llama_cloud/types/managed_open_ai_embedding.py +0 -36
  593. llama_cloud/types/managed_open_ai_embedding_config.py +0 -34
  594. llama_cloud/types/message_annotation.py +0 -33
  595. llama_cloud/types/metadata_filter.py +0 -44
  596. llama_cloud/types/metadata_filter_value.py +0 -5
  597. llama_cloud/types/metadata_filters_filters_item.py +0 -8
  598. llama_cloud/types/multimodal_parse_resolution.py +0 -17
  599. llama_cloud/types/node_relationship.py +0 -44
  600. llama_cloud/types/none_chunking_config.py +0 -29
  601. llama_cloud/types/none_segmentation_config.py +0 -29
  602. llama_cloud/types/object_type.py +0 -33
  603. llama_cloud/types/open_ai_embedding.py +0 -47
  604. llama_cloud/types/open_ai_embedding_config.py +0 -34
  605. llama_cloud/types/organization.py +0 -43
  606. llama_cloud/types/organization_create.py +0 -35
  607. llama_cloud/types/page_figure_metadata.py +0 -37
  608. llama_cloud/types/page_screenshot_metadata.py +0 -34
  609. llama_cloud/types/page_segmentation_config.py +0 -31
  610. llama_cloud/types/paginated_extract_runs_response.py +0 -39
  611. llama_cloud/types/paginated_jobs_history_with_metrics.py +0 -35
  612. llama_cloud/types/paginated_list_cloud_documents_response.py +0 -35
  613. llama_cloud/types/paginated_list_pipeline_files_response.py +0 -35
  614. llama_cloud/types/paginated_response_agent_data.py +0 -34
  615. llama_cloud/types/paginated_response_aggregate_group.py +0 -34
  616. llama_cloud/types/paginated_response_classify_job.py +0 -34
  617. llama_cloud/types/paginated_response_quota_configuration.py +0 -36
  618. llama_cloud/types/parse_configuration.py +0 -44
  619. llama_cloud/types/parse_configuration_create.py +0 -41
  620. llama_cloud/types/parse_configuration_filter.py +0 -40
  621. llama_cloud/types/parse_configuration_query_response.py +0 -38
  622. llama_cloud/types/parse_job_config.py +0 -149
  623. llama_cloud/types/parse_job_config_priority.py +0 -29
  624. llama_cloud/types/parse_plan_level.py +0 -21
  625. llama_cloud/types/parser_languages.py +0 -361
  626. llama_cloud/types/parsing_history_item.py +0 -39
  627. llama_cloud/types/parsing_job.py +0 -35
  628. llama_cloud/types/parsing_job_json_result.py +0 -32
  629. llama_cloud/types/parsing_job_markdown_result.py +0 -32
  630. llama_cloud/types/parsing_job_structured_result.py +0 -32
  631. llama_cloud/types/parsing_job_text_result.py +0 -32
  632. llama_cloud/types/partition_names.py +0 -45
  633. llama_cloud/types/permission.py +0 -40
  634. llama_cloud/types/pg_vector_distance_method.py +0 -43
  635. llama_cloud/types/pg_vector_hnsw_settings.py +0 -45
  636. llama_cloud/types/pg_vector_vector_type.py +0 -35
  637. llama_cloud/types/pipeline_configuration_hashes.py +0 -37
  638. llama_cloud/types/pipeline_create.py +0 -65
  639. llama_cloud/types/pipeline_create_embedding_config.py +0 -89
  640. llama_cloud/types/pipeline_create_transform_config.py +0 -8
  641. llama_cloud/types/pipeline_data_source.py +0 -55
  642. llama_cloud/types/pipeline_data_source_component.py +0 -28
  643. llama_cloud/types/pipeline_data_source_create.py +0 -36
  644. llama_cloud/types/pipeline_data_source_custom_metadata_value.py +0 -7
  645. llama_cloud/types/pipeline_data_source_status.py +0 -33
  646. llama_cloud/types/pipeline_deployment.py +0 -37
  647. llama_cloud/types/pipeline_embedding_config.py +0 -100
  648. llama_cloud/types/pipeline_file.py +0 -58
  649. llama_cloud/types/pipeline_file_config_hash_value.py +0 -5
  650. llama_cloud/types/pipeline_file_create.py +0 -37
  651. llama_cloud/types/pipeline_file_create_custom_metadata_value.py +0 -7
  652. llama_cloud/types/pipeline_file_custom_metadata_value.py +0 -7
  653. llama_cloud/types/pipeline_file_permission_info_value.py +0 -7
  654. llama_cloud/types/pipeline_file_resource_info_value.py +0 -7
  655. llama_cloud/types/pipeline_file_status.py +0 -33
  656. llama_cloud/types/pipeline_file_update_dispatcher_config.py +0 -38
  657. llama_cloud/types/pipeline_file_updater_config.py +0 -44
  658. llama_cloud/types/pipeline_managed_ingestion_job_params.py +0 -37
  659. llama_cloud/types/pipeline_status.py +0 -17
  660. llama_cloud/types/pipeline_transform_config.py +0 -31
  661. llama_cloud/types/plan_limits.py +0 -53
  662. llama_cloud/types/playground_session.py +0 -51
  663. llama_cloud/types/pooling.py +0 -29
  664. llama_cloud/types/preset_composite_retrieval_params.py +0 -37
  665. llama_cloud/types/preset_retrieval_params_search_filters_inference_schema_value.py +0 -7
  666. llama_cloud/types/project_create.py +0 -35
  667. llama_cloud/types/prompt_conf.py +0 -38
  668. llama_cloud/types/public_model_name.py +0 -97
  669. llama_cloud/types/quota_configuration.py +0 -53
  670. llama_cloud/types/quota_configuration_configuration_type.py +0 -33
  671. llama_cloud/types/quota_configuration_status.py +0 -21
  672. llama_cloud/types/quota_rate_limit_configuration_value.py +0 -38
  673. llama_cloud/types/quota_rate_limit_configuration_value_denominator_units.py +0 -29
  674. llama_cloud/types/re_rank_config.py +0 -35
  675. llama_cloud/types/re_ranker_type.py +0 -41
  676. llama_cloud/types/recurring_credit_grant.py +0 -44
  677. llama_cloud/types/related_node_info.py +0 -36
  678. llama_cloud/types/related_node_info_node_type.py +0 -7
  679. llama_cloud/types/retrieve_results.py +0 -56
  680. llama_cloud/types/retriever_create.py +0 -37
  681. llama_cloud/types/role.py +0 -40
  682. llama_cloud/types/schema_generation_availability.py +0 -33
  683. llama_cloud/types/schema_generation_availability_status.py +0 -17
  684. llama_cloud/types/schema_relax_mode.py +0 -25
  685. llama_cloud/types/semantic_chunking_config.py +0 -32
  686. llama_cloud/types/sentence_chunking_config.py +0 -34
  687. llama_cloud/types/sparse_model_type.py +0 -33
  688. llama_cloud/types/struct_mode.py +0 -33
  689. llama_cloud/types/struct_parse_conf.py +0 -63
  690. llama_cloud/types/supported_llm_model.py +0 -40
  691. llama_cloud/types/supported_llm_model_names.py +0 -69
  692. llama_cloud/types/text_node.py +0 -67
  693. llama_cloud/types/text_node_relationships_value.py +0 -7
  694. llama_cloud/types/text_node_with_score.py +0 -39
  695. llama_cloud/types/token_chunking_config.py +0 -33
  696. llama_cloud/types/update_user_response.py +0 -33
  697. llama_cloud/types/usage_and_plan.py +0 -34
  698. llama_cloud/types/usage_metric_response.py +0 -34
  699. llama_cloud/types/usage_response.py +0 -43
  700. llama_cloud/types/usage_response_active_alerts_item.py +0 -37
  701. llama_cloud/types/user_job_record.py +0 -32
  702. llama_cloud/types/user_organization.py +0 -47
  703. llama_cloud/types/user_organization_create.py +0 -38
  704. llama_cloud/types/user_organization_delete.py +0 -37
  705. llama_cloud/types/user_organization_role.py +0 -42
  706. llama_cloud/types/user_summary.py +0 -38
  707. llama_cloud/types/validation_error.py +0 -34
  708. llama_cloud/types/validation_error_loc_item.py +0 -5
  709. llama_cloud/types/vertex_embedding_mode.py +0 -38
  710. llama_cloud/types/webhook_configuration.py +0 -39
  711. llama_cloud/types/webhook_configuration_webhook_events_item.py +0 -57
  712. llama_cloud-0.1.41.dist-info/LICENSE +0 -21
  713. llama_cloud-0.1.41.dist-info/METADATA +0 -106
  714. llama_cloud-0.1.41.dist-info/RECORD +0 -385
@@ -1,837 +0,0 @@
1
- # This file was auto-generated by Fern from our API Definition.
2
-
3
- import typing
4
- import urllib.parse
5
- from json.decoder import JSONDecodeError
6
-
7
- from ...core.api_error import ApiError
8
- from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
9
- from ...core.jsonable_encoder import jsonable_encoder
10
- from ...core.remove_none_from_dict import remove_none_from_dict
11
- from ...errors.unprocessable_entity_error import UnprocessableEntityError
12
- from ...types.composite_retrieval_mode import CompositeRetrievalMode
13
- from ...types.composite_retrieval_result import CompositeRetrievalResult
14
- from ...types.http_validation_error import HttpValidationError
15
- from ...types.re_rank_config import ReRankConfig
16
- from ...types.retriever import Retriever
17
- from ...types.retriever_create import RetrieverCreate
18
- from ...types.retriever_pipeline import RetrieverPipeline
19
-
20
- try:
21
- import pydantic
22
- if pydantic.__version__.startswith("1."):
23
- raise ImportError
24
- import pydantic.v1 as pydantic # type: ignore
25
- except ImportError:
26
- import pydantic # type: ignore
27
-
28
- # this is used as the default value for optional parameters
29
- OMIT = typing.cast(typing.Any, ...)
30
-
31
-
32
- class RetrieversClient:
33
- def __init__(self, *, client_wrapper: SyncClientWrapper):
34
- self._client_wrapper = client_wrapper
35
-
36
- def list_retrievers(
37
- self,
38
- *,
39
- name: typing.Optional[str] = None,
40
- project_id: typing.Optional[str] = None,
41
- organization_id: typing.Optional[str] = None,
42
- ) -> typing.List[Retriever]:
43
- """
44
- List Retrievers for a project.
45
-
46
- Parameters:
47
- - name: typing.Optional[str].
48
-
49
- - project_id: typing.Optional[str].
50
-
51
- - organization_id: typing.Optional[str].
52
- ---
53
- from llama_cloud.client import LlamaCloud
54
-
55
- client = LlamaCloud(
56
- token="YOUR_TOKEN",
57
- )
58
- client.retrievers.list_retrievers()
59
- """
60
- _response = self._client_wrapper.httpx_client.request(
61
- "GET",
62
- urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/retrievers"),
63
- params=remove_none_from_dict({"name": name, "project_id": project_id, "organization_id": organization_id}),
64
- headers=self._client_wrapper.get_headers(),
65
- timeout=60,
66
- )
67
- if 200 <= _response.status_code < 300:
68
- return pydantic.parse_obj_as(typing.List[Retriever], _response.json()) # type: ignore
69
- if _response.status_code == 422:
70
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
71
- try:
72
- _response_json = _response.json()
73
- except JSONDecodeError:
74
- raise ApiError(status_code=_response.status_code, body=_response.text)
75
- raise ApiError(status_code=_response.status_code, body=_response_json)
76
-
77
- def create_retriever(
78
- self,
79
- *,
80
- project_id: typing.Optional[str] = None,
81
- organization_id: typing.Optional[str] = None,
82
- request: RetrieverCreate,
83
- ) -> Retriever:
84
- """
85
- Create a new Retriever.
86
-
87
- Parameters:
88
- - project_id: typing.Optional[str].
89
-
90
- - organization_id: typing.Optional[str].
91
-
92
- - request: RetrieverCreate.
93
- ---
94
- from llama_cloud import RetrieverCreate
95
- from llama_cloud.client import LlamaCloud
96
-
97
- client = LlamaCloud(
98
- token="YOUR_TOKEN",
99
- )
100
- client.retrievers.create_retriever(
101
- request=RetrieverCreate(
102
- name="string",
103
- ),
104
- )
105
- """
106
- _response = self._client_wrapper.httpx_client.request(
107
- "POST",
108
- urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/retrievers"),
109
- params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
110
- json=jsonable_encoder(request),
111
- headers=self._client_wrapper.get_headers(),
112
- timeout=60,
113
- )
114
- if 200 <= _response.status_code < 300:
115
- return pydantic.parse_obj_as(Retriever, _response.json()) # type: ignore
116
- if _response.status_code == 422:
117
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
118
- try:
119
- _response_json = _response.json()
120
- except JSONDecodeError:
121
- raise ApiError(status_code=_response.status_code, body=_response.text)
122
- raise ApiError(status_code=_response.status_code, body=_response_json)
123
-
124
- def upsert_retriever(
125
- self,
126
- *,
127
- project_id: typing.Optional[str] = None,
128
- organization_id: typing.Optional[str] = None,
129
- request: RetrieverCreate,
130
- ) -> Retriever:
131
- """
132
- Upsert a new Retriever.
133
-
134
- Parameters:
135
- - project_id: typing.Optional[str].
136
-
137
- - organization_id: typing.Optional[str].
138
-
139
- - request: RetrieverCreate.
140
- ---
141
- from llama_cloud import RetrieverCreate
142
- from llama_cloud.client import LlamaCloud
143
-
144
- client = LlamaCloud(
145
- token="YOUR_TOKEN",
146
- )
147
- client.retrievers.upsert_retriever(
148
- request=RetrieverCreate(
149
- name="string",
150
- ),
151
- )
152
- """
153
- _response = self._client_wrapper.httpx_client.request(
154
- "PUT",
155
- urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/retrievers"),
156
- params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
157
- json=jsonable_encoder(request),
158
- headers=self._client_wrapper.get_headers(),
159
- timeout=60,
160
- )
161
- if 200 <= _response.status_code < 300:
162
- return pydantic.parse_obj_as(Retriever, _response.json()) # type: ignore
163
- if _response.status_code == 422:
164
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
165
- try:
166
- _response_json = _response.json()
167
- except JSONDecodeError:
168
- raise ApiError(status_code=_response.status_code, body=_response.text)
169
- raise ApiError(status_code=_response.status_code, body=_response_json)
170
-
171
- def get_retriever(
172
- self,
173
- retriever_id: str,
174
- *,
175
- project_id: typing.Optional[str] = None,
176
- organization_id: typing.Optional[str] = None,
177
- ) -> Retriever:
178
- """
179
- Get a Retriever by ID.
180
-
181
- Parameters:
182
- - retriever_id: str.
183
-
184
- - project_id: typing.Optional[str].
185
-
186
- - organization_id: typing.Optional[str].
187
- ---
188
- from llama_cloud.client import LlamaCloud
189
-
190
- client = LlamaCloud(
191
- token="YOUR_TOKEN",
192
- )
193
- client.retrievers.get_retriever(
194
- retriever_id="string",
195
- )
196
- """
197
- _response = self._client_wrapper.httpx_client.request(
198
- "GET",
199
- urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/retrievers/{retriever_id}"),
200
- params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
201
- headers=self._client_wrapper.get_headers(),
202
- timeout=60,
203
- )
204
- if 200 <= _response.status_code < 300:
205
- return pydantic.parse_obj_as(Retriever, _response.json()) # type: ignore
206
- if _response.status_code == 422:
207
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
208
- try:
209
- _response_json = _response.json()
210
- except JSONDecodeError:
211
- raise ApiError(status_code=_response.status_code, body=_response.text)
212
- raise ApiError(status_code=_response.status_code, body=_response_json)
213
-
214
- def update_retriever(
215
- self,
216
- retriever_id: str,
217
- *,
218
- name: typing.Optional[str] = OMIT,
219
- pipelines: typing.Optional[typing.List[RetrieverPipeline]] = OMIT,
220
- ) -> Retriever:
221
- """
222
- Update an existing Retriever.
223
-
224
- Parameters:
225
- - retriever_id: str.
226
-
227
- - name: typing.Optional[str].
228
-
229
- - pipelines: typing.Optional[typing.List[RetrieverPipeline]].
230
- ---
231
- from llama_cloud.client import LlamaCloud
232
-
233
- client = LlamaCloud(
234
- token="YOUR_TOKEN",
235
- )
236
- client.retrievers.update_retriever(
237
- retriever_id="string",
238
- )
239
- """
240
- _request: typing.Dict[str, typing.Any] = {}
241
- if name is not OMIT:
242
- _request["name"] = name
243
- if pipelines is not OMIT:
244
- _request["pipelines"] = pipelines
245
- _response = self._client_wrapper.httpx_client.request(
246
- "PUT",
247
- urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/retrievers/{retriever_id}"),
248
- json=jsonable_encoder(_request),
249
- headers=self._client_wrapper.get_headers(),
250
- timeout=60,
251
- )
252
- if 200 <= _response.status_code < 300:
253
- return pydantic.parse_obj_as(Retriever, _response.json()) # type: ignore
254
- if _response.status_code == 422:
255
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
256
- try:
257
- _response_json = _response.json()
258
- except JSONDecodeError:
259
- raise ApiError(status_code=_response.status_code, body=_response.text)
260
- raise ApiError(status_code=_response.status_code, body=_response_json)
261
-
262
- def delete_retriever(self, retriever_id: str) -> None:
263
- """
264
- Delete a Retriever by ID.
265
-
266
- Parameters:
267
- - retriever_id: str.
268
- ---
269
- from llama_cloud.client import LlamaCloud
270
-
271
- client = LlamaCloud(
272
- token="YOUR_TOKEN",
273
- )
274
- client.retrievers.delete_retriever(
275
- retriever_id="string",
276
- )
277
- """
278
- _response = self._client_wrapper.httpx_client.request(
279
- "DELETE",
280
- urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/retrievers/{retriever_id}"),
281
- headers=self._client_wrapper.get_headers(),
282
- timeout=60,
283
- )
284
- if 200 <= _response.status_code < 300:
285
- return
286
- if _response.status_code == 422:
287
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
288
- try:
289
- _response_json = _response.json()
290
- except JSONDecodeError:
291
- raise ApiError(status_code=_response.status_code, body=_response.text)
292
- raise ApiError(status_code=_response.status_code, body=_response_json)
293
-
294
- def retrieve(
295
- self,
296
- retriever_id: str,
297
- *,
298
- project_id: typing.Optional[str] = None,
299
- organization_id: typing.Optional[str] = None,
300
- mode: typing.Optional[CompositeRetrievalMode] = OMIT,
301
- rerank_top_n: typing.Optional[int] = OMIT,
302
- rerank_config: typing.Optional[ReRankConfig] = OMIT,
303
- query: str,
304
- ) -> CompositeRetrievalResult:
305
- """
306
- Retrieve data using a Retriever.
307
-
308
- Parameters:
309
- - retriever_id: str.
310
-
311
- - project_id: typing.Optional[str].
312
-
313
- - organization_id: typing.Optional[str].
314
-
315
- - mode: typing.Optional[CompositeRetrievalMode]. The mode of composite retrieval.
316
-
317
- - rerank_top_n: typing.Optional[int].
318
-
319
- - rerank_config: typing.Optional[ReRankConfig]. The rerank configuration for composite retrieval.
320
-
321
- - query: str. The query to retrieve against.
322
- ---
323
- from llama_cloud import CompositeRetrievalMode, ReRankConfig, ReRankerType
324
- from llama_cloud.client import LlamaCloud
325
-
326
- client = LlamaCloud(
327
- token="YOUR_TOKEN",
328
- )
329
- client.retrievers.retrieve(
330
- retriever_id="string",
331
- mode=CompositeRetrievalMode.ROUTING,
332
- rerank_config=ReRankConfig(
333
- type=ReRankerType.SYSTEM_DEFAULT,
334
- ),
335
- query="string",
336
- )
337
- """
338
- _request: typing.Dict[str, typing.Any] = {"query": query}
339
- if mode is not OMIT:
340
- _request["mode"] = mode
341
- if rerank_top_n is not OMIT:
342
- _request["rerank_top_n"] = rerank_top_n
343
- if rerank_config is not OMIT:
344
- _request["rerank_config"] = rerank_config
345
- _response = self._client_wrapper.httpx_client.request(
346
- "POST",
347
- urllib.parse.urljoin(
348
- f"{self._client_wrapper.get_base_url()}/", f"api/v1/retrievers/{retriever_id}/retrieve"
349
- ),
350
- params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
351
- json=jsonable_encoder(_request),
352
- headers=self._client_wrapper.get_headers(),
353
- timeout=60,
354
- )
355
- if 200 <= _response.status_code < 300:
356
- return pydantic.parse_obj_as(CompositeRetrievalResult, _response.json()) # type: ignore
357
- if _response.status_code == 422:
358
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
359
- try:
360
- _response_json = _response.json()
361
- except JSONDecodeError:
362
- raise ApiError(status_code=_response.status_code, body=_response.text)
363
- raise ApiError(status_code=_response.status_code, body=_response_json)
364
-
365
- def direct_retrieve(
366
- self,
367
- *,
368
- project_id: typing.Optional[str] = None,
369
- organization_id: typing.Optional[str] = None,
370
- mode: typing.Optional[CompositeRetrievalMode] = OMIT,
371
- rerank_top_n: typing.Optional[int] = OMIT,
372
- rerank_config: typing.Optional[ReRankConfig] = OMIT,
373
- query: str,
374
- pipelines: typing.Optional[typing.List[RetrieverPipeline]] = OMIT,
375
- ) -> CompositeRetrievalResult:
376
- """
377
- Retrieve data using specified pipelines without creating a persistent retriever.
378
-
379
- Parameters:
380
- - project_id: typing.Optional[str].
381
-
382
- - organization_id: typing.Optional[str].
383
-
384
- - mode: typing.Optional[CompositeRetrievalMode]. The mode of composite retrieval.
385
-
386
- - rerank_top_n: typing.Optional[int].
387
-
388
- - rerank_config: typing.Optional[ReRankConfig]. The rerank configuration for composite retrieval.
389
-
390
- - query: str. The query to retrieve against.
391
-
392
- - pipelines: typing.Optional[typing.List[RetrieverPipeline]]. The pipelines to use for retrieval.
393
- ---
394
- from llama_cloud import CompositeRetrievalMode, ReRankConfig, ReRankerType
395
- from llama_cloud.client import LlamaCloud
396
-
397
- client = LlamaCloud(
398
- token="YOUR_TOKEN",
399
- )
400
- client.retrievers.direct_retrieve(
401
- mode=CompositeRetrievalMode.ROUTING,
402
- rerank_config=ReRankConfig(
403
- type=ReRankerType.SYSTEM_DEFAULT,
404
- ),
405
- query="string",
406
- )
407
- """
408
- _request: typing.Dict[str, typing.Any] = {"query": query}
409
- if mode is not OMIT:
410
- _request["mode"] = mode
411
- if rerank_top_n is not OMIT:
412
- _request["rerank_top_n"] = rerank_top_n
413
- if rerank_config is not OMIT:
414
- _request["rerank_config"] = rerank_config
415
- if pipelines is not OMIT:
416
- _request["pipelines"] = pipelines
417
- _response = self._client_wrapper.httpx_client.request(
418
- "POST",
419
- urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/retrievers/retrieve"),
420
- params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
421
- json=jsonable_encoder(_request),
422
- headers=self._client_wrapper.get_headers(),
423
- timeout=60,
424
- )
425
- if 200 <= _response.status_code < 300:
426
- return pydantic.parse_obj_as(CompositeRetrievalResult, _response.json()) # type: ignore
427
- if _response.status_code == 422:
428
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
429
- try:
430
- _response_json = _response.json()
431
- except JSONDecodeError:
432
- raise ApiError(status_code=_response.status_code, body=_response.text)
433
- raise ApiError(status_code=_response.status_code, body=_response_json)
434
-
435
-
436
- class AsyncRetrieversClient:
437
- def __init__(self, *, client_wrapper: AsyncClientWrapper):
438
- self._client_wrapper = client_wrapper
439
-
440
- async def list_retrievers(
441
- self,
442
- *,
443
- name: typing.Optional[str] = None,
444
- project_id: typing.Optional[str] = None,
445
- organization_id: typing.Optional[str] = None,
446
- ) -> typing.List[Retriever]:
447
- """
448
- List Retrievers for a project.
449
-
450
- Parameters:
451
- - name: typing.Optional[str].
452
-
453
- - project_id: typing.Optional[str].
454
-
455
- - organization_id: typing.Optional[str].
456
- ---
457
- from llama_cloud.client import AsyncLlamaCloud
458
-
459
- client = AsyncLlamaCloud(
460
- token="YOUR_TOKEN",
461
- )
462
- await client.retrievers.list_retrievers()
463
- """
464
- _response = await self._client_wrapper.httpx_client.request(
465
- "GET",
466
- urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/retrievers"),
467
- params=remove_none_from_dict({"name": name, "project_id": project_id, "organization_id": organization_id}),
468
- headers=self._client_wrapper.get_headers(),
469
- timeout=60,
470
- )
471
- if 200 <= _response.status_code < 300:
472
- return pydantic.parse_obj_as(typing.List[Retriever], _response.json()) # type: ignore
473
- if _response.status_code == 422:
474
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
475
- try:
476
- _response_json = _response.json()
477
- except JSONDecodeError:
478
- raise ApiError(status_code=_response.status_code, body=_response.text)
479
- raise ApiError(status_code=_response.status_code, body=_response_json)
480
-
481
- async def create_retriever(
482
- self,
483
- *,
484
- project_id: typing.Optional[str] = None,
485
- organization_id: typing.Optional[str] = None,
486
- request: RetrieverCreate,
487
- ) -> Retriever:
488
- """
489
- Create a new Retriever.
490
-
491
- Parameters:
492
- - project_id: typing.Optional[str].
493
-
494
- - organization_id: typing.Optional[str].
495
-
496
- - request: RetrieverCreate.
497
- ---
498
- from llama_cloud import RetrieverCreate
499
- from llama_cloud.client import AsyncLlamaCloud
500
-
501
- client = AsyncLlamaCloud(
502
- token="YOUR_TOKEN",
503
- )
504
- await client.retrievers.create_retriever(
505
- request=RetrieverCreate(
506
- name="string",
507
- ),
508
- )
509
- """
510
- _response = await self._client_wrapper.httpx_client.request(
511
- "POST",
512
- urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/retrievers"),
513
- params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
514
- json=jsonable_encoder(request),
515
- headers=self._client_wrapper.get_headers(),
516
- timeout=60,
517
- )
518
- if 200 <= _response.status_code < 300:
519
- return pydantic.parse_obj_as(Retriever, _response.json()) # type: ignore
520
- if _response.status_code == 422:
521
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
522
- try:
523
- _response_json = _response.json()
524
- except JSONDecodeError:
525
- raise ApiError(status_code=_response.status_code, body=_response.text)
526
- raise ApiError(status_code=_response.status_code, body=_response_json)
527
-
528
- async def upsert_retriever(
529
- self,
530
- *,
531
- project_id: typing.Optional[str] = None,
532
- organization_id: typing.Optional[str] = None,
533
- request: RetrieverCreate,
534
- ) -> Retriever:
535
- """
536
- Upsert a new Retriever.
537
-
538
- Parameters:
539
- - project_id: typing.Optional[str].
540
-
541
- - organization_id: typing.Optional[str].
542
-
543
- - request: RetrieverCreate.
544
- ---
545
- from llama_cloud import RetrieverCreate
546
- from llama_cloud.client import AsyncLlamaCloud
547
-
548
- client = AsyncLlamaCloud(
549
- token="YOUR_TOKEN",
550
- )
551
- await client.retrievers.upsert_retriever(
552
- request=RetrieverCreate(
553
- name="string",
554
- ),
555
- )
556
- """
557
- _response = await self._client_wrapper.httpx_client.request(
558
- "PUT",
559
- urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/retrievers"),
560
- params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
561
- json=jsonable_encoder(request),
562
- headers=self._client_wrapper.get_headers(),
563
- timeout=60,
564
- )
565
- if 200 <= _response.status_code < 300:
566
- return pydantic.parse_obj_as(Retriever, _response.json()) # type: ignore
567
- if _response.status_code == 422:
568
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
569
- try:
570
- _response_json = _response.json()
571
- except JSONDecodeError:
572
- raise ApiError(status_code=_response.status_code, body=_response.text)
573
- raise ApiError(status_code=_response.status_code, body=_response_json)
574
-
575
- async def get_retriever(
576
- self,
577
- retriever_id: str,
578
- *,
579
- project_id: typing.Optional[str] = None,
580
- organization_id: typing.Optional[str] = None,
581
- ) -> Retriever:
582
- """
583
- Get a Retriever by ID.
584
-
585
- Parameters:
586
- - retriever_id: str.
587
-
588
- - project_id: typing.Optional[str].
589
-
590
- - organization_id: typing.Optional[str].
591
- ---
592
- from llama_cloud.client import AsyncLlamaCloud
593
-
594
- client = AsyncLlamaCloud(
595
- token="YOUR_TOKEN",
596
- )
597
- await client.retrievers.get_retriever(
598
- retriever_id="string",
599
- )
600
- """
601
- _response = await self._client_wrapper.httpx_client.request(
602
- "GET",
603
- urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/retrievers/{retriever_id}"),
604
- params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
605
- headers=self._client_wrapper.get_headers(),
606
- timeout=60,
607
- )
608
- if 200 <= _response.status_code < 300:
609
- return pydantic.parse_obj_as(Retriever, _response.json()) # type: ignore
610
- if _response.status_code == 422:
611
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
612
- try:
613
- _response_json = _response.json()
614
- except JSONDecodeError:
615
- raise ApiError(status_code=_response.status_code, body=_response.text)
616
- raise ApiError(status_code=_response.status_code, body=_response_json)
617
-
618
- async def update_retriever(
619
- self,
620
- retriever_id: str,
621
- *,
622
- name: typing.Optional[str] = OMIT,
623
- pipelines: typing.Optional[typing.List[RetrieverPipeline]] = OMIT,
624
- ) -> Retriever:
625
- """
626
- Update an existing Retriever.
627
-
628
- Parameters:
629
- - retriever_id: str.
630
-
631
- - name: typing.Optional[str].
632
-
633
- - pipelines: typing.Optional[typing.List[RetrieverPipeline]].
634
- ---
635
- from llama_cloud.client import AsyncLlamaCloud
636
-
637
- client = AsyncLlamaCloud(
638
- token="YOUR_TOKEN",
639
- )
640
- await client.retrievers.update_retriever(
641
- retriever_id="string",
642
- )
643
- """
644
- _request: typing.Dict[str, typing.Any] = {}
645
- if name is not OMIT:
646
- _request["name"] = name
647
- if pipelines is not OMIT:
648
- _request["pipelines"] = pipelines
649
- _response = await self._client_wrapper.httpx_client.request(
650
- "PUT",
651
- urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/retrievers/{retriever_id}"),
652
- json=jsonable_encoder(_request),
653
- headers=self._client_wrapper.get_headers(),
654
- timeout=60,
655
- )
656
- if 200 <= _response.status_code < 300:
657
- return pydantic.parse_obj_as(Retriever, _response.json()) # type: ignore
658
- if _response.status_code == 422:
659
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
660
- try:
661
- _response_json = _response.json()
662
- except JSONDecodeError:
663
- raise ApiError(status_code=_response.status_code, body=_response.text)
664
- raise ApiError(status_code=_response.status_code, body=_response_json)
665
-
666
- async def delete_retriever(self, retriever_id: str) -> None:
667
- """
668
- Delete a Retriever by ID.
669
-
670
- Parameters:
671
- - retriever_id: str.
672
- ---
673
- from llama_cloud.client import AsyncLlamaCloud
674
-
675
- client = AsyncLlamaCloud(
676
- token="YOUR_TOKEN",
677
- )
678
- await client.retrievers.delete_retriever(
679
- retriever_id="string",
680
- )
681
- """
682
- _response = await self._client_wrapper.httpx_client.request(
683
- "DELETE",
684
- urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/retrievers/{retriever_id}"),
685
- headers=self._client_wrapper.get_headers(),
686
- timeout=60,
687
- )
688
- if 200 <= _response.status_code < 300:
689
- return
690
- if _response.status_code == 422:
691
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
692
- try:
693
- _response_json = _response.json()
694
- except JSONDecodeError:
695
- raise ApiError(status_code=_response.status_code, body=_response.text)
696
- raise ApiError(status_code=_response.status_code, body=_response_json)
697
-
698
- async def retrieve(
699
- self,
700
- retriever_id: str,
701
- *,
702
- project_id: typing.Optional[str] = None,
703
- organization_id: typing.Optional[str] = None,
704
- mode: typing.Optional[CompositeRetrievalMode] = OMIT,
705
- rerank_top_n: typing.Optional[int] = OMIT,
706
- rerank_config: typing.Optional[ReRankConfig] = OMIT,
707
- query: str,
708
- ) -> CompositeRetrievalResult:
709
- """
710
- Retrieve data using a Retriever.
711
-
712
- Parameters:
713
- - retriever_id: str.
714
-
715
- - project_id: typing.Optional[str].
716
-
717
- - organization_id: typing.Optional[str].
718
-
719
- - mode: typing.Optional[CompositeRetrievalMode]. The mode of composite retrieval.
720
-
721
- - rerank_top_n: typing.Optional[int].
722
-
723
- - rerank_config: typing.Optional[ReRankConfig]. The rerank configuration for composite retrieval.
724
-
725
- - query: str. The query to retrieve against.
726
- ---
727
- from llama_cloud import CompositeRetrievalMode, ReRankConfig, ReRankerType
728
- from llama_cloud.client import AsyncLlamaCloud
729
-
730
- client = AsyncLlamaCloud(
731
- token="YOUR_TOKEN",
732
- )
733
- await client.retrievers.retrieve(
734
- retriever_id="string",
735
- mode=CompositeRetrievalMode.ROUTING,
736
- rerank_config=ReRankConfig(
737
- type=ReRankerType.SYSTEM_DEFAULT,
738
- ),
739
- query="string",
740
- )
741
- """
742
- _request: typing.Dict[str, typing.Any] = {"query": query}
743
- if mode is not OMIT:
744
- _request["mode"] = mode
745
- if rerank_top_n is not OMIT:
746
- _request["rerank_top_n"] = rerank_top_n
747
- if rerank_config is not OMIT:
748
- _request["rerank_config"] = rerank_config
749
- _response = await self._client_wrapper.httpx_client.request(
750
- "POST",
751
- urllib.parse.urljoin(
752
- f"{self._client_wrapper.get_base_url()}/", f"api/v1/retrievers/{retriever_id}/retrieve"
753
- ),
754
- params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
755
- json=jsonable_encoder(_request),
756
- headers=self._client_wrapper.get_headers(),
757
- timeout=60,
758
- )
759
- if 200 <= _response.status_code < 300:
760
- return pydantic.parse_obj_as(CompositeRetrievalResult, _response.json()) # type: ignore
761
- if _response.status_code == 422:
762
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
763
- try:
764
- _response_json = _response.json()
765
- except JSONDecodeError:
766
- raise ApiError(status_code=_response.status_code, body=_response.text)
767
- raise ApiError(status_code=_response.status_code, body=_response_json)
768
-
769
- async def direct_retrieve(
770
- self,
771
- *,
772
- project_id: typing.Optional[str] = None,
773
- organization_id: typing.Optional[str] = None,
774
- mode: typing.Optional[CompositeRetrievalMode] = OMIT,
775
- rerank_top_n: typing.Optional[int] = OMIT,
776
- rerank_config: typing.Optional[ReRankConfig] = OMIT,
777
- query: str,
778
- pipelines: typing.Optional[typing.List[RetrieverPipeline]] = OMIT,
779
- ) -> CompositeRetrievalResult:
780
- """
781
- Retrieve data using specified pipelines without creating a persistent retriever.
782
-
783
- Parameters:
784
- - project_id: typing.Optional[str].
785
-
786
- - organization_id: typing.Optional[str].
787
-
788
- - mode: typing.Optional[CompositeRetrievalMode]. The mode of composite retrieval.
789
-
790
- - rerank_top_n: typing.Optional[int].
791
-
792
- - rerank_config: typing.Optional[ReRankConfig]. The rerank configuration for composite retrieval.
793
-
794
- - query: str. The query to retrieve against.
795
-
796
- - pipelines: typing.Optional[typing.List[RetrieverPipeline]]. The pipelines to use for retrieval.
797
- ---
798
- from llama_cloud import CompositeRetrievalMode, ReRankConfig, ReRankerType
799
- from llama_cloud.client import AsyncLlamaCloud
800
-
801
- client = AsyncLlamaCloud(
802
- token="YOUR_TOKEN",
803
- )
804
- await client.retrievers.direct_retrieve(
805
- mode=CompositeRetrievalMode.ROUTING,
806
- rerank_config=ReRankConfig(
807
- type=ReRankerType.SYSTEM_DEFAULT,
808
- ),
809
- query="string",
810
- )
811
- """
812
- _request: typing.Dict[str, typing.Any] = {"query": query}
813
- if mode is not OMIT:
814
- _request["mode"] = mode
815
- if rerank_top_n is not OMIT:
816
- _request["rerank_top_n"] = rerank_top_n
817
- if rerank_config is not OMIT:
818
- _request["rerank_config"] = rerank_config
819
- if pipelines is not OMIT:
820
- _request["pipelines"] = pipelines
821
- _response = await self._client_wrapper.httpx_client.request(
822
- "POST",
823
- urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/retrievers/retrieve"),
824
- params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
825
- json=jsonable_encoder(_request),
826
- headers=self._client_wrapper.get_headers(),
827
- timeout=60,
828
- )
829
- if 200 <= _response.status_code < 300:
830
- return pydantic.parse_obj_as(CompositeRetrievalResult, _response.json()) # type: ignore
831
- if _response.status_code == 422:
832
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
833
- try:
834
- _response_json = _response.json()
835
- except JSONDecodeError:
836
- raise ApiError(status_code=_response.status_code, body=_response.text)
837
- raise ApiError(status_code=_response.status_code, body=_response_json)