truefoundry 0.5.10rc1__py3-none-any.whl → 0.5.10rc2__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.

Potentially problematic release.


This version of truefoundry might be problematic. Click here for more details.

Files changed (221) hide show
  1. truefoundry/cli/display_util.py +1 -1
  2. truefoundry/deploy/auto_gen/models.py +1 -1
  3. truefoundry/deploy/io/output_callback.py +13 -6
  4. truefoundry/deploy/io/rich_output_callback.py +8 -4
  5. truefoundry/deploy/lib/clients/servicefoundry_client.py +78 -105
  6. truefoundry/deploy/lib/dao/apply.py +2 -2
  7. truefoundry/deploy/lib/dao/delete.py +2 -2
  8. truefoundry/deploy/lib/model/entity.py +37 -24
  9. truefoundry/ml/autogen/client/__init__.py +22 -14
  10. truefoundry/ml/autogen/client/api/auth_api.py +2 -2
  11. truefoundry/ml/autogen/client/api/deprecated_api.py +5 -338
  12. truefoundry/ml/autogen/client/api/experiments_api.py +77 -32
  13. truefoundry/ml/autogen/client/api/generate_code_snippet_api.py +2 -2
  14. truefoundry/ml/autogen/client/api/health_api.py +2 -2
  15. truefoundry/ml/autogen/client/api/metrics_api.py +4 -4
  16. truefoundry/ml/autogen/client/api/mlfoundry_artifacts_api.py +530 -140
  17. truefoundry/ml/autogen/client/api/run_artifacts_api.py +10 -5
  18. truefoundry/ml/autogen/client/api/runs_api.py +20 -20
  19. truefoundry/ml/autogen/client/api_client.py +2 -2
  20. truefoundry/ml/autogen/client/configuration.py +29 -57
  21. truefoundry/ml/autogen/client/exceptions.py +2 -2
  22. truefoundry/ml/autogen/client/models/__init__.py +22 -14
  23. truefoundry/ml/autogen/client/models/agent_app.py +13 -5
  24. truefoundry/ml/autogen/client/models/{agent.py → agent_manifest.py} +39 -30
  25. truefoundry/ml/autogen/client/models/{agent_open_api_tool.py → agent_open_api_tool_manifest.py} +50 -31
  26. truefoundry/ml/autogen/client/models/agent_open_api_tool_with_fqn.py +47 -26
  27. truefoundry/ml/autogen/client/models/agent_with_fqn.py +36 -25
  28. truefoundry/ml/autogen/client/models/{add_custom_metrics_to_model_version_request_dto.py → apply_request_dto.py} +23 -17
  29. truefoundry/ml/autogen/client/models/apply_response_dto.py +79 -0
  30. truefoundry/ml/autogen/client/models/artifact_dto.py +2 -2
  31. truefoundry/ml/autogen/client/models/{add_features_to_model_version_request_dto.py → artifact_path.py} +20 -17
  32. truefoundry/ml/autogen/client/models/artifact_response_dto.py +2 -2
  33. truefoundry/ml/autogen/client/models/artifact_type.py +3 -3
  34. truefoundry/ml/autogen/client/models/artifact_version_dto.py +7 -15
  35. truefoundry/ml/autogen/client/models/artifact_version_manifest.py +32 -19
  36. truefoundry/ml/autogen/client/models/artifact_version_response_dto.py +2 -2
  37. truefoundry/ml/autogen/client/models/artifact_version_serialization_format.py +2 -2
  38. truefoundry/ml/autogen/client/models/artifact_version_status.py +2 -2
  39. truefoundry/ml/autogen/client/models/assistant_message.py +9 -5
  40. truefoundry/ml/autogen/client/models/authorize_user_for_model_request_dto.py +2 -2
  41. truefoundry/ml/autogen/client/models/authorize_user_for_model_version_request_dto.py +2 -2
  42. truefoundry/ml/autogen/client/models/blob_storage_reference.py +4 -4
  43. truefoundry/ml/autogen/client/models/body_get_search_runs_get.py +2 -2
  44. truefoundry/ml/autogen/client/models/{chat_prompt.py → chat_prompt_manifest.py} +39 -32
  45. truefoundry/ml/autogen/client/models/{chat_prompt_messages_inner.py → chat_prompt_manifest_messages_inner.py} +12 -12
  46. truefoundry/ml/autogen/client/models/columns_dto.py +2 -2
  47. truefoundry/ml/autogen/client/models/command.py +2 -2
  48. truefoundry/ml/autogen/client/models/content.py +3 -3
  49. truefoundry/ml/autogen/client/models/content1.py +3 -3
  50. truefoundry/ml/autogen/client/models/content2.py +3 -3
  51. truefoundry/ml/autogen/client/models/content2_any_of_inner.py +2 -2
  52. truefoundry/ml/autogen/client/models/create_artifact_request_dto.py +2 -2
  53. truefoundry/ml/autogen/client/models/create_artifact_response_dto.py +2 -2
  54. truefoundry/ml/autogen/client/models/create_artifact_version_request_dto.py +2 -2
  55. truefoundry/ml/autogen/client/models/create_artifact_version_response_dto.py +2 -2
  56. truefoundry/ml/autogen/client/models/create_dataset_request_dto.py +21 -5
  57. truefoundry/ml/autogen/client/models/create_experiment_request_dto.py +2 -2
  58. truefoundry/ml/autogen/client/models/create_experiment_response_dto.py +2 -2
  59. truefoundry/ml/autogen/client/models/create_model_version_request_dto.py +4 -12
  60. truefoundry/ml/autogen/client/models/create_multi_part_upload_for_dataset_request_dto.py +2 -2
  61. truefoundry/ml/autogen/client/models/create_multi_part_upload_for_dataset_response_dto.py +2 -2
  62. truefoundry/ml/autogen/client/models/create_multi_part_upload_request_dto.py +2 -2
  63. truefoundry/ml/autogen/client/models/create_python_deployment_config_request_dto.py +2 -2
  64. truefoundry/ml/autogen/client/models/create_python_deployment_config_response_dto.py +2 -2
  65. truefoundry/ml/autogen/client/models/create_run_request_dto.py +2 -2
  66. truefoundry/ml/autogen/client/models/create_run_response_dto.py +2 -2
  67. truefoundry/ml/autogen/client/models/create_workflow_task_config_request_dto.py +2 -2
  68. truefoundry/ml/autogen/client/models/data_directory_manifest.py +126 -0
  69. truefoundry/ml/autogen/client/models/dataset_dto.py +13 -2
  70. truefoundry/ml/autogen/client/models/dataset_response_dto.py +2 -2
  71. truefoundry/ml/autogen/client/models/delete_artifact_versions_request_dto.py +2 -2
  72. truefoundry/ml/autogen/client/models/delete_dataset_request_dto.py +2 -2
  73. truefoundry/ml/autogen/client/models/delete_files_for_dataset_request_dto.py +2 -2
  74. truefoundry/ml/autogen/client/models/delete_model_version_request_dto.py +2 -2
  75. truefoundry/ml/autogen/client/models/delete_run_request.py +2 -2
  76. truefoundry/ml/autogen/client/models/delete_tag_request_dto.py +2 -2
  77. truefoundry/ml/autogen/client/models/experiment_dto.py +3 -3
  78. truefoundry/ml/autogen/client/models/experiment_id_request_dto.py +2 -2
  79. truefoundry/ml/autogen/client/models/experiment_response_dto.py +2 -2
  80. truefoundry/ml/autogen/client/models/experiment_tag_dto.py +2 -2
  81. truefoundry/ml/autogen/client/models/export_deployment_files_request_dto.py +2 -2
  82. truefoundry/ml/autogen/client/models/external_blob_storage_source.py +5 -4
  83. truefoundry/ml/autogen/client/models/fast_ai_framework.py +3 -3
  84. truefoundry/ml/autogen/client/models/file_info_dto.py +2 -2
  85. truefoundry/ml/autogen/client/models/finalize_artifact_version_request_dto.py +7 -15
  86. truefoundry/ml/autogen/client/models/framework.py +3 -3
  87. truefoundry/ml/autogen/client/models/get_artifact_version_aliases_response_dto.py +2 -2
  88. truefoundry/ml/autogen/client/models/get_experiment_response_dto.py +2 -2
  89. truefoundry/ml/autogen/client/models/get_latest_run_log_response_dto.py +2 -2
  90. truefoundry/ml/autogen/client/models/get_metric_history_response.py +2 -2
  91. truefoundry/ml/autogen/client/models/get_signed_url_for_dataset_write_request_dto.py +2 -2
  92. truefoundry/ml/autogen/client/models/get_signed_urls_for_artifact_version_read_request_dto.py +2 -2
  93. truefoundry/ml/autogen/client/models/get_signed_urls_for_artifact_version_read_response_dto.py +2 -2
  94. truefoundry/ml/autogen/client/models/get_signed_urls_for_artifact_version_write_request_dto.py +2 -2
  95. truefoundry/ml/autogen/client/models/get_signed_urls_for_artifact_version_write_response_dto.py +2 -2
  96. truefoundry/ml/autogen/client/models/get_signed_urls_for_dataset_read_request_dto.py +2 -2
  97. truefoundry/ml/autogen/client/models/get_signed_urls_for_dataset_read_response_dto.py +2 -2
  98. truefoundry/ml/autogen/client/models/get_signed_urls_for_dataset_write_response_dto.py +2 -2
  99. truefoundry/ml/autogen/client/models/get_tenant_id_response_dto.py +2 -2
  100. truefoundry/ml/autogen/client/models/gluon_framework.py +3 -3
  101. truefoundry/ml/autogen/client/models/h2_o_framework.py +3 -3
  102. truefoundry/ml/autogen/client/models/http_validation_error.py +2 -2
  103. truefoundry/ml/autogen/client/models/image_content_part.py +7 -5
  104. truefoundry/ml/autogen/client/models/image_url.py +3 -3
  105. truefoundry/ml/autogen/client/models/infer_method_name.py +2 -2
  106. truefoundry/ml/autogen/client/models/keras_framework.py +3 -3
  107. truefoundry/ml/autogen/client/models/latest_run_log_dto.py +2 -2
  108. truefoundry/ml/autogen/client/models/library_name.py +2 -2
  109. truefoundry/ml/autogen/client/models/light_gbm_framework.py +3 -3
  110. truefoundry/ml/autogen/client/models/list_artifact_versions_request_dto.py +2 -2
  111. truefoundry/ml/autogen/client/models/list_artifact_versions_response_dto.py +2 -2
  112. truefoundry/ml/autogen/client/models/list_artifacts_request_dto.py +2 -2
  113. truefoundry/ml/autogen/client/models/list_artifacts_response_dto.py +2 -2
  114. truefoundry/ml/autogen/client/models/list_colums_response_dto.py +2 -2
  115. truefoundry/ml/autogen/client/models/list_datasets_request_dto.py +2 -2
  116. truefoundry/ml/autogen/client/models/list_datasets_response_dto.py +2 -2
  117. truefoundry/ml/autogen/client/models/list_experiments_response_dto.py +2 -2
  118. truefoundry/ml/autogen/client/models/list_files_for_artifact_version_request_dto.py +2 -2
  119. truefoundry/ml/autogen/client/models/list_files_for_artifact_versions_response_dto.py +2 -2
  120. truefoundry/ml/autogen/client/models/list_files_for_dataset_request_dto.py +2 -2
  121. truefoundry/ml/autogen/client/models/list_files_for_dataset_response_dto.py +2 -2
  122. truefoundry/ml/autogen/client/models/list_latest_run_logs_response_dto.py +2 -2
  123. truefoundry/ml/autogen/client/models/list_metric_history_request_dto.py +2 -2
  124. truefoundry/ml/autogen/client/models/list_metric_history_response_dto.py +2 -2
  125. truefoundry/ml/autogen/client/models/list_model_version_response_dto.py +2 -2
  126. truefoundry/ml/autogen/client/models/list_model_versions_request_dto.py +2 -2
  127. truefoundry/ml/autogen/client/models/list_models_request_dto.py +2 -2
  128. truefoundry/ml/autogen/client/models/list_models_response_dto.py +2 -2
  129. truefoundry/ml/autogen/client/models/list_run_artifacts_response_dto.py +2 -2
  130. truefoundry/ml/autogen/client/models/list_run_logs_response_dto.py +2 -2
  131. truefoundry/ml/autogen/client/models/local_artifact_source.py +93 -0
  132. truefoundry/ml/autogen/client/models/local_model_source.py +80 -0
  133. truefoundry/ml/autogen/client/models/log_batch_request_dto.py +2 -2
  134. truefoundry/ml/autogen/client/models/log_metric_request_dto.py +2 -2
  135. truefoundry/ml/autogen/client/models/log_param_request_dto.py +2 -2
  136. truefoundry/ml/autogen/client/models/manifest.py +100 -23
  137. truefoundry/ml/autogen/client/models/manifest1.py +219 -0
  138. truefoundry/ml/autogen/client/models/manifest2.py +198 -0
  139. truefoundry/ml/autogen/client/models/method.py +3 -3
  140. truefoundry/ml/autogen/client/models/metric_collection_dto.py +2 -2
  141. truefoundry/ml/autogen/client/models/metric_dto.py +2 -2
  142. truefoundry/ml/autogen/client/models/mime_type.py +2 -2
  143. truefoundry/ml/autogen/client/models/model_configuration.py +7 -7
  144. truefoundry/ml/autogen/client/models/model_dto.py +2 -2
  145. truefoundry/ml/autogen/client/models/model_response_dto.py +2 -2
  146. truefoundry/ml/autogen/client/models/model_server.py +2 -2
  147. truefoundry/ml/autogen/client/models/model_version_dto.py +2 -2
  148. truefoundry/ml/autogen/client/models/model_version_environment.py +5 -5
  149. truefoundry/ml/autogen/client/models/model_version_manifest.py +35 -22
  150. truefoundry/ml/autogen/client/models/model_version_response_dto.py +2 -2
  151. truefoundry/ml/autogen/client/models/multi_part_upload_dto.py +2 -2
  152. truefoundry/ml/autogen/client/models/multi_part_upload_response_dto.py +2 -2
  153. truefoundry/ml/autogen/client/models/multi_part_upload_storage_provider.py +2 -2
  154. truefoundry/ml/autogen/client/models/notify_artifact_version_failure_dto.py +2 -2
  155. truefoundry/ml/autogen/client/models/onnx_framework.py +3 -3
  156. truefoundry/ml/autogen/client/models/openapi_spec.py +3 -3
  157. truefoundry/ml/autogen/client/models/paddle_framework.py +3 -3
  158. truefoundry/ml/autogen/client/models/param_dto.py +2 -2
  159. truefoundry/ml/autogen/client/models/parameters.py +3 -3
  160. truefoundry/ml/autogen/client/models/py_torch_framework.py +3 -3
  161. truefoundry/ml/autogen/client/models/resolve_agent_app_response_dto.py +2 -2
  162. truefoundry/ml/autogen/client/models/restore_run_request_dto.py +2 -2
  163. truefoundry/ml/autogen/client/models/run_data_dto.py +2 -2
  164. truefoundry/ml/autogen/client/models/run_dto.py +2 -2
  165. truefoundry/ml/autogen/client/models/run_info_dto.py +2 -2
  166. truefoundry/ml/autogen/client/models/run_log_dto.py +2 -2
  167. truefoundry/ml/autogen/client/models/run_log_input_dto.py +2 -2
  168. truefoundry/ml/autogen/client/models/run_response_dto.py +2 -2
  169. truefoundry/ml/autogen/client/models/run_tag_dto.py +2 -2
  170. truefoundry/ml/autogen/client/models/search_runs_request_dto.py +2 -2
  171. truefoundry/ml/autogen/client/models/search_runs_response_dto.py +2 -2
  172. truefoundry/ml/autogen/client/models/set_experiment_tag_request_dto.py +2 -2
  173. truefoundry/ml/autogen/client/models/set_tag_request_dto.py +2 -2
  174. truefoundry/ml/autogen/client/models/signed_url_dto.py +2 -2
  175. truefoundry/ml/autogen/client/models/sklearn_framework.py +4 -4
  176. truefoundry/ml/autogen/client/models/sklearn_model_schema.py +7 -5
  177. truefoundry/ml/autogen/client/models/sklearn_serialization_format.py +3 -3
  178. truefoundry/ml/autogen/client/models/source.py +32 -7
  179. truefoundry/ml/autogen/client/models/source1.py +3 -3
  180. truefoundry/ml/autogen/client/models/{internal_metadata.py → source2.py} +52 -55
  181. truefoundry/ml/autogen/client/models/spa_cy_framework.py +3 -3
  182. truefoundry/ml/autogen/client/models/stats_models_framework.py +3 -3
  183. truefoundry/ml/autogen/client/models/stop.py +2 -2
  184. truefoundry/ml/autogen/client/models/store_run_logs_request_dto.py +2 -2
  185. truefoundry/ml/autogen/client/models/subject.py +2 -2
  186. truefoundry/ml/autogen/client/models/subject_type.py +2 -2
  187. truefoundry/ml/autogen/client/models/system_message.py +7 -4
  188. truefoundry/ml/autogen/client/models/tensor_flow_framework.py +3 -3
  189. truefoundry/ml/autogen/client/models/text.py +3 -3
  190. truefoundry/ml/autogen/client/models/text_content_part.py +6 -4
  191. truefoundry/ml/autogen/client/models/transformers_framework.py +5 -5
  192. truefoundry/ml/autogen/client/models/trigger_job_run_config_request_dto.py +2 -2
  193. truefoundry/ml/autogen/client/models/trigger_job_run_config_response_dto.py +2 -2
  194. truefoundry/ml/autogen/client/models/true_foundry_managed_source.py +3 -3
  195. truefoundry/ml/autogen/client/models/update_artifact_version_request_dto.py +5 -7
  196. truefoundry/ml/autogen/client/models/update_dataset_request_dto.py +13 -3
  197. truefoundry/ml/autogen/client/models/update_experiment_request_dto.py +2 -2
  198. truefoundry/ml/autogen/client/models/update_model_version_request_dto.py +2 -2
  199. truefoundry/ml/autogen/client/models/update_run_request_dto.py +2 -2
  200. truefoundry/ml/autogen/client/models/update_run_response_dto.py +2 -2
  201. truefoundry/ml/autogen/client/models/url.py +3 -3
  202. truefoundry/ml/autogen/client/models/user_message.py +8 -5
  203. truefoundry/ml/autogen/client/models/validate_external_storage_root_request_dto.py +2 -2
  204. truefoundry/ml/autogen/client/models/validate_external_storage_root_response_dto.py +2 -2
  205. truefoundry/ml/autogen/client/models/validation_error.py +2 -2
  206. truefoundry/ml/autogen/client/models/validation_error_loc_inner.py +2 -2
  207. truefoundry/ml/autogen/client/models/xg_boost_framework.py +4 -4
  208. truefoundry/ml/autogen/client/models/xg_boost_model_schema.py +7 -5
  209. truefoundry/ml/autogen/client/models/xg_boost_serialization_format.py +3 -3
  210. truefoundry/ml/autogen/client/rest.py +2 -2
  211. truefoundry/ml/autogen/client_README.md +17 -19
  212. truefoundry/ml/autogen/entities/artifacts.py +264 -139
  213. truefoundry/ml/clients/servicefoundry_client.py +3 -2
  214. truefoundry/ml/log_types/artifacts/artifact.py +15 -78
  215. truefoundry/ml/log_types/artifacts/model.py +4 -5
  216. truefoundry/ml/log_types/artifacts/utils.py +18 -2
  217. truefoundry/ml/mlfoundry_api.py +0 -40
  218. {truefoundry-0.5.10rc1.dist-info → truefoundry-0.5.10rc2.dist-info}/METADATA +1 -1
  219. {truefoundry-0.5.10rc1.dist-info → truefoundry-0.5.10rc2.dist-info}/RECORD +221 -215
  220. {truefoundry-0.5.10rc1.dist-info → truefoundry-0.5.10rc2.dist-info}/WHEEL +0 -0
  221. {truefoundry-0.5.10rc1.dist-info → truefoundry-0.5.10rc2.dist-info}/entry_points.txt +0 -0
@@ -1,11 +1,11 @@
1
1
  # coding: utf-8
2
2
 
3
3
  """
4
- FastAPI
4
+ TrueFoundry ML API
5
5
 
6
6
  No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
7
 
8
- The version of the OpenAPI document: 0.1.0
8
+ The version of the OpenAPI document: 1.0.0
9
9
  Generated by OpenAPI Generator (https://openapi-generator.tech)
10
10
 
11
11
  Do not edit the class manually.
@@ -1,11 +1,11 @@
1
1
  # coding: utf-8
2
2
 
3
3
  """
4
- FastAPI
4
+ TrueFoundry ML API
5
5
 
6
6
  No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
7
 
8
- The version of the OpenAPI document: 0.1.0
8
+ The version of the OpenAPI document: 1.0.0
9
9
  Generated by OpenAPI Generator (https://openapi-generator.tech)
10
10
 
11
11
  Do not edit the class manually.
@@ -24,13 +24,17 @@ from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
24
24
 
25
25
  class AssistantMessage(BaseModel):
26
26
  """
27
- +usage=Assistant message +docs=Defines the structure of an assistant message, including role and content +label=Assistant Message # noqa: E501
27
+ +label=Assistant Message +usage=Assistant message turn # noqa: E501
28
28
  """
29
29
 
30
- role: StrictStr = Field(...)
30
+ role: StrictStr = Field(
31
+ default=...,
32
+ description="+value=assistant +label=Role +usage=Role of the message +uiType=Ignore",
33
+ )
31
34
  content: Content = Field(...)
32
35
  name: Optional[StrictStr] = Field(
33
- default=None, description="+label=Name +usage=Name of the assistant message"
36
+ default=None,
37
+ description="+label=Name +usage=Name of the assistant this message is from",
34
38
  )
35
39
  __properties = ["role", "content", "name"]
36
40
 
@@ -1,11 +1,11 @@
1
1
  # coding: utf-8
2
2
 
3
3
  """
4
- FastAPI
4
+ TrueFoundry ML API
5
5
 
6
6
  No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
7
 
8
- The version of the OpenAPI document: 0.1.0
8
+ The version of the OpenAPI document: 1.0.0
9
9
  Generated by OpenAPI Generator (https://openapi-generator.tech)
10
10
 
11
11
  Do not edit the class manually.
@@ -1,11 +1,11 @@
1
1
  # coding: utf-8
2
2
 
3
3
  """
4
- FastAPI
4
+ TrueFoundry ML API
5
5
 
6
6
  No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
7
 
8
- The version of the OpenAPI document: 0.1.0
8
+ The version of the OpenAPI document: 1.0.0
9
9
  Generated by OpenAPI Generator (https://openapi-generator.tech)
10
10
 
11
11
  Do not edit the class manually.
@@ -1,11 +1,11 @@
1
1
  # coding: utf-8
2
2
 
3
3
  """
4
- FastAPI
4
+ TrueFoundry ML API
5
5
 
6
6
  No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
7
 
8
- The version of the OpenAPI document: 0.1.0
8
+ The version of the OpenAPI document: 1.0.0
9
9
  Generated by OpenAPI Generator (https://openapi-generator.tech)
10
10
 
11
11
  Do not edit the class manually.
@@ -23,10 +23,10 @@ from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, constr, validat
23
23
 
24
24
  class BlobStorageReference(BaseModel):
25
25
  """
26
- +usage=Blob Storage Location +label=Blob Storage Location +docs=Defines the structure for blob storage content references, including type, path, and mime_type # noqa: E501
26
+ +label=Blob Storage Reference +usage=A reference to content offloaded to blob storage +docs=Defines the structure for blob storage content references, including type, path, and mime_type # noqa: E501
27
27
  """
28
28
 
29
- type: StrictStr = Field(...)
29
+ type: StrictStr = Field(default=..., description="+value=blob-storage")
30
30
  path: constr(strict=True) = Field(
31
31
  default=...,
32
32
  description="+label=Path +usage=Path to the content in blob storage",
@@ -1,11 +1,11 @@
1
1
  # coding: utf-8
2
2
 
3
3
  """
4
- FastAPI
4
+ TrueFoundry ML API
5
5
 
6
6
  No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
7
 
8
- The version of the OpenAPI document: 0.1.0
8
+ The version of the OpenAPI document: 1.0.0
9
9
  Generated by OpenAPI Generator (https://openapi-generator.tech)
10
10
 
11
11
  Do not edit the class manually.
@@ -1,11 +1,11 @@
1
1
  # coding: utf-8
2
2
 
3
3
  """
4
- FastAPI
4
+ TrueFoundry ML API
5
5
 
6
6
  No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
7
 
8
- The version of the OpenAPI document: 0.1.0
8
+ The version of the OpenAPI document: 1.0.0
9
9
  Generated by OpenAPI Generator (https://openapi-generator.tech)
10
10
 
11
11
  Do not edit the class manually.
@@ -18,87 +18,93 @@ import pprint
18
18
  import re # noqa: F401
19
19
  from typing import Any, Dict, Optional
20
20
 
21
- from truefoundry.ml.autogen.client.models.chat_prompt_messages_inner import (
22
- ChatPromptMessagesInner,
21
+ from truefoundry.ml.autogen.client.models.chat_prompt_manifest_messages_inner import (
22
+ ChatPromptManifestMessagesInner,
23
23
  )
24
24
  from truefoundry.ml.autogen.client.models.model_configuration import ModelConfiguration
25
25
  from truefoundry.pydantic_v1 import (
26
26
  BaseModel,
27
27
  Field,
28
28
  StrictStr,
29
+ conint,
29
30
  conlist,
30
31
  constr,
31
32
  validator,
32
33
  )
33
34
 
34
35
 
35
- class ChatPrompt(BaseModel):
36
+ class ChatPromptManifest(BaseModel):
36
37
  """
37
- Chat Prompt artifact. # noqa: E501
38
+ Chat Prompt manifest. # noqa: E501
38
39
  """
39
40
 
41
+ name: Optional[constr(strict=True, max_length=256)] = Field(
42
+ default=None,
43
+ description="+sort=1 +label=Name +message=The name should start with lowercase alphabets and can contain alphanumeric and can include '-' in between +usage=Name of the entity",
44
+ )
40
45
  description: Optional[constr(strict=True, max_length=512)] = Field(
41
46
  default=None, description="+sort=2 +label=Description"
42
47
  )
43
48
  metadata: Dict[str, Any] = Field(
44
49
  default=...,
45
- description="+label=Metadata +docs=Key value pairs to store additional metadata +usage=Key value pairs to store additional metadata +uiType=JsonInput",
50
+ description='+label=Metadata +usage=Key value metadata. Should be valid JSON. For e.g. `{"business-unit": "sales", "quality": "good", "rating": 4.5}` +uiType=JsonInput +uiProps={"descriptionInline":true}',
46
51
  )
47
52
  version_alias: Optional[constr(strict=True, max_length=128)] = Field(
48
53
  default=None,
49
- description="+label=Version Alias +usage=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc) +docs=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc) +message=The version alias should start with 'v' followed by alphanumeric and it can include '.' and '-' in between (e.g. v1.0.0, v1-prod, v3-dev, etc)",
54
+ description='+label=Version Alias +usage=Version alias is alternate, ideally human readable, version string to reference an artifact version. It should start with `v` followed by alphanumeric and it can include `.` and `-` in between (e.g. `v1.0.0`, `v1-prod`, `v3-dev`, etc) +message=The version alias should start with `v` followed by alphanumeric and can include `.` and `-` in between (e.g. `v1.0.0`, `v1-prod`, `v3-dev`, etc) +uiProps={"descriptionInline":true} +placeholder=v1.0.0',
50
55
  )
51
- name: Optional[constr(strict=True, max_length=256)] = Field(
56
+ ml_repo: Optional[constr(strict=True)] = Field(
52
57
  default=None,
53
- description="+sort=1 +label=Name +message=The name should start with lowercase alphabets and can contain alphanumeric and can include '-' in between",
58
+ description="+label=ML Repo +usage=Name of the ML Repo +uiType=Hidden",
54
59
  )
55
- ml_repo: Optional[constr(strict=True)] = Field(
60
+ version: Optional[conint(strict=True, ge=1)] = Field(
56
61
  default=None,
57
- description="+label=ML Repo +usage=name of the ML Repo +docs=name of the ML Repo +uiType=Hidden",
62
+ description="+label=Version of the entity +usage=Version of the entity +uiType=Hidden",
58
63
  )
59
64
  type: Optional[StrictStr] = "chat_prompt"
60
- messages: conlist(ChatPromptMessagesInner) = Field(
65
+ messages: conlist(ChatPromptManifestMessagesInner) = Field(
61
66
  default=...,
62
- description="+sort=4000 +usage=Chat completion messages +label=Messages in the chat conversation +message=Chat completion messages +usage=List of messages in the chat conversation, must be non-empty +docs=Messages that define the chat conversation, including system, assistant, and user messages.",
67
+ description="+sort=4000 +label=Messages in the chat conversation +usage=List of messages in the chat conversation, must be non-empty",
63
68
  )
64
69
  variables: Optional[Dict[str, StrictStr]] = Field(
65
70
  default=None,
66
- description="+label=Variables +usage=Variables for the chat completion messages to be used in the prompt messages +sort=5000",
71
+ description="+label=Variables +usage=Variables referenced in messages and that can be replaced when running generation +sort=5000",
67
72
  )
68
73
  model_configuration: ModelConfiguration = Field(...)
69
74
  __properties = [
75
+ "name",
70
76
  "description",
71
77
  "metadata",
72
78
  "version_alias",
73
- "name",
74
79
  "ml_repo",
80
+ "version",
75
81
  "type",
76
82
  "messages",
77
83
  "variables",
78
84
  "model_configuration",
79
85
  ]
80
86
 
81
- @validator("version_alias")
82
- def version_alias_validate_regular_expression(cls, value):
87
+ @validator("name")
88
+ def name_validate_regular_expression(cls, value):
83
89
  """Validates the regular expression"""
84
90
  if value is None:
85
91
  return value
86
92
 
87
- if not re.match(r"^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$", value):
93
+ if not re.match(r"^[A-Za-z0-9_\-]+$", value):
88
94
  raise ValueError(
89
- r"must validate the regular expression /^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$/"
95
+ r"must validate the regular expression /^[A-Za-z0-9_\-]+$/"
90
96
  )
91
97
  return value
92
98
 
93
- @validator("name")
94
- def name_validate_regular_expression(cls, value):
99
+ @validator("version_alias")
100
+ def version_alias_validate_regular_expression(cls, value):
95
101
  """Validates the regular expression"""
96
102
  if value is None:
97
103
  return value
98
104
 
99
- if not re.match(r"^[A-Za-z0-9_\-]+$", value):
105
+ if not re.match(r"^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$", value):
100
106
  raise ValueError(
101
- r"must validate the regular expression /^[A-Za-z0-9_\-]+$/"
107
+ r"must validate the regular expression /^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$/"
102
108
  )
103
109
  return value
104
110
 
@@ -139,8 +145,8 @@ class ChatPrompt(BaseModel):
139
145
  return json.dumps(self.to_dict())
140
146
 
141
147
  @classmethod
142
- def from_json(cls, json_str: str) -> ChatPrompt:
143
- """Create an instance of ChatPrompt from a JSON string"""
148
+ def from_json(cls, json_str: str) -> ChatPromptManifest:
149
+ """Create an instance of ChatPromptManifest from a JSON string"""
144
150
  return cls.from_dict(json.loads(json_str))
145
151
 
146
152
  def to_dict(self):
@@ -159,26 +165,27 @@ class ChatPrompt(BaseModel):
159
165
  return _dict
160
166
 
161
167
  @classmethod
162
- def from_dict(cls, obj: dict) -> ChatPrompt:
163
- """Create an instance of ChatPrompt from a dict"""
168
+ def from_dict(cls, obj: dict) -> ChatPromptManifest:
169
+ """Create an instance of ChatPromptManifest from a dict"""
164
170
  if obj is None:
165
171
  return None
166
172
 
167
173
  if not isinstance(obj, dict):
168
- return ChatPrompt.parse_obj(obj)
174
+ return ChatPromptManifest.parse_obj(obj)
169
175
 
170
- _obj = ChatPrompt.parse_obj(
176
+ _obj = ChatPromptManifest.parse_obj(
171
177
  {
178
+ "name": obj.get("name"),
172
179
  "description": obj.get("description"),
173
180
  "metadata": obj.get("metadata"),
174
181
  "version_alias": obj.get("version_alias"),
175
- "name": obj.get("name"),
176
182
  "ml_repo": obj.get("ml_repo"),
183
+ "version": obj.get("version"),
177
184
  "type": obj.get("type")
178
185
  if obj.get("type") is not None
179
186
  else "chat_prompt",
180
187
  "messages": [
181
- ChatPromptMessagesInner.from_dict(_item)
188
+ ChatPromptManifestMessagesInner.from_dict(_item)
182
189
  for _item in obj.get("messages")
183
190
  ]
184
191
  if obj.get("messages") is not None
@@ -1,11 +1,11 @@
1
1
  # coding: utf-8
2
2
 
3
3
  """
4
- FastAPI
4
+ TrueFoundry ML API
5
5
 
6
6
  No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
7
 
8
- The version of the OpenAPI document: 0.1.0
8
+ The version of the OpenAPI document: 1.0.0
9
9
  Generated by OpenAPI Generator (https://openapi-generator.tech)
10
10
 
11
11
  Do not edit the class manually.
@@ -28,16 +28,16 @@ from truefoundry.pydantic_v1 import (
28
28
  validator,
29
29
  )
30
30
 
31
- CHATPROMPTMESSAGESINNER_ANY_OF_SCHEMAS = [
31
+ CHATPROMPTMANIFESTMESSAGESINNER_ANY_OF_SCHEMAS = [
32
32
  "AssistantMessage",
33
33
  "SystemMessage",
34
34
  "UserMessage",
35
35
  ]
36
36
 
37
37
 
38
- class ChatPromptMessagesInner(BaseModel):
38
+ class ChatPromptManifestMessagesInner(BaseModel):
39
39
  """
40
- ChatPromptMessagesInner
40
+ ChatPromptManifestMessagesInner
41
41
  """
42
42
 
43
43
  # data type: SystemMessage
@@ -51,7 +51,7 @@ class ChatPromptMessagesInner(BaseModel):
51
51
  else:
52
52
  actual_instance: Any
53
53
  any_of_schemas: List[str] = Field(
54
- CHATPROMPTMESSAGESINNER_ANY_OF_SCHEMAS, const=True
54
+ CHATPROMPTMANIFESTMESSAGESINNER_ANY_OF_SCHEMAS, const=True
55
55
  )
56
56
 
57
57
  class Config:
@@ -73,7 +73,7 @@ class ChatPromptMessagesInner(BaseModel):
73
73
 
74
74
  @validator("actual_instance")
75
75
  def actual_instance_must_validate_anyof(cls, v):
76
- instance = ChatPromptMessagesInner.construct()
76
+ instance = ChatPromptManifestMessagesInner.construct()
77
77
  error_messages = []
78
78
  # validate data type: SystemMessage
79
79
  if not isinstance(v, SystemMessage):
@@ -100,20 +100,20 @@ class ChatPromptMessagesInner(BaseModel):
100
100
  if error_messages:
101
101
  # no match
102
102
  raise ValueError(
103
- "No match found when setting the actual_instance in ChatPromptMessagesInner with anyOf schemas: AssistantMessage, SystemMessage, UserMessage. Details: "
103
+ "No match found when setting the actual_instance in ChatPromptManifestMessagesInner with anyOf schemas: AssistantMessage, SystemMessage, UserMessage. Details: "
104
104
  + ", ".join(error_messages)
105
105
  )
106
106
  else:
107
107
  return v
108
108
 
109
109
  @classmethod
110
- def from_dict(cls, obj: dict) -> ChatPromptMessagesInner:
110
+ def from_dict(cls, obj: dict) -> ChatPromptManifestMessagesInner:
111
111
  return cls.from_json(json.dumps(obj))
112
112
 
113
113
  @classmethod
114
- def from_json(cls, json_str: str) -> ChatPromptMessagesInner:
114
+ def from_json(cls, json_str: str) -> ChatPromptManifestMessagesInner:
115
115
  """Returns the object represented by the json string"""
116
- instance = ChatPromptMessagesInner.construct()
116
+ instance = ChatPromptManifestMessagesInner.construct()
117
117
  error_messages = []
118
118
  # anyof_schema_1_validator: Optional[SystemMessage] = None
119
119
  try:
@@ -137,7 +137,7 @@ class ChatPromptMessagesInner(BaseModel):
137
137
  if error_messages:
138
138
  # no match
139
139
  raise ValueError(
140
- "No match found when deserializing the JSON string into ChatPromptMessagesInner with anyOf schemas: AssistantMessage, SystemMessage, UserMessage. Details: "
140
+ "No match found when deserializing the JSON string into ChatPromptManifestMessagesInner with anyOf schemas: AssistantMessage, SystemMessage, UserMessage. Details: "
141
141
  + ", ".join(error_messages)
142
142
  )
143
143
  else:
@@ -1,11 +1,11 @@
1
1
  # coding: utf-8
2
2
 
3
3
  """
4
- FastAPI
4
+ TrueFoundry ML API
5
5
 
6
6
  No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
7
 
8
- The version of the OpenAPI document: 0.1.0
8
+ The version of the OpenAPI document: 1.0.0
9
9
  Generated by OpenAPI Generator (https://openapi-generator.tech)
10
10
 
11
11
  Do not edit the class manually.
@@ -1,11 +1,11 @@
1
1
  # coding: utf-8
2
2
 
3
3
  """
4
- FastAPI
4
+ TrueFoundry ML API
5
5
 
6
6
  No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
7
 
8
- The version of the OpenAPI document: 0.1.0
8
+ The version of the OpenAPI document: 1.0.0
9
9
  Generated by OpenAPI Generator (https://openapi-generator.tech)
10
10
 
11
11
  Do not edit the class manually.
@@ -1,11 +1,11 @@
1
1
  # coding: utf-8
2
2
 
3
3
  """
4
- FastAPI
4
+ TrueFoundry ML API
5
5
 
6
6
  No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
7
 
8
- The version of the OpenAPI document: 0.1.0
8
+ The version of the OpenAPI document: 1.0.0
9
9
  Generated by OpenAPI Generator (https://openapi-generator.tech)
10
10
 
11
11
  Do not edit the class manually.
@@ -34,7 +34,7 @@ CONTENT_ANY_OF_SCHEMAS = ["BlobStorageReference", "str"]
34
34
 
35
35
  class Content(BaseModel):
36
36
  """
37
- +label=Content +usage=Text content for the assistant message
37
+ +label=Content +usage=Text content for the assistant message +uiType=TextArea
38
38
  """
39
39
 
40
40
  # data type: str
@@ -1,11 +1,11 @@
1
1
  # coding: utf-8
2
2
 
3
3
  """
4
- FastAPI
4
+ TrueFoundry ML API
5
5
 
6
6
  No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
7
 
8
- The version of the OpenAPI document: 0.1.0
8
+ The version of the OpenAPI document: 1.0.0
9
9
  Generated by OpenAPI Generator (https://openapi-generator.tech)
10
10
 
11
11
  Do not edit the class manually.
@@ -34,7 +34,7 @@ CONTENT1_ANY_OF_SCHEMAS = ["BlobStorageReference", "str"]
34
34
 
35
35
  class Content1(BaseModel):
36
36
  """
37
- +label=Content +usage=Text content for the system message
37
+ +label=Content +usage=Text content for the system message +uiType=TextArea
38
38
  """
39
39
 
40
40
  # data type: str
@@ -1,11 +1,11 @@
1
1
  # coding: utf-8
2
2
 
3
3
  """
4
- FastAPI
4
+ TrueFoundry ML API
5
5
 
6
6
  No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
7
 
8
- The version of the OpenAPI document: 0.1.0
8
+ The version of the OpenAPI document: 1.0.0
9
9
  Generated by OpenAPI Generator (https://openapi-generator.tech)
10
10
 
11
11
  Do not edit the class manually.
@@ -38,7 +38,7 @@ CONTENT2_ANY_OF_SCHEMAS = ["BlobStorageReference", "List[Content2AnyOfInner]", "
38
38
 
39
39
  class Content2(BaseModel):
40
40
  """
41
- +label=Content +usage=Text content for the user message
41
+ +label=Content +usage=Content of the user message. can be a mix of text and images +uiType=TextArea
42
42
  """
43
43
 
44
44
  # data type: str
@@ -1,11 +1,11 @@
1
1
  # coding: utf-8
2
2
 
3
3
  """
4
- FastAPI
4
+ TrueFoundry ML API
5
5
 
6
6
  No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
7
 
8
- The version of the OpenAPI document: 0.1.0
8
+ The version of the OpenAPI document: 1.0.0
9
9
  Generated by OpenAPI Generator (https://openapi-generator.tech)
10
10
 
11
11
  Do not edit the class manually.
@@ -1,11 +1,11 @@
1
1
  # coding: utf-8
2
2
 
3
3
  """
4
- FastAPI
4
+ TrueFoundry ML API
5
5
 
6
6
  No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
7
 
8
- The version of the OpenAPI document: 0.1.0
8
+ The version of the OpenAPI document: 1.0.0
9
9
  Generated by OpenAPI Generator (https://openapi-generator.tech)
10
10
 
11
11
  Do not edit the class manually.
@@ -1,11 +1,11 @@
1
1
  # coding: utf-8
2
2
 
3
3
  """
4
- FastAPI
4
+ TrueFoundry ML API
5
5
 
6
6
  No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
7
 
8
- The version of the OpenAPI document: 0.1.0
8
+ The version of the OpenAPI document: 1.0.0
9
9
  Generated by OpenAPI Generator (https://openapi-generator.tech)
10
10
 
11
11
  Do not edit the class manually.
@@ -1,11 +1,11 @@
1
1
  # coding: utf-8
2
2
 
3
3
  """
4
- FastAPI
4
+ TrueFoundry ML API
5
5
 
6
6
  No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
7
 
8
- The version of the OpenAPI document: 0.1.0
8
+ The version of the OpenAPI document: 1.0.0
9
9
  Generated by OpenAPI Generator (https://openapi-generator.tech)
10
10
 
11
11
  Do not edit the class manually.
@@ -1,11 +1,11 @@
1
1
  # coding: utf-8
2
2
 
3
3
  """
4
- FastAPI
4
+ TrueFoundry ML API
5
5
 
6
6
  No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
7
 
8
- The version of the OpenAPI document: 0.1.0
8
+ The version of the OpenAPI document: 1.0.0
9
9
  Generated by OpenAPI Generator (https://openapi-generator.tech)
10
10
 
11
11
  Do not edit the class manually.
@@ -1,11 +1,11 @@
1
1
  # coding: utf-8
2
2
 
3
3
  """
4
- FastAPI
4
+ TrueFoundry ML API
5
5
 
6
6
  No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
7
 
8
- The version of the OpenAPI document: 0.1.0
8
+ The version of the OpenAPI document: 1.0.0
9
9
  Generated by OpenAPI Generator (https://openapi-generator.tech)
10
10
 
11
11
  Do not edit the class manually.
@@ -18,7 +18,10 @@ import pprint
18
18
  import re # noqa: F401
19
19
  from typing import Any, Dict, Optional
20
20
 
21
- from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr
21
+ from truefoundry.ml.autogen.client.models.data_directory_manifest import (
22
+ DataDirectoryManifest,
23
+ )
24
+ from truefoundry.pydantic_v1 import BaseModel, StrictStr
22
25
 
23
26
 
24
27
  class CreateDatasetRequestDto(BaseModel):
@@ -26,11 +29,18 @@ class CreateDatasetRequestDto(BaseModel):
26
29
  CreateDatasetRequestDto
27
30
  """
28
31
 
29
- name: StrictStr = Field(...)
32
+ name: Optional[StrictStr] = None
30
33
  description: Optional[StrictStr] = None
31
34
  experiment_id: Optional[StrictStr] = None
32
35
  dataset_metadata: Optional[Dict[str, Any]] = None
33
- __properties = ["name", "description", "experiment_id", "dataset_metadata"]
36
+ manifest: Optional[DataDirectoryManifest] = None
37
+ __properties = [
38
+ "name",
39
+ "description",
40
+ "experiment_id",
41
+ "dataset_metadata",
42
+ "manifest",
43
+ ]
34
44
 
35
45
  class Config:
36
46
  """Pydantic configuration"""
@@ -54,6 +64,9 @@ class CreateDatasetRequestDto(BaseModel):
54
64
  def to_dict(self):
55
65
  """Returns the dictionary representation of the model using alias"""
56
66
  _dict = self.dict(by_alias=True, exclude={}, exclude_none=True)
67
+ # override the default output from truefoundry.pydantic_v1 by calling `to_dict()` of manifest
68
+ if self.manifest:
69
+ _dict["manifest"] = self.manifest.to_dict()
57
70
  return _dict
58
71
 
59
72
  @classmethod
@@ -71,6 +84,9 @@ class CreateDatasetRequestDto(BaseModel):
71
84
  "description": obj.get("description"),
72
85
  "experiment_id": obj.get("experiment_id"),
73
86
  "dataset_metadata": obj.get("dataset_metadata"),
87
+ "manifest": DataDirectoryManifest.from_dict(obj.get("manifest"))
88
+ if obj.get("manifest") is not None
89
+ else None,
74
90
  }
75
91
  )
76
92
  return _obj
@@ -1,11 +1,11 @@
1
1
  # coding: utf-8
2
2
 
3
3
  """
4
- FastAPI
4
+ TrueFoundry ML API
5
5
 
6
6
  No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
7
 
8
- The version of the OpenAPI document: 0.1.0
8
+ The version of the OpenAPI document: 1.0.0
9
9
  Generated by OpenAPI Generator (https://openapi-generator.tech)
10
10
 
11
11
  Do not edit the class manually.
@@ -1,11 +1,11 @@
1
1
  # coding: utf-8
2
2
 
3
3
  """
4
- FastAPI
4
+ TrueFoundry ML API
5
5
 
6
6
  No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
7
 
8
- The version of the OpenAPI document: 0.1.0
8
+ The version of the OpenAPI document: 1.0.0
9
9
  Generated by OpenAPI Generator (https://openapi-generator.tech)
10
10
 
11
11
  Do not edit the class manually.