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,1448 +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.http_validation_error import HttpValidationError
13
- from ...types.organization import Organization
14
- from ...types.organization_create import OrganizationCreate
15
- from ...types.project import Project
16
- from ...types.role import Role
17
- from ...types.usage_and_plan import UsageAndPlan
18
- from ...types.user_organization import UserOrganization
19
- from ...types.user_organization_create import UserOrganizationCreate
20
- from ...types.user_organization_delete import UserOrganizationDelete
21
- from ...types.user_organization_role import UserOrganizationRole
22
-
23
- try:
24
- import pydantic
25
- if pydantic.__version__.startswith("1."):
26
- raise ImportError
27
- import pydantic.v1 as pydantic # type: ignore
28
- except ImportError:
29
- import pydantic # type: ignore
30
-
31
- # this is used as the default value for optional parameters
32
- OMIT = typing.cast(typing.Any, ...)
33
-
34
-
35
- class OrganizationsClient:
36
- def __init__(self, *, client_wrapper: SyncClientWrapper):
37
- self._client_wrapper = client_wrapper
38
-
39
- def list_organizations(self) -> typing.List[Organization]:
40
- """
41
- List organizations for a user.
42
-
43
- ---
44
- from llama_cloud.client import LlamaCloud
45
-
46
- client = LlamaCloud(
47
- token="YOUR_TOKEN",
48
- )
49
- client.organizations.list_organizations()
50
- """
51
- _response = self._client_wrapper.httpx_client.request(
52
- "GET",
53
- urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/organizations"),
54
- headers=self._client_wrapper.get_headers(),
55
- timeout=60,
56
- )
57
- if 200 <= _response.status_code < 300:
58
- return pydantic.parse_obj_as(typing.List[Organization], _response.json()) # type: ignore
59
- if _response.status_code == 422:
60
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
61
- try:
62
- _response_json = _response.json()
63
- except JSONDecodeError:
64
- raise ApiError(status_code=_response.status_code, body=_response.text)
65
- raise ApiError(status_code=_response.status_code, body=_response_json)
66
-
67
- def create_organization(self, *, request: OrganizationCreate) -> Organization:
68
- """
69
- Create a new organization.
70
-
71
- Parameters:
72
- - request: OrganizationCreate.
73
- ---
74
- from llama_cloud import OrganizationCreate
75
- from llama_cloud.client import LlamaCloud
76
-
77
- client = LlamaCloud(
78
- token="YOUR_TOKEN",
79
- )
80
- client.organizations.create_organization(
81
- request=OrganizationCreate(
82
- name="string",
83
- ),
84
- )
85
- """
86
- _response = self._client_wrapper.httpx_client.request(
87
- "POST",
88
- urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/organizations"),
89
- json=jsonable_encoder(request),
90
- headers=self._client_wrapper.get_headers(),
91
- timeout=60,
92
- )
93
- if 200 <= _response.status_code < 300:
94
- return pydantic.parse_obj_as(Organization, _response.json()) # type: ignore
95
- if _response.status_code == 422:
96
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
97
- try:
98
- _response_json = _response.json()
99
- except JSONDecodeError:
100
- raise ApiError(status_code=_response.status_code, body=_response.text)
101
- raise ApiError(status_code=_response.status_code, body=_response_json)
102
-
103
- def upsert_organization(self, *, request: OrganizationCreate) -> Organization:
104
- """
105
- Upsert a new organization.
106
-
107
- Parameters:
108
- - request: OrganizationCreate.
109
- ---
110
- from llama_cloud import OrganizationCreate
111
- from llama_cloud.client import LlamaCloud
112
-
113
- client = LlamaCloud(
114
- token="YOUR_TOKEN",
115
- )
116
- client.organizations.upsert_organization(
117
- request=OrganizationCreate(
118
- name="string",
119
- ),
120
- )
121
- """
122
- _response = self._client_wrapper.httpx_client.request(
123
- "PUT",
124
- urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/organizations"),
125
- json=jsonable_encoder(request),
126
- headers=self._client_wrapper.get_headers(),
127
- timeout=60,
128
- )
129
- if 200 <= _response.status_code < 300:
130
- return pydantic.parse_obj_as(Organization, _response.json()) # type: ignore
131
- if _response.status_code == 422:
132
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
133
- try:
134
- _response_json = _response.json()
135
- except JSONDecodeError:
136
- raise ApiError(status_code=_response.status_code, body=_response.text)
137
- raise ApiError(status_code=_response.status_code, body=_response_json)
138
-
139
- def get_default_organization(self) -> Organization:
140
- """
141
- Get the default organization for the user.
142
-
143
- ---
144
- from llama_cloud.client import LlamaCloud
145
-
146
- client = LlamaCloud(
147
- token="YOUR_TOKEN",
148
- )
149
- client.organizations.get_default_organization()
150
- """
151
- _response = self._client_wrapper.httpx_client.request(
152
- "GET",
153
- urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/organizations/default"),
154
- headers=self._client_wrapper.get_headers(),
155
- timeout=60,
156
- )
157
- if 200 <= _response.status_code < 300:
158
- return pydantic.parse_obj_as(Organization, _response.json()) # type: ignore
159
- if _response.status_code == 422:
160
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
161
- try:
162
- _response_json = _response.json()
163
- except JSONDecodeError:
164
- raise ApiError(status_code=_response.status_code, body=_response.text)
165
- raise ApiError(status_code=_response.status_code, body=_response_json)
166
-
167
- def set_default_organization(self, *, organization_id: str) -> Organization:
168
- """
169
- Set the default organization for the user.
170
-
171
- Parameters:
172
- - organization_id: str. The organization's ID.
173
- ---
174
- from llama_cloud.client import LlamaCloud
175
-
176
- client = LlamaCloud(
177
- token="YOUR_TOKEN",
178
- )
179
- client.organizations.set_default_organization(
180
- organization_id="string",
181
- )
182
- """
183
- _response = self._client_wrapper.httpx_client.request(
184
- "PUT",
185
- urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/organizations/default"),
186
- json=jsonable_encoder({"organization_id": organization_id}),
187
- headers=self._client_wrapper.get_headers(),
188
- timeout=60,
189
- )
190
- if 200 <= _response.status_code < 300:
191
- return pydantic.parse_obj_as(Organization, _response.json()) # type: ignore
192
- if _response.status_code == 422:
193
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
194
- try:
195
- _response_json = _response.json()
196
- except JSONDecodeError:
197
- raise ApiError(status_code=_response.status_code, body=_response.text)
198
- raise ApiError(status_code=_response.status_code, body=_response_json)
199
-
200
- def get_organization(self, organization_id: str) -> Organization:
201
- """
202
- Get an organization by ID.
203
-
204
- Parameters:
205
- - organization_id: str.
206
- ---
207
- from llama_cloud.client import LlamaCloud
208
-
209
- client = LlamaCloud(
210
- token="YOUR_TOKEN",
211
- )
212
- client.organizations.get_organization(
213
- organization_id="string",
214
- )
215
- """
216
- _response = self._client_wrapper.httpx_client.request(
217
- "GET",
218
- urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}"),
219
- headers=self._client_wrapper.get_headers(),
220
- timeout=60,
221
- )
222
- if 200 <= _response.status_code < 300:
223
- return pydantic.parse_obj_as(Organization, _response.json()) # type: ignore
224
- if _response.status_code == 422:
225
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
226
- try:
227
- _response_json = _response.json()
228
- except JSONDecodeError:
229
- raise ApiError(status_code=_response.status_code, body=_response.text)
230
- raise ApiError(status_code=_response.status_code, body=_response_json)
231
-
232
- def update_organization(self, organization_id: str, *, name: str) -> Organization:
233
- """
234
- Update an existing organization.
235
-
236
- Parameters:
237
- - organization_id: str.
238
-
239
- - name: str. A name for the organization.
240
- ---
241
- from llama_cloud.client import LlamaCloud
242
-
243
- client = LlamaCloud(
244
- token="YOUR_TOKEN",
245
- )
246
- client.organizations.update_organization(
247
- organization_id="string",
248
- name="string",
249
- )
250
- """
251
- _response = self._client_wrapper.httpx_client.request(
252
- "PUT",
253
- urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}"),
254
- json=jsonable_encoder({"name": name}),
255
- headers=self._client_wrapper.get_headers(),
256
- timeout=60,
257
- )
258
- if 200 <= _response.status_code < 300:
259
- return pydantic.parse_obj_as(Organization, _response.json()) # type: ignore
260
- if _response.status_code == 422:
261
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
262
- try:
263
- _response_json = _response.json()
264
- except JSONDecodeError:
265
- raise ApiError(status_code=_response.status_code, body=_response.text)
266
- raise ApiError(status_code=_response.status_code, body=_response_json)
267
-
268
- def delete_organization(self, organization_id: str) -> None:
269
- """
270
- Delete an organization by ID.
271
-
272
- Parameters:
273
- - organization_id: str.
274
- ---
275
- from llama_cloud.client import LlamaCloud
276
-
277
- client = LlamaCloud(
278
- token="YOUR_TOKEN",
279
- )
280
- client.organizations.delete_organization(
281
- organization_id="string",
282
- )
283
- """
284
- _response = self._client_wrapper.httpx_client.request(
285
- "DELETE",
286
- urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}"),
287
- headers=self._client_wrapper.get_headers(),
288
- timeout=60,
289
- )
290
- if 200 <= _response.status_code < 300:
291
- return
292
- if _response.status_code == 422:
293
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
294
- try:
295
- _response_json = _response.json()
296
- except JSONDecodeError:
297
- raise ApiError(status_code=_response.status_code, body=_response.text)
298
- raise ApiError(status_code=_response.status_code, body=_response_json)
299
-
300
- def get_organization_usage(
301
- self, organization_id: str, *, get_current_invoice_total: typing.Optional[bool] = None
302
- ) -> UsageAndPlan:
303
- """
304
- Get usage for a specific organization.
305
-
306
- Parameters:
307
- - organization_id: str.
308
-
309
- - get_current_invoice_total: typing.Optional[bool].
310
- ---
311
- from llama_cloud.client import LlamaCloud
312
-
313
- client = LlamaCloud(
314
- token="YOUR_TOKEN",
315
- )
316
- client.organizations.get_organization_usage(
317
- organization_id="string",
318
- )
319
- """
320
- _response = self._client_wrapper.httpx_client.request(
321
- "GET",
322
- urllib.parse.urljoin(
323
- f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/usage"
324
- ),
325
- params=remove_none_from_dict({"get_current_invoice_total": get_current_invoice_total}),
326
- headers=self._client_wrapper.get_headers(),
327
- timeout=60,
328
- )
329
- if 200 <= _response.status_code < 300:
330
- return pydantic.parse_obj_as(UsageAndPlan, _response.json()) # type: ignore
331
- if _response.status_code == 422:
332
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
333
- try:
334
- _response_json = _response.json()
335
- except JSONDecodeError:
336
- raise ApiError(status_code=_response.status_code, body=_response.text)
337
- raise ApiError(status_code=_response.status_code, body=_response_json)
338
-
339
- def list_organization_users(self, organization_id: str) -> typing.List[UserOrganization]:
340
- """
341
- Get all users in an organization.
342
-
343
- Parameters:
344
- - organization_id: str.
345
- ---
346
- from llama_cloud.client import LlamaCloud
347
-
348
- client = LlamaCloud(
349
- token="YOUR_TOKEN",
350
- )
351
- client.organizations.list_organization_users(
352
- organization_id="string",
353
- )
354
- """
355
- _response = self._client_wrapper.httpx_client.request(
356
- "GET",
357
- urllib.parse.urljoin(
358
- f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/users"
359
- ),
360
- headers=self._client_wrapper.get_headers(),
361
- timeout=60,
362
- )
363
- if 200 <= _response.status_code < 300:
364
- return pydantic.parse_obj_as(typing.List[UserOrganization], _response.json()) # type: ignore
365
- if _response.status_code == 422:
366
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
367
- try:
368
- _response_json = _response.json()
369
- except JSONDecodeError:
370
- raise ApiError(status_code=_response.status_code, body=_response.text)
371
- raise ApiError(status_code=_response.status_code, body=_response_json)
372
-
373
- def add_users_to_organization(
374
- self, organization_id: str, *, request: typing.List[UserOrganizationCreate]
375
- ) -> typing.List[UserOrganization]:
376
- """
377
- Add a user to an organization.
378
-
379
- Parameters:
380
- - organization_id: str.
381
-
382
- - request: typing.List[UserOrganizationCreate].
383
- ---
384
- from llama_cloud.client import LlamaCloud
385
-
386
- client = LlamaCloud(
387
- token="YOUR_TOKEN",
388
- )
389
- client.organizations.add_users_to_organization(
390
- organization_id="string",
391
- request=[],
392
- )
393
- """
394
- _response = self._client_wrapper.httpx_client.request(
395
- "PUT",
396
- urllib.parse.urljoin(
397
- f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/users"
398
- ),
399
- json=jsonable_encoder(request),
400
- headers=self._client_wrapper.get_headers(),
401
- timeout=60,
402
- )
403
- if 200 <= _response.status_code < 300:
404
- return pydantic.parse_obj_as(typing.List[UserOrganization], _response.json()) # type: ignore
405
- if _response.status_code == 422:
406
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
407
- try:
408
- _response_json = _response.json()
409
- except JSONDecodeError:
410
- raise ApiError(status_code=_response.status_code, body=_response.text)
411
- raise ApiError(status_code=_response.status_code, body=_response_json)
412
-
413
- def remove_users_from_organization(
414
- self, organization_id: str, member_user_id: str, *, request: typing.Optional[typing.List[str]] = None
415
- ) -> None:
416
- """
417
- Remove users from an organization.
418
-
419
- Parameters:
420
- - organization_id: str.
421
-
422
- - member_user_id: str.
423
-
424
- - request: typing.Optional[typing.List[str]].
425
- ---
426
- from llama_cloud.client import LlamaCloud
427
-
428
- client = LlamaCloud(
429
- token="YOUR_TOKEN",
430
- )
431
- client.organizations.remove_users_from_organization(
432
- organization_id="string",
433
- member_user_id="string",
434
- )
435
- """
436
- _response = self._client_wrapper.httpx_client.request(
437
- "DELETE",
438
- urllib.parse.urljoin(
439
- f"{self._client_wrapper.get_base_url()}/",
440
- f"api/v1/organizations/{organization_id}/users/{member_user_id}",
441
- ),
442
- json=jsonable_encoder(request),
443
- headers=self._client_wrapper.get_headers(),
444
- timeout=60,
445
- )
446
- if 200 <= _response.status_code < 300:
447
- return
448
- if _response.status_code == 422:
449
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
450
- try:
451
- _response_json = _response.json()
452
- except JSONDecodeError:
453
- raise ApiError(status_code=_response.status_code, body=_response.text)
454
- raise ApiError(status_code=_response.status_code, body=_response_json)
455
-
456
- def batch_remove_users_from_organization(
457
- self, organization_id: str, *, request: typing.List[UserOrganizationDelete]
458
- ) -> None:
459
- """
460
- Remove a batch of users from an organization.
461
-
462
- Parameters:
463
- - organization_id: str.
464
-
465
- - request: typing.List[UserOrganizationDelete].
466
- ---
467
- from llama_cloud.client import LlamaCloud
468
-
469
- client = LlamaCloud(
470
- token="YOUR_TOKEN",
471
- )
472
- client.organizations.batch_remove_users_from_organization(
473
- organization_id="string",
474
- request=[],
475
- )
476
- """
477
- _response = self._client_wrapper.httpx_client.request(
478
- "PUT",
479
- urllib.parse.urljoin(
480
- f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/users/remove"
481
- ),
482
- json=jsonable_encoder(request),
483
- headers=self._client_wrapper.get_headers(),
484
- timeout=60,
485
- )
486
- if 200 <= _response.status_code < 300:
487
- return
488
- if _response.status_code == 422:
489
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
490
- try:
491
- _response_json = _response.json()
492
- except JSONDecodeError:
493
- raise ApiError(status_code=_response.status_code, body=_response.text)
494
- raise ApiError(status_code=_response.status_code, body=_response_json)
495
-
496
- def list_roles(self, organization_id: str) -> typing.List[Role]:
497
- """
498
- List all roles in an organization.
499
-
500
- Parameters:
501
- - organization_id: str.
502
- ---
503
- from llama_cloud.client import LlamaCloud
504
-
505
- client = LlamaCloud(
506
- token="YOUR_TOKEN",
507
- )
508
- client.organizations.list_roles(
509
- organization_id="string",
510
- )
511
- """
512
- _response = self._client_wrapper.httpx_client.request(
513
- "GET",
514
- urllib.parse.urljoin(
515
- f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/roles"
516
- ),
517
- headers=self._client_wrapper.get_headers(),
518
- timeout=60,
519
- )
520
- if 200 <= _response.status_code < 300:
521
- return pydantic.parse_obj_as(typing.List[Role], _response.json()) # type: ignore
522
- if _response.status_code == 422:
523
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
524
- try:
525
- _response_json = _response.json()
526
- except JSONDecodeError:
527
- raise ApiError(status_code=_response.status_code, body=_response.text)
528
- raise ApiError(status_code=_response.status_code, body=_response_json)
529
-
530
- def get_user_role(
531
- self, organization_id: str, *, project_id: typing.Optional[str] = None
532
- ) -> typing.Optional[UserOrganizationRole]:
533
- """
534
- Get the role of a user in an organization.
535
-
536
- Parameters:
537
- - organization_id: str.
538
-
539
- - project_id: typing.Optional[str].
540
- ---
541
- from llama_cloud.client import LlamaCloud
542
-
543
- client = LlamaCloud(
544
- token="YOUR_TOKEN",
545
- )
546
- client.organizations.get_user_role(
547
- organization_id="string",
548
- )
549
- """
550
- _response = self._client_wrapper.httpx_client.request(
551
- "GET",
552
- urllib.parse.urljoin(
553
- f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/users/roles"
554
- ),
555
- params=remove_none_from_dict({"project_id": project_id}),
556
- headers=self._client_wrapper.get_headers(),
557
- timeout=60,
558
- )
559
- if 200 <= _response.status_code < 300:
560
- return pydantic.parse_obj_as(typing.Optional[UserOrganizationRole], _response.json()) # type: ignore
561
- if _response.status_code == 422:
562
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
563
- try:
564
- _response_json = _response.json()
565
- except JSONDecodeError:
566
- raise ApiError(status_code=_response.status_code, body=_response.text)
567
- raise ApiError(status_code=_response.status_code, body=_response_json)
568
-
569
- def assign_role_to_user_in_organization(
570
- self, organization_id: str, *, user_id: str, user_organization_role_create_organization_id: str, role_id: str
571
- ) -> UserOrganizationRole:
572
- """
573
- Assign a role to a user in an organization.
574
-
575
- Parameters:
576
- - organization_id: str.
577
-
578
- - user_id: str. The user's ID.
579
-
580
- - user_organization_role_create_organization_id: str. The organization's ID.
581
-
582
- - role_id: str. The role's ID.
583
- ---
584
- from llama_cloud.client import LlamaCloud
585
-
586
- client = LlamaCloud(
587
- token="YOUR_TOKEN",
588
- )
589
- client.organizations.assign_role_to_user_in_organization(
590
- organization_id="string",
591
- user_id="string",
592
- user_organization_role_create_organization_id="string",
593
- role_id="string",
594
- )
595
- """
596
- _response = self._client_wrapper.httpx_client.request(
597
- "PUT",
598
- urllib.parse.urljoin(
599
- f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/users/roles"
600
- ),
601
- json=jsonable_encoder(
602
- {
603
- "user_id": user_id,
604
- "organization_id": user_organization_role_create_organization_id,
605
- "role_id": role_id,
606
- }
607
- ),
608
- headers=self._client_wrapper.get_headers(),
609
- timeout=60,
610
- )
611
- if 200 <= _response.status_code < 300:
612
- return pydantic.parse_obj_as(UserOrganizationRole, _response.json()) # type: ignore
613
- if _response.status_code == 422:
614
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
615
- try:
616
- _response_json = _response.json()
617
- except JSONDecodeError:
618
- raise ApiError(status_code=_response.status_code, body=_response.text)
619
- raise ApiError(status_code=_response.status_code, body=_response_json)
620
-
621
- def list_projects_by_user(self, organization_id: str, user_id: str) -> typing.List[Project]:
622
- """
623
- List all projects for a user in an organization.
624
-
625
- Parameters:
626
- - organization_id: str.
627
-
628
- - user_id: str.
629
- ---
630
- from llama_cloud.client import LlamaCloud
631
-
632
- client = LlamaCloud(
633
- token="YOUR_TOKEN",
634
- )
635
- client.organizations.list_projects_by_user(
636
- organization_id="string",
637
- user_id="string",
638
- )
639
- """
640
- _response = self._client_wrapper.httpx_client.request(
641
- "GET",
642
- urllib.parse.urljoin(
643
- f"{self._client_wrapper.get_base_url()}/",
644
- f"api/v1/organizations/{organization_id}/users/{user_id}/projects",
645
- ),
646
- headers=self._client_wrapper.get_headers(),
647
- timeout=60,
648
- )
649
- if 200 <= _response.status_code < 300:
650
- return pydantic.parse_obj_as(typing.List[Project], _response.json()) # type: ignore
651
- if _response.status_code == 422:
652
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
653
- try:
654
- _response_json = _response.json()
655
- except JSONDecodeError:
656
- raise ApiError(status_code=_response.status_code, body=_response.text)
657
- raise ApiError(status_code=_response.status_code, body=_response_json)
658
-
659
- def add_user_to_project(
660
- self, organization_id: typing.Optional[str], user_id: str, *, project_id: typing.Optional[str] = None
661
- ) -> typing.Any:
662
- """
663
- Add a user to a project.
664
-
665
- Parameters:
666
- - organization_id: typing.Optional[str].
667
-
668
- - user_id: str.
669
-
670
- - project_id: typing.Optional[str].
671
- ---
672
- from llama_cloud.client import LlamaCloud
673
-
674
- client = LlamaCloud(
675
- token="YOUR_TOKEN",
676
- )
677
- client.organizations.add_user_to_project(
678
- user_id="string",
679
- )
680
- """
681
- _response = self._client_wrapper.httpx_client.request(
682
- "PUT",
683
- urllib.parse.urljoin(
684
- f"{self._client_wrapper.get_base_url()}/",
685
- f"api/v1/organizations/{organization_id}/users/{user_id}/projects",
686
- ),
687
- params=remove_none_from_dict({"project_id": project_id}),
688
- headers=self._client_wrapper.get_headers(),
689
- timeout=60,
690
- )
691
- if 200 <= _response.status_code < 300:
692
- return pydantic.parse_obj_as(typing.Any, _response.json()) # type: ignore
693
- if _response.status_code == 422:
694
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
695
- try:
696
- _response_json = _response.json()
697
- except JSONDecodeError:
698
- raise ApiError(status_code=_response.status_code, body=_response.text)
699
- raise ApiError(status_code=_response.status_code, body=_response_json)
700
-
701
- def remove_user_from_project(self, organization_id: str, user_id: str, project_id: str) -> typing.Any:
702
- """
703
- Remove a user from a project.
704
-
705
- Parameters:
706
- - organization_id: str.
707
-
708
- - user_id: str.
709
-
710
- - project_id: str.
711
- ---
712
- from llama_cloud.client import LlamaCloud
713
-
714
- client = LlamaCloud(
715
- token="YOUR_TOKEN",
716
- )
717
- client.organizations.remove_user_from_project(
718
- organization_id="string",
719
- user_id="string",
720
- project_id="string",
721
- )
722
- """
723
- _response = self._client_wrapper.httpx_client.request(
724
- "DELETE",
725
- urllib.parse.urljoin(
726
- f"{self._client_wrapper.get_base_url()}/",
727
- f"api/v1/organizations/{organization_id}/users/{user_id}/projects/{project_id}",
728
- ),
729
- headers=self._client_wrapper.get_headers(),
730
- timeout=60,
731
- )
732
- if 200 <= _response.status_code < 300:
733
- return pydantic.parse_obj_as(typing.Any, _response.json()) # type: ignore
734
- if _response.status_code == 422:
735
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
736
- try:
737
- _response_json = _response.json()
738
- except JSONDecodeError:
739
- raise ApiError(status_code=_response.status_code, body=_response.text)
740
- raise ApiError(status_code=_response.status_code, body=_response_json)
741
-
742
-
743
- class AsyncOrganizationsClient:
744
- def __init__(self, *, client_wrapper: AsyncClientWrapper):
745
- self._client_wrapper = client_wrapper
746
-
747
- async def list_organizations(self) -> typing.List[Organization]:
748
- """
749
- List organizations for a user.
750
-
751
- ---
752
- from llama_cloud.client import AsyncLlamaCloud
753
-
754
- client = AsyncLlamaCloud(
755
- token="YOUR_TOKEN",
756
- )
757
- await client.organizations.list_organizations()
758
- """
759
- _response = await self._client_wrapper.httpx_client.request(
760
- "GET",
761
- urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/organizations"),
762
- headers=self._client_wrapper.get_headers(),
763
- timeout=60,
764
- )
765
- if 200 <= _response.status_code < 300:
766
- return pydantic.parse_obj_as(typing.List[Organization], _response.json()) # type: ignore
767
- if _response.status_code == 422:
768
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
769
- try:
770
- _response_json = _response.json()
771
- except JSONDecodeError:
772
- raise ApiError(status_code=_response.status_code, body=_response.text)
773
- raise ApiError(status_code=_response.status_code, body=_response_json)
774
-
775
- async def create_organization(self, *, request: OrganizationCreate) -> Organization:
776
- """
777
- Create a new organization.
778
-
779
- Parameters:
780
- - request: OrganizationCreate.
781
- ---
782
- from llama_cloud import OrganizationCreate
783
- from llama_cloud.client import AsyncLlamaCloud
784
-
785
- client = AsyncLlamaCloud(
786
- token="YOUR_TOKEN",
787
- )
788
- await client.organizations.create_organization(
789
- request=OrganizationCreate(
790
- name="string",
791
- ),
792
- )
793
- """
794
- _response = await self._client_wrapper.httpx_client.request(
795
- "POST",
796
- urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/organizations"),
797
- json=jsonable_encoder(request),
798
- headers=self._client_wrapper.get_headers(),
799
- timeout=60,
800
- )
801
- if 200 <= _response.status_code < 300:
802
- return pydantic.parse_obj_as(Organization, _response.json()) # type: ignore
803
- if _response.status_code == 422:
804
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
805
- try:
806
- _response_json = _response.json()
807
- except JSONDecodeError:
808
- raise ApiError(status_code=_response.status_code, body=_response.text)
809
- raise ApiError(status_code=_response.status_code, body=_response_json)
810
-
811
- async def upsert_organization(self, *, request: OrganizationCreate) -> Organization:
812
- """
813
- Upsert a new organization.
814
-
815
- Parameters:
816
- - request: OrganizationCreate.
817
- ---
818
- from llama_cloud import OrganizationCreate
819
- from llama_cloud.client import AsyncLlamaCloud
820
-
821
- client = AsyncLlamaCloud(
822
- token="YOUR_TOKEN",
823
- )
824
- await client.organizations.upsert_organization(
825
- request=OrganizationCreate(
826
- name="string",
827
- ),
828
- )
829
- """
830
- _response = await self._client_wrapper.httpx_client.request(
831
- "PUT",
832
- urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/organizations"),
833
- json=jsonable_encoder(request),
834
- headers=self._client_wrapper.get_headers(),
835
- timeout=60,
836
- )
837
- if 200 <= _response.status_code < 300:
838
- return pydantic.parse_obj_as(Organization, _response.json()) # type: ignore
839
- if _response.status_code == 422:
840
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
841
- try:
842
- _response_json = _response.json()
843
- except JSONDecodeError:
844
- raise ApiError(status_code=_response.status_code, body=_response.text)
845
- raise ApiError(status_code=_response.status_code, body=_response_json)
846
-
847
- async def get_default_organization(self) -> Organization:
848
- """
849
- Get the default organization for the user.
850
-
851
- ---
852
- from llama_cloud.client import AsyncLlamaCloud
853
-
854
- client = AsyncLlamaCloud(
855
- token="YOUR_TOKEN",
856
- )
857
- await client.organizations.get_default_organization()
858
- """
859
- _response = await self._client_wrapper.httpx_client.request(
860
- "GET",
861
- urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/organizations/default"),
862
- headers=self._client_wrapper.get_headers(),
863
- timeout=60,
864
- )
865
- if 200 <= _response.status_code < 300:
866
- return pydantic.parse_obj_as(Organization, _response.json()) # type: ignore
867
- if _response.status_code == 422:
868
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
869
- try:
870
- _response_json = _response.json()
871
- except JSONDecodeError:
872
- raise ApiError(status_code=_response.status_code, body=_response.text)
873
- raise ApiError(status_code=_response.status_code, body=_response_json)
874
-
875
- async def set_default_organization(self, *, organization_id: str) -> Organization:
876
- """
877
- Set the default organization for the user.
878
-
879
- Parameters:
880
- - organization_id: str. The organization's ID.
881
- ---
882
- from llama_cloud.client import AsyncLlamaCloud
883
-
884
- client = AsyncLlamaCloud(
885
- token="YOUR_TOKEN",
886
- )
887
- await client.organizations.set_default_organization(
888
- organization_id="string",
889
- )
890
- """
891
- _response = await self._client_wrapper.httpx_client.request(
892
- "PUT",
893
- urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/organizations/default"),
894
- json=jsonable_encoder({"organization_id": organization_id}),
895
- headers=self._client_wrapper.get_headers(),
896
- timeout=60,
897
- )
898
- if 200 <= _response.status_code < 300:
899
- return pydantic.parse_obj_as(Organization, _response.json()) # type: ignore
900
- if _response.status_code == 422:
901
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
902
- try:
903
- _response_json = _response.json()
904
- except JSONDecodeError:
905
- raise ApiError(status_code=_response.status_code, body=_response.text)
906
- raise ApiError(status_code=_response.status_code, body=_response_json)
907
-
908
- async def get_organization(self, organization_id: str) -> Organization:
909
- """
910
- Get an organization by ID.
911
-
912
- Parameters:
913
- - organization_id: str.
914
- ---
915
- from llama_cloud.client import AsyncLlamaCloud
916
-
917
- client = AsyncLlamaCloud(
918
- token="YOUR_TOKEN",
919
- )
920
- await client.organizations.get_organization(
921
- organization_id="string",
922
- )
923
- """
924
- _response = await self._client_wrapper.httpx_client.request(
925
- "GET",
926
- urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}"),
927
- headers=self._client_wrapper.get_headers(),
928
- timeout=60,
929
- )
930
- if 200 <= _response.status_code < 300:
931
- return pydantic.parse_obj_as(Organization, _response.json()) # type: ignore
932
- if _response.status_code == 422:
933
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
934
- try:
935
- _response_json = _response.json()
936
- except JSONDecodeError:
937
- raise ApiError(status_code=_response.status_code, body=_response.text)
938
- raise ApiError(status_code=_response.status_code, body=_response_json)
939
-
940
- async def update_organization(self, organization_id: str, *, name: str) -> Organization:
941
- """
942
- Update an existing organization.
943
-
944
- Parameters:
945
- - organization_id: str.
946
-
947
- - name: str. A name for the organization.
948
- ---
949
- from llama_cloud.client import AsyncLlamaCloud
950
-
951
- client = AsyncLlamaCloud(
952
- token="YOUR_TOKEN",
953
- )
954
- await client.organizations.update_organization(
955
- organization_id="string",
956
- name="string",
957
- )
958
- """
959
- _response = await self._client_wrapper.httpx_client.request(
960
- "PUT",
961
- urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}"),
962
- json=jsonable_encoder({"name": name}),
963
- headers=self._client_wrapper.get_headers(),
964
- timeout=60,
965
- )
966
- if 200 <= _response.status_code < 300:
967
- return pydantic.parse_obj_as(Organization, _response.json()) # type: ignore
968
- if _response.status_code == 422:
969
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
970
- try:
971
- _response_json = _response.json()
972
- except JSONDecodeError:
973
- raise ApiError(status_code=_response.status_code, body=_response.text)
974
- raise ApiError(status_code=_response.status_code, body=_response_json)
975
-
976
- async def delete_organization(self, organization_id: str) -> None:
977
- """
978
- Delete an organization by ID.
979
-
980
- Parameters:
981
- - organization_id: str.
982
- ---
983
- from llama_cloud.client import AsyncLlamaCloud
984
-
985
- client = AsyncLlamaCloud(
986
- token="YOUR_TOKEN",
987
- )
988
- await client.organizations.delete_organization(
989
- organization_id="string",
990
- )
991
- """
992
- _response = await self._client_wrapper.httpx_client.request(
993
- "DELETE",
994
- urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}"),
995
- headers=self._client_wrapper.get_headers(),
996
- timeout=60,
997
- )
998
- if 200 <= _response.status_code < 300:
999
- return
1000
- if _response.status_code == 422:
1001
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
1002
- try:
1003
- _response_json = _response.json()
1004
- except JSONDecodeError:
1005
- raise ApiError(status_code=_response.status_code, body=_response.text)
1006
- raise ApiError(status_code=_response.status_code, body=_response_json)
1007
-
1008
- async def get_organization_usage(
1009
- self, organization_id: str, *, get_current_invoice_total: typing.Optional[bool] = None
1010
- ) -> UsageAndPlan:
1011
- """
1012
- Get usage for a specific organization.
1013
-
1014
- Parameters:
1015
- - organization_id: str.
1016
-
1017
- - get_current_invoice_total: typing.Optional[bool].
1018
- ---
1019
- from llama_cloud.client import AsyncLlamaCloud
1020
-
1021
- client = AsyncLlamaCloud(
1022
- token="YOUR_TOKEN",
1023
- )
1024
- await client.organizations.get_organization_usage(
1025
- organization_id="string",
1026
- )
1027
- """
1028
- _response = await self._client_wrapper.httpx_client.request(
1029
- "GET",
1030
- urllib.parse.urljoin(
1031
- f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/usage"
1032
- ),
1033
- params=remove_none_from_dict({"get_current_invoice_total": get_current_invoice_total}),
1034
- headers=self._client_wrapper.get_headers(),
1035
- timeout=60,
1036
- )
1037
- if 200 <= _response.status_code < 300:
1038
- return pydantic.parse_obj_as(UsageAndPlan, _response.json()) # type: ignore
1039
- if _response.status_code == 422:
1040
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
1041
- try:
1042
- _response_json = _response.json()
1043
- except JSONDecodeError:
1044
- raise ApiError(status_code=_response.status_code, body=_response.text)
1045
- raise ApiError(status_code=_response.status_code, body=_response_json)
1046
-
1047
- async def list_organization_users(self, organization_id: str) -> typing.List[UserOrganization]:
1048
- """
1049
- Get all users in an organization.
1050
-
1051
- Parameters:
1052
- - organization_id: str.
1053
- ---
1054
- from llama_cloud.client import AsyncLlamaCloud
1055
-
1056
- client = AsyncLlamaCloud(
1057
- token="YOUR_TOKEN",
1058
- )
1059
- await client.organizations.list_organization_users(
1060
- organization_id="string",
1061
- )
1062
- """
1063
- _response = await self._client_wrapper.httpx_client.request(
1064
- "GET",
1065
- urllib.parse.urljoin(
1066
- f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/users"
1067
- ),
1068
- headers=self._client_wrapper.get_headers(),
1069
- timeout=60,
1070
- )
1071
- if 200 <= _response.status_code < 300:
1072
- return pydantic.parse_obj_as(typing.List[UserOrganization], _response.json()) # type: ignore
1073
- if _response.status_code == 422:
1074
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
1075
- try:
1076
- _response_json = _response.json()
1077
- except JSONDecodeError:
1078
- raise ApiError(status_code=_response.status_code, body=_response.text)
1079
- raise ApiError(status_code=_response.status_code, body=_response_json)
1080
-
1081
- async def add_users_to_organization(
1082
- self, organization_id: str, *, request: typing.List[UserOrganizationCreate]
1083
- ) -> typing.List[UserOrganization]:
1084
- """
1085
- Add a user to an organization.
1086
-
1087
- Parameters:
1088
- - organization_id: str.
1089
-
1090
- - request: typing.List[UserOrganizationCreate].
1091
- ---
1092
- from llama_cloud.client import AsyncLlamaCloud
1093
-
1094
- client = AsyncLlamaCloud(
1095
- token="YOUR_TOKEN",
1096
- )
1097
- await client.organizations.add_users_to_organization(
1098
- organization_id="string",
1099
- request=[],
1100
- )
1101
- """
1102
- _response = await self._client_wrapper.httpx_client.request(
1103
- "PUT",
1104
- urllib.parse.urljoin(
1105
- f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/users"
1106
- ),
1107
- json=jsonable_encoder(request),
1108
- headers=self._client_wrapper.get_headers(),
1109
- timeout=60,
1110
- )
1111
- if 200 <= _response.status_code < 300:
1112
- return pydantic.parse_obj_as(typing.List[UserOrganization], _response.json()) # type: ignore
1113
- if _response.status_code == 422:
1114
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
1115
- try:
1116
- _response_json = _response.json()
1117
- except JSONDecodeError:
1118
- raise ApiError(status_code=_response.status_code, body=_response.text)
1119
- raise ApiError(status_code=_response.status_code, body=_response_json)
1120
-
1121
- async def remove_users_from_organization(
1122
- self, organization_id: str, member_user_id: str, *, request: typing.Optional[typing.List[str]] = None
1123
- ) -> None:
1124
- """
1125
- Remove users from an organization.
1126
-
1127
- Parameters:
1128
- - organization_id: str.
1129
-
1130
- - member_user_id: str.
1131
-
1132
- - request: typing.Optional[typing.List[str]].
1133
- ---
1134
- from llama_cloud.client import AsyncLlamaCloud
1135
-
1136
- client = AsyncLlamaCloud(
1137
- token="YOUR_TOKEN",
1138
- )
1139
- await client.organizations.remove_users_from_organization(
1140
- organization_id="string",
1141
- member_user_id="string",
1142
- )
1143
- """
1144
- _response = await self._client_wrapper.httpx_client.request(
1145
- "DELETE",
1146
- urllib.parse.urljoin(
1147
- f"{self._client_wrapper.get_base_url()}/",
1148
- f"api/v1/organizations/{organization_id}/users/{member_user_id}",
1149
- ),
1150
- json=jsonable_encoder(request),
1151
- headers=self._client_wrapper.get_headers(),
1152
- timeout=60,
1153
- )
1154
- if 200 <= _response.status_code < 300:
1155
- return
1156
- if _response.status_code == 422:
1157
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
1158
- try:
1159
- _response_json = _response.json()
1160
- except JSONDecodeError:
1161
- raise ApiError(status_code=_response.status_code, body=_response.text)
1162
- raise ApiError(status_code=_response.status_code, body=_response_json)
1163
-
1164
- async def batch_remove_users_from_organization(
1165
- self, organization_id: str, *, request: typing.List[UserOrganizationDelete]
1166
- ) -> None:
1167
- """
1168
- Remove a batch of users from an organization.
1169
-
1170
- Parameters:
1171
- - organization_id: str.
1172
-
1173
- - request: typing.List[UserOrganizationDelete].
1174
- ---
1175
- from llama_cloud.client import AsyncLlamaCloud
1176
-
1177
- client = AsyncLlamaCloud(
1178
- token="YOUR_TOKEN",
1179
- )
1180
- await client.organizations.batch_remove_users_from_organization(
1181
- organization_id="string",
1182
- request=[],
1183
- )
1184
- """
1185
- _response = await self._client_wrapper.httpx_client.request(
1186
- "PUT",
1187
- urllib.parse.urljoin(
1188
- f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/users/remove"
1189
- ),
1190
- json=jsonable_encoder(request),
1191
- headers=self._client_wrapper.get_headers(),
1192
- timeout=60,
1193
- )
1194
- if 200 <= _response.status_code < 300:
1195
- return
1196
- if _response.status_code == 422:
1197
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
1198
- try:
1199
- _response_json = _response.json()
1200
- except JSONDecodeError:
1201
- raise ApiError(status_code=_response.status_code, body=_response.text)
1202
- raise ApiError(status_code=_response.status_code, body=_response_json)
1203
-
1204
- async def list_roles(self, organization_id: str) -> typing.List[Role]:
1205
- """
1206
- List all roles in an organization.
1207
-
1208
- Parameters:
1209
- - organization_id: str.
1210
- ---
1211
- from llama_cloud.client import AsyncLlamaCloud
1212
-
1213
- client = AsyncLlamaCloud(
1214
- token="YOUR_TOKEN",
1215
- )
1216
- await client.organizations.list_roles(
1217
- organization_id="string",
1218
- )
1219
- """
1220
- _response = await self._client_wrapper.httpx_client.request(
1221
- "GET",
1222
- urllib.parse.urljoin(
1223
- f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/roles"
1224
- ),
1225
- headers=self._client_wrapper.get_headers(),
1226
- timeout=60,
1227
- )
1228
- if 200 <= _response.status_code < 300:
1229
- return pydantic.parse_obj_as(typing.List[Role], _response.json()) # type: ignore
1230
- if _response.status_code == 422:
1231
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
1232
- try:
1233
- _response_json = _response.json()
1234
- except JSONDecodeError:
1235
- raise ApiError(status_code=_response.status_code, body=_response.text)
1236
- raise ApiError(status_code=_response.status_code, body=_response_json)
1237
-
1238
- async def get_user_role(
1239
- self, organization_id: str, *, project_id: typing.Optional[str] = None
1240
- ) -> typing.Optional[UserOrganizationRole]:
1241
- """
1242
- Get the role of a user in an organization.
1243
-
1244
- Parameters:
1245
- - organization_id: str.
1246
-
1247
- - project_id: typing.Optional[str].
1248
- ---
1249
- from llama_cloud.client import AsyncLlamaCloud
1250
-
1251
- client = AsyncLlamaCloud(
1252
- token="YOUR_TOKEN",
1253
- )
1254
- await client.organizations.get_user_role(
1255
- organization_id="string",
1256
- )
1257
- """
1258
- _response = await self._client_wrapper.httpx_client.request(
1259
- "GET",
1260
- urllib.parse.urljoin(
1261
- f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/users/roles"
1262
- ),
1263
- params=remove_none_from_dict({"project_id": project_id}),
1264
- headers=self._client_wrapper.get_headers(),
1265
- timeout=60,
1266
- )
1267
- if 200 <= _response.status_code < 300:
1268
- return pydantic.parse_obj_as(typing.Optional[UserOrganizationRole], _response.json()) # type: ignore
1269
- if _response.status_code == 422:
1270
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
1271
- try:
1272
- _response_json = _response.json()
1273
- except JSONDecodeError:
1274
- raise ApiError(status_code=_response.status_code, body=_response.text)
1275
- raise ApiError(status_code=_response.status_code, body=_response_json)
1276
-
1277
- async def assign_role_to_user_in_organization(
1278
- self, organization_id: str, *, user_id: str, user_organization_role_create_organization_id: str, role_id: str
1279
- ) -> UserOrganizationRole:
1280
- """
1281
- Assign a role to a user in an organization.
1282
-
1283
- Parameters:
1284
- - organization_id: str.
1285
-
1286
- - user_id: str. The user's ID.
1287
-
1288
- - user_organization_role_create_organization_id: str. The organization's ID.
1289
-
1290
- - role_id: str. The role's ID.
1291
- ---
1292
- from llama_cloud.client import AsyncLlamaCloud
1293
-
1294
- client = AsyncLlamaCloud(
1295
- token="YOUR_TOKEN",
1296
- )
1297
- await client.organizations.assign_role_to_user_in_organization(
1298
- organization_id="string",
1299
- user_id="string",
1300
- user_organization_role_create_organization_id="string",
1301
- role_id="string",
1302
- )
1303
- """
1304
- _response = await self._client_wrapper.httpx_client.request(
1305
- "PUT",
1306
- urllib.parse.urljoin(
1307
- f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/users/roles"
1308
- ),
1309
- json=jsonable_encoder(
1310
- {
1311
- "user_id": user_id,
1312
- "organization_id": user_organization_role_create_organization_id,
1313
- "role_id": role_id,
1314
- }
1315
- ),
1316
- headers=self._client_wrapper.get_headers(),
1317
- timeout=60,
1318
- )
1319
- if 200 <= _response.status_code < 300:
1320
- return pydantic.parse_obj_as(UserOrganizationRole, _response.json()) # type: ignore
1321
- if _response.status_code == 422:
1322
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
1323
- try:
1324
- _response_json = _response.json()
1325
- except JSONDecodeError:
1326
- raise ApiError(status_code=_response.status_code, body=_response.text)
1327
- raise ApiError(status_code=_response.status_code, body=_response_json)
1328
-
1329
- async def list_projects_by_user(self, organization_id: str, user_id: str) -> typing.List[Project]:
1330
- """
1331
- List all projects for a user in an organization.
1332
-
1333
- Parameters:
1334
- - organization_id: str.
1335
-
1336
- - user_id: str.
1337
- ---
1338
- from llama_cloud.client import AsyncLlamaCloud
1339
-
1340
- client = AsyncLlamaCloud(
1341
- token="YOUR_TOKEN",
1342
- )
1343
- await client.organizations.list_projects_by_user(
1344
- organization_id="string",
1345
- user_id="string",
1346
- )
1347
- """
1348
- _response = await self._client_wrapper.httpx_client.request(
1349
- "GET",
1350
- urllib.parse.urljoin(
1351
- f"{self._client_wrapper.get_base_url()}/",
1352
- f"api/v1/organizations/{organization_id}/users/{user_id}/projects",
1353
- ),
1354
- headers=self._client_wrapper.get_headers(),
1355
- timeout=60,
1356
- )
1357
- if 200 <= _response.status_code < 300:
1358
- return pydantic.parse_obj_as(typing.List[Project], _response.json()) # type: ignore
1359
- if _response.status_code == 422:
1360
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
1361
- try:
1362
- _response_json = _response.json()
1363
- except JSONDecodeError:
1364
- raise ApiError(status_code=_response.status_code, body=_response.text)
1365
- raise ApiError(status_code=_response.status_code, body=_response_json)
1366
-
1367
- async def add_user_to_project(
1368
- self, organization_id: typing.Optional[str], user_id: str, *, project_id: typing.Optional[str] = None
1369
- ) -> typing.Any:
1370
- """
1371
- Add a user to a project.
1372
-
1373
- Parameters:
1374
- - organization_id: typing.Optional[str].
1375
-
1376
- - user_id: str.
1377
-
1378
- - project_id: typing.Optional[str].
1379
- ---
1380
- from llama_cloud.client import AsyncLlamaCloud
1381
-
1382
- client = AsyncLlamaCloud(
1383
- token="YOUR_TOKEN",
1384
- )
1385
- await client.organizations.add_user_to_project(
1386
- user_id="string",
1387
- )
1388
- """
1389
- _response = await self._client_wrapper.httpx_client.request(
1390
- "PUT",
1391
- urllib.parse.urljoin(
1392
- f"{self._client_wrapper.get_base_url()}/",
1393
- f"api/v1/organizations/{organization_id}/users/{user_id}/projects",
1394
- ),
1395
- params=remove_none_from_dict({"project_id": project_id}),
1396
- headers=self._client_wrapper.get_headers(),
1397
- timeout=60,
1398
- )
1399
- if 200 <= _response.status_code < 300:
1400
- return pydantic.parse_obj_as(typing.Any, _response.json()) # type: ignore
1401
- if _response.status_code == 422:
1402
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
1403
- try:
1404
- _response_json = _response.json()
1405
- except JSONDecodeError:
1406
- raise ApiError(status_code=_response.status_code, body=_response.text)
1407
- raise ApiError(status_code=_response.status_code, body=_response_json)
1408
-
1409
- async def remove_user_from_project(self, organization_id: str, user_id: str, project_id: str) -> typing.Any:
1410
- """
1411
- Remove a user from a project.
1412
-
1413
- Parameters:
1414
- - organization_id: str.
1415
-
1416
- - user_id: str.
1417
-
1418
- - project_id: str.
1419
- ---
1420
- from llama_cloud.client import AsyncLlamaCloud
1421
-
1422
- client = AsyncLlamaCloud(
1423
- token="YOUR_TOKEN",
1424
- )
1425
- await client.organizations.remove_user_from_project(
1426
- organization_id="string",
1427
- user_id="string",
1428
- project_id="string",
1429
- )
1430
- """
1431
- _response = await self._client_wrapper.httpx_client.request(
1432
- "DELETE",
1433
- urllib.parse.urljoin(
1434
- f"{self._client_wrapper.get_base_url()}/",
1435
- f"api/v1/organizations/{organization_id}/users/{user_id}/projects/{project_id}",
1436
- ),
1437
- headers=self._client_wrapper.get_headers(),
1438
- timeout=60,
1439
- )
1440
- if 200 <= _response.status_code < 300:
1441
- return pydantic.parse_obj_as(typing.Any, _response.json()) # type: ignore
1442
- if _response.status_code == 422:
1443
- raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
1444
- try:
1445
- _response_json = _response.json()
1446
- except JSONDecodeError:
1447
- raise ApiError(status_code=_response.status_code, body=_response.text)
1448
- raise ApiError(status_code=_response.status_code, body=_response_json)