c63a5cfe-b235-4fbe-8bbb-82a9e02a482a-python 0.1.0a3__py3-none-any.whl → 0.1.0a5__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 (248) hide show
  1. {c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a3.dist-info → c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a5.dist-info}/METADATA +124 -67
  2. c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a5.dist-info/RECORD +230 -0
  3. {c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a3.dist-info → c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a5.dist-info}/licenses/LICENSE +1 -1
  4. {digitalocean_genai_sdk → do_gradientai}/__init__.py +10 -9
  5. {digitalocean_genai_sdk → do_gradientai}/_base_client.py +39 -3
  6. {digitalocean_genai_sdk → do_gradientai}/_client.py +132 -209
  7. {digitalocean_genai_sdk → do_gradientai}/_exceptions.py +2 -2
  8. {digitalocean_genai_sdk → do_gradientai}/_resource.py +5 -5
  9. {digitalocean_genai_sdk → do_gradientai}/_response.py +6 -6
  10. {digitalocean_genai_sdk → do_gradientai}/_streaming.py +3 -3
  11. {digitalocean_genai_sdk → do_gradientai}/_types.py +1 -1
  12. {digitalocean_genai_sdk → do_gradientai}/_utils/_logs.py +3 -3
  13. do_gradientai/_utils/_resources_proxy.py +24 -0
  14. do_gradientai/_version.py +4 -0
  15. {digitalocean_genai_sdk → do_gradientai}/resources/__init__.py +27 -69
  16. do_gradientai/resources/agents/__init__.py +145 -0
  17. {digitalocean_genai_sdk → do_gradientai}/resources/agents/agents.py +188 -36
  18. {digitalocean_genai_sdk → do_gradientai}/resources/agents/api_keys.py +34 -14
  19. do_gradientai/resources/agents/evaluation_datasets.py +292 -0
  20. do_gradientai/resources/agents/evaluation_metrics/__init__.py +33 -0
  21. do_gradientai/resources/agents/evaluation_metrics/evaluation_metrics.py +177 -0
  22. {digitalocean_genai_sdk/resources/auth/agents → do_gradientai/resources/agents/evaluation_metrics/workspaces}/__init__.py +14 -14
  23. do_gradientai/resources/agents/evaluation_metrics/workspaces/agents.py +324 -0
  24. do_gradientai/resources/agents/evaluation_metrics/workspaces/workspaces.py +654 -0
  25. do_gradientai/resources/agents/evaluation_runs.py +466 -0
  26. do_gradientai/resources/agents/evaluation_test_cases.py +639 -0
  27. {digitalocean_genai_sdk → do_gradientai}/resources/agents/functions.py +22 -10
  28. {digitalocean_genai_sdk → do_gradientai}/resources/agents/knowledge_bases.py +22 -10
  29. digitalocean_genai_sdk/resources/agents/child_agents.py → do_gradientai/resources/agents/routes.py +92 -76
  30. {digitalocean_genai_sdk → do_gradientai}/resources/agents/versions.py +16 -8
  31. do_gradientai/resources/chat/__init__.py +33 -0
  32. do_gradientai/resources/chat/chat.py +102 -0
  33. digitalocean_genai_sdk/resources/chat.py → do_gradientai/resources/chat/completions.py +81 -59
  34. {digitalocean_genai_sdk/resources/api_keys → do_gradientai/resources/inference}/__init__.py +14 -0
  35. digitalocean_genai_sdk/resources/api_keys/api_keys_.py → do_gradientai/resources/inference/api_keys.py +40 -20
  36. do_gradientai/resources/inference/inference.py +102 -0
  37. {digitalocean_genai_sdk → do_gradientai}/resources/knowledge_bases/__init__.py +14 -0
  38. {digitalocean_genai_sdk → do_gradientai}/resources/knowledge_bases/data_sources.py +25 -12
  39. {digitalocean_genai_sdk/resources → do_gradientai/resources/knowledge_bases}/indexing_jobs.py +50 -26
  40. {digitalocean_genai_sdk → do_gradientai}/resources/knowledge_bases/knowledge_bases.py +66 -14
  41. {digitalocean_genai_sdk/resources/providers → do_gradientai/resources/model_providers}/__init__.py +13 -13
  42. {digitalocean_genai_sdk/resources/providers → do_gradientai/resources/model_providers}/anthropic/anthropic.py +4 -4
  43. {digitalocean_genai_sdk/resources/providers → do_gradientai/resources/model_providers}/anthropic/keys.py +52 -23
  44. digitalocean_genai_sdk/resources/providers/providers.py → do_gradientai/resources/model_providers/model_providers.py +35 -35
  45. {digitalocean_genai_sdk/resources/providers → do_gradientai/resources/model_providers}/openai/keys.py +52 -23
  46. {digitalocean_genai_sdk/resources/providers → do_gradientai/resources/model_providers}/openai/openai.py +4 -4
  47. {digitalocean_genai_sdk → do_gradientai}/resources/models.py +12 -8
  48. {digitalocean_genai_sdk → do_gradientai}/resources/regions.py +10 -6
  49. {digitalocean_genai_sdk → do_gradientai}/types/__init__.py +3 -22
  50. {digitalocean_genai_sdk → do_gradientai}/types/agent_list_response.py +5 -5
  51. do_gradientai/types/agents/__init__.py +67 -0
  52. do_gradientai/types/agents/api_evaluation_metric.py +24 -0
  53. do_gradientai/types/agents/api_evaluation_metric_result.py +17 -0
  54. do_gradientai/types/agents/api_evaluation_prompt.py +42 -0
  55. do_gradientai/types/agents/api_evaluation_run.py +68 -0
  56. do_gradientai/types/agents/api_evaluation_test_case.py +46 -0
  57. {digitalocean_genai_sdk → do_gradientai}/types/agents/api_key_list_response.py +2 -2
  58. do_gradientai/types/agents/api_star_metric.py +19 -0
  59. do_gradientai/types/agents/api_star_metric_param.py +19 -0
  60. do_gradientai/types/agents/evaluation_dataset_create_file_upload_presigned_urls_params.py +20 -0
  61. do_gradientai/types/agents/evaluation_dataset_create_file_upload_presigned_urls_response.py +30 -0
  62. do_gradientai/types/agents/evaluation_dataset_create_params.py +17 -0
  63. do_gradientai/types/agents/evaluation_dataset_create_response.py +12 -0
  64. do_gradientai/types/agents/evaluation_metric_list_response.py +12 -0
  65. do_gradientai/types/agents/evaluation_metrics/__init__.py +14 -0
  66. do_gradientai/types/agents/evaluation_metrics/workspace_create_params.py +16 -0
  67. do_gradientai/types/agents/evaluation_metrics/workspace_create_response.py +16 -0
  68. digitalocean_genai_sdk/types/auth/agents/token_create_response.py → do_gradientai/types/agents/evaluation_metrics/workspace_delete_response.py +3 -5
  69. do_gradientai/types/agents/evaluation_metrics/workspace_list_evaluation_test_cases_response.py +12 -0
  70. do_gradientai/types/agents/evaluation_metrics/workspace_list_response.py +16 -0
  71. do_gradientai/types/agents/evaluation_metrics/workspace_retrieve_response.py +16 -0
  72. do_gradientai/types/agents/evaluation_metrics/workspace_update_params.py +18 -0
  73. do_gradientai/types/agents/evaluation_metrics/workspace_update_response.py +16 -0
  74. do_gradientai/types/agents/evaluation_metrics/workspaces/__init__.py +8 -0
  75. do_gradientai/types/agents/evaluation_metrics/workspaces/agent_list_params.py +26 -0
  76. do_gradientai/types/agents/evaluation_metrics/workspaces/agent_list_response.py +22 -0
  77. do_gradientai/types/agents/evaluation_metrics/workspaces/agent_move_params.py +16 -0
  78. do_gradientai/types/agents/evaluation_metrics/workspaces/agent_move_response.py +16 -0
  79. do_gradientai/types/agents/evaluation_run_create_params.py +18 -0
  80. do_gradientai/types/agents/evaluation_run_create_response.py +11 -0
  81. do_gradientai/types/agents/evaluation_run_list_results_response.py +16 -0
  82. do_gradientai/types/agents/evaluation_run_retrieve_response.py +12 -0
  83. do_gradientai/types/agents/evaluation_run_retrieve_results_response.py +12 -0
  84. do_gradientai/types/agents/evaluation_test_case_create_params.py +29 -0
  85. do_gradientai/types/agents/evaluation_test_case_create_response.py +12 -0
  86. do_gradientai/types/agents/evaluation_test_case_list_evaluation_runs_params.py +12 -0
  87. do_gradientai/types/agents/evaluation_test_case_list_evaluation_runs_response.py +13 -0
  88. do_gradientai/types/agents/evaluation_test_case_list_response.py +12 -0
  89. do_gradientai/types/agents/evaluation_test_case_retrieve_params.py +12 -0
  90. do_gradientai/types/agents/evaluation_test_case_retrieve_response.py +12 -0
  91. do_gradientai/types/agents/evaluation_test_case_update_params.py +32 -0
  92. do_gradientai/types/agents/evaluation_test_case_update_response.py +14 -0
  93. digitalocean_genai_sdk/types/agents/child_agent_add_params.py → do_gradientai/types/agents/route_add_params.py +2 -2
  94. digitalocean_genai_sdk/types/agents/child_agent_add_response.py → do_gradientai/types/agents/route_add_response.py +2 -2
  95. digitalocean_genai_sdk/types/agents/child_agent_delete_response.py → do_gradientai/types/agents/route_delete_response.py +2 -2
  96. digitalocean_genai_sdk/types/agents/child_agent_update_params.py → do_gradientai/types/agents/route_update_params.py +2 -2
  97. digitalocean_genai_sdk/types/agents/child_agent_update_response.py → do_gradientai/types/agents/route_update_response.py +2 -2
  98. digitalocean_genai_sdk/types/agents/child_agent_view_response.py → do_gradientai/types/agents/route_view_response.py +2 -2
  99. {digitalocean_genai_sdk → do_gradientai}/types/agents/version_list_response.py +2 -2
  100. {digitalocean_genai_sdk → do_gradientai}/types/api_agent.py +8 -3
  101. digitalocean_genai_sdk/types/api_model.py → do_gradientai/types/api_agent_model.py +2 -2
  102. {digitalocean_genai_sdk → do_gradientai}/types/api_knowledge_base.py +1 -1
  103. {digitalocean_genai_sdk → do_gradientai}/types/api_openai_api_key_info.py +2 -2
  104. do_gradientai/types/api_workspace.py +36 -0
  105. do_gradientai/types/chat/__init__.py +7 -0
  106. {digitalocean_genai_sdk/types → do_gradientai/types/chat}/chat_completion_token_logprob.py +1 -1
  107. digitalocean_genai_sdk/types/chat_create_completion_params.py → do_gradientai/types/chat/completion_create_params.py +6 -29
  108. digitalocean_genai_sdk/types/chat_create_completion_response.py → do_gradientai/types/chat/completion_create_response.py +3 -3
  109. {digitalocean_genai_sdk/types/api_keys → do_gradientai/types/inference}/api_key_list_response.py +2 -2
  110. {digitalocean_genai_sdk → do_gradientai}/types/knowledge_base_create_params.py +3 -0
  111. {digitalocean_genai_sdk → do_gradientai}/types/knowledge_base_list_response.py +2 -2
  112. {digitalocean_genai_sdk → do_gradientai}/types/knowledge_bases/__init__.py +13 -0
  113. digitalocean_genai_sdk/types/indexing_job_retrieve_data_sources_response.py → do_gradientai/types/knowledge_bases/api_indexed_data_source.py +4 -8
  114. {digitalocean_genai_sdk/types → do_gradientai/types/knowledge_bases}/api_indexing_job.py +13 -1
  115. {digitalocean_genai_sdk → do_gradientai}/types/knowledge_bases/api_knowledge_base_data_source.py +15 -2
  116. do_gradientai/types/knowledge_bases/aws_data_source_param.py +19 -0
  117. {digitalocean_genai_sdk → do_gradientai}/types/knowledge_bases/data_source_create_params.py +3 -14
  118. {digitalocean_genai_sdk → do_gradientai}/types/knowledge_bases/data_source_list_response.py +2 -2
  119. {digitalocean_genai_sdk/types → do_gradientai/types/knowledge_bases}/indexing_job_create_response.py +1 -1
  120. {digitalocean_genai_sdk/types → do_gradientai/types/knowledge_bases}/indexing_job_list_response.py +3 -3
  121. do_gradientai/types/knowledge_bases/indexing_job_retrieve_data_sources_response.py +12 -0
  122. {digitalocean_genai_sdk/types → do_gradientai/types/knowledge_bases}/indexing_job_retrieve_response.py +1 -1
  123. {digitalocean_genai_sdk/types → do_gradientai/types/knowledge_bases}/indexing_job_update_cancel_params.py +1 -1
  124. {digitalocean_genai_sdk/types → do_gradientai/types/knowledge_bases}/indexing_job_update_cancel_response.py +1 -1
  125. {digitalocean_genai_sdk/types/providers → do_gradientai/types/model_providers}/anthropic/key_list_agents_response.py +2 -2
  126. {digitalocean_genai_sdk/types/providers → do_gradientai/types/model_providers}/anthropic/key_list_response.py +2 -2
  127. {digitalocean_genai_sdk/types/providers → do_gradientai/types/model_providers}/openai/key_list_response.py +2 -2
  128. {digitalocean_genai_sdk/types/providers → do_gradientai/types/model_providers}/openai/key_retrieve_agents_response.py +2 -2
  129. {digitalocean_genai_sdk/types/providers → do_gradientai/types/shared}/__init__.py +2 -1
  130. c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a3.dist-info/RECORD +0 -190
  131. digitalocean_genai_sdk/_utils/_resources_proxy.py +0 -24
  132. digitalocean_genai_sdk/_version.py +0 -4
  133. digitalocean_genai_sdk/resources/agents/__init__.py +0 -89
  134. digitalocean_genai_sdk/resources/api_keys/api_keys.py +0 -275
  135. digitalocean_genai_sdk/resources/auth/__init__.py +0 -33
  136. digitalocean_genai_sdk/resources/auth/agents/agents.py +0 -102
  137. digitalocean_genai_sdk/resources/auth/agents/token.py +0 -173
  138. digitalocean_genai_sdk/resources/auth/auth.py +0 -102
  139. digitalocean_genai_sdk/resources/embeddings.py +0 -201
  140. digitalocean_genai_sdk/types/agents/__init__.py +0 -31
  141. digitalocean_genai_sdk/types/api_key_list_params.py +0 -42
  142. digitalocean_genai_sdk/types/api_key_list_response.py +0 -42
  143. digitalocean_genai_sdk/types/auth/agents/__init__.py +0 -6
  144. digitalocean_genai_sdk/types/auth/agents/token_create_params.py +0 -13
  145. digitalocean_genai_sdk/types/chat_completion_request_message_content_part_text_param.py +0 -15
  146. digitalocean_genai_sdk/types/embedding_create_params.py +0 -28
  147. digitalocean_genai_sdk/types/embedding_create_response.py +0 -41
  148. {c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a3.dist-info → c63a5cfe_b235_4fbe_8bbb_82a9e02a482a_python-0.1.0a5.dist-info}/WHEEL +0 -0
  149. {digitalocean_genai_sdk → do_gradientai}/_compat.py +0 -0
  150. {digitalocean_genai_sdk → do_gradientai}/_constants.py +0 -0
  151. {digitalocean_genai_sdk → do_gradientai}/_files.py +0 -0
  152. {digitalocean_genai_sdk → do_gradientai}/_models.py +0 -0
  153. {digitalocean_genai_sdk → do_gradientai}/_qs.py +0 -0
  154. {digitalocean_genai_sdk → do_gradientai}/_utils/__init__.py +0 -0
  155. {digitalocean_genai_sdk → do_gradientai}/_utils/_proxy.py +0 -0
  156. {digitalocean_genai_sdk → do_gradientai}/_utils/_reflection.py +0 -0
  157. {digitalocean_genai_sdk → do_gradientai}/_utils/_streams.py +0 -0
  158. {digitalocean_genai_sdk → do_gradientai}/_utils/_sync.py +0 -0
  159. {digitalocean_genai_sdk → do_gradientai}/_utils/_transform.py +0 -0
  160. {digitalocean_genai_sdk → do_gradientai}/_utils/_typing.py +0 -0
  161. {digitalocean_genai_sdk → do_gradientai}/_utils/_utils.py +0 -0
  162. {digitalocean_genai_sdk → do_gradientai}/lib/.keep +0 -0
  163. {digitalocean_genai_sdk → do_gradientai}/py.typed +0 -0
  164. {digitalocean_genai_sdk/resources/providers → do_gradientai/resources/model_providers}/anthropic/__init__.py +0 -0
  165. {digitalocean_genai_sdk/resources/providers → do_gradientai/resources/model_providers}/openai/__init__.py +0 -0
  166. {digitalocean_genai_sdk → do_gradientai}/types/agent_create_params.py +0 -0
  167. {digitalocean_genai_sdk → do_gradientai}/types/agent_create_response.py +0 -0
  168. {digitalocean_genai_sdk → do_gradientai}/types/agent_delete_response.py +0 -0
  169. {digitalocean_genai_sdk → do_gradientai}/types/agent_list_params.py +0 -0
  170. {digitalocean_genai_sdk → do_gradientai}/types/agent_retrieve_response.py +0 -0
  171. {digitalocean_genai_sdk → do_gradientai}/types/agent_update_params.py +0 -0
  172. {digitalocean_genai_sdk → do_gradientai}/types/agent_update_response.py +0 -0
  173. {digitalocean_genai_sdk → do_gradientai}/types/agent_update_status_params.py +0 -0
  174. {digitalocean_genai_sdk → do_gradientai}/types/agent_update_status_response.py +0 -0
  175. {digitalocean_genai_sdk → do_gradientai}/types/agents/api_key_create_params.py +0 -0
  176. {digitalocean_genai_sdk → do_gradientai}/types/agents/api_key_create_response.py +0 -0
  177. {digitalocean_genai_sdk → do_gradientai}/types/agents/api_key_delete_response.py +0 -0
  178. {digitalocean_genai_sdk → do_gradientai}/types/agents/api_key_list_params.py +0 -0
  179. {digitalocean_genai_sdk → do_gradientai}/types/agents/api_key_regenerate_response.py +0 -0
  180. {digitalocean_genai_sdk → do_gradientai}/types/agents/api_key_update_params.py +0 -0
  181. {digitalocean_genai_sdk → do_gradientai}/types/agents/api_key_update_response.py +0 -0
  182. {digitalocean_genai_sdk → do_gradientai}/types/agents/api_link_knowledge_base_output.py +0 -0
  183. {digitalocean_genai_sdk → do_gradientai}/types/agents/function_create_params.py +0 -0
  184. {digitalocean_genai_sdk → do_gradientai}/types/agents/function_create_response.py +0 -0
  185. {digitalocean_genai_sdk → do_gradientai}/types/agents/function_delete_response.py +0 -0
  186. {digitalocean_genai_sdk → do_gradientai}/types/agents/function_update_params.py +0 -0
  187. {digitalocean_genai_sdk → do_gradientai}/types/agents/function_update_response.py +0 -0
  188. {digitalocean_genai_sdk → do_gradientai}/types/agents/knowledge_base_detach_response.py +0 -0
  189. {digitalocean_genai_sdk → do_gradientai}/types/agents/version_list_params.py +0 -0
  190. {digitalocean_genai_sdk → do_gradientai}/types/agents/version_update_params.py +0 -0
  191. {digitalocean_genai_sdk → do_gradientai}/types/agents/version_update_response.py +0 -0
  192. {digitalocean_genai_sdk → do_gradientai}/types/api_agent_api_key_info.py +0 -0
  193. {digitalocean_genai_sdk → do_gradientai}/types/api_agreement.py +0 -0
  194. {digitalocean_genai_sdk → do_gradientai}/types/api_anthropic_api_key_info.py +0 -0
  195. {digitalocean_genai_sdk → do_gradientai}/types/api_deployment_visibility.py +0 -0
  196. {digitalocean_genai_sdk → do_gradientai}/types/api_model_version.py +0 -0
  197. {digitalocean_genai_sdk → do_gradientai}/types/api_retrieval_method.py +0 -0
  198. {digitalocean_genai_sdk/types/api_keys → do_gradientai/types/inference}/__init__.py +0 -0
  199. {digitalocean_genai_sdk/types/api_keys → do_gradientai/types/inference}/api_key_create_params.py +0 -0
  200. {digitalocean_genai_sdk/types/api_keys → do_gradientai/types/inference}/api_key_create_response.py +0 -0
  201. {digitalocean_genai_sdk/types/api_keys → do_gradientai/types/inference}/api_key_delete_response.py +0 -0
  202. {digitalocean_genai_sdk/types/api_keys → do_gradientai/types/inference}/api_key_list_params.py +0 -0
  203. {digitalocean_genai_sdk/types/api_keys → do_gradientai/types/inference}/api_key_update_params.py +0 -0
  204. {digitalocean_genai_sdk/types/api_keys → do_gradientai/types/inference}/api_key_update_regenerate_response.py +0 -0
  205. {digitalocean_genai_sdk/types/api_keys → do_gradientai/types/inference}/api_key_update_response.py +0 -0
  206. {digitalocean_genai_sdk/types/api_keys → do_gradientai/types/inference}/api_model_api_key_info.py +0 -0
  207. {digitalocean_genai_sdk → do_gradientai}/types/knowledge_base_create_response.py +0 -0
  208. {digitalocean_genai_sdk → do_gradientai}/types/knowledge_base_delete_response.py +0 -0
  209. {digitalocean_genai_sdk → do_gradientai}/types/knowledge_base_list_params.py +0 -0
  210. {digitalocean_genai_sdk → do_gradientai}/types/knowledge_base_retrieve_response.py +0 -0
  211. {digitalocean_genai_sdk → do_gradientai}/types/knowledge_base_update_params.py +0 -0
  212. {digitalocean_genai_sdk → do_gradientai}/types/knowledge_base_update_response.py +0 -0
  213. {digitalocean_genai_sdk → do_gradientai}/types/knowledge_bases/api_file_upload_data_source.py +0 -0
  214. {digitalocean_genai_sdk → do_gradientai}/types/knowledge_bases/api_file_upload_data_source_param.py +0 -0
  215. {digitalocean_genai_sdk → do_gradientai}/types/knowledge_bases/api_spaces_data_source.py +0 -0
  216. {digitalocean_genai_sdk → do_gradientai}/types/knowledge_bases/api_spaces_data_source_param.py +0 -0
  217. {digitalocean_genai_sdk → do_gradientai}/types/knowledge_bases/api_web_crawler_data_source.py +0 -0
  218. {digitalocean_genai_sdk → do_gradientai}/types/knowledge_bases/api_web_crawler_data_source_param.py +0 -0
  219. {digitalocean_genai_sdk → do_gradientai}/types/knowledge_bases/data_source_create_response.py +0 -0
  220. {digitalocean_genai_sdk → do_gradientai}/types/knowledge_bases/data_source_delete_response.py +0 -0
  221. {digitalocean_genai_sdk → do_gradientai}/types/knowledge_bases/data_source_list_params.py +0 -0
  222. {digitalocean_genai_sdk/types → do_gradientai/types/knowledge_bases}/indexing_job_create_params.py +0 -0
  223. {digitalocean_genai_sdk/types → do_gradientai/types/knowledge_bases}/indexing_job_list_params.py +0 -0
  224. {digitalocean_genai_sdk → do_gradientai}/types/model.py +0 -0
  225. {digitalocean_genai_sdk → do_gradientai}/types/model_list_response.py +0 -0
  226. {digitalocean_genai_sdk/types/auth → do_gradientai/types/model_providers}/__init__.py +0 -0
  227. {digitalocean_genai_sdk/types/providers → do_gradientai/types/model_providers}/anthropic/__init__.py +0 -0
  228. {digitalocean_genai_sdk/types/providers → do_gradientai/types/model_providers}/anthropic/key_create_params.py +0 -0
  229. {digitalocean_genai_sdk/types/providers → do_gradientai/types/model_providers}/anthropic/key_create_response.py +0 -0
  230. {digitalocean_genai_sdk/types/providers → do_gradientai/types/model_providers}/anthropic/key_delete_response.py +0 -0
  231. {digitalocean_genai_sdk/types/providers → do_gradientai/types/model_providers}/anthropic/key_list_agents_params.py +0 -0
  232. {digitalocean_genai_sdk/types/providers → do_gradientai/types/model_providers}/anthropic/key_list_params.py +0 -0
  233. {digitalocean_genai_sdk/types/providers → do_gradientai/types/model_providers}/anthropic/key_retrieve_response.py +0 -0
  234. {digitalocean_genai_sdk/types/providers → do_gradientai/types/model_providers}/anthropic/key_update_params.py +0 -0
  235. {digitalocean_genai_sdk/types/providers → do_gradientai/types/model_providers}/anthropic/key_update_response.py +0 -0
  236. {digitalocean_genai_sdk/types/providers → do_gradientai/types/model_providers}/openai/__init__.py +0 -0
  237. {digitalocean_genai_sdk/types/providers → do_gradientai/types/model_providers}/openai/key_create_params.py +0 -0
  238. {digitalocean_genai_sdk/types/providers → do_gradientai/types/model_providers}/openai/key_create_response.py +0 -0
  239. {digitalocean_genai_sdk/types/providers → do_gradientai/types/model_providers}/openai/key_delete_response.py +0 -0
  240. {digitalocean_genai_sdk/types/providers → do_gradientai/types/model_providers}/openai/key_list_params.py +0 -0
  241. {digitalocean_genai_sdk/types/providers → do_gradientai/types/model_providers}/openai/key_retrieve_agents_params.py +0 -0
  242. {digitalocean_genai_sdk/types/providers → do_gradientai/types/model_providers}/openai/key_retrieve_response.py +0 -0
  243. {digitalocean_genai_sdk/types/providers → do_gradientai/types/model_providers}/openai/key_update_params.py +0 -0
  244. {digitalocean_genai_sdk/types/providers → do_gradientai/types/model_providers}/openai/key_update_response.py +0 -0
  245. {digitalocean_genai_sdk → do_gradientai}/types/region_list_params.py +0 -0
  246. {digitalocean_genai_sdk → do_gradientai}/types/region_list_response.py +0 -0
  247. {digitalocean_genai_sdk/types/agents → do_gradientai/types/shared}/api_links.py +0 -0
  248. {digitalocean_genai_sdk/types/agents → do_gradientai/types/shared}/api_meta.py +0 -0
@@ -15,12 +15,12 @@ from ..._response import (
15
15
  async_to_streamed_response_wrapper,
16
16
  )
17
17
  from ..._base_client import make_request_options
18
- from ...types.api_keys import api_key_list_params, api_key_create_params, api_key_update_params
19
- from ...types.api_keys.api_key_list_response import APIKeyListResponse
20
- from ...types.api_keys.api_key_create_response import APIKeyCreateResponse
21
- from ...types.api_keys.api_key_delete_response import APIKeyDeleteResponse
22
- from ...types.api_keys.api_key_update_response import APIKeyUpdateResponse
23
- from ...types.api_keys.api_key_update_regenerate_response import APIKeyUpdateRegenerateResponse
18
+ from ...types.inference import api_key_list_params, api_key_create_params, api_key_update_params
19
+ from ...types.inference.api_key_list_response import APIKeyListResponse
20
+ from ...types.inference.api_key_create_response import APIKeyCreateResponse
21
+ from ...types.inference.api_key_delete_response import APIKeyDeleteResponse
22
+ from ...types.inference.api_key_update_response import APIKeyUpdateResponse
23
+ from ...types.inference.api_key_update_regenerate_response import APIKeyUpdateRegenerateResponse
24
24
 
25
25
  __all__ = ["APIKeysResource", "AsyncAPIKeysResource"]
26
26
 
@@ -32,7 +32,7 @@ class APIKeysResource(SyncAPIResource):
32
32
  This property can be used as a prefix for any HTTP method call to return
33
33
  the raw response object instead of the parsed content.
34
34
 
35
- For more information, see https://www.github.com/digitalocean/genai-python#accessing-raw-response-data-eg-headers
35
+ For more information, see https://www.github.com/digitalocean/gradientai-python#accessing-raw-response-data-eg-headers
36
36
  """
37
37
  return APIKeysResourceWithRawResponse(self)
38
38
 
@@ -41,7 +41,7 @@ class APIKeysResource(SyncAPIResource):
41
41
  """
42
42
  An alternative to `.with_raw_response` that doesn't eagerly read the response body.
43
43
 
44
- For more information, see https://www.github.com/digitalocean/genai-python#with_streaming_response
44
+ For more information, see https://www.github.com/digitalocean/gradientai-python#with_streaming_response
45
45
  """
46
46
  return APIKeysResourceWithStreamingResponse(self)
47
47
 
@@ -69,7 +69,9 @@ class APIKeysResource(SyncAPIResource):
69
69
  timeout: Override the client-level default timeout for this request, in seconds
70
70
  """
71
71
  return self._post(
72
- "/v2/genai/models/api_keys",
72
+ "/v2/gen-ai/models/api_keys"
73
+ if self._client._base_url_overridden
74
+ else "https://api.digitalocean.com/v2/gen-ai/models/api_keys",
73
75
  body=maybe_transform({"name": name}, api_key_create_params.APIKeyCreateParams),
74
76
  options=make_request_options(
75
77
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
@@ -106,7 +108,9 @@ class APIKeysResource(SyncAPIResource):
106
108
  if not path_api_key_uuid:
107
109
  raise ValueError(f"Expected a non-empty value for `path_api_key_uuid` but received {path_api_key_uuid!r}")
108
110
  return self._put(
109
- f"/v2/genai/models/api_keys/{path_api_key_uuid}",
111
+ f"/v2/gen-ai/models/api_keys/{path_api_key_uuid}"
112
+ if self._client._base_url_overridden
113
+ else f"https://api.digitalocean.com/v2/gen-ai/models/api_keys/{path_api_key_uuid}",
110
114
  body=maybe_transform(
111
115
  {
112
116
  "body_api_key_uuid": body_api_key_uuid,
@@ -149,7 +153,9 @@ class APIKeysResource(SyncAPIResource):
149
153
  timeout: Override the client-level default timeout for this request, in seconds
150
154
  """
151
155
  return self._get(
152
- "/v2/genai/models/api_keys",
156
+ "/v2/gen-ai/models/api_keys"
157
+ if self._client._base_url_overridden
158
+ else "https://api.digitalocean.com/v2/gen-ai/models/api_keys",
153
159
  options=make_request_options(
154
160
  extra_headers=extra_headers,
155
161
  extra_query=extra_query,
@@ -193,7 +199,9 @@ class APIKeysResource(SyncAPIResource):
193
199
  if not api_key_uuid:
194
200
  raise ValueError(f"Expected a non-empty value for `api_key_uuid` but received {api_key_uuid!r}")
195
201
  return self._delete(
196
- f"/v2/genai/models/api_keys/{api_key_uuid}",
202
+ f"/v2/gen-ai/models/api_keys/{api_key_uuid}"
203
+ if self._client._base_url_overridden
204
+ else f"https://api.digitalocean.com/v2/gen-ai/models/api_keys/{api_key_uuid}",
197
205
  options=make_request_options(
198
206
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
199
207
  ),
@@ -227,7 +235,9 @@ class APIKeysResource(SyncAPIResource):
227
235
  if not api_key_uuid:
228
236
  raise ValueError(f"Expected a non-empty value for `api_key_uuid` but received {api_key_uuid!r}")
229
237
  return self._put(
230
- f"/v2/genai/models/api_keys/{api_key_uuid}/regenerate",
238
+ f"/v2/gen-ai/models/api_keys/{api_key_uuid}/regenerate"
239
+ if self._client._base_url_overridden
240
+ else f"https://api.digitalocean.com/v2/gen-ai/models/api_keys/{api_key_uuid}/regenerate",
231
241
  options=make_request_options(
232
242
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
233
243
  ),
@@ -242,7 +252,7 @@ class AsyncAPIKeysResource(AsyncAPIResource):
242
252
  This property can be used as a prefix for any HTTP method call to return
243
253
  the raw response object instead of the parsed content.
244
254
 
245
- For more information, see https://www.github.com/digitalocean/genai-python#accessing-raw-response-data-eg-headers
255
+ For more information, see https://www.github.com/digitalocean/gradientai-python#accessing-raw-response-data-eg-headers
246
256
  """
247
257
  return AsyncAPIKeysResourceWithRawResponse(self)
248
258
 
@@ -251,7 +261,7 @@ class AsyncAPIKeysResource(AsyncAPIResource):
251
261
  """
252
262
  An alternative to `.with_raw_response` that doesn't eagerly read the response body.
253
263
 
254
- For more information, see https://www.github.com/digitalocean/genai-python#with_streaming_response
264
+ For more information, see https://www.github.com/digitalocean/gradientai-python#with_streaming_response
255
265
  """
256
266
  return AsyncAPIKeysResourceWithStreamingResponse(self)
257
267
 
@@ -279,7 +289,9 @@ class AsyncAPIKeysResource(AsyncAPIResource):
279
289
  timeout: Override the client-level default timeout for this request, in seconds
280
290
  """
281
291
  return await self._post(
282
- "/v2/genai/models/api_keys",
292
+ "/v2/gen-ai/models/api_keys"
293
+ if self._client._base_url_overridden
294
+ else "https://api.digitalocean.com/v2/gen-ai/models/api_keys",
283
295
  body=await async_maybe_transform({"name": name}, api_key_create_params.APIKeyCreateParams),
284
296
  options=make_request_options(
285
297
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
@@ -316,7 +328,9 @@ class AsyncAPIKeysResource(AsyncAPIResource):
316
328
  if not path_api_key_uuid:
317
329
  raise ValueError(f"Expected a non-empty value for `path_api_key_uuid` but received {path_api_key_uuid!r}")
318
330
  return await self._put(
319
- f"/v2/genai/models/api_keys/{path_api_key_uuid}",
331
+ f"/v2/gen-ai/models/api_keys/{path_api_key_uuid}"
332
+ if self._client._base_url_overridden
333
+ else f"https://api.digitalocean.com/v2/gen-ai/models/api_keys/{path_api_key_uuid}",
320
334
  body=await async_maybe_transform(
321
335
  {
322
336
  "body_api_key_uuid": body_api_key_uuid,
@@ -359,7 +373,9 @@ class AsyncAPIKeysResource(AsyncAPIResource):
359
373
  timeout: Override the client-level default timeout for this request, in seconds
360
374
  """
361
375
  return await self._get(
362
- "/v2/genai/models/api_keys",
376
+ "/v2/gen-ai/models/api_keys"
377
+ if self._client._base_url_overridden
378
+ else "https://api.digitalocean.com/v2/gen-ai/models/api_keys",
363
379
  options=make_request_options(
364
380
  extra_headers=extra_headers,
365
381
  extra_query=extra_query,
@@ -403,7 +419,9 @@ class AsyncAPIKeysResource(AsyncAPIResource):
403
419
  if not api_key_uuid:
404
420
  raise ValueError(f"Expected a non-empty value for `api_key_uuid` but received {api_key_uuid!r}")
405
421
  return await self._delete(
406
- f"/v2/genai/models/api_keys/{api_key_uuid}",
422
+ f"/v2/gen-ai/models/api_keys/{api_key_uuid}"
423
+ if self._client._base_url_overridden
424
+ else f"https://api.digitalocean.com/v2/gen-ai/models/api_keys/{api_key_uuid}",
407
425
  options=make_request_options(
408
426
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
409
427
  ),
@@ -437,7 +455,9 @@ class AsyncAPIKeysResource(AsyncAPIResource):
437
455
  if not api_key_uuid:
438
456
  raise ValueError(f"Expected a non-empty value for `api_key_uuid` but received {api_key_uuid!r}")
439
457
  return await self._put(
440
- f"/v2/genai/models/api_keys/{api_key_uuid}/regenerate",
458
+ f"/v2/gen-ai/models/api_keys/{api_key_uuid}/regenerate"
459
+ if self._client._base_url_overridden
460
+ else f"https://api.digitalocean.com/v2/gen-ai/models/api_keys/{api_key_uuid}/regenerate",
441
461
  options=make_request_options(
442
462
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
443
463
  ),
@@ -0,0 +1,102 @@
1
+ # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2
+
3
+ from __future__ import annotations
4
+
5
+ from .api_keys import (
6
+ APIKeysResource,
7
+ AsyncAPIKeysResource,
8
+ APIKeysResourceWithRawResponse,
9
+ AsyncAPIKeysResourceWithRawResponse,
10
+ APIKeysResourceWithStreamingResponse,
11
+ AsyncAPIKeysResourceWithStreamingResponse,
12
+ )
13
+ from ..._compat import cached_property
14
+ from ..._resource import SyncAPIResource, AsyncAPIResource
15
+
16
+ __all__ = ["InferenceResource", "AsyncInferenceResource"]
17
+
18
+
19
+ class InferenceResource(SyncAPIResource):
20
+ @cached_property
21
+ def api_keys(self) -> APIKeysResource:
22
+ return APIKeysResource(self._client)
23
+
24
+ @cached_property
25
+ def with_raw_response(self) -> InferenceResourceWithRawResponse:
26
+ """
27
+ This property can be used as a prefix for any HTTP method call to return
28
+ the raw response object instead of the parsed content.
29
+
30
+ For more information, see https://www.github.com/digitalocean/gradientai-python#accessing-raw-response-data-eg-headers
31
+ """
32
+ return InferenceResourceWithRawResponse(self)
33
+
34
+ @cached_property
35
+ def with_streaming_response(self) -> InferenceResourceWithStreamingResponse:
36
+ """
37
+ An alternative to `.with_raw_response` that doesn't eagerly read the response body.
38
+
39
+ For more information, see https://www.github.com/digitalocean/gradientai-python#with_streaming_response
40
+ """
41
+ return InferenceResourceWithStreamingResponse(self)
42
+
43
+
44
+ class AsyncInferenceResource(AsyncAPIResource):
45
+ @cached_property
46
+ def api_keys(self) -> AsyncAPIKeysResource:
47
+ return AsyncAPIKeysResource(self._client)
48
+
49
+ @cached_property
50
+ def with_raw_response(self) -> AsyncInferenceResourceWithRawResponse:
51
+ """
52
+ This property can be used as a prefix for any HTTP method call to return
53
+ the raw response object instead of the parsed content.
54
+
55
+ For more information, see https://www.github.com/digitalocean/gradientai-python#accessing-raw-response-data-eg-headers
56
+ """
57
+ return AsyncInferenceResourceWithRawResponse(self)
58
+
59
+ @cached_property
60
+ def with_streaming_response(self) -> AsyncInferenceResourceWithStreamingResponse:
61
+ """
62
+ An alternative to `.with_raw_response` that doesn't eagerly read the response body.
63
+
64
+ For more information, see https://www.github.com/digitalocean/gradientai-python#with_streaming_response
65
+ """
66
+ return AsyncInferenceResourceWithStreamingResponse(self)
67
+
68
+
69
+ class InferenceResourceWithRawResponse:
70
+ def __init__(self, inference: InferenceResource) -> None:
71
+ self._inference = inference
72
+
73
+ @cached_property
74
+ def api_keys(self) -> APIKeysResourceWithRawResponse:
75
+ return APIKeysResourceWithRawResponse(self._inference.api_keys)
76
+
77
+
78
+ class AsyncInferenceResourceWithRawResponse:
79
+ def __init__(self, inference: AsyncInferenceResource) -> None:
80
+ self._inference = inference
81
+
82
+ @cached_property
83
+ def api_keys(self) -> AsyncAPIKeysResourceWithRawResponse:
84
+ return AsyncAPIKeysResourceWithRawResponse(self._inference.api_keys)
85
+
86
+
87
+ class InferenceResourceWithStreamingResponse:
88
+ def __init__(self, inference: InferenceResource) -> None:
89
+ self._inference = inference
90
+
91
+ @cached_property
92
+ def api_keys(self) -> APIKeysResourceWithStreamingResponse:
93
+ return APIKeysResourceWithStreamingResponse(self._inference.api_keys)
94
+
95
+
96
+ class AsyncInferenceResourceWithStreamingResponse:
97
+ def __init__(self, inference: AsyncInferenceResource) -> None:
98
+ self._inference = inference
99
+
100
+ @cached_property
101
+ def api_keys(self) -> AsyncAPIKeysResourceWithStreamingResponse:
102
+ return AsyncAPIKeysResourceWithStreamingResponse(self._inference.api_keys)
@@ -8,6 +8,14 @@ from .data_sources import (
8
8
  DataSourcesResourceWithStreamingResponse,
9
9
  AsyncDataSourcesResourceWithStreamingResponse,
10
10
  )
11
+ from .indexing_jobs import (
12
+ IndexingJobsResource,
13
+ AsyncIndexingJobsResource,
14
+ IndexingJobsResourceWithRawResponse,
15
+ AsyncIndexingJobsResourceWithRawResponse,
16
+ IndexingJobsResourceWithStreamingResponse,
17
+ AsyncIndexingJobsResourceWithStreamingResponse,
18
+ )
11
19
  from .knowledge_bases import (
12
20
  KnowledgeBasesResource,
13
21
  AsyncKnowledgeBasesResource,
@@ -24,6 +32,12 @@ __all__ = [
24
32
  "AsyncDataSourcesResourceWithRawResponse",
25
33
  "DataSourcesResourceWithStreamingResponse",
26
34
  "AsyncDataSourcesResourceWithStreamingResponse",
35
+ "IndexingJobsResource",
36
+ "AsyncIndexingJobsResource",
37
+ "IndexingJobsResourceWithRawResponse",
38
+ "AsyncIndexingJobsResourceWithRawResponse",
39
+ "IndexingJobsResourceWithStreamingResponse",
40
+ "AsyncIndexingJobsResourceWithStreamingResponse",
27
41
  "KnowledgeBasesResource",
28
42
  "AsyncKnowledgeBasesResource",
29
43
  "KnowledgeBasesResourceWithRawResponse",
@@ -19,6 +19,7 @@ from ...types.knowledge_bases import (
19
19
  data_source_list_params,
20
20
  data_source_create_params,
21
21
  )
22
+ from ...types.knowledge_bases.aws_data_source_param import AwsDataSourceParam
22
23
  from ...types.knowledge_bases.data_source_list_response import DataSourceListResponse
23
24
  from ...types.knowledge_bases.data_source_create_response import DataSourceCreateResponse
24
25
  from ...types.knowledge_bases.data_source_delete_response import DataSourceDeleteResponse
@@ -35,7 +36,7 @@ class DataSourcesResource(SyncAPIResource):
35
36
  This property can be used as a prefix for any HTTP method call to return
36
37
  the raw response object instead of the parsed content.
37
38
 
38
- For more information, see https://www.github.com/digitalocean/genai-python#accessing-raw-response-data-eg-headers
39
+ For more information, see https://www.github.com/digitalocean/gradientai-python#accessing-raw-response-data-eg-headers
39
40
  """
40
41
  return DataSourcesResourceWithRawResponse(self)
41
42
 
@@ -44,7 +45,7 @@ class DataSourcesResource(SyncAPIResource):
44
45
  """
45
46
  An alternative to `.with_raw_response` that doesn't eagerly read the response body.
46
47
 
47
- For more information, see https://www.github.com/digitalocean/genai-python#with_streaming_response
48
+ For more information, see https://www.github.com/digitalocean/gradientai-python#with_streaming_response
48
49
  """
49
50
  return DataSourcesResourceWithStreamingResponse(self)
50
51
 
@@ -52,7 +53,7 @@ class DataSourcesResource(SyncAPIResource):
52
53
  self,
53
54
  path_knowledge_base_uuid: str,
54
55
  *,
55
- aws_data_source: data_source_create_params.AwsDataSource | NotGiven = NOT_GIVEN,
56
+ aws_data_source: AwsDataSourceParam | NotGiven = NOT_GIVEN,
56
57
  body_knowledge_base_uuid: str | NotGiven = NOT_GIVEN,
57
58
  spaces_data_source: APISpacesDataSourceParam | NotGiven = NOT_GIVEN,
58
59
  web_crawler_data_source: APIWebCrawlerDataSourceParam | NotGiven = NOT_GIVEN,
@@ -81,7 +82,9 @@ class DataSourcesResource(SyncAPIResource):
81
82
  f"Expected a non-empty value for `path_knowledge_base_uuid` but received {path_knowledge_base_uuid!r}"
82
83
  )
83
84
  return self._post(
84
- f"/v2/genai/knowledge_bases/{path_knowledge_base_uuid}/data_sources",
85
+ f"/v2/gen-ai/knowledge_bases/{path_knowledge_base_uuid}/data_sources"
86
+ if self._client._base_url_overridden
87
+ else f"https://api.digitalocean.com/v2/gen-ai/knowledge_bases/{path_knowledge_base_uuid}/data_sources",
85
88
  body=maybe_transform(
86
89
  {
87
90
  "aws_data_source": aws_data_source,
@@ -132,7 +135,9 @@ class DataSourcesResource(SyncAPIResource):
132
135
  f"Expected a non-empty value for `knowledge_base_uuid` but received {knowledge_base_uuid!r}"
133
136
  )
134
137
  return self._get(
135
- f"/v2/genai/knowledge_bases/{knowledge_base_uuid}/data_sources",
138
+ f"/v2/gen-ai/knowledge_bases/{knowledge_base_uuid}/data_sources"
139
+ if self._client._base_url_overridden
140
+ else f"https://api.digitalocean.com/v2/gen-ai/knowledge_bases/{knowledge_base_uuid}/data_sources",
136
141
  options=make_request_options(
137
142
  extra_headers=extra_headers,
138
143
  extra_query=extra_query,
@@ -181,7 +186,9 @@ class DataSourcesResource(SyncAPIResource):
181
186
  if not data_source_uuid:
182
187
  raise ValueError(f"Expected a non-empty value for `data_source_uuid` but received {data_source_uuid!r}")
183
188
  return self._delete(
184
- f"/v2/genai/knowledge_bases/{knowledge_base_uuid}/data_sources/{data_source_uuid}",
189
+ f"/v2/gen-ai/knowledge_bases/{knowledge_base_uuid}/data_sources/{data_source_uuid}"
190
+ if self._client._base_url_overridden
191
+ else f"https://api.digitalocean.com/v2/gen-ai/knowledge_bases/{knowledge_base_uuid}/data_sources/{data_source_uuid}",
185
192
  options=make_request_options(
186
193
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
187
194
  ),
@@ -196,7 +203,7 @@ class AsyncDataSourcesResource(AsyncAPIResource):
196
203
  This property can be used as a prefix for any HTTP method call to return
197
204
  the raw response object instead of the parsed content.
198
205
 
199
- For more information, see https://www.github.com/digitalocean/genai-python#accessing-raw-response-data-eg-headers
206
+ For more information, see https://www.github.com/digitalocean/gradientai-python#accessing-raw-response-data-eg-headers
200
207
  """
201
208
  return AsyncDataSourcesResourceWithRawResponse(self)
202
209
 
@@ -205,7 +212,7 @@ class AsyncDataSourcesResource(AsyncAPIResource):
205
212
  """
206
213
  An alternative to `.with_raw_response` that doesn't eagerly read the response body.
207
214
 
208
- For more information, see https://www.github.com/digitalocean/genai-python#with_streaming_response
215
+ For more information, see https://www.github.com/digitalocean/gradientai-python#with_streaming_response
209
216
  """
210
217
  return AsyncDataSourcesResourceWithStreamingResponse(self)
211
218
 
@@ -213,7 +220,7 @@ class AsyncDataSourcesResource(AsyncAPIResource):
213
220
  self,
214
221
  path_knowledge_base_uuid: str,
215
222
  *,
216
- aws_data_source: data_source_create_params.AwsDataSource | NotGiven = NOT_GIVEN,
223
+ aws_data_source: AwsDataSourceParam | NotGiven = NOT_GIVEN,
217
224
  body_knowledge_base_uuid: str | NotGiven = NOT_GIVEN,
218
225
  spaces_data_source: APISpacesDataSourceParam | NotGiven = NOT_GIVEN,
219
226
  web_crawler_data_source: APIWebCrawlerDataSourceParam | NotGiven = NOT_GIVEN,
@@ -242,7 +249,9 @@ class AsyncDataSourcesResource(AsyncAPIResource):
242
249
  f"Expected a non-empty value for `path_knowledge_base_uuid` but received {path_knowledge_base_uuid!r}"
243
250
  )
244
251
  return await self._post(
245
- f"/v2/genai/knowledge_bases/{path_knowledge_base_uuid}/data_sources",
252
+ f"/v2/gen-ai/knowledge_bases/{path_knowledge_base_uuid}/data_sources"
253
+ if self._client._base_url_overridden
254
+ else f"https://api.digitalocean.com/v2/gen-ai/knowledge_bases/{path_knowledge_base_uuid}/data_sources",
246
255
  body=await async_maybe_transform(
247
256
  {
248
257
  "aws_data_source": aws_data_source,
@@ -293,7 +302,9 @@ class AsyncDataSourcesResource(AsyncAPIResource):
293
302
  f"Expected a non-empty value for `knowledge_base_uuid` but received {knowledge_base_uuid!r}"
294
303
  )
295
304
  return await self._get(
296
- f"/v2/genai/knowledge_bases/{knowledge_base_uuid}/data_sources",
305
+ f"/v2/gen-ai/knowledge_bases/{knowledge_base_uuid}/data_sources"
306
+ if self._client._base_url_overridden
307
+ else f"https://api.digitalocean.com/v2/gen-ai/knowledge_bases/{knowledge_base_uuid}/data_sources",
297
308
  options=make_request_options(
298
309
  extra_headers=extra_headers,
299
310
  extra_query=extra_query,
@@ -342,7 +353,9 @@ class AsyncDataSourcesResource(AsyncAPIResource):
342
353
  if not data_source_uuid:
343
354
  raise ValueError(f"Expected a non-empty value for `data_source_uuid` but received {data_source_uuid!r}")
344
355
  return await self._delete(
345
- f"/v2/genai/knowledge_bases/{knowledge_base_uuid}/data_sources/{data_source_uuid}",
356
+ f"/v2/gen-ai/knowledge_bases/{knowledge_base_uuid}/data_sources/{data_source_uuid}"
357
+ if self._client._base_url_overridden
358
+ else f"https://api.digitalocean.com/v2/gen-ai/knowledge_bases/{knowledge_base_uuid}/data_sources/{data_source_uuid}",
346
359
  options=make_request_options(
347
360
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
348
361
  ),
@@ -6,23 +6,27 @@ from typing import List
6
6
 
7
7
  import httpx
8
8
 
9
- from ..types import indexing_job_list_params, indexing_job_create_params, indexing_job_update_cancel_params
10
- from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
11
- from .._utils import maybe_transform, async_maybe_transform
12
- from .._compat import cached_property
13
- from .._resource import SyncAPIResource, AsyncAPIResource
14
- from .._response import (
9
+ from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven
10
+ from ..._utils import maybe_transform, async_maybe_transform
11
+ from ..._compat import cached_property
12
+ from ..._resource import SyncAPIResource, AsyncAPIResource
13
+ from ..._response import (
15
14
  to_raw_response_wrapper,
16
15
  to_streamed_response_wrapper,
17
16
  async_to_raw_response_wrapper,
18
17
  async_to_streamed_response_wrapper,
19
18
  )
20
- from .._base_client import make_request_options
21
- from ..types.indexing_job_list_response import IndexingJobListResponse
22
- from ..types.indexing_job_create_response import IndexingJobCreateResponse
23
- from ..types.indexing_job_retrieve_response import IndexingJobRetrieveResponse
24
- from ..types.indexing_job_update_cancel_response import IndexingJobUpdateCancelResponse
25
- from ..types.indexing_job_retrieve_data_sources_response import IndexingJobRetrieveDataSourcesResponse
19
+ from ..._base_client import make_request_options
20
+ from ...types.knowledge_bases import (
21
+ indexing_job_list_params,
22
+ indexing_job_create_params,
23
+ indexing_job_update_cancel_params,
24
+ )
25
+ from ...types.knowledge_bases.indexing_job_list_response import IndexingJobListResponse
26
+ from ...types.knowledge_bases.indexing_job_create_response import IndexingJobCreateResponse
27
+ from ...types.knowledge_bases.indexing_job_retrieve_response import IndexingJobRetrieveResponse
28
+ from ...types.knowledge_bases.indexing_job_update_cancel_response import IndexingJobUpdateCancelResponse
29
+ from ...types.knowledge_bases.indexing_job_retrieve_data_sources_response import IndexingJobRetrieveDataSourcesResponse
26
30
 
27
31
  __all__ = ["IndexingJobsResource", "AsyncIndexingJobsResource"]
28
32
 
@@ -34,7 +38,7 @@ class IndexingJobsResource(SyncAPIResource):
34
38
  This property can be used as a prefix for any HTTP method call to return
35
39
  the raw response object instead of the parsed content.
36
40
 
37
- For more information, see https://www.github.com/digitalocean/genai-python#accessing-raw-response-data-eg-headers
41
+ For more information, see https://www.github.com/digitalocean/gradientai-python#accessing-raw-response-data-eg-headers
38
42
  """
39
43
  return IndexingJobsResourceWithRawResponse(self)
40
44
 
@@ -43,7 +47,7 @@ class IndexingJobsResource(SyncAPIResource):
43
47
  """
44
48
  An alternative to `.with_raw_response` that doesn't eagerly read the response body.
45
49
 
46
- For more information, see https://www.github.com/digitalocean/genai-python#with_streaming_response
50
+ For more information, see https://www.github.com/digitalocean/gradientai-python#with_streaming_response
47
51
  """
48
52
  return IndexingJobsResourceWithStreamingResponse(self)
49
53
 
@@ -73,7 +77,9 @@ class IndexingJobsResource(SyncAPIResource):
73
77
  timeout: Override the client-level default timeout for this request, in seconds
74
78
  """
75
79
  return self._post(
76
- "/v2/genai/indexing_jobs",
80
+ "/v2/gen-ai/indexing_jobs"
81
+ if self._client._base_url_overridden
82
+ else "https://api.digitalocean.com/v2/gen-ai/indexing_jobs",
77
83
  body=maybe_transform(
78
84
  {
79
85
  "data_source_uuids": data_source_uuids,
@@ -114,7 +120,9 @@ class IndexingJobsResource(SyncAPIResource):
114
120
  if not uuid:
115
121
  raise ValueError(f"Expected a non-empty value for `uuid` but received {uuid!r}")
116
122
  return self._get(
117
- f"/v2/genai/indexing_jobs/{uuid}",
123
+ f"/v2/gen-ai/indexing_jobs/{uuid}"
124
+ if self._client._base_url_overridden
125
+ else f"https://api.digitalocean.com/v2/gen-ai/indexing_jobs/{uuid}",
118
126
  options=make_request_options(
119
127
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
120
128
  ),
@@ -151,7 +159,9 @@ class IndexingJobsResource(SyncAPIResource):
151
159
  timeout: Override the client-level default timeout for this request, in seconds
152
160
  """
153
161
  return self._get(
154
- "/v2/genai/indexing_jobs",
162
+ "/v2/gen-ai/indexing_jobs"
163
+ if self._client._base_url_overridden
164
+ else "https://api.digitalocean.com/v2/gen-ai/indexing_jobs",
155
165
  options=make_request_options(
156
166
  extra_headers=extra_headers,
157
167
  extra_query=extra_query,
@@ -195,7 +205,9 @@ class IndexingJobsResource(SyncAPIResource):
195
205
  if not indexing_job_uuid:
196
206
  raise ValueError(f"Expected a non-empty value for `indexing_job_uuid` but received {indexing_job_uuid!r}")
197
207
  return self._get(
198
- f"/v2/genai/indexing_jobs/{indexing_job_uuid}/data_sources",
208
+ f"/v2/gen-ai/indexing_jobs/{indexing_job_uuid}/data_sources"
209
+ if self._client._base_url_overridden
210
+ else f"https://api.digitalocean.com/v2/gen-ai/indexing_jobs/{indexing_job_uuid}/data_sources",
199
211
  options=make_request_options(
200
212
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
201
213
  ),
@@ -232,7 +244,9 @@ class IndexingJobsResource(SyncAPIResource):
232
244
  if not path_uuid:
233
245
  raise ValueError(f"Expected a non-empty value for `path_uuid` but received {path_uuid!r}")
234
246
  return self._put(
235
- f"/v2/genai/indexing_jobs/{path_uuid}/cancel",
247
+ f"/v2/gen-ai/indexing_jobs/{path_uuid}/cancel"
248
+ if self._client._base_url_overridden
249
+ else f"https://api.digitalocean.com/v2/gen-ai/indexing_jobs/{path_uuid}/cancel",
236
250
  body=maybe_transform(
237
251
  {"body_uuid": body_uuid}, indexing_job_update_cancel_params.IndexingJobUpdateCancelParams
238
252
  ),
@@ -250,7 +264,7 @@ class AsyncIndexingJobsResource(AsyncAPIResource):
250
264
  This property can be used as a prefix for any HTTP method call to return
251
265
  the raw response object instead of the parsed content.
252
266
 
253
- For more information, see https://www.github.com/digitalocean/genai-python#accessing-raw-response-data-eg-headers
267
+ For more information, see https://www.github.com/digitalocean/gradientai-python#accessing-raw-response-data-eg-headers
254
268
  """
255
269
  return AsyncIndexingJobsResourceWithRawResponse(self)
256
270
 
@@ -259,7 +273,7 @@ class AsyncIndexingJobsResource(AsyncAPIResource):
259
273
  """
260
274
  An alternative to `.with_raw_response` that doesn't eagerly read the response body.
261
275
 
262
- For more information, see https://www.github.com/digitalocean/genai-python#with_streaming_response
276
+ For more information, see https://www.github.com/digitalocean/gradientai-python#with_streaming_response
263
277
  """
264
278
  return AsyncIndexingJobsResourceWithStreamingResponse(self)
265
279
 
@@ -289,7 +303,9 @@ class AsyncIndexingJobsResource(AsyncAPIResource):
289
303
  timeout: Override the client-level default timeout for this request, in seconds
290
304
  """
291
305
  return await self._post(
292
- "/v2/genai/indexing_jobs",
306
+ "/v2/gen-ai/indexing_jobs"
307
+ if self._client._base_url_overridden
308
+ else "https://api.digitalocean.com/v2/gen-ai/indexing_jobs",
293
309
  body=await async_maybe_transform(
294
310
  {
295
311
  "data_source_uuids": data_source_uuids,
@@ -330,7 +346,9 @@ class AsyncIndexingJobsResource(AsyncAPIResource):
330
346
  if not uuid:
331
347
  raise ValueError(f"Expected a non-empty value for `uuid` but received {uuid!r}")
332
348
  return await self._get(
333
- f"/v2/genai/indexing_jobs/{uuid}",
349
+ f"/v2/gen-ai/indexing_jobs/{uuid}"
350
+ if self._client._base_url_overridden
351
+ else f"https://api.digitalocean.com/v2/gen-ai/indexing_jobs/{uuid}",
334
352
  options=make_request_options(
335
353
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
336
354
  ),
@@ -367,7 +385,9 @@ class AsyncIndexingJobsResource(AsyncAPIResource):
367
385
  timeout: Override the client-level default timeout for this request, in seconds
368
386
  """
369
387
  return await self._get(
370
- "/v2/genai/indexing_jobs",
388
+ "/v2/gen-ai/indexing_jobs"
389
+ if self._client._base_url_overridden
390
+ else "https://api.digitalocean.com/v2/gen-ai/indexing_jobs",
371
391
  options=make_request_options(
372
392
  extra_headers=extra_headers,
373
393
  extra_query=extra_query,
@@ -411,7 +431,9 @@ class AsyncIndexingJobsResource(AsyncAPIResource):
411
431
  if not indexing_job_uuid:
412
432
  raise ValueError(f"Expected a non-empty value for `indexing_job_uuid` but received {indexing_job_uuid!r}")
413
433
  return await self._get(
414
- f"/v2/genai/indexing_jobs/{indexing_job_uuid}/data_sources",
434
+ f"/v2/gen-ai/indexing_jobs/{indexing_job_uuid}/data_sources"
435
+ if self._client._base_url_overridden
436
+ else f"https://api.digitalocean.com/v2/gen-ai/indexing_jobs/{indexing_job_uuid}/data_sources",
415
437
  options=make_request_options(
416
438
  extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
417
439
  ),
@@ -448,7 +470,9 @@ class AsyncIndexingJobsResource(AsyncAPIResource):
448
470
  if not path_uuid:
449
471
  raise ValueError(f"Expected a non-empty value for `path_uuid` but received {path_uuid!r}")
450
472
  return await self._put(
451
- f"/v2/genai/indexing_jobs/{path_uuid}/cancel",
473
+ f"/v2/gen-ai/indexing_jobs/{path_uuid}/cancel"
474
+ if self._client._base_url_overridden
475
+ else f"https://api.digitalocean.com/v2/gen-ai/indexing_jobs/{path_uuid}/cancel",
452
476
  body=await async_maybe_transform(
453
477
  {"body_uuid": body_uuid}, indexing_job_update_cancel_params.IndexingJobUpdateCancelParams
454
478
  ),