oracle-ads 2.13.9rc0__py3-none-any.whl → 2.13.10__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 (858) hide show
  1. ads/aqua/__init__.py +40 -0
  2. ads/aqua/app.py +507 -0
  3. ads/aqua/cli.py +96 -0
  4. ads/aqua/client/__init__.py +3 -0
  5. ads/aqua/client/client.py +836 -0
  6. ads/aqua/client/openai_client.py +305 -0
  7. ads/aqua/common/__init__.py +5 -0
  8. ads/aqua/common/decorator.py +125 -0
  9. ads/aqua/common/entities.py +274 -0
  10. ads/aqua/common/enums.py +134 -0
  11. ads/aqua/common/errors.py +109 -0
  12. ads/aqua/common/utils.py +1295 -0
  13. ads/aqua/config/__init__.py +4 -0
  14. ads/aqua/config/container_config.py +246 -0
  15. ads/aqua/config/evaluation/__init__.py +4 -0
  16. ads/aqua/config/evaluation/evaluation_service_config.py +147 -0
  17. ads/aqua/config/utils/__init__.py +4 -0
  18. ads/aqua/config/utils/serializer.py +339 -0
  19. ads/aqua/constants.py +116 -0
  20. ads/aqua/data.py +14 -0
  21. ads/aqua/dummy_data/icon.txt +1 -0
  22. ads/aqua/dummy_data/oci_model_deployments.json +56 -0
  23. ads/aqua/dummy_data/oci_models.json +1 -0
  24. ads/aqua/dummy_data/readme.md +26 -0
  25. ads/aqua/evaluation/__init__.py +8 -0
  26. ads/aqua/evaluation/constants.py +53 -0
  27. ads/aqua/evaluation/entities.py +186 -0
  28. ads/aqua/evaluation/errors.py +70 -0
  29. ads/aqua/evaluation/evaluation.py +1814 -0
  30. ads/aqua/extension/__init__.py +42 -0
  31. ads/aqua/extension/aqua_ws_msg_handler.py +76 -0
  32. ads/aqua/extension/base_handler.py +90 -0
  33. ads/aqua/extension/common_handler.py +121 -0
  34. ads/aqua/extension/common_ws_msg_handler.py +36 -0
  35. ads/aqua/extension/deployment_handler.py +381 -0
  36. ads/aqua/extension/deployment_ws_msg_handler.py +54 -0
  37. ads/aqua/extension/errors.py +30 -0
  38. ads/aqua/extension/evaluation_handler.py +129 -0
  39. ads/aqua/extension/evaluation_ws_msg_handler.py +61 -0
  40. ads/aqua/extension/finetune_handler.py +96 -0
  41. ads/aqua/extension/model_handler.py +390 -0
  42. ads/aqua/extension/models/__init__.py +0 -0
  43. ads/aqua/extension/models/ws_models.py +145 -0
  44. ads/aqua/extension/models_ws_msg_handler.py +50 -0
  45. ads/aqua/extension/ui_handler.py +300 -0
  46. ads/aqua/extension/ui_websocket_handler.py +130 -0
  47. ads/aqua/extension/utils.py +133 -0
  48. ads/aqua/finetuning/__init__.py +7 -0
  49. ads/aqua/finetuning/constants.py +23 -0
  50. ads/aqua/finetuning/entities.py +181 -0
  51. ads/aqua/finetuning/finetuning.py +749 -0
  52. ads/aqua/model/__init__.py +8 -0
  53. ads/aqua/model/constants.py +60 -0
  54. ads/aqua/model/entities.py +385 -0
  55. ads/aqua/model/enums.py +32 -0
  56. ads/aqua/model/model.py +2134 -0
  57. ads/aqua/model/utils.py +52 -0
  58. ads/aqua/modeldeployment/__init__.py +6 -0
  59. ads/aqua/modeldeployment/constants.py +10 -0
  60. ads/aqua/modeldeployment/deployment.py +1315 -0
  61. ads/aqua/modeldeployment/entities.py +653 -0
  62. ads/aqua/modeldeployment/utils.py +543 -0
  63. ads/aqua/resources/gpu_shapes_index.json +94 -0
  64. ads/aqua/server/__init__.py +4 -0
  65. ads/aqua/server/__main__.py +24 -0
  66. ads/aqua/server/app.py +47 -0
  67. ads/aqua/server/aqua_spec.yml +1291 -0
  68. ads/aqua/training/__init__.py +4 -0
  69. ads/aqua/training/exceptions.py +476 -0
  70. ads/aqua/ui.py +519 -0
  71. ads/automl/__init__.py +9 -0
  72. ads/automl/driver.py +330 -0
  73. ads/automl/provider.py +975 -0
  74. ads/bds/__init__.py +5 -0
  75. ads/bds/auth.py +127 -0
  76. ads/bds/big_data_service.py +255 -0
  77. ads/catalog/__init__.py +19 -0
  78. ads/catalog/model.py +1576 -0
  79. ads/catalog/notebook.py +461 -0
  80. ads/catalog/project.py +468 -0
  81. ads/catalog/summary.py +178 -0
  82. ads/common/__init__.py +11 -0
  83. ads/common/analyzer.py +65 -0
  84. ads/common/artifact/.model-ignore +63 -0
  85. ads/common/artifact/__init__.py +10 -0
  86. ads/common/auth.py +1122 -0
  87. ads/common/card_identifier.py +83 -0
  88. ads/common/config.py +647 -0
  89. ads/common/data.py +165 -0
  90. ads/common/decorator/__init__.py +9 -0
  91. ads/common/decorator/argument_to_case.py +88 -0
  92. ads/common/decorator/deprecate.py +69 -0
  93. ads/common/decorator/require_nonempty_arg.py +65 -0
  94. ads/common/decorator/runtime_dependency.py +178 -0
  95. ads/common/decorator/threaded.py +97 -0
  96. ads/common/decorator/utils.py +35 -0
  97. ads/common/dsc_file_system.py +303 -0
  98. ads/common/error.py +14 -0
  99. ads/common/extended_enum.py +81 -0
  100. ads/common/function/__init__.py +5 -0
  101. ads/common/function/fn_util.py +142 -0
  102. ads/common/function/func_conf.yaml +25 -0
  103. ads/common/ipython.py +76 -0
  104. ads/common/model.py +679 -0
  105. ads/common/model_artifact.py +1759 -0
  106. ads/common/model_artifact_schema.json +107 -0
  107. ads/common/model_export_util.py +664 -0
  108. ads/common/model_metadata.py +24 -0
  109. ads/common/object_storage_details.py +296 -0
  110. ads/common/oci_client.py +179 -0
  111. ads/common/oci_datascience.py +46 -0
  112. ads/common/oci_logging.py +1144 -0
  113. ads/common/oci_mixin.py +957 -0
  114. ads/common/oci_resource.py +136 -0
  115. ads/common/serializer.py +559 -0
  116. ads/common/utils.py +1852 -0
  117. ads/common/word_lists.py +1491 -0
  118. ads/common/work_request.py +189 -0
  119. ads/config.py +1 -0
  120. ads/data_labeling/__init__.py +13 -0
  121. ads/data_labeling/boundingbox.py +253 -0
  122. ads/data_labeling/constants.py +47 -0
  123. ads/data_labeling/data_labeling_service.py +244 -0
  124. ads/data_labeling/interface/__init__.py +5 -0
  125. ads/data_labeling/interface/loader.py +16 -0
  126. ads/data_labeling/interface/parser.py +16 -0
  127. ads/data_labeling/interface/reader.py +23 -0
  128. ads/data_labeling/loader/__init__.py +5 -0
  129. ads/data_labeling/loader/file_loader.py +241 -0
  130. ads/data_labeling/metadata.py +110 -0
  131. ads/data_labeling/mixin/__init__.py +5 -0
  132. ads/data_labeling/mixin/data_labeling.py +232 -0
  133. ads/data_labeling/ner.py +129 -0
  134. ads/data_labeling/parser/__init__.py +5 -0
  135. ads/data_labeling/parser/dls_record_parser.py +388 -0
  136. ads/data_labeling/parser/export_metadata_parser.py +94 -0
  137. ads/data_labeling/parser/export_record_parser.py +473 -0
  138. ads/data_labeling/reader/__init__.py +5 -0
  139. ads/data_labeling/reader/dataset_reader.py +574 -0
  140. ads/data_labeling/reader/dls_record_reader.py +121 -0
  141. ads/data_labeling/reader/export_record_reader.py +62 -0
  142. ads/data_labeling/reader/jsonl_reader.py +75 -0
  143. ads/data_labeling/reader/metadata_reader.py +203 -0
  144. ads/data_labeling/reader/record_reader.py +263 -0
  145. ads/data_labeling/record.py +52 -0
  146. ads/data_labeling/visualizer/__init__.py +5 -0
  147. ads/data_labeling/visualizer/image_visualizer.py +525 -0
  148. ads/data_labeling/visualizer/text_visualizer.py +357 -0
  149. ads/database/__init__.py +5 -0
  150. ads/database/connection.py +338 -0
  151. ads/dataset/__init__.py +10 -0
  152. ads/dataset/capabilities.md +51 -0
  153. ads/dataset/classification_dataset.py +339 -0
  154. ads/dataset/correlation.py +226 -0
  155. ads/dataset/correlation_plot.py +563 -0
  156. ads/dataset/dask_series.py +173 -0
  157. ads/dataset/dataframe_transformer.py +110 -0
  158. ads/dataset/dataset.py +1979 -0
  159. ads/dataset/dataset_browser.py +360 -0
  160. ads/dataset/dataset_with_target.py +995 -0
  161. ads/dataset/exception.py +25 -0
  162. ads/dataset/factory.py +987 -0
  163. ads/dataset/feature_engineering_transformer.py +35 -0
  164. ads/dataset/feature_selection.py +107 -0
  165. ads/dataset/forecasting_dataset.py +26 -0
  166. ads/dataset/helper.py +1450 -0
  167. ads/dataset/label_encoder.py +99 -0
  168. ads/dataset/mixin/__init__.py +5 -0
  169. ads/dataset/mixin/dataset_accessor.py +134 -0
  170. ads/dataset/pipeline.py +58 -0
  171. ads/dataset/plot.py +710 -0
  172. ads/dataset/progress.py +86 -0
  173. ads/dataset/recommendation.py +297 -0
  174. ads/dataset/recommendation_transformer.py +502 -0
  175. ads/dataset/regression_dataset.py +14 -0
  176. ads/dataset/sampled_dataset.py +1050 -0
  177. ads/dataset/target.py +98 -0
  178. ads/dataset/timeseries.py +18 -0
  179. ads/dbmixin/__init__.py +5 -0
  180. ads/dbmixin/db_pandas_accessor.py +153 -0
  181. ads/environment/__init__.py +9 -0
  182. ads/environment/ml_runtime.py +66 -0
  183. ads/evaluations/README.md +14 -0
  184. ads/evaluations/__init__.py +109 -0
  185. ads/evaluations/evaluation_plot.py +983 -0
  186. ads/evaluations/evaluator.py +1334 -0
  187. ads/evaluations/statistical_metrics.py +543 -0
  188. ads/experiments/__init__.py +9 -0
  189. ads/experiments/capabilities.md +0 -0
  190. ads/explanations/__init__.py +21 -0
  191. ads/explanations/base_explainer.py +142 -0
  192. ads/explanations/capabilities.md +83 -0
  193. ads/explanations/explainer.py +190 -0
  194. ads/explanations/mlx_global_explainer.py +1050 -0
  195. ads/explanations/mlx_interface.py +386 -0
  196. ads/explanations/mlx_local_explainer.py +287 -0
  197. ads/explanations/mlx_whatif_explainer.py +201 -0
  198. ads/feature_engineering/__init__.py +20 -0
  199. ads/feature_engineering/accessor/__init__.py +5 -0
  200. ads/feature_engineering/accessor/dataframe_accessor.py +535 -0
  201. ads/feature_engineering/accessor/mixin/__init__.py +5 -0
  202. ads/feature_engineering/accessor/mixin/correlation.py +166 -0
  203. ads/feature_engineering/accessor/mixin/eda_mixin.py +266 -0
  204. ads/feature_engineering/accessor/mixin/eda_mixin_series.py +85 -0
  205. ads/feature_engineering/accessor/mixin/feature_types_mixin.py +211 -0
  206. ads/feature_engineering/accessor/mixin/utils.py +65 -0
  207. ads/feature_engineering/accessor/series_accessor.py +431 -0
  208. ads/feature_engineering/adsimage/__init__.py +5 -0
  209. ads/feature_engineering/adsimage/image.py +192 -0
  210. ads/feature_engineering/adsimage/image_reader.py +170 -0
  211. ads/feature_engineering/adsimage/interface/__init__.py +5 -0
  212. ads/feature_engineering/adsimage/interface/reader.py +19 -0
  213. ads/feature_engineering/adsstring/__init__.py +7 -0
  214. ads/feature_engineering/adsstring/oci_language/__init__.py +8 -0
  215. ads/feature_engineering/adsstring/string/__init__.py +8 -0
  216. ads/feature_engineering/data_schema.json +57 -0
  217. ads/feature_engineering/dataset/__init__.py +5 -0
  218. ads/feature_engineering/dataset/zip_code_data.py +42062 -0
  219. ads/feature_engineering/exceptions.py +40 -0
  220. ads/feature_engineering/feature_type/__init__.py +133 -0
  221. ads/feature_engineering/feature_type/address.py +184 -0
  222. ads/feature_engineering/feature_type/adsstring/__init__.py +5 -0
  223. ads/feature_engineering/feature_type/adsstring/common_regex_mixin.py +164 -0
  224. ads/feature_engineering/feature_type/adsstring/oci_language.py +93 -0
  225. ads/feature_engineering/feature_type/adsstring/parsers/__init__.py +5 -0
  226. ads/feature_engineering/feature_type/adsstring/parsers/base.py +47 -0
  227. ads/feature_engineering/feature_type/adsstring/parsers/nltk_parser.py +96 -0
  228. ads/feature_engineering/feature_type/adsstring/parsers/spacy_parser.py +221 -0
  229. ads/feature_engineering/feature_type/adsstring/string.py +258 -0
  230. ads/feature_engineering/feature_type/base.py +58 -0
  231. ads/feature_engineering/feature_type/boolean.py +183 -0
  232. ads/feature_engineering/feature_type/category.py +146 -0
  233. ads/feature_engineering/feature_type/constant.py +137 -0
  234. ads/feature_engineering/feature_type/continuous.py +151 -0
  235. ads/feature_engineering/feature_type/creditcard.py +314 -0
  236. ads/feature_engineering/feature_type/datetime.py +190 -0
  237. ads/feature_engineering/feature_type/discrete.py +134 -0
  238. ads/feature_engineering/feature_type/document.py +43 -0
  239. ads/feature_engineering/feature_type/gis.py +251 -0
  240. ads/feature_engineering/feature_type/handler/__init__.py +5 -0
  241. ads/feature_engineering/feature_type/handler/feature_validator.py +524 -0
  242. ads/feature_engineering/feature_type/handler/feature_warning.py +319 -0
  243. ads/feature_engineering/feature_type/handler/warnings.py +128 -0
  244. ads/feature_engineering/feature_type/integer.py +142 -0
  245. ads/feature_engineering/feature_type/ip_address.py +144 -0
  246. ads/feature_engineering/feature_type/ip_address_v4.py +138 -0
  247. ads/feature_engineering/feature_type/ip_address_v6.py +138 -0
  248. ads/feature_engineering/feature_type/lat_long.py +256 -0
  249. ads/feature_engineering/feature_type/object.py +43 -0
  250. ads/feature_engineering/feature_type/ordinal.py +132 -0
  251. ads/feature_engineering/feature_type/phone_number.py +135 -0
  252. ads/feature_engineering/feature_type/string.py +171 -0
  253. ads/feature_engineering/feature_type/text.py +93 -0
  254. ads/feature_engineering/feature_type/unknown.py +43 -0
  255. ads/feature_engineering/feature_type/zip_code.py +164 -0
  256. ads/feature_engineering/feature_type_manager.py +406 -0
  257. ads/feature_engineering/schema.py +795 -0
  258. ads/feature_engineering/utils.py +245 -0
  259. ads/feature_store/.readthedocs.yaml +19 -0
  260. ads/feature_store/README.md +65 -0
  261. ads/feature_store/__init__.py +9 -0
  262. ads/feature_store/common/__init__.py +0 -0
  263. ads/feature_store/common/enums.py +339 -0
  264. ads/feature_store/common/exceptions.py +18 -0
  265. ads/feature_store/common/spark_session_singleton.py +125 -0
  266. ads/feature_store/common/utils/__init__.py +0 -0
  267. ads/feature_store/common/utils/base64_encoder_decoder.py +72 -0
  268. ads/feature_store/common/utils/feature_schema_mapper.py +283 -0
  269. ads/feature_store/common/utils/transformation_utils.py +82 -0
  270. ads/feature_store/common/utils/utility.py +403 -0
  271. ads/feature_store/data_validation/__init__.py +0 -0
  272. ads/feature_store/data_validation/great_expectation.py +129 -0
  273. ads/feature_store/dataset.py +1230 -0
  274. ads/feature_store/dataset_job.py +530 -0
  275. ads/feature_store/docs/Dockerfile +7 -0
  276. ads/feature_store/docs/Makefile +44 -0
  277. ads/feature_store/docs/conf.py +28 -0
  278. ads/feature_store/docs/requirements.txt +14 -0
  279. ads/feature_store/docs/source/ads.feature_store.query.rst +20 -0
  280. ads/feature_store/docs/source/cicd.rst +137 -0
  281. ads/feature_store/docs/source/conf.py +86 -0
  282. ads/feature_store/docs/source/data_versioning.rst +33 -0
  283. ads/feature_store/docs/source/dataset.rst +388 -0
  284. ads/feature_store/docs/source/dataset_job.rst +27 -0
  285. ads/feature_store/docs/source/demo.rst +70 -0
  286. ads/feature_store/docs/source/entity.rst +78 -0
  287. ads/feature_store/docs/source/feature_group.rst +624 -0
  288. ads/feature_store/docs/source/feature_group_job.rst +29 -0
  289. ads/feature_store/docs/source/feature_store.rst +122 -0
  290. ads/feature_store/docs/source/feature_store_class.rst +123 -0
  291. ads/feature_store/docs/source/feature_validation.rst +66 -0
  292. ads/feature_store/docs/source/figures/cicd.png +0 -0
  293. ads/feature_store/docs/source/figures/data_validation.png +0 -0
  294. ads/feature_store/docs/source/figures/data_versioning.png +0 -0
  295. ads/feature_store/docs/source/figures/dataset.gif +0 -0
  296. ads/feature_store/docs/source/figures/dataset.png +0 -0
  297. ads/feature_store/docs/source/figures/dataset_lineage.png +0 -0
  298. ads/feature_store/docs/source/figures/dataset_statistics.png +0 -0
  299. ads/feature_store/docs/source/figures/dataset_statistics_viz.png +0 -0
  300. ads/feature_store/docs/source/figures/dataset_validation_results.png +0 -0
  301. ads/feature_store/docs/source/figures/dataset_validation_summary.png +0 -0
  302. ads/feature_store/docs/source/figures/drift_monitoring.png +0 -0
  303. ads/feature_store/docs/source/figures/entity.png +0 -0
  304. ads/feature_store/docs/source/figures/feature_group.png +0 -0
  305. ads/feature_store/docs/source/figures/feature_group_lineage.png +0 -0
  306. ads/feature_store/docs/source/figures/feature_group_statistics_viz.png +0 -0
  307. ads/feature_store/docs/source/figures/feature_store_deployment.png +0 -0
  308. ads/feature_store/docs/source/figures/feature_store_overview.png +0 -0
  309. ads/feature_store/docs/source/figures/featuregroup.gif +0 -0
  310. ads/feature_store/docs/source/figures/lineage_d1.png +0 -0
  311. ads/feature_store/docs/source/figures/lineage_d2.png +0 -0
  312. ads/feature_store/docs/source/figures/lineage_fg.png +0 -0
  313. ads/feature_store/docs/source/figures/logo-dark-mode.png +0 -0
  314. ads/feature_store/docs/source/figures/logo-light-mode.png +0 -0
  315. ads/feature_store/docs/source/figures/overview.png +0 -0
  316. ads/feature_store/docs/source/figures/resource_manager.png +0 -0
  317. ads/feature_store/docs/source/figures/resource_manager_feature_store_stack.png +0 -0
  318. ads/feature_store/docs/source/figures/resource_manager_home.png +0 -0
  319. ads/feature_store/docs/source/figures/stats_1.png +0 -0
  320. ads/feature_store/docs/source/figures/stats_2.png +0 -0
  321. ads/feature_store/docs/source/figures/stats_d.png +0 -0
  322. ads/feature_store/docs/source/figures/stats_fg.png +0 -0
  323. ads/feature_store/docs/source/figures/transformation.png +0 -0
  324. ads/feature_store/docs/source/figures/transformations.gif +0 -0
  325. ads/feature_store/docs/source/figures/validation.png +0 -0
  326. ads/feature_store/docs/source/figures/validation_fg.png +0 -0
  327. ads/feature_store/docs/source/figures/validation_results.png +0 -0
  328. ads/feature_store/docs/source/figures/validation_summary.png +0 -0
  329. ads/feature_store/docs/source/index.rst +81 -0
  330. ads/feature_store/docs/source/module.rst +8 -0
  331. ads/feature_store/docs/source/notebook.rst +94 -0
  332. ads/feature_store/docs/source/overview.rst +47 -0
  333. ads/feature_store/docs/source/quickstart.rst +176 -0
  334. ads/feature_store/docs/source/release_notes.rst +194 -0
  335. ads/feature_store/docs/source/setup_feature_store.rst +81 -0
  336. ads/feature_store/docs/source/statistics.rst +58 -0
  337. ads/feature_store/docs/source/transformation.rst +199 -0
  338. ads/feature_store/docs/source/ui.rst +65 -0
  339. ads/feature_store/docs/source/user_guides.setup.feature_store_operator.rst +66 -0
  340. ads/feature_store/docs/source/user_guides.setup.helm_chart.rst +192 -0
  341. ads/feature_store/docs/source/user_guides.setup.terraform.rst +338 -0
  342. ads/feature_store/entity.py +718 -0
  343. ads/feature_store/execution_strategy/__init__.py +0 -0
  344. ads/feature_store/execution_strategy/delta_lake/__init__.py +0 -0
  345. ads/feature_store/execution_strategy/delta_lake/delta_lake_service.py +375 -0
  346. ads/feature_store/execution_strategy/engine/__init__.py +0 -0
  347. ads/feature_store/execution_strategy/engine/spark_engine.py +316 -0
  348. ads/feature_store/execution_strategy/execution_strategy.py +113 -0
  349. ads/feature_store/execution_strategy/execution_strategy_provider.py +47 -0
  350. ads/feature_store/execution_strategy/spark/__init__.py +0 -0
  351. ads/feature_store/execution_strategy/spark/spark_execution.py +618 -0
  352. ads/feature_store/feature.py +192 -0
  353. ads/feature_store/feature_group.py +1494 -0
  354. ads/feature_store/feature_group_expectation.py +346 -0
  355. ads/feature_store/feature_group_job.py +602 -0
  356. ads/feature_store/feature_lineage/__init__.py +0 -0
  357. ads/feature_store/feature_lineage/graphviz_service.py +180 -0
  358. ads/feature_store/feature_option_details.py +50 -0
  359. ads/feature_store/feature_statistics/__init__.py +0 -0
  360. ads/feature_store/feature_statistics/statistics_service.py +99 -0
  361. ads/feature_store/feature_store.py +699 -0
  362. ads/feature_store/feature_store_registrar.py +518 -0
  363. ads/feature_store/input_feature_detail.py +149 -0
  364. ads/feature_store/mixin/__init__.py +4 -0
  365. ads/feature_store/mixin/oci_feature_store.py +145 -0
  366. ads/feature_store/model_details.py +73 -0
  367. ads/feature_store/query/__init__.py +0 -0
  368. ads/feature_store/query/filter.py +266 -0
  369. ads/feature_store/query/generator/__init__.py +0 -0
  370. ads/feature_store/query/generator/query_generator.py +298 -0
  371. ads/feature_store/query/join.py +161 -0
  372. ads/feature_store/query/query.py +403 -0
  373. ads/feature_store/query/validator/__init__.py +0 -0
  374. ads/feature_store/query/validator/query_validator.py +57 -0
  375. ads/feature_store/response/__init__.py +0 -0
  376. ads/feature_store/response/response_builder.py +68 -0
  377. ads/feature_store/service/__init__.py +0 -0
  378. ads/feature_store/service/oci_dataset.py +139 -0
  379. ads/feature_store/service/oci_dataset_job.py +199 -0
  380. ads/feature_store/service/oci_entity.py +125 -0
  381. ads/feature_store/service/oci_feature_group.py +164 -0
  382. ads/feature_store/service/oci_feature_group_job.py +214 -0
  383. ads/feature_store/service/oci_feature_store.py +182 -0
  384. ads/feature_store/service/oci_lineage.py +87 -0
  385. ads/feature_store/service/oci_transformation.py +104 -0
  386. ads/feature_store/statistics/__init__.py +0 -0
  387. ads/feature_store/statistics/abs_feature_value.py +49 -0
  388. ads/feature_store/statistics/charts/__init__.py +0 -0
  389. ads/feature_store/statistics/charts/abstract_feature_plot.py +37 -0
  390. ads/feature_store/statistics/charts/box_plot.py +148 -0
  391. ads/feature_store/statistics/charts/frequency_distribution.py +65 -0
  392. ads/feature_store/statistics/charts/probability_distribution.py +68 -0
  393. ads/feature_store/statistics/charts/top_k_frequent_elements.py +98 -0
  394. ads/feature_store/statistics/feature_stat.py +126 -0
  395. ads/feature_store/statistics/generic_feature_value.py +33 -0
  396. ads/feature_store/statistics/statistics.py +41 -0
  397. ads/feature_store/statistics_config.py +101 -0
  398. ads/feature_store/templates/feature_store_template.yaml +45 -0
  399. ads/feature_store/transformation.py +499 -0
  400. ads/feature_store/validation_output.py +57 -0
  401. ads/hpo/__init__.py +9 -0
  402. ads/hpo/_imports.py +91 -0
  403. ads/hpo/ads_search_space.py +439 -0
  404. ads/hpo/distributions.py +325 -0
  405. ads/hpo/objective.py +280 -0
  406. ads/hpo/search_cv.py +1657 -0
  407. ads/hpo/stopping_criterion.py +75 -0
  408. ads/hpo/tuner_artifact.py +413 -0
  409. ads/hpo/utils.py +91 -0
  410. ads/hpo/validation.py +140 -0
  411. ads/hpo/visualization/__init__.py +5 -0
  412. ads/hpo/visualization/_contour.py +23 -0
  413. ads/hpo/visualization/_edf.py +20 -0
  414. ads/hpo/visualization/_intermediate_values.py +21 -0
  415. ads/hpo/visualization/_optimization_history.py +25 -0
  416. ads/hpo/visualization/_parallel_coordinate.py +169 -0
  417. ads/hpo/visualization/_param_importances.py +26 -0
  418. ads/jobs/__init__.py +53 -0
  419. ads/jobs/ads_job.py +663 -0
  420. ads/jobs/builders/__init__.py +5 -0
  421. ads/jobs/builders/base.py +156 -0
  422. ads/jobs/builders/infrastructure/__init__.py +6 -0
  423. ads/jobs/builders/infrastructure/base.py +165 -0
  424. ads/jobs/builders/infrastructure/dataflow.py +1252 -0
  425. ads/jobs/builders/infrastructure/dsc_job.py +1894 -0
  426. ads/jobs/builders/infrastructure/dsc_job_runtime.py +1233 -0
  427. ads/jobs/builders/infrastructure/utils.py +65 -0
  428. ads/jobs/builders/runtimes/__init__.py +5 -0
  429. ads/jobs/builders/runtimes/artifact.py +338 -0
  430. ads/jobs/builders/runtimes/base.py +325 -0
  431. ads/jobs/builders/runtimes/container_runtime.py +242 -0
  432. ads/jobs/builders/runtimes/python_runtime.py +1016 -0
  433. ads/jobs/builders/runtimes/pytorch_runtime.py +204 -0
  434. ads/jobs/cli.py +104 -0
  435. ads/jobs/env_var_parser.py +131 -0
  436. ads/jobs/extension.py +160 -0
  437. ads/jobs/schema/__init__.py +5 -0
  438. ads/jobs/schema/infrastructure_schema.json +116 -0
  439. ads/jobs/schema/job_schema.json +42 -0
  440. ads/jobs/schema/runtime_schema.json +183 -0
  441. ads/jobs/schema/validator.py +141 -0
  442. ads/jobs/serializer.py +296 -0
  443. ads/jobs/templates/__init__.py +5 -0
  444. ads/jobs/templates/container.py +6 -0
  445. ads/jobs/templates/driver_notebook.py +177 -0
  446. ads/jobs/templates/driver_oci.py +500 -0
  447. ads/jobs/templates/driver_python.py +48 -0
  448. ads/jobs/templates/driver_pytorch.py +852 -0
  449. ads/jobs/templates/driver_utils.py +615 -0
  450. ads/jobs/templates/hostname_from_env.c +55 -0
  451. ads/jobs/templates/oci_metrics.py +181 -0
  452. ads/jobs/utils.py +104 -0
  453. ads/llm/__init__.py +28 -0
  454. ads/llm/autogen/__init__.py +2 -0
  455. ads/llm/autogen/constants.py +15 -0
  456. ads/llm/autogen/reports/__init__.py +2 -0
  457. ads/llm/autogen/reports/base.py +67 -0
  458. ads/llm/autogen/reports/data.py +103 -0
  459. ads/llm/autogen/reports/session.py +526 -0
  460. ads/llm/autogen/reports/templates/chat_box.html +13 -0
  461. ads/llm/autogen/reports/templates/chat_box_lt.html +5 -0
  462. ads/llm/autogen/reports/templates/chat_box_rt.html +6 -0
  463. ads/llm/autogen/reports/utils.py +56 -0
  464. ads/llm/autogen/v02/__init__.py +4 -0
  465. ads/llm/autogen/v02/client.py +295 -0
  466. ads/llm/autogen/v02/log_handlers/__init__.py +2 -0
  467. ads/llm/autogen/v02/log_handlers/oci_file_handler.py +83 -0
  468. ads/llm/autogen/v02/loggers/__init__.py +6 -0
  469. ads/llm/autogen/v02/loggers/metric_logger.py +320 -0
  470. ads/llm/autogen/v02/loggers/session_logger.py +580 -0
  471. ads/llm/autogen/v02/loggers/utils.py +86 -0
  472. ads/llm/autogen/v02/runtime_logging.py +163 -0
  473. ads/llm/chain.py +268 -0
  474. ads/llm/chat_template.py +31 -0
  475. ads/llm/deploy.py +63 -0
  476. ads/llm/guardrails/__init__.py +5 -0
  477. ads/llm/guardrails/base.py +442 -0
  478. ads/llm/guardrails/huggingface.py +44 -0
  479. ads/llm/langchain/__init__.py +5 -0
  480. ads/llm/langchain/plugins/__init__.py +5 -0
  481. ads/llm/langchain/plugins/chat_models/__init__.py +5 -0
  482. ads/llm/langchain/plugins/chat_models/oci_data_science.py +1027 -0
  483. ads/llm/langchain/plugins/embeddings/__init__.py +4 -0
  484. ads/llm/langchain/plugins/embeddings/oci_data_science_model_deployment_endpoint.py +184 -0
  485. ads/llm/langchain/plugins/llms/__init__.py +5 -0
  486. ads/llm/langchain/plugins/llms/oci_data_science_model_deployment_endpoint.py +979 -0
  487. ads/llm/requirements.txt +3 -0
  488. ads/llm/serialize.py +219 -0
  489. ads/llm/serializers/__init__.py +0 -0
  490. ads/llm/serializers/retrieval_qa.py +153 -0
  491. ads/llm/serializers/runnable_parallel.py +27 -0
  492. ads/llm/templates/score_chain.jinja2 +155 -0
  493. ads/llm/templates/tool_chat_template_hermes.jinja +130 -0
  494. ads/llm/templates/tool_chat_template_mistral_parallel.jinja +94 -0
  495. ads/model/__init__.py +52 -0
  496. ads/model/artifact.py +573 -0
  497. ads/model/artifact_downloader.py +254 -0
  498. ads/model/artifact_uploader.py +267 -0
  499. ads/model/base_properties.py +238 -0
  500. ads/model/common/.model-ignore +66 -0
  501. ads/model/common/__init__.py +5 -0
  502. ads/model/common/utils.py +142 -0
  503. ads/model/datascience_model.py +2635 -0
  504. ads/model/deployment/__init__.py +20 -0
  505. ads/model/deployment/common/__init__.py +5 -0
  506. ads/model/deployment/common/utils.py +308 -0
  507. ads/model/deployment/model_deployer.py +466 -0
  508. ads/model/deployment/model_deployment.py +1846 -0
  509. ads/model/deployment/model_deployment_infrastructure.py +671 -0
  510. ads/model/deployment/model_deployment_properties.py +493 -0
  511. ads/model/deployment/model_deployment_runtime.py +838 -0
  512. ads/model/extractor/__init__.py +5 -0
  513. ads/model/extractor/automl_extractor.py +74 -0
  514. ads/model/extractor/embedding_onnx_extractor.py +80 -0
  515. ads/model/extractor/huggingface_extractor.py +88 -0
  516. ads/model/extractor/keras_extractor.py +84 -0
  517. ads/model/extractor/lightgbm_extractor.py +93 -0
  518. ads/model/extractor/model_info_extractor.py +114 -0
  519. ads/model/extractor/model_info_extractor_factory.py +105 -0
  520. ads/model/extractor/pytorch_extractor.py +87 -0
  521. ads/model/extractor/sklearn_extractor.py +112 -0
  522. ads/model/extractor/spark_extractor.py +89 -0
  523. ads/model/extractor/tensorflow_extractor.py +85 -0
  524. ads/model/extractor/xgboost_extractor.py +94 -0
  525. ads/model/framework/__init__.py +5 -0
  526. ads/model/framework/automl_model.py +178 -0
  527. ads/model/framework/embedding_onnx_model.py +438 -0
  528. ads/model/framework/huggingface_model.py +399 -0
  529. ads/model/framework/lightgbm_model.py +266 -0
  530. ads/model/framework/pytorch_model.py +266 -0
  531. ads/model/framework/sklearn_model.py +250 -0
  532. ads/model/framework/spark_model.py +326 -0
  533. ads/model/framework/tensorflow_model.py +254 -0
  534. ads/model/framework/xgboost_model.py +258 -0
  535. ads/model/generic_model.py +3518 -0
  536. ads/model/model_artifact_boilerplate/README.md +381 -0
  537. ads/model/model_artifact_boilerplate/__init__.py +5 -0
  538. ads/model/model_artifact_boilerplate/artifact_introspection_test/__init__.py +5 -0
  539. ads/model/model_artifact_boilerplate/artifact_introspection_test/model_artifact_validate.py +427 -0
  540. ads/model/model_artifact_boilerplate/artifact_introspection_test/requirements.txt +2 -0
  541. ads/model/model_artifact_boilerplate/runtime.yaml +7 -0
  542. ads/model/model_artifact_boilerplate/score.py +61 -0
  543. ads/model/model_file_description_schema.json +68 -0
  544. ads/model/model_introspect.py +331 -0
  545. ads/model/model_metadata.py +1810 -0
  546. ads/model/model_metadata_mixin.py +460 -0
  547. ads/model/model_properties.py +63 -0
  548. ads/model/model_version_set.py +739 -0
  549. ads/model/runtime/__init__.py +5 -0
  550. ads/model/runtime/env_info.py +306 -0
  551. ads/model/runtime/model_deployment_details.py +37 -0
  552. ads/model/runtime/model_provenance_details.py +58 -0
  553. ads/model/runtime/runtime_info.py +81 -0
  554. ads/model/runtime/schemas/inference_env_info_schema.yaml +16 -0
  555. ads/model/runtime/schemas/model_provenance_schema.yaml +36 -0
  556. ads/model/runtime/schemas/training_env_info_schema.yaml +16 -0
  557. ads/model/runtime/utils.py +201 -0
  558. ads/model/serde/__init__.py +5 -0
  559. ads/model/serde/common.py +40 -0
  560. ads/model/serde/model_input.py +547 -0
  561. ads/model/serde/model_serializer.py +1184 -0
  562. ads/model/service/__init__.py +5 -0
  563. ads/model/service/oci_datascience_model.py +1076 -0
  564. ads/model/service/oci_datascience_model_deployment.py +500 -0
  565. ads/model/service/oci_datascience_model_version_set.py +176 -0
  566. ads/model/transformer/__init__.py +5 -0
  567. ads/model/transformer/onnx_transformer.py +324 -0
  568. ads/mysqldb/__init__.py +5 -0
  569. ads/mysqldb/mysql_db.py +227 -0
  570. ads/opctl/__init__.py +18 -0
  571. ads/opctl/anomaly_detection.py +11 -0
  572. ads/opctl/backend/__init__.py +5 -0
  573. ads/opctl/backend/ads_dataflow.py +353 -0
  574. ads/opctl/backend/ads_ml_job.py +710 -0
  575. ads/opctl/backend/ads_ml_pipeline.py +164 -0
  576. ads/opctl/backend/ads_model_deployment.py +209 -0
  577. ads/opctl/backend/base.py +146 -0
  578. ads/opctl/backend/local.py +1053 -0
  579. ads/opctl/backend/marketplace/__init__.py +9 -0
  580. ads/opctl/backend/marketplace/helm_helper.py +173 -0
  581. ads/opctl/backend/marketplace/local_marketplace.py +271 -0
  582. ads/opctl/backend/marketplace/marketplace_backend_runner.py +71 -0
  583. ads/opctl/backend/marketplace/marketplace_operator_interface.py +44 -0
  584. ads/opctl/backend/marketplace/marketplace_operator_runner.py +24 -0
  585. ads/opctl/backend/marketplace/marketplace_utils.py +212 -0
  586. ads/opctl/backend/marketplace/models/__init__.py +5 -0
  587. ads/opctl/backend/marketplace/models/bearer_token.py +94 -0
  588. ads/opctl/backend/marketplace/models/marketplace_type.py +70 -0
  589. ads/opctl/backend/marketplace/models/ocir_details.py +56 -0
  590. ads/opctl/backend/marketplace/prerequisite_checker.py +238 -0
  591. ads/opctl/cli.py +707 -0
  592. ads/opctl/cmds.py +869 -0
  593. ads/opctl/conda/__init__.py +5 -0
  594. ads/opctl/conda/cli.py +193 -0
  595. ads/opctl/conda/cmds.py +749 -0
  596. ads/opctl/conda/config.yaml +34 -0
  597. ads/opctl/conda/manifest_template.yaml +13 -0
  598. ads/opctl/conda/multipart_uploader.py +188 -0
  599. ads/opctl/conda/pack.py +89 -0
  600. ads/opctl/config/__init__.py +5 -0
  601. ads/opctl/config/base.py +57 -0
  602. ads/opctl/config/diagnostics/__init__.py +5 -0
  603. ads/opctl/config/diagnostics/distributed/default_requirements_config.yaml +62 -0
  604. ads/opctl/config/merger.py +255 -0
  605. ads/opctl/config/resolver.py +297 -0
  606. ads/opctl/config/utils.py +79 -0
  607. ads/opctl/config/validator.py +17 -0
  608. ads/opctl/config/versioner.py +68 -0
  609. ads/opctl/config/yaml_parsers/__init__.py +7 -0
  610. ads/opctl/config/yaml_parsers/base.py +58 -0
  611. ads/opctl/config/yaml_parsers/distributed/__init__.py +7 -0
  612. ads/opctl/config/yaml_parsers/distributed/yaml_parser.py +201 -0
  613. ads/opctl/constants.py +66 -0
  614. ads/opctl/decorator/__init__.py +5 -0
  615. ads/opctl/decorator/common.py +129 -0
  616. ads/opctl/diagnostics/__init__.py +5 -0
  617. ads/opctl/diagnostics/__main__.py +25 -0
  618. ads/opctl/diagnostics/check_distributed_job_requirements.py +212 -0
  619. ads/opctl/diagnostics/check_requirements.py +144 -0
  620. ads/opctl/diagnostics/requirement_exception.py +9 -0
  621. ads/opctl/distributed/README.md +109 -0
  622. ads/opctl/distributed/__init__.py +5 -0
  623. ads/opctl/distributed/certificates.py +32 -0
  624. ads/opctl/distributed/cli.py +207 -0
  625. ads/opctl/distributed/cmds.py +731 -0
  626. ads/opctl/distributed/common/__init__.py +5 -0
  627. ads/opctl/distributed/common/abstract_cluster_provider.py +449 -0
  628. ads/opctl/distributed/common/abstract_framework_spec_builder.py +88 -0
  629. ads/opctl/distributed/common/cluster_config_helper.py +103 -0
  630. ads/opctl/distributed/common/cluster_provider_factory.py +21 -0
  631. ads/opctl/distributed/common/cluster_runner.py +54 -0
  632. ads/opctl/distributed/common/framework_factory.py +29 -0
  633. ads/opctl/docker/Dockerfile.job +103 -0
  634. ads/opctl/docker/Dockerfile.job.arm +107 -0
  635. ads/opctl/docker/Dockerfile.job.gpu +175 -0
  636. ads/opctl/docker/base-env.yaml +13 -0
  637. ads/opctl/docker/cuda.repo +6 -0
  638. ads/opctl/docker/operator/.dockerignore +0 -0
  639. ads/opctl/docker/operator/Dockerfile +41 -0
  640. ads/opctl/docker/operator/Dockerfile.gpu +85 -0
  641. ads/opctl/docker/operator/cuda.repo +6 -0
  642. ads/opctl/docker/operator/environment.yaml +8 -0
  643. ads/opctl/forecast.py +11 -0
  644. ads/opctl/index.yaml +3 -0
  645. ads/opctl/model/__init__.py +5 -0
  646. ads/opctl/model/cli.py +65 -0
  647. ads/opctl/model/cmds.py +73 -0
  648. ads/opctl/operator/README.md +4 -0
  649. ads/opctl/operator/__init__.py +31 -0
  650. ads/opctl/operator/cli.py +344 -0
  651. ads/opctl/operator/cmd.py +596 -0
  652. ads/opctl/operator/common/__init__.py +5 -0
  653. ads/opctl/operator/common/backend_factory.py +460 -0
  654. ads/opctl/operator/common/const.py +27 -0
  655. ads/opctl/operator/common/data/synthetic.csv +16001 -0
  656. ads/opctl/operator/common/dictionary_merger.py +148 -0
  657. ads/opctl/operator/common/errors.py +42 -0
  658. ads/opctl/operator/common/operator_config.py +99 -0
  659. ads/opctl/operator/common/operator_loader.py +811 -0
  660. ads/opctl/operator/common/operator_schema.yaml +130 -0
  661. ads/opctl/operator/common/operator_yaml_generator.py +152 -0
  662. ads/opctl/operator/common/utils.py +208 -0
  663. ads/opctl/operator/lowcode/__init__.py +5 -0
  664. ads/opctl/operator/lowcode/anomaly/MLoperator +16 -0
  665. ads/opctl/operator/lowcode/anomaly/README.md +207 -0
  666. ads/opctl/operator/lowcode/anomaly/__init__.py +5 -0
  667. ads/opctl/operator/lowcode/anomaly/__main__.py +103 -0
  668. ads/opctl/operator/lowcode/anomaly/cmd.py +35 -0
  669. ads/opctl/operator/lowcode/anomaly/const.py +167 -0
  670. ads/opctl/operator/lowcode/anomaly/environment.yaml +10 -0
  671. ads/opctl/operator/lowcode/anomaly/model/__init__.py +5 -0
  672. ads/opctl/operator/lowcode/anomaly/model/anomaly_dataset.py +146 -0
  673. ads/opctl/operator/lowcode/anomaly/model/anomaly_merlion.py +162 -0
  674. ads/opctl/operator/lowcode/anomaly/model/automlx.py +99 -0
  675. ads/opctl/operator/lowcode/anomaly/model/autots.py +115 -0
  676. ads/opctl/operator/lowcode/anomaly/model/base_model.py +404 -0
  677. ads/opctl/operator/lowcode/anomaly/model/factory.py +110 -0
  678. ads/opctl/operator/lowcode/anomaly/model/isolationforest.py +78 -0
  679. ads/opctl/operator/lowcode/anomaly/model/oneclasssvm.py +78 -0
  680. ads/opctl/operator/lowcode/anomaly/model/randomcutforest.py +120 -0
  681. ads/opctl/operator/lowcode/anomaly/model/tods.py +119 -0
  682. ads/opctl/operator/lowcode/anomaly/operator_config.py +127 -0
  683. ads/opctl/operator/lowcode/anomaly/schema.yaml +401 -0
  684. ads/opctl/operator/lowcode/anomaly/utils.py +88 -0
  685. ads/opctl/operator/lowcode/common/__init__.py +5 -0
  686. ads/opctl/operator/lowcode/common/const.py +10 -0
  687. ads/opctl/operator/lowcode/common/data.py +116 -0
  688. ads/opctl/operator/lowcode/common/errors.py +47 -0
  689. ads/opctl/operator/lowcode/common/transformations.py +296 -0
  690. ads/opctl/operator/lowcode/common/utils.py +384 -0
  691. ads/opctl/operator/lowcode/feature_store_marketplace/MLoperator +13 -0
  692. ads/opctl/operator/lowcode/feature_store_marketplace/README.md +30 -0
  693. ads/opctl/operator/lowcode/feature_store_marketplace/__init__.py +5 -0
  694. ads/opctl/operator/lowcode/feature_store_marketplace/__main__.py +116 -0
  695. ads/opctl/operator/lowcode/feature_store_marketplace/cmd.py +85 -0
  696. ads/opctl/operator/lowcode/feature_store_marketplace/const.py +15 -0
  697. ads/opctl/operator/lowcode/feature_store_marketplace/environment.yaml +0 -0
  698. ads/opctl/operator/lowcode/feature_store_marketplace/models/__init__.py +4 -0
  699. ads/opctl/operator/lowcode/feature_store_marketplace/models/apigw_config.py +32 -0
  700. ads/opctl/operator/lowcode/feature_store_marketplace/models/db_config.py +43 -0
  701. ads/opctl/operator/lowcode/feature_store_marketplace/models/mysql_config.py +120 -0
  702. ads/opctl/operator/lowcode/feature_store_marketplace/models/serializable_yaml_model.py +34 -0
  703. ads/opctl/operator/lowcode/feature_store_marketplace/operator_utils.py +386 -0
  704. ads/opctl/operator/lowcode/feature_store_marketplace/schema.yaml +160 -0
  705. ads/opctl/operator/lowcode/forecast/MLoperator +25 -0
  706. ads/opctl/operator/lowcode/forecast/README.md +209 -0
  707. ads/opctl/operator/lowcode/forecast/__init__.py +5 -0
  708. ads/opctl/operator/lowcode/forecast/__main__.py +89 -0
  709. ads/opctl/operator/lowcode/forecast/cmd.py +40 -0
  710. ads/opctl/operator/lowcode/forecast/const.py +92 -0
  711. ads/opctl/operator/lowcode/forecast/environment.yaml +20 -0
  712. ads/opctl/operator/lowcode/forecast/errors.py +26 -0
  713. ads/opctl/operator/lowcode/forecast/model/__init__.py +5 -0
  714. ads/opctl/operator/lowcode/forecast/model/arima.py +279 -0
  715. ads/opctl/operator/lowcode/forecast/model/automlx.py +553 -0
  716. ads/opctl/operator/lowcode/forecast/model/autots.py +312 -0
  717. ads/opctl/operator/lowcode/forecast/model/base_model.py +875 -0
  718. ads/opctl/operator/lowcode/forecast/model/factory.py +106 -0
  719. ads/opctl/operator/lowcode/forecast/model/forecast_datasets.py +492 -0
  720. ads/opctl/operator/lowcode/forecast/model/ml_forecast.py +243 -0
  721. ads/opctl/operator/lowcode/forecast/model/neuralprophet.py +482 -0
  722. ads/opctl/operator/lowcode/forecast/model/prophet.py +450 -0
  723. ads/opctl/operator/lowcode/forecast/model_evaluator.py +244 -0
  724. ads/opctl/operator/lowcode/forecast/operator_config.py +234 -0
  725. ads/opctl/operator/lowcode/forecast/schema.yaml +506 -0
  726. ads/opctl/operator/lowcode/forecast/utils.py +397 -0
  727. ads/opctl/operator/lowcode/forecast/whatifserve/__init__.py +7 -0
  728. ads/opctl/operator/lowcode/forecast/whatifserve/deployment_manager.py +285 -0
  729. ads/opctl/operator/lowcode/forecast/whatifserve/score.py +246 -0
  730. ads/opctl/operator/lowcode/pii/MLoperator +17 -0
  731. ads/opctl/operator/lowcode/pii/README.md +208 -0
  732. ads/opctl/operator/lowcode/pii/__init__.py +5 -0
  733. ads/opctl/operator/lowcode/pii/__main__.py +78 -0
  734. ads/opctl/operator/lowcode/pii/cmd.py +39 -0
  735. ads/opctl/operator/lowcode/pii/constant.py +84 -0
  736. ads/opctl/operator/lowcode/pii/environment.yaml +17 -0
  737. ads/opctl/operator/lowcode/pii/errors.py +27 -0
  738. ads/opctl/operator/lowcode/pii/model/__init__.py +5 -0
  739. ads/opctl/operator/lowcode/pii/model/factory.py +82 -0
  740. ads/opctl/operator/lowcode/pii/model/guardrails.py +167 -0
  741. ads/opctl/operator/lowcode/pii/model/pii.py +145 -0
  742. ads/opctl/operator/lowcode/pii/model/processor/__init__.py +34 -0
  743. ads/opctl/operator/lowcode/pii/model/processor/email_replacer.py +34 -0
  744. ads/opctl/operator/lowcode/pii/model/processor/mbi_replacer.py +35 -0
  745. ads/opctl/operator/lowcode/pii/model/processor/name_replacer.py +225 -0
  746. ads/opctl/operator/lowcode/pii/model/processor/number_replacer.py +73 -0
  747. ads/opctl/operator/lowcode/pii/model/processor/remover.py +26 -0
  748. ads/opctl/operator/lowcode/pii/model/report.py +487 -0
  749. ads/opctl/operator/lowcode/pii/operator_config.py +95 -0
  750. ads/opctl/operator/lowcode/pii/schema.yaml +108 -0
  751. ads/opctl/operator/lowcode/pii/utils.py +43 -0
  752. ads/opctl/operator/lowcode/recommender/MLoperator +16 -0
  753. ads/opctl/operator/lowcode/recommender/README.md +206 -0
  754. ads/opctl/operator/lowcode/recommender/__init__.py +5 -0
  755. ads/opctl/operator/lowcode/recommender/__main__.py +82 -0
  756. ads/opctl/operator/lowcode/recommender/cmd.py +33 -0
  757. ads/opctl/operator/lowcode/recommender/constant.py +30 -0
  758. ads/opctl/operator/lowcode/recommender/environment.yaml +11 -0
  759. ads/opctl/operator/lowcode/recommender/model/base_model.py +212 -0
  760. ads/opctl/operator/lowcode/recommender/model/factory.py +56 -0
  761. ads/opctl/operator/lowcode/recommender/model/recommender_dataset.py +25 -0
  762. ads/opctl/operator/lowcode/recommender/model/svd.py +106 -0
  763. ads/opctl/operator/lowcode/recommender/operator_config.py +81 -0
  764. ads/opctl/operator/lowcode/recommender/schema.yaml +265 -0
  765. ads/opctl/operator/lowcode/recommender/utils.py +13 -0
  766. ads/opctl/operator/runtime/__init__.py +5 -0
  767. ads/opctl/operator/runtime/const.py +17 -0
  768. ads/opctl/operator/runtime/container_runtime_schema.yaml +50 -0
  769. ads/opctl/operator/runtime/marketplace_runtime.py +50 -0
  770. ads/opctl/operator/runtime/python_marketplace_runtime_schema.yaml +21 -0
  771. ads/opctl/operator/runtime/python_runtime_schema.yaml +21 -0
  772. ads/opctl/operator/runtime/runtime.py +115 -0
  773. ads/opctl/schema.yaml.yml +36 -0
  774. ads/opctl/script.py +40 -0
  775. ads/opctl/spark/__init__.py +5 -0
  776. ads/opctl/spark/cli.py +43 -0
  777. ads/opctl/spark/cmds.py +147 -0
  778. ads/opctl/templates/diagnostic_report_template.jinja2 +102 -0
  779. ads/opctl/utils.py +344 -0
  780. ads/oracledb/__init__.py +5 -0
  781. ads/oracledb/oracle_db.py +346 -0
  782. ads/pipeline/__init__.py +39 -0
  783. ads/pipeline/ads_pipeline.py +2279 -0
  784. ads/pipeline/ads_pipeline_run.py +772 -0
  785. ads/pipeline/ads_pipeline_step.py +605 -0
  786. ads/pipeline/builders/__init__.py +5 -0
  787. ads/pipeline/builders/infrastructure/__init__.py +5 -0
  788. ads/pipeline/builders/infrastructure/custom_script.py +32 -0
  789. ads/pipeline/cli.py +119 -0
  790. ads/pipeline/extension.py +291 -0
  791. ads/pipeline/schema/__init__.py +5 -0
  792. ads/pipeline/schema/cs_step_schema.json +35 -0
  793. ads/pipeline/schema/ml_step_schema.json +31 -0
  794. ads/pipeline/schema/pipeline_schema.json +71 -0
  795. ads/pipeline/visualizer/__init__.py +5 -0
  796. ads/pipeline/visualizer/base.py +570 -0
  797. ads/pipeline/visualizer/graph_renderer.py +272 -0
  798. ads/pipeline/visualizer/text_renderer.py +84 -0
  799. ads/secrets/__init__.py +11 -0
  800. ads/secrets/adb.py +386 -0
  801. ads/secrets/auth_token.py +86 -0
  802. ads/secrets/big_data_service.py +365 -0
  803. ads/secrets/mysqldb.py +149 -0
  804. ads/secrets/oracledb.py +160 -0
  805. ads/secrets/secrets.py +407 -0
  806. ads/telemetry/__init__.py +7 -0
  807. ads/telemetry/base.py +69 -0
  808. ads/telemetry/client.py +122 -0
  809. ads/telemetry/telemetry.py +257 -0
  810. ads/templates/dataflow_pyspark.jinja2 +13 -0
  811. ads/templates/dataflow_sparksql.jinja2 +22 -0
  812. ads/templates/func.jinja2 +20 -0
  813. ads/templates/schemas/openapi.json +1740 -0
  814. ads/templates/score-pkl.jinja2 +173 -0
  815. ads/templates/score.jinja2 +322 -0
  816. ads/templates/score_embedding_onnx.jinja2 +202 -0
  817. ads/templates/score_generic.jinja2 +165 -0
  818. ads/templates/score_huggingface_pipeline.jinja2 +217 -0
  819. ads/templates/score_lightgbm.jinja2 +185 -0
  820. ads/templates/score_onnx.jinja2 +407 -0
  821. ads/templates/score_onnx_new.jinja2 +473 -0
  822. ads/templates/score_oracle_automl.jinja2 +185 -0
  823. ads/templates/score_pyspark.jinja2 +154 -0
  824. ads/templates/score_pytorch.jinja2 +219 -0
  825. ads/templates/score_scikit-learn.jinja2 +184 -0
  826. ads/templates/score_tensorflow.jinja2 +184 -0
  827. ads/templates/score_xgboost.jinja2 +178 -0
  828. ads/text_dataset/__init__.py +5 -0
  829. ads/text_dataset/backends.py +211 -0
  830. ads/text_dataset/dataset.py +445 -0
  831. ads/text_dataset/extractor.py +207 -0
  832. ads/text_dataset/options.py +53 -0
  833. ads/text_dataset/udfs.py +22 -0
  834. ads/text_dataset/utils.py +49 -0
  835. ads/type_discovery/__init__.py +9 -0
  836. ads/type_discovery/abstract_detector.py +21 -0
  837. ads/type_discovery/constant_detector.py +41 -0
  838. ads/type_discovery/continuous_detector.py +54 -0
  839. ads/type_discovery/credit_card_detector.py +99 -0
  840. ads/type_discovery/datetime_detector.py +92 -0
  841. ads/type_discovery/discrete_detector.py +118 -0
  842. ads/type_discovery/document_detector.py +146 -0
  843. ads/type_discovery/ip_detector.py +68 -0
  844. ads/type_discovery/latlon_detector.py +90 -0
  845. ads/type_discovery/phone_number_detector.py +63 -0
  846. ads/type_discovery/type_discovery_driver.py +87 -0
  847. ads/type_discovery/typed_feature.py +594 -0
  848. ads/type_discovery/unknown_detector.py +41 -0
  849. ads/type_discovery/zipcode_detector.py +48 -0
  850. ads/vault/__init__.py +7 -0
  851. ads/vault/vault.py +237 -0
  852. {oracle_ads-2.13.9rc0.dist-info → oracle_ads-2.13.10.dist-info}/METADATA +150 -149
  853. oracle_ads-2.13.10.dist-info/RECORD +858 -0
  854. {oracle_ads-2.13.9rc0.dist-info → oracle_ads-2.13.10.dist-info}/WHEEL +1 -2
  855. {oracle_ads-2.13.9rc0.dist-info → oracle_ads-2.13.10.dist-info}/entry_points.txt +2 -1
  856. oracle_ads-2.13.9rc0.dist-info/RECORD +0 -9
  857. oracle_ads-2.13.9rc0.dist-info/top_level.txt +0 -1
  858. {oracle_ads-2.13.9rc0.dist-info → oracle_ads-2.13.10.dist-info}/licenses/LICENSE.txt +0 -0
@@ -0,0 +1,1740 @@
1
+ {
2
+ "components": {
3
+ "schemas": {
4
+ "ClassifierModel": {
5
+ "properties": {
6
+ "id2label": {
7
+ "additionalProperties": {
8
+ "type": "string"
9
+ },
10
+ "example": {
11
+ "0": "LABEL"
12
+ },
13
+ "type": "object"
14
+ },
15
+ "label2id": {
16
+ "additionalProperties": {
17
+ "minimum": 0,
18
+ "type": "integer"
19
+ },
20
+ "example": {
21
+ "LABEL": 0
22
+ },
23
+ "type": "object"
24
+ }
25
+ },
26
+ "required": [
27
+ "id2label",
28
+ "label2id"
29
+ ],
30
+ "type": "object"
31
+ },
32
+ "DecodeRequest": {
33
+ "properties": {
34
+ "ids": {
35
+ "$ref": "#/components/schemas/InputIds"
36
+ },
37
+ "skip_special_tokens": {
38
+ "default": "true",
39
+ "example": "true",
40
+ "type": "boolean"
41
+ }
42
+ },
43
+ "required": [
44
+ "ids"
45
+ ],
46
+ "type": "object"
47
+ },
48
+ "DecodeResponse": {
49
+ "example": [
50
+ "test"
51
+ ],
52
+ "items": {
53
+ "type": "string"
54
+ },
55
+ "type": "array"
56
+ },
57
+ "EmbedAllRequest": {
58
+ "properties": {
59
+ "inputs": {
60
+ "$ref": "#/components/schemas/Input"
61
+ },
62
+ "prompt_name": {
63
+ "default": "null",
64
+ "description": "The name of the prompt that should be used by for encoding. If not set, no prompt\nwill be applied.\n\nMust be a key in the `sentence-transformers` configuration `prompts` dictionary.\n\nFor example if ``prompt_name`` is \"query\" and the ``prompts`` is {\"query\": \"query: \", ...},\nthen the sentence \"What is the capital of France?\" will be encoded as\n\"query: What is the capital of France?\" because the prompt text will be prepended before\nany text to encode.",
65
+ "example": "null",
66
+ "nullable": true,
67
+ "type": "string"
68
+ },
69
+ "truncate": {
70
+ "default": "false",
71
+ "example": "false",
72
+ "nullable": true,
73
+ "type": "boolean"
74
+ },
75
+ "truncation_direction": {
76
+ "allOf": [
77
+ {
78
+ "$ref": "#/components/schemas/TruncationDirection"
79
+ }
80
+ ],
81
+ "default": "right"
82
+ }
83
+ },
84
+ "required": [
85
+ "inputs"
86
+ ],
87
+ "type": "object"
88
+ },
89
+ "EmbedAllResponse": {
90
+ "example": [
91
+ [
92
+ [
93
+ 0.0,
94
+ 1.0,
95
+ 2.0
96
+ ]
97
+ ]
98
+ ],
99
+ "items": {
100
+ "items": {
101
+ "items": {
102
+ "format": "float",
103
+ "type": "number"
104
+ },
105
+ "type": "array"
106
+ },
107
+ "type": "array"
108
+ },
109
+ "type": "array"
110
+ },
111
+ "EmbedRequest": {
112
+ "properties": {
113
+ "inputs": {
114
+ "$ref": "#/components/schemas/Input"
115
+ },
116
+ "normalize": {
117
+ "default": "true",
118
+ "example": "true",
119
+ "type": "boolean"
120
+ },
121
+ "prompt_name": {
122
+ "default": "null",
123
+ "description": "The name of the prompt that should be used by for encoding. If not set, no prompt\nwill be applied.\n\nMust be a key in the `sentence-transformers` configuration `prompts` dictionary.\n\nFor example if ``prompt_name`` is \"query\" and the ``prompts`` is {\"query\": \"query: \", ...},\nthen the sentence \"What is the capital of France?\" will be encoded as\n\"query: What is the capital of France?\" because the prompt text will be prepended before\nany text to encode.",
124
+ "example": "null",
125
+ "nullable": true,
126
+ "type": "string"
127
+ },
128
+ "truncate": {
129
+ "default": "false",
130
+ "example": "false",
131
+ "nullable": true,
132
+ "type": "boolean"
133
+ },
134
+ "truncation_direction": {
135
+ "allOf": [
136
+ {
137
+ "$ref": "#/components/schemas/TruncationDirection"
138
+ }
139
+ ],
140
+ "default": "right"
141
+ }
142
+ },
143
+ "required": [
144
+ "inputs"
145
+ ],
146
+ "type": "object"
147
+ },
148
+ "EmbedResponse": {
149
+ "example": [
150
+ [
151
+ 0.0,
152
+ 1.0,
153
+ 2.0
154
+ ]
155
+ ],
156
+ "items": {
157
+ "items": {
158
+ "format": "float",
159
+ "type": "number"
160
+ },
161
+ "type": "array"
162
+ },
163
+ "type": "array"
164
+ },
165
+ "EmbedSparseRequest": {
166
+ "properties": {
167
+ "inputs": {
168
+ "$ref": "#/components/schemas/Input"
169
+ },
170
+ "prompt_name": {
171
+ "default": "null",
172
+ "description": "The name of the prompt that should be used by for encoding. If not set, no prompt\nwill be applied.\n\nMust be a key in the `sentence-transformers` configuration `prompts` dictionary.\n\nFor example if ``prompt_name`` is \"query\" and the ``prompts`` is {\"query\": \"query: \", ...},\nthen the sentence \"What is the capital of France?\" will be encoded as\n\"query: What is the capital of France?\" because the prompt text will be prepended before\nany text to encode.",
173
+ "example": "null",
174
+ "nullable": true,
175
+ "type": "string"
176
+ },
177
+ "truncate": {
178
+ "default": "false",
179
+ "example": "false",
180
+ "nullable": true,
181
+ "type": "boolean"
182
+ },
183
+ "truncation_direction": {
184
+ "allOf": [
185
+ {
186
+ "$ref": "#/components/schemas/TruncationDirection"
187
+ }
188
+ ],
189
+ "default": "right"
190
+ }
191
+ },
192
+ "required": [
193
+ "inputs"
194
+ ],
195
+ "type": "object"
196
+ },
197
+ "EmbedSparseResponse": {
198
+ "items": {
199
+ "items": {
200
+ "$ref": "#/components/schemas/SparseValue"
201
+ },
202
+ "type": "array"
203
+ },
204
+ "type": "array"
205
+ },
206
+ "Embedding": {
207
+ "oneOf": [
208
+ {
209
+ "items": {
210
+ "format": "float",
211
+ "type": "number"
212
+ },
213
+ "type": "array"
214
+ },
215
+ {
216
+ "type": "string"
217
+ }
218
+ ]
219
+ },
220
+ "EmbeddingModel": {
221
+ "properties": {
222
+ "pooling": {
223
+ "example": "cls",
224
+ "type": "string"
225
+ }
226
+ },
227
+ "required": [
228
+ "pooling"
229
+ ],
230
+ "type": "object"
231
+ },
232
+ "EncodingFormat": {
233
+ "enum": [
234
+ "float",
235
+ "base64"
236
+ ],
237
+ "type": "string"
238
+ },
239
+ "ErrorResponse": {
240
+ "properties": {
241
+ "error": {
242
+ "type": "string"
243
+ },
244
+ "error_type": {
245
+ "$ref": "#/components/schemas/ErrorType"
246
+ }
247
+ },
248
+ "required": [
249
+ "error",
250
+ "error_type"
251
+ ],
252
+ "type": "object"
253
+ },
254
+ "ErrorType": {
255
+ "enum": [
256
+ "Unhealthy",
257
+ "Backend",
258
+ "Overloaded",
259
+ "Validation",
260
+ "Tokenizer"
261
+ ],
262
+ "type": "string"
263
+ },
264
+ "Info": {
265
+ "properties": {
266
+ "auto_truncate": {
267
+ "type": "boolean"
268
+ },
269
+ "docker_label": {
270
+ "example": "null",
271
+ "nullable": true,
272
+ "type": "string"
273
+ },
274
+ "max_batch_requests": {
275
+ "default": "null",
276
+ "example": "null",
277
+ "minimum": 0,
278
+ "nullable": true,
279
+ "type": "integer"
280
+ },
281
+ "max_batch_tokens": {
282
+ "example": "2048",
283
+ "minimum": 0,
284
+ "type": "integer"
285
+ },
286
+ "max_client_batch_size": {
287
+ "example": "32",
288
+ "minimum": 0,
289
+ "type": "integer"
290
+ },
291
+ "max_concurrent_requests": {
292
+ "description": "Router Parameters",
293
+ "example": "128",
294
+ "minimum": 0,
295
+ "type": "integer"
296
+ },
297
+ "max_input_length": {
298
+ "example": "512",
299
+ "minimum": 0,
300
+ "type": "integer"
301
+ },
302
+ "model_dtype": {
303
+ "example": "float16",
304
+ "type": "string"
305
+ },
306
+ "model_id": {
307
+ "description": "Model info",
308
+ "example": "thenlper/gte-base",
309
+ "type": "string"
310
+ },
311
+ "model_sha": {
312
+ "example": "fca14538aa9956a46526bd1d0d11d69e19b5a101",
313
+ "nullable": true,
314
+ "type": "string"
315
+ },
316
+ "model_type": {
317
+ "$ref": "#/components/schemas/ModelType"
318
+ },
319
+ "sha": {
320
+ "example": "null",
321
+ "nullable": true,
322
+ "type": "string"
323
+ },
324
+ "tokenization_workers": {
325
+ "example": "4",
326
+ "minimum": 0,
327
+ "type": "integer"
328
+ },
329
+ "version": {
330
+ "description": "Router Info",
331
+ "example": "0.5.0",
332
+ "type": "string"
333
+ }
334
+ },
335
+ "required": [
336
+ "model_id",
337
+ "model_dtype",
338
+ "model_type",
339
+ "max_concurrent_requests",
340
+ "max_input_length",
341
+ "max_batch_tokens",
342
+ "max_client_batch_size",
343
+ "auto_truncate",
344
+ "tokenization_workers",
345
+ "version"
346
+ ],
347
+ "type": "object"
348
+ },
349
+ "Input": {
350
+ "oneOf": [
351
+ {
352
+ "$ref": "#/components/schemas/InputType"
353
+ },
354
+ {
355
+ "items": {
356
+ "$ref": "#/components/schemas/InputType"
357
+ },
358
+ "type": "array"
359
+ }
360
+ ]
361
+ },
362
+ "InputIds": {
363
+ "oneOf": [
364
+ {
365
+ "items": {
366
+ "format": "int32",
367
+ "minimum": 0,
368
+ "type": "integer"
369
+ },
370
+ "type": "array"
371
+ },
372
+ {
373
+ "items": {
374
+ "items": {
375
+ "format": "int32",
376
+ "minimum": 0,
377
+ "type": "integer"
378
+ },
379
+ "type": "array"
380
+ },
381
+ "type": "array"
382
+ }
383
+ ]
384
+ },
385
+ "InputType": {
386
+ "oneOf": [
387
+ {
388
+ "type": "string"
389
+ },
390
+ {
391
+ "items": {
392
+ "format": "int32",
393
+ "minimum": 0,
394
+ "type": "integer"
395
+ },
396
+ "type": "array"
397
+ }
398
+ ]
399
+ },
400
+ "ModelType": {
401
+ "oneOf": [
402
+ {
403
+ "properties": {
404
+ "classifier": {
405
+ "$ref": "#/components/schemas/ClassifierModel"
406
+ }
407
+ },
408
+ "required": [
409
+ "classifier"
410
+ ],
411
+ "type": "object"
412
+ },
413
+ {
414
+ "properties": {
415
+ "embedding": {
416
+ "$ref": "#/components/schemas/EmbeddingModel"
417
+ }
418
+ },
419
+ "required": [
420
+ "embedding"
421
+ ],
422
+ "type": "object"
423
+ },
424
+ {
425
+ "properties": {
426
+ "reranker": {
427
+ "$ref": "#/components/schemas/ClassifierModel"
428
+ }
429
+ },
430
+ "required": [
431
+ "reranker"
432
+ ],
433
+ "type": "object"
434
+ }
435
+ ]
436
+ },
437
+ "OpenAICompatEmbedding": {
438
+ "properties": {
439
+ "embedding": {
440
+ "$ref": "#/components/schemas/Embedding"
441
+ },
442
+ "index": {
443
+ "example": "0",
444
+ "minimum": 0,
445
+ "type": "integer"
446
+ },
447
+ "object": {
448
+ "example": "embedding",
449
+ "type": "string"
450
+ }
451
+ },
452
+ "required": [
453
+ "object",
454
+ "embedding",
455
+ "index"
456
+ ],
457
+ "type": "object"
458
+ },
459
+ "OpenAICompatErrorResponse": {
460
+ "properties": {
461
+ "code": {
462
+ "format": "int32",
463
+ "minimum": 0,
464
+ "type": "integer"
465
+ },
466
+ "error_type": {
467
+ "$ref": "#/components/schemas/ErrorType"
468
+ },
469
+ "message": {
470
+ "type": "string"
471
+ }
472
+ },
473
+ "required": [
474
+ "message",
475
+ "code",
476
+ "error_type"
477
+ ],
478
+ "type": "object"
479
+ },
480
+ "OpenAICompatRequest": {
481
+ "properties": {
482
+ "encoding_format": {
483
+ "allOf": [
484
+ {
485
+ "$ref": "#/components/schemas/EncodingFormat"
486
+ }
487
+ ],
488
+ "default": "float"
489
+ },
490
+ "input": {
491
+ "$ref": "#/components/schemas/Input"
492
+ },
493
+ "model": {
494
+ "example": "null",
495
+ "nullable": true,
496
+ "type": "string"
497
+ },
498
+ "user": {
499
+ "example": "null",
500
+ "nullable": true,
501
+ "type": "string"
502
+ }
503
+ },
504
+ "required": [
505
+ "input"
506
+ ],
507
+ "type": "object"
508
+ },
509
+ "OpenAICompatResponse": {
510
+ "properties": {
511
+ "data": {
512
+ "items": {
513
+ "$ref": "#/components/schemas/OpenAICompatEmbedding"
514
+ },
515
+ "type": "array"
516
+ },
517
+ "model": {
518
+ "example": "thenlper/gte-base",
519
+ "type": "string"
520
+ },
521
+ "object": {
522
+ "example": "list",
523
+ "type": "string"
524
+ },
525
+ "usage": {
526
+ "$ref": "#/components/schemas/OpenAICompatUsage"
527
+ }
528
+ },
529
+ "required": [
530
+ "object",
531
+ "data",
532
+ "model",
533
+ "usage"
534
+ ],
535
+ "type": "object"
536
+ },
537
+ "OpenAICompatUsage": {
538
+ "properties": {
539
+ "prompt_tokens": {
540
+ "example": "512",
541
+ "minimum": 0,
542
+ "type": "integer"
543
+ },
544
+ "total_tokens": {
545
+ "example": "512",
546
+ "minimum": 0,
547
+ "type": "integer"
548
+ }
549
+ },
550
+ "required": [
551
+ "prompt_tokens",
552
+ "total_tokens"
553
+ ],
554
+ "type": "object"
555
+ },
556
+ "PredictInput": {
557
+ "description": "Model input. Can be either a single string, a pair of strings or a batch of mixed single and pairs of strings.",
558
+ "example": "What is Deep Learning?",
559
+ "oneOf": [
560
+ {
561
+ "description": "A single string",
562
+ "type": "string"
563
+ },
564
+ {
565
+ "description": "A pair of strings",
566
+ "items": {
567
+ "type": "string"
568
+ },
569
+ "maxItems": 2,
570
+ "minItems": 2,
571
+ "type": "array"
572
+ },
573
+ {
574
+ "description": "A batch",
575
+ "items": {
576
+ "oneOf": [
577
+ {
578
+ "description": "A single string",
579
+ "items": {
580
+ "type": "string"
581
+ },
582
+ "maxItems": 1,
583
+ "minItems": 1,
584
+ "type": "array"
585
+ },
586
+ {
587
+ "description": "A pair of strings",
588
+ "items": {
589
+ "type": "string"
590
+ },
591
+ "maxItems": 2,
592
+ "minItems": 2,
593
+ "type": "array"
594
+ }
595
+ ]
596
+ },
597
+ "type": "array"
598
+ }
599
+ ]
600
+ },
601
+ "PredictRequest": {
602
+ "properties": {
603
+ "inputs": {
604
+ "$ref": "#/components/schemas/PredictInput"
605
+ },
606
+ "raw_scores": {
607
+ "default": "false",
608
+ "example": "false",
609
+ "type": "boolean"
610
+ },
611
+ "truncate": {
612
+ "default": "false",
613
+ "example": "false",
614
+ "nullable": true,
615
+ "type": "boolean"
616
+ },
617
+ "truncation_direction": {
618
+ "allOf": [
619
+ {
620
+ "$ref": "#/components/schemas/TruncationDirection"
621
+ }
622
+ ],
623
+ "default": "right"
624
+ }
625
+ },
626
+ "required": [
627
+ "inputs"
628
+ ],
629
+ "type": "object"
630
+ },
631
+ "PredictResponse": {
632
+ "oneOf": [
633
+ {
634
+ "items": {
635
+ "$ref": "#/components/schemas/Prediction"
636
+ },
637
+ "type": "array"
638
+ },
639
+ {
640
+ "items": {
641
+ "items": {
642
+ "$ref": "#/components/schemas/Prediction"
643
+ },
644
+ "type": "array"
645
+ },
646
+ "type": "array"
647
+ }
648
+ ]
649
+ },
650
+ "Prediction": {
651
+ "properties": {
652
+ "label": {
653
+ "example": "admiration",
654
+ "type": "string"
655
+ },
656
+ "score": {
657
+ "example": "0.5",
658
+ "format": "float",
659
+ "type": "number"
660
+ }
661
+ },
662
+ "required": [
663
+ "score",
664
+ "label"
665
+ ],
666
+ "type": "object"
667
+ },
668
+ "Rank": {
669
+ "properties": {
670
+ "index": {
671
+ "example": "0",
672
+ "minimum": 0,
673
+ "type": "integer"
674
+ },
675
+ "score": {
676
+ "example": "1.0",
677
+ "format": "float",
678
+ "type": "number"
679
+ },
680
+ "text": {
681
+ "default": "null",
682
+ "example": "Deep Learning is ...",
683
+ "nullable": true,
684
+ "type": "string"
685
+ }
686
+ },
687
+ "required": [
688
+ "index",
689
+ "score"
690
+ ],
691
+ "type": "object"
692
+ },
693
+ "RerankRequest": {
694
+ "properties": {
695
+ "query": {
696
+ "example": "What is Deep Learning?",
697
+ "type": "string"
698
+ },
699
+ "raw_scores": {
700
+ "default": "false",
701
+ "example": "false",
702
+ "type": "boolean"
703
+ },
704
+ "return_text": {
705
+ "default": "false",
706
+ "example": "false",
707
+ "type": "boolean"
708
+ },
709
+ "texts": {
710
+ "example": [
711
+ "Deep Learning is ..."
712
+ ],
713
+ "items": {
714
+ "type": "string"
715
+ },
716
+ "type": "array"
717
+ },
718
+ "truncate": {
719
+ "default": "false",
720
+ "example": "false",
721
+ "nullable": true,
722
+ "type": "boolean"
723
+ },
724
+ "truncation_direction": {
725
+ "allOf": [
726
+ {
727
+ "$ref": "#/components/schemas/TruncationDirection"
728
+ }
729
+ ],
730
+ "default": "right"
731
+ }
732
+ },
733
+ "required": [
734
+ "query",
735
+ "texts"
736
+ ],
737
+ "type": "object"
738
+ },
739
+ "RerankResponse": {
740
+ "items": {
741
+ "$ref": "#/components/schemas/Rank"
742
+ },
743
+ "type": "array"
744
+ },
745
+ "SimilarityInput": {
746
+ "properties": {
747
+ "sentences": {
748
+ "description": "A list of strings which will be compared against the source_sentence.",
749
+ "example": [
750
+ "What is Machine Learning?"
751
+ ],
752
+ "items": {
753
+ "type": "string"
754
+ },
755
+ "type": "array"
756
+ },
757
+ "source_sentence": {
758
+ "description": "The string that you wish to compare the other strings with. This can be a phrase, sentence,\nor longer passage, depending on the model being used.",
759
+ "example": "What is Deep Learning?",
760
+ "type": "string"
761
+ }
762
+ },
763
+ "required": [
764
+ "source_sentence",
765
+ "sentences"
766
+ ],
767
+ "type": "object"
768
+ },
769
+ "SimilarityParameters": {
770
+ "properties": {
771
+ "prompt_name": {
772
+ "default": "null",
773
+ "description": "The name of the prompt that should be used by for encoding. If not set, no prompt\nwill be applied.\n\nMust be a key in the `sentence-transformers` configuration `prompts` dictionary.\n\nFor example if ``prompt_name`` is \"query\" and the ``prompts`` is {\"query\": \"query: \", ...},\nthen the sentence \"What is the capital of France?\" will be encoded as\n\"query: What is the capital of France?\" because the prompt text will be prepended before\nany text to encode.",
774
+ "example": "null",
775
+ "nullable": true,
776
+ "type": "string"
777
+ },
778
+ "truncate": {
779
+ "default": "false",
780
+ "example": "false",
781
+ "nullable": true,
782
+ "type": "boolean"
783
+ },
784
+ "truncation_direction": {
785
+ "allOf": [
786
+ {
787
+ "$ref": "#/components/schemas/TruncationDirection"
788
+ }
789
+ ],
790
+ "default": "right"
791
+ }
792
+ },
793
+ "required": [
794
+ "truncation_direction"
795
+ ],
796
+ "type": "object"
797
+ },
798
+ "SimilarityRequest": {
799
+ "properties": {
800
+ "inputs": {
801
+ "$ref": "#/components/schemas/SimilarityInput"
802
+ },
803
+ "parameters": {
804
+ "allOf": [
805
+ {
806
+ "$ref": "#/components/schemas/SimilarityParameters"
807
+ }
808
+ ],
809
+ "default": "null",
810
+ "nullable": true
811
+ }
812
+ },
813
+ "required": [
814
+ "inputs"
815
+ ],
816
+ "type": "object"
817
+ },
818
+ "SimilarityResponse": {
819
+ "example": [
820
+ 0.0,
821
+ 1.0,
822
+ 0.5
823
+ ],
824
+ "items": {
825
+ "format": "float",
826
+ "type": "number"
827
+ },
828
+ "type": "array"
829
+ },
830
+ "SimpleToken": {
831
+ "properties": {
832
+ "id": {
833
+ "example": 0,
834
+ "format": "int32",
835
+ "minimum": 0,
836
+ "type": "integer"
837
+ },
838
+ "special": {
839
+ "example": "false",
840
+ "type": "boolean"
841
+ },
842
+ "start": {
843
+ "example": 0,
844
+ "minimum": 0,
845
+ "nullable": true,
846
+ "type": "integer"
847
+ },
848
+ "stop": {
849
+ "example": 2,
850
+ "minimum": 0,
851
+ "nullable": true,
852
+ "type": "integer"
853
+ },
854
+ "text": {
855
+ "example": "test",
856
+ "type": "string"
857
+ }
858
+ },
859
+ "required": [
860
+ "id",
861
+ "text",
862
+ "special"
863
+ ],
864
+ "type": "object"
865
+ },
866
+ "SparseValue": {
867
+ "properties": {
868
+ "index": {
869
+ "minimum": 0,
870
+ "type": "integer"
871
+ },
872
+ "value": {
873
+ "format": "float",
874
+ "type": "number"
875
+ }
876
+ },
877
+ "required": [
878
+ "index",
879
+ "value"
880
+ ],
881
+ "type": "object"
882
+ },
883
+ "TokenizeInput": {
884
+ "oneOf": [
885
+ {
886
+ "type": "string"
887
+ },
888
+ {
889
+ "items": {
890
+ "type": "string"
891
+ },
892
+ "type": "array"
893
+ }
894
+ ]
895
+ },
896
+ "TokenizeRequest": {
897
+ "properties": {
898
+ "add_special_tokens": {
899
+ "default": "true",
900
+ "example": "true",
901
+ "type": "boolean"
902
+ },
903
+ "inputs": {
904
+ "$ref": "#/components/schemas/TokenizeInput"
905
+ },
906
+ "prompt_name": {
907
+ "default": "null",
908
+ "description": "The name of the prompt that should be used by for encoding. If not set, no prompt\nwill be applied.\n\nMust be a key in the `sentence-transformers` configuration `prompts` dictionary.\n\nFor example if ``prompt_name`` is \"query\" and the ``prompts`` is {\"query\": \"query: \", ...},\nthen the sentence \"What is the capital of France?\" will be encoded as\n\"query: What is the capital of France?\" because the prompt text will be prepended before\nany text to encode.",
909
+ "example": "null",
910
+ "nullable": true,
911
+ "type": "string"
912
+ }
913
+ },
914
+ "required": [
915
+ "inputs"
916
+ ],
917
+ "type": "object"
918
+ },
919
+ "TokenizeResponse": {
920
+ "example": [
921
+ [
922
+ {
923
+ "id": 0,
924
+ "special": false,
925
+ "start": 0,
926
+ "stop": 2,
927
+ "text": "test"
928
+ }
929
+ ]
930
+ ],
931
+ "items": {
932
+ "items": {
933
+ "$ref": "#/components/schemas/SimpleToken"
934
+ },
935
+ "type": "array"
936
+ },
937
+ "type": "array"
938
+ },
939
+ "TruncationDirection": {
940
+ "enum": [
941
+ "Left",
942
+ "Right"
943
+ ],
944
+ "type": "string"
945
+ }
946
+ }
947
+ },
948
+ "info": {
949
+ "contact": {
950
+ "name": "Olivier Dehaene"
951
+ },
952
+ "description": "Text Embedding Webserver",
953
+ "license": {
954
+ "name": "Apache 2.0",
955
+ "url": "https://www.apache.org/licenses/LICENSE-2.0"
956
+ },
957
+ "title": "Text Embeddings Inference",
958
+ "version": "1.6.0"
959
+ },
960
+ "openapi": "3.0.3",
961
+ "paths": {
962
+ "/decode": {
963
+ "post": {
964
+ "operationId": "decode",
965
+ "requestBody": {
966
+ "content": {
967
+ "application/json": {
968
+ "schema": {
969
+ "$ref": "#/components/schemas/DecodeRequest"
970
+ }
971
+ }
972
+ },
973
+ "required": true
974
+ },
975
+ "responses": {
976
+ "200": {
977
+ "content": {
978
+ "application/json": {
979
+ "schema": {
980
+ "$ref": "#/components/schemas/DecodeResponse"
981
+ }
982
+ }
983
+ },
984
+ "description": "Decoded ids"
985
+ },
986
+ "422": {
987
+ "content": {
988
+ "application/json": {
989
+ "example": {
990
+ "message": "Tokenization error",
991
+ "type": "tokenizer"
992
+ },
993
+ "schema": {
994
+ "$ref": "#/components/schemas/ErrorResponse"
995
+ }
996
+ }
997
+ },
998
+ "description": "Tokenization error"
999
+ }
1000
+ },
1001
+ "summary": "Decode input ids",
1002
+ "tags": [
1003
+ "Text Embeddings Inference"
1004
+ ]
1005
+ }
1006
+ },
1007
+ "/embed": {
1008
+ "post": {
1009
+ "operationId": "embed",
1010
+ "requestBody": {
1011
+ "content": {
1012
+ "application/json": {
1013
+ "schema": {
1014
+ "$ref": "#/components/schemas/EmbedRequest"
1015
+ }
1016
+ }
1017
+ },
1018
+ "required": true
1019
+ },
1020
+ "responses": {
1021
+ "200": {
1022
+ "content": {
1023
+ "application/json": {
1024
+ "schema": {
1025
+ "$ref": "#/components/schemas/EmbedResponse"
1026
+ }
1027
+ }
1028
+ },
1029
+ "description": "Embeddings"
1030
+ },
1031
+ "413": {
1032
+ "content": {
1033
+ "application/json": {
1034
+ "example": {
1035
+ "error": "Batch size error",
1036
+ "error_type": "validation"
1037
+ },
1038
+ "schema": {
1039
+ "$ref": "#/components/schemas/ErrorResponse"
1040
+ }
1041
+ }
1042
+ },
1043
+ "description": "Batch size error"
1044
+ },
1045
+ "422": {
1046
+ "content": {
1047
+ "application/json": {
1048
+ "example": {
1049
+ "error": "Tokenization error",
1050
+ "error_type": "tokenizer"
1051
+ },
1052
+ "schema": {
1053
+ "$ref": "#/components/schemas/ErrorResponse"
1054
+ }
1055
+ }
1056
+ },
1057
+ "description": "Tokenization error"
1058
+ },
1059
+ "424": {
1060
+ "content": {
1061
+ "application/json": {
1062
+ "example": {
1063
+ "error": "Inference failed",
1064
+ "error_type": "backend"
1065
+ },
1066
+ "schema": {
1067
+ "$ref": "#/components/schemas/ErrorResponse"
1068
+ }
1069
+ }
1070
+ },
1071
+ "description": "Embedding Error"
1072
+ },
1073
+ "429": {
1074
+ "content": {
1075
+ "application/json": {
1076
+ "example": {
1077
+ "error": "Model is overloaded",
1078
+ "error_type": "overloaded"
1079
+ },
1080
+ "schema": {
1081
+ "$ref": "#/components/schemas/ErrorResponse"
1082
+ }
1083
+ }
1084
+ },
1085
+ "description": "Model is overloaded"
1086
+ }
1087
+ },
1088
+ "summary": "Get Embeddings. Returns a 424 status code if the model is not an embedding model.",
1089
+ "tags": [
1090
+ "Text Embeddings Inference"
1091
+ ]
1092
+ }
1093
+ },
1094
+ "/embed_all": {
1095
+ "post": {
1096
+ "description": "Returns a 424 status code if the model is not an embedding model.",
1097
+ "operationId": "embed_all",
1098
+ "requestBody": {
1099
+ "content": {
1100
+ "application/json": {
1101
+ "schema": {
1102
+ "$ref": "#/components/schemas/EmbedAllRequest"
1103
+ }
1104
+ }
1105
+ },
1106
+ "required": true
1107
+ },
1108
+ "responses": {
1109
+ "200": {
1110
+ "content": {
1111
+ "application/json": {
1112
+ "schema": {
1113
+ "$ref": "#/components/schemas/EmbedAllResponse"
1114
+ }
1115
+ }
1116
+ },
1117
+ "description": "Embeddings"
1118
+ },
1119
+ "413": {
1120
+ "content": {
1121
+ "application/json": {
1122
+ "example": {
1123
+ "error": "Batch size error",
1124
+ "error_type": "validation"
1125
+ },
1126
+ "schema": {
1127
+ "$ref": "#/components/schemas/ErrorResponse"
1128
+ }
1129
+ }
1130
+ },
1131
+ "description": "Batch size error"
1132
+ },
1133
+ "422": {
1134
+ "content": {
1135
+ "application/json": {
1136
+ "example": {
1137
+ "error": "Tokenization error",
1138
+ "error_type": "tokenizer"
1139
+ },
1140
+ "schema": {
1141
+ "$ref": "#/components/schemas/ErrorResponse"
1142
+ }
1143
+ }
1144
+ },
1145
+ "description": "Tokenization error"
1146
+ },
1147
+ "424": {
1148
+ "content": {
1149
+ "application/json": {
1150
+ "example": {
1151
+ "error": "Inference failed",
1152
+ "error_type": "backend"
1153
+ },
1154
+ "schema": {
1155
+ "$ref": "#/components/schemas/ErrorResponse"
1156
+ }
1157
+ }
1158
+ },
1159
+ "description": "Embedding Error"
1160
+ },
1161
+ "429": {
1162
+ "content": {
1163
+ "application/json": {
1164
+ "example": {
1165
+ "error": "Model is overloaded",
1166
+ "error_type": "overloaded"
1167
+ },
1168
+ "schema": {
1169
+ "$ref": "#/components/schemas/ErrorResponse"
1170
+ }
1171
+ }
1172
+ },
1173
+ "description": "Model is overloaded"
1174
+ }
1175
+ },
1176
+ "summary": "Get all Embeddings without Pooling.",
1177
+ "tags": [
1178
+ "Text Embeddings Inference"
1179
+ ]
1180
+ }
1181
+ },
1182
+ "/embed_sparse": {
1183
+ "post": {
1184
+ "operationId": "embed_sparse",
1185
+ "requestBody": {
1186
+ "content": {
1187
+ "application/json": {
1188
+ "schema": {
1189
+ "$ref": "#/components/schemas/EmbedSparseRequest"
1190
+ }
1191
+ }
1192
+ },
1193
+ "required": true
1194
+ },
1195
+ "responses": {
1196
+ "200": {
1197
+ "content": {
1198
+ "application/json": {
1199
+ "schema": {
1200
+ "$ref": "#/components/schemas/EmbedSparseResponse"
1201
+ }
1202
+ }
1203
+ },
1204
+ "description": "Embeddings"
1205
+ },
1206
+ "413": {
1207
+ "content": {
1208
+ "application/json": {
1209
+ "example": {
1210
+ "error": "Batch size error",
1211
+ "error_type": "validation"
1212
+ },
1213
+ "schema": {
1214
+ "$ref": "#/components/schemas/ErrorResponse"
1215
+ }
1216
+ }
1217
+ },
1218
+ "description": "Batch size error"
1219
+ },
1220
+ "422": {
1221
+ "content": {
1222
+ "application/json": {
1223
+ "example": {
1224
+ "error": "Tokenization error",
1225
+ "error_type": "tokenizer"
1226
+ },
1227
+ "schema": {
1228
+ "$ref": "#/components/schemas/ErrorResponse"
1229
+ }
1230
+ }
1231
+ },
1232
+ "description": "Tokenization error"
1233
+ },
1234
+ "424": {
1235
+ "content": {
1236
+ "application/json": {
1237
+ "example": {
1238
+ "error": "Inference failed",
1239
+ "error_type": "backend"
1240
+ },
1241
+ "schema": {
1242
+ "$ref": "#/components/schemas/ErrorResponse"
1243
+ }
1244
+ }
1245
+ },
1246
+ "description": "Embedding Error"
1247
+ },
1248
+ "429": {
1249
+ "content": {
1250
+ "application/json": {
1251
+ "example": {
1252
+ "error": "Model is overloaded",
1253
+ "error_type": "overloaded"
1254
+ },
1255
+ "schema": {
1256
+ "$ref": "#/components/schemas/ErrorResponse"
1257
+ }
1258
+ }
1259
+ },
1260
+ "description": "Model is overloaded"
1261
+ }
1262
+ },
1263
+ "summary": "Get Sparse Embeddings. Returns a 424 status code if the model is not an embedding model with SPLADE pooling.",
1264
+ "tags": [
1265
+ "Text Embeddings Inference"
1266
+ ]
1267
+ }
1268
+ },
1269
+ "/health": {
1270
+ "get": {
1271
+ "operationId": "health",
1272
+ "responses": {
1273
+ "200": {
1274
+ "description": "Everything is working fine"
1275
+ },
1276
+ "503": {
1277
+ "content": {
1278
+ "application/json": {
1279
+ "example": {
1280
+ "error": "unhealthy",
1281
+ "error_type": "unhealthy"
1282
+ },
1283
+ "schema": {
1284
+ "$ref": "#/components/schemas/ErrorResponse"
1285
+ }
1286
+ }
1287
+ },
1288
+ "description": "Text embeddings Inference is down"
1289
+ }
1290
+ },
1291
+ "summary": "Health check method",
1292
+ "tags": [
1293
+ "Text Embeddings Inference"
1294
+ ]
1295
+ }
1296
+ },
1297
+ "/info": {
1298
+ "get": {
1299
+ "operationId": "get_model_info",
1300
+ "responses": {
1301
+ "200": {
1302
+ "content": {
1303
+ "application/json": {
1304
+ "schema": {
1305
+ "$ref": "#/components/schemas/Info"
1306
+ }
1307
+ }
1308
+ },
1309
+ "description": "Served model info"
1310
+ }
1311
+ },
1312
+ "summary": "Text Embeddings Inference endpoint info",
1313
+ "tags": [
1314
+ "Text Embeddings Inference"
1315
+ ]
1316
+ }
1317
+ },
1318
+ "/metrics": {
1319
+ "get": {
1320
+ "operationId": "metrics",
1321
+ "responses": {
1322
+ "200": {
1323
+ "content": {
1324
+ "text/plain": {
1325
+ "schema": {
1326
+ "type": "string"
1327
+ }
1328
+ }
1329
+ },
1330
+ "description": "Prometheus Metrics"
1331
+ }
1332
+ },
1333
+ "summary": "Prometheus metrics scrape endpoint",
1334
+ "tags": [
1335
+ "Text Embeddings Inference"
1336
+ ]
1337
+ }
1338
+ },
1339
+ "/predict": {
1340
+ "post": {
1341
+ "operationId": "predict",
1342
+ "requestBody": {
1343
+ "content": {
1344
+ "application/json": {
1345
+ "schema": {
1346
+ "$ref": "#/components/schemas/PredictRequest"
1347
+ }
1348
+ }
1349
+ },
1350
+ "required": true
1351
+ },
1352
+ "responses": {
1353
+ "200": {
1354
+ "content": {
1355
+ "application/json": {
1356
+ "schema": {
1357
+ "$ref": "#/components/schemas/PredictResponse"
1358
+ }
1359
+ }
1360
+ },
1361
+ "description": "Predictions"
1362
+ },
1363
+ "413": {
1364
+ "content": {
1365
+ "application/json": {
1366
+ "example": {
1367
+ "error": "Batch size error",
1368
+ "error_type": "validation"
1369
+ },
1370
+ "schema": {
1371
+ "$ref": "#/components/schemas/ErrorResponse"
1372
+ }
1373
+ }
1374
+ },
1375
+ "description": "Batch size error"
1376
+ },
1377
+ "422": {
1378
+ "content": {
1379
+ "application/json": {
1380
+ "example": {
1381
+ "error": "Tokenization error",
1382
+ "error_type": "tokenizer"
1383
+ },
1384
+ "schema": {
1385
+ "$ref": "#/components/schemas/ErrorResponse"
1386
+ }
1387
+ }
1388
+ },
1389
+ "description": "Tokenization error"
1390
+ },
1391
+ "424": {
1392
+ "content": {
1393
+ "application/json": {
1394
+ "example": {
1395
+ "error": "Inference failed",
1396
+ "error_type": "backend"
1397
+ },
1398
+ "schema": {
1399
+ "$ref": "#/components/schemas/ErrorResponse"
1400
+ }
1401
+ }
1402
+ },
1403
+ "description": "Prediction Error"
1404
+ },
1405
+ "429": {
1406
+ "content": {
1407
+ "application/json": {
1408
+ "example": {
1409
+ "error": "Model is overloaded",
1410
+ "error_type": "overloaded"
1411
+ },
1412
+ "schema": {
1413
+ "$ref": "#/components/schemas/ErrorResponse"
1414
+ }
1415
+ }
1416
+ },
1417
+ "description": "Model is overloaded"
1418
+ }
1419
+ },
1420
+ "summary": "Get Predictions. Returns a 424 status code if the model is not a Sequence Classification model",
1421
+ "tags": [
1422
+ "Text Embeddings Inference"
1423
+ ]
1424
+ }
1425
+ },
1426
+ "/rerank": {
1427
+ "post": {
1428
+ "description": "a single class.",
1429
+ "operationId": "rerank",
1430
+ "requestBody": {
1431
+ "content": {
1432
+ "application/json": {
1433
+ "schema": {
1434
+ "$ref": "#/components/schemas/RerankRequest"
1435
+ }
1436
+ }
1437
+ },
1438
+ "required": true
1439
+ },
1440
+ "responses": {
1441
+ "200": {
1442
+ "content": {
1443
+ "application/json": {
1444
+ "schema": {
1445
+ "$ref": "#/components/schemas/RerankResponse"
1446
+ }
1447
+ }
1448
+ },
1449
+ "description": "Ranks"
1450
+ },
1451
+ "413": {
1452
+ "content": {
1453
+ "application/json": {
1454
+ "example": {
1455
+ "error": "Batch size error",
1456
+ "error_type": "validation"
1457
+ },
1458
+ "schema": {
1459
+ "$ref": "#/components/schemas/ErrorResponse"
1460
+ }
1461
+ }
1462
+ },
1463
+ "description": "Batch size error"
1464
+ },
1465
+ "422": {
1466
+ "content": {
1467
+ "application/json": {
1468
+ "example": {
1469
+ "error": "Tokenization error",
1470
+ "error_type": "tokenizer"
1471
+ },
1472
+ "schema": {
1473
+ "$ref": "#/components/schemas/ErrorResponse"
1474
+ }
1475
+ }
1476
+ },
1477
+ "description": "Tokenization error"
1478
+ },
1479
+ "424": {
1480
+ "content": {
1481
+ "application/json": {
1482
+ "example": {
1483
+ "error": "Inference failed",
1484
+ "error_type": "backend"
1485
+ },
1486
+ "schema": {
1487
+ "$ref": "#/components/schemas/ErrorResponse"
1488
+ }
1489
+ }
1490
+ },
1491
+ "description": "Rerank Error"
1492
+ },
1493
+ "429": {
1494
+ "content": {
1495
+ "application/json": {
1496
+ "example": {
1497
+ "error": "Model is overloaded",
1498
+ "error_type": "overloaded"
1499
+ },
1500
+ "schema": {
1501
+ "$ref": "#/components/schemas/ErrorResponse"
1502
+ }
1503
+ }
1504
+ },
1505
+ "description": "Model is overloaded"
1506
+ }
1507
+ },
1508
+ "summary": "Get Ranks. Returns a 424 status code if the model is not a Sequence Classification model with",
1509
+ "tags": [
1510
+ "Text Embeddings Inference"
1511
+ ]
1512
+ }
1513
+ },
1514
+ "/similarity": {
1515
+ "post": {
1516
+ "operationId": "similarity",
1517
+ "requestBody": {
1518
+ "content": {
1519
+ "application/json": {
1520
+ "schema": {
1521
+ "$ref": "#/components/schemas/SimilarityRequest"
1522
+ }
1523
+ }
1524
+ },
1525
+ "required": true
1526
+ },
1527
+ "responses": {
1528
+ "200": {
1529
+ "content": {
1530
+ "application/json": {
1531
+ "schema": {
1532
+ "$ref": "#/components/schemas/SimilarityResponse"
1533
+ }
1534
+ }
1535
+ },
1536
+ "description": "Sentence Similarity"
1537
+ },
1538
+ "413": {
1539
+ "content": {
1540
+ "application/json": {
1541
+ "example": {
1542
+ "error": "Batch size error",
1543
+ "error_type": "validation"
1544
+ },
1545
+ "schema": {
1546
+ "$ref": "#/components/schemas/ErrorResponse"
1547
+ }
1548
+ }
1549
+ },
1550
+ "description": "Batch size error"
1551
+ },
1552
+ "422": {
1553
+ "content": {
1554
+ "application/json": {
1555
+ "example": {
1556
+ "error": "Tokenization error",
1557
+ "error_type": "tokenizer"
1558
+ },
1559
+ "schema": {
1560
+ "$ref": "#/components/schemas/ErrorResponse"
1561
+ }
1562
+ }
1563
+ },
1564
+ "description": "Tokenization error"
1565
+ },
1566
+ "424": {
1567
+ "content": {
1568
+ "application/json": {
1569
+ "example": {
1570
+ "error": "Inference failed",
1571
+ "error_type": "backend"
1572
+ },
1573
+ "schema": {
1574
+ "$ref": "#/components/schemas/ErrorResponse"
1575
+ }
1576
+ }
1577
+ },
1578
+ "description": "Embedding Error"
1579
+ },
1580
+ "429": {
1581
+ "content": {
1582
+ "application/json": {
1583
+ "example": {
1584
+ "error": "Model is overloaded",
1585
+ "error_type": "overloaded"
1586
+ },
1587
+ "schema": {
1588
+ "$ref": "#/components/schemas/ErrorResponse"
1589
+ }
1590
+ }
1591
+ },
1592
+ "description": "Model is overloaded"
1593
+ }
1594
+ },
1595
+ "summary": "Get Sentence Similarity. Returns a 424 status code if the model is not an embedding model.",
1596
+ "tags": [
1597
+ "Text Embeddings Inference"
1598
+ ]
1599
+ }
1600
+ },
1601
+ "/tokenize": {
1602
+ "post": {
1603
+ "operationId": "tokenize",
1604
+ "requestBody": {
1605
+ "content": {
1606
+ "application/json": {
1607
+ "schema": {
1608
+ "$ref": "#/components/schemas/TokenizeRequest"
1609
+ }
1610
+ }
1611
+ },
1612
+ "required": true
1613
+ },
1614
+ "responses": {
1615
+ "200": {
1616
+ "content": {
1617
+ "application/json": {
1618
+ "schema": {
1619
+ "$ref": "#/components/schemas/TokenizeResponse"
1620
+ }
1621
+ }
1622
+ },
1623
+ "description": "Tokenized ids"
1624
+ },
1625
+ "422": {
1626
+ "content": {
1627
+ "application/json": {
1628
+ "example": {
1629
+ "message": "Tokenization error",
1630
+ "type": "tokenizer"
1631
+ },
1632
+ "schema": {
1633
+ "$ref": "#/components/schemas/ErrorResponse"
1634
+ }
1635
+ }
1636
+ },
1637
+ "description": "Tokenization error"
1638
+ }
1639
+ },
1640
+ "summary": "Tokenize inputs",
1641
+ "tags": [
1642
+ "Text Embeddings Inference"
1643
+ ]
1644
+ }
1645
+ },
1646
+ "/v1/embeddings": {
1647
+ "post": {
1648
+ "operationId": "openai_embed",
1649
+ "requestBody": {
1650
+ "content": {
1651
+ "application/json": {
1652
+ "schema": {
1653
+ "$ref": "#/components/schemas/OpenAICompatRequest"
1654
+ }
1655
+ }
1656
+ },
1657
+ "required": true
1658
+ },
1659
+ "responses": {
1660
+ "200": {
1661
+ "content": {
1662
+ "application/json": {
1663
+ "schema": {
1664
+ "$ref": "#/components/schemas/OpenAICompatResponse"
1665
+ }
1666
+ }
1667
+ },
1668
+ "description": "Embeddings"
1669
+ },
1670
+ "413": {
1671
+ "content": {
1672
+ "application/json": {
1673
+ "example": {
1674
+ "message": "Batch size error",
1675
+ "type": "validation"
1676
+ },
1677
+ "schema": {
1678
+ "$ref": "#/components/schemas/OpenAICompatErrorResponse"
1679
+ }
1680
+ }
1681
+ },
1682
+ "description": "Batch size error"
1683
+ },
1684
+ "422": {
1685
+ "content": {
1686
+ "application/json": {
1687
+ "example": {
1688
+ "message": "Tokenization error",
1689
+ "type": "tokenizer"
1690
+ },
1691
+ "schema": {
1692
+ "$ref": "#/components/schemas/OpenAICompatErrorResponse"
1693
+ }
1694
+ }
1695
+ },
1696
+ "description": "Tokenization error"
1697
+ },
1698
+ "424": {
1699
+ "content": {
1700
+ "application/json": {
1701
+ "example": {
1702
+ "message": "Inference failed",
1703
+ "type": "backend"
1704
+ },
1705
+ "schema": {
1706
+ "$ref": "#/components/schemas/OpenAICompatErrorResponse"
1707
+ }
1708
+ }
1709
+ },
1710
+ "description": "Embedding Error"
1711
+ },
1712
+ "429": {
1713
+ "content": {
1714
+ "application/json": {
1715
+ "example": {
1716
+ "message": "Model is overloaded",
1717
+ "type": "overloaded"
1718
+ },
1719
+ "schema": {
1720
+ "$ref": "#/components/schemas/OpenAICompatErrorResponse"
1721
+ }
1722
+ }
1723
+ },
1724
+ "description": "Model is overloaded"
1725
+ }
1726
+ },
1727
+ "summary": "OpenAI compatible route. Returns a 424 status code if the model is not an embedding model.",
1728
+ "tags": [
1729
+ "Text Embeddings Inference"
1730
+ ]
1731
+ }
1732
+ }
1733
+ },
1734
+ "tags": [
1735
+ {
1736
+ "description": "Hugging Face Text Embeddings Inference API",
1737
+ "name": "Text Embeddings Inference"
1738
+ }
1739
+ ]
1740
+ }