truefoundry 0.5.9__py3-none-any.whl → 0.5.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.

Potentially problematic release.


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

Files changed (223) hide show
  1. truefoundry/cli/display_util.py +1 -1
  2. truefoundry/common/utils.py +9 -0
  3. truefoundry/deploy/auto_gen/models.py +1 -1
  4. truefoundry/deploy/io/output_callback.py +13 -6
  5. truefoundry/deploy/io/rich_output_callback.py +8 -4
  6. truefoundry/deploy/lib/clients/servicefoundry_client.py +115 -136
  7. truefoundry/deploy/lib/dao/apply.py +2 -2
  8. truefoundry/deploy/lib/dao/delete.py +2 -2
  9. truefoundry/deploy/lib/model/entity.py +37 -24
  10. truefoundry/ml/autogen/client/__init__.py +22 -14
  11. truefoundry/ml/autogen/client/api/auth_api.py +2 -2
  12. truefoundry/ml/autogen/client/api/deprecated_api.py +5 -338
  13. truefoundry/ml/autogen/client/api/experiments_api.py +77 -32
  14. truefoundry/ml/autogen/client/api/generate_code_snippet_api.py +2 -2
  15. truefoundry/ml/autogen/client/api/health_api.py +2 -2
  16. truefoundry/ml/autogen/client/api/metrics_api.py +4 -4
  17. truefoundry/ml/autogen/client/api/mlfoundry_artifacts_api.py +530 -140
  18. truefoundry/ml/autogen/client/api/run_artifacts_api.py +10 -5
  19. truefoundry/ml/autogen/client/api/runs_api.py +20 -20
  20. truefoundry/ml/autogen/client/api_client.py +2 -2
  21. truefoundry/ml/autogen/client/configuration.py +29 -57
  22. truefoundry/ml/autogen/client/exceptions.py +2 -2
  23. truefoundry/ml/autogen/client/models/__init__.py +22 -14
  24. truefoundry/ml/autogen/client/models/agent_app.py +13 -5
  25. truefoundry/ml/autogen/client/models/{agent.py → agent_manifest.py} +39 -30
  26. truefoundry/ml/autogen/client/models/{agent_open_api_tool.py → agent_open_api_tool_manifest.py} +50 -31
  27. truefoundry/ml/autogen/client/models/agent_open_api_tool_with_fqn.py +47 -26
  28. truefoundry/ml/autogen/client/models/agent_with_fqn.py +36 -25
  29. truefoundry/ml/autogen/client/models/{add_custom_metrics_to_model_version_request_dto.py → apply_request_dto.py} +23 -17
  30. truefoundry/ml/autogen/client/models/apply_response_dto.py +79 -0
  31. truefoundry/ml/autogen/client/models/artifact_dto.py +2 -2
  32. truefoundry/ml/autogen/client/models/{add_features_to_model_version_request_dto.py → artifact_path.py} +20 -17
  33. truefoundry/ml/autogen/client/models/artifact_response_dto.py +2 -2
  34. truefoundry/ml/autogen/client/models/artifact_type.py +3 -3
  35. truefoundry/ml/autogen/client/models/artifact_version_dto.py +7 -15
  36. truefoundry/ml/autogen/client/models/artifact_version_manifest.py +32 -19
  37. truefoundry/ml/autogen/client/models/artifact_version_response_dto.py +2 -2
  38. truefoundry/ml/autogen/client/models/artifact_version_serialization_format.py +2 -2
  39. truefoundry/ml/autogen/client/models/artifact_version_status.py +2 -2
  40. truefoundry/ml/autogen/client/models/assistant_message.py +9 -5
  41. truefoundry/ml/autogen/client/models/authorize_user_for_model_request_dto.py +2 -2
  42. truefoundry/ml/autogen/client/models/authorize_user_for_model_version_request_dto.py +2 -2
  43. truefoundry/ml/autogen/client/models/blob_storage_reference.py +4 -4
  44. truefoundry/ml/autogen/client/models/body_get_search_runs_get.py +2 -2
  45. truefoundry/ml/autogen/client/models/{chat_prompt.py → chat_prompt_manifest.py} +39 -32
  46. truefoundry/ml/autogen/client/models/{chat_prompt_messages_inner.py → chat_prompt_manifest_messages_inner.py} +12 -12
  47. truefoundry/ml/autogen/client/models/columns_dto.py +2 -2
  48. truefoundry/ml/autogen/client/models/command.py +2 -2
  49. truefoundry/ml/autogen/client/models/content.py +3 -3
  50. truefoundry/ml/autogen/client/models/content1.py +3 -3
  51. truefoundry/ml/autogen/client/models/content2.py +3 -3
  52. truefoundry/ml/autogen/client/models/content2_any_of_inner.py +2 -2
  53. truefoundry/ml/autogen/client/models/create_artifact_request_dto.py +2 -2
  54. truefoundry/ml/autogen/client/models/create_artifact_response_dto.py +2 -2
  55. truefoundry/ml/autogen/client/models/create_artifact_version_request_dto.py +2 -2
  56. truefoundry/ml/autogen/client/models/create_artifact_version_response_dto.py +2 -2
  57. truefoundry/ml/autogen/client/models/create_dataset_request_dto.py +21 -5
  58. truefoundry/ml/autogen/client/models/create_experiment_request_dto.py +2 -2
  59. truefoundry/ml/autogen/client/models/create_experiment_response_dto.py +2 -2
  60. truefoundry/ml/autogen/client/models/create_model_version_request_dto.py +4 -12
  61. truefoundry/ml/autogen/client/models/create_multi_part_upload_for_dataset_request_dto.py +2 -2
  62. truefoundry/ml/autogen/client/models/create_multi_part_upload_for_dataset_response_dto.py +2 -2
  63. truefoundry/ml/autogen/client/models/create_multi_part_upload_request_dto.py +2 -2
  64. truefoundry/ml/autogen/client/models/create_python_deployment_config_request_dto.py +2 -2
  65. truefoundry/ml/autogen/client/models/create_python_deployment_config_response_dto.py +2 -2
  66. truefoundry/ml/autogen/client/models/create_run_request_dto.py +2 -2
  67. truefoundry/ml/autogen/client/models/create_run_response_dto.py +2 -2
  68. truefoundry/ml/autogen/client/models/create_workflow_task_config_request_dto.py +2 -2
  69. truefoundry/ml/autogen/client/models/data_directory_manifest.py +126 -0
  70. truefoundry/ml/autogen/client/models/dataset_dto.py +13 -2
  71. truefoundry/ml/autogen/client/models/dataset_response_dto.py +2 -2
  72. truefoundry/ml/autogen/client/models/delete_artifact_versions_request_dto.py +2 -2
  73. truefoundry/ml/autogen/client/models/delete_dataset_request_dto.py +2 -2
  74. truefoundry/ml/autogen/client/models/delete_files_for_dataset_request_dto.py +2 -2
  75. truefoundry/ml/autogen/client/models/delete_model_version_request_dto.py +2 -2
  76. truefoundry/ml/autogen/client/models/delete_run_request.py +2 -2
  77. truefoundry/ml/autogen/client/models/delete_tag_request_dto.py +2 -2
  78. truefoundry/ml/autogen/client/models/experiment_dto.py +3 -3
  79. truefoundry/ml/autogen/client/models/experiment_id_request_dto.py +2 -2
  80. truefoundry/ml/autogen/client/models/experiment_response_dto.py +2 -2
  81. truefoundry/ml/autogen/client/models/experiment_tag_dto.py +2 -2
  82. truefoundry/ml/autogen/client/models/export_deployment_files_request_dto.py +2 -2
  83. truefoundry/ml/autogen/client/models/external_blob_storage_source.py +5 -4
  84. truefoundry/ml/autogen/client/models/fast_ai_framework.py +3 -3
  85. truefoundry/ml/autogen/client/models/file_info_dto.py +2 -2
  86. truefoundry/ml/autogen/client/models/finalize_artifact_version_request_dto.py +7 -15
  87. truefoundry/ml/autogen/client/models/framework.py +3 -3
  88. truefoundry/ml/autogen/client/models/get_artifact_version_aliases_response_dto.py +2 -2
  89. truefoundry/ml/autogen/client/models/get_experiment_response_dto.py +2 -2
  90. truefoundry/ml/autogen/client/models/get_latest_run_log_response_dto.py +2 -2
  91. truefoundry/ml/autogen/client/models/get_metric_history_response.py +2 -2
  92. truefoundry/ml/autogen/client/models/get_signed_url_for_dataset_write_request_dto.py +2 -2
  93. truefoundry/ml/autogen/client/models/get_signed_urls_for_artifact_version_read_request_dto.py +2 -2
  94. truefoundry/ml/autogen/client/models/get_signed_urls_for_artifact_version_read_response_dto.py +2 -2
  95. truefoundry/ml/autogen/client/models/get_signed_urls_for_artifact_version_write_request_dto.py +2 -2
  96. truefoundry/ml/autogen/client/models/get_signed_urls_for_artifact_version_write_response_dto.py +2 -2
  97. truefoundry/ml/autogen/client/models/get_signed_urls_for_dataset_read_request_dto.py +2 -2
  98. truefoundry/ml/autogen/client/models/get_signed_urls_for_dataset_read_response_dto.py +2 -2
  99. truefoundry/ml/autogen/client/models/get_signed_urls_for_dataset_write_response_dto.py +2 -2
  100. truefoundry/ml/autogen/client/models/get_tenant_id_response_dto.py +2 -2
  101. truefoundry/ml/autogen/client/models/gluon_framework.py +3 -3
  102. truefoundry/ml/autogen/client/models/h2_o_framework.py +3 -3
  103. truefoundry/ml/autogen/client/models/http_validation_error.py +2 -2
  104. truefoundry/ml/autogen/client/models/image_content_part.py +7 -5
  105. truefoundry/ml/autogen/client/models/image_url.py +3 -3
  106. truefoundry/ml/autogen/client/models/infer_method_name.py +2 -2
  107. truefoundry/ml/autogen/client/models/keras_framework.py +3 -3
  108. truefoundry/ml/autogen/client/models/latest_run_log_dto.py +2 -2
  109. truefoundry/ml/autogen/client/models/library_name.py +2 -2
  110. truefoundry/ml/autogen/client/models/light_gbm_framework.py +3 -3
  111. truefoundry/ml/autogen/client/models/list_artifact_versions_request_dto.py +2 -2
  112. truefoundry/ml/autogen/client/models/list_artifact_versions_response_dto.py +2 -2
  113. truefoundry/ml/autogen/client/models/list_artifacts_request_dto.py +2 -2
  114. truefoundry/ml/autogen/client/models/list_artifacts_response_dto.py +2 -2
  115. truefoundry/ml/autogen/client/models/list_colums_response_dto.py +2 -2
  116. truefoundry/ml/autogen/client/models/list_datasets_request_dto.py +2 -2
  117. truefoundry/ml/autogen/client/models/list_datasets_response_dto.py +2 -2
  118. truefoundry/ml/autogen/client/models/list_experiments_response_dto.py +2 -2
  119. truefoundry/ml/autogen/client/models/list_files_for_artifact_version_request_dto.py +2 -2
  120. truefoundry/ml/autogen/client/models/list_files_for_artifact_versions_response_dto.py +2 -2
  121. truefoundry/ml/autogen/client/models/list_files_for_dataset_request_dto.py +2 -2
  122. truefoundry/ml/autogen/client/models/list_files_for_dataset_response_dto.py +2 -2
  123. truefoundry/ml/autogen/client/models/list_latest_run_logs_response_dto.py +2 -2
  124. truefoundry/ml/autogen/client/models/list_metric_history_request_dto.py +2 -2
  125. truefoundry/ml/autogen/client/models/list_metric_history_response_dto.py +2 -2
  126. truefoundry/ml/autogen/client/models/list_model_version_response_dto.py +2 -2
  127. truefoundry/ml/autogen/client/models/list_model_versions_request_dto.py +2 -2
  128. truefoundry/ml/autogen/client/models/list_models_request_dto.py +2 -2
  129. truefoundry/ml/autogen/client/models/list_models_response_dto.py +2 -2
  130. truefoundry/ml/autogen/client/models/list_run_artifacts_response_dto.py +2 -2
  131. truefoundry/ml/autogen/client/models/list_run_logs_response_dto.py +2 -2
  132. truefoundry/ml/autogen/client/models/local_artifact_source.py +93 -0
  133. truefoundry/ml/autogen/client/models/local_model_source.py +80 -0
  134. truefoundry/ml/autogen/client/models/log_batch_request_dto.py +2 -2
  135. truefoundry/ml/autogen/client/models/log_metric_request_dto.py +2 -2
  136. truefoundry/ml/autogen/client/models/log_param_request_dto.py +2 -2
  137. truefoundry/ml/autogen/client/models/manifest.py +100 -23
  138. truefoundry/ml/autogen/client/models/manifest1.py +219 -0
  139. truefoundry/ml/autogen/client/models/manifest2.py +198 -0
  140. truefoundry/ml/autogen/client/models/method.py +3 -3
  141. truefoundry/ml/autogen/client/models/metric_collection_dto.py +2 -2
  142. truefoundry/ml/autogen/client/models/metric_dto.py +2 -2
  143. truefoundry/ml/autogen/client/models/mime_type.py +2 -2
  144. truefoundry/ml/autogen/client/models/model_configuration.py +7 -7
  145. truefoundry/ml/autogen/client/models/model_dto.py +2 -2
  146. truefoundry/ml/autogen/client/models/model_response_dto.py +2 -2
  147. truefoundry/ml/autogen/client/models/model_server.py +2 -2
  148. truefoundry/ml/autogen/client/models/model_version_dto.py +2 -2
  149. truefoundry/ml/autogen/client/models/model_version_environment.py +5 -5
  150. truefoundry/ml/autogen/client/models/model_version_manifest.py +35 -22
  151. truefoundry/ml/autogen/client/models/model_version_response_dto.py +2 -2
  152. truefoundry/ml/autogen/client/models/multi_part_upload_dto.py +2 -2
  153. truefoundry/ml/autogen/client/models/multi_part_upload_response_dto.py +2 -2
  154. truefoundry/ml/autogen/client/models/multi_part_upload_storage_provider.py +2 -2
  155. truefoundry/ml/autogen/client/models/notify_artifact_version_failure_dto.py +2 -2
  156. truefoundry/ml/autogen/client/models/onnx_framework.py +3 -3
  157. truefoundry/ml/autogen/client/models/openapi_spec.py +3 -3
  158. truefoundry/ml/autogen/client/models/paddle_framework.py +3 -3
  159. truefoundry/ml/autogen/client/models/param_dto.py +2 -2
  160. truefoundry/ml/autogen/client/models/parameters.py +3 -3
  161. truefoundry/ml/autogen/client/models/py_torch_framework.py +3 -3
  162. truefoundry/ml/autogen/client/models/resolve_agent_app_response_dto.py +2 -2
  163. truefoundry/ml/autogen/client/models/restore_run_request_dto.py +2 -2
  164. truefoundry/ml/autogen/client/models/run_data_dto.py +2 -2
  165. truefoundry/ml/autogen/client/models/run_dto.py +2 -2
  166. truefoundry/ml/autogen/client/models/run_info_dto.py +2 -2
  167. truefoundry/ml/autogen/client/models/run_log_dto.py +2 -2
  168. truefoundry/ml/autogen/client/models/run_log_input_dto.py +2 -2
  169. truefoundry/ml/autogen/client/models/run_response_dto.py +2 -2
  170. truefoundry/ml/autogen/client/models/run_tag_dto.py +2 -2
  171. truefoundry/ml/autogen/client/models/search_runs_request_dto.py +2 -2
  172. truefoundry/ml/autogen/client/models/search_runs_response_dto.py +2 -2
  173. truefoundry/ml/autogen/client/models/set_experiment_tag_request_dto.py +2 -2
  174. truefoundry/ml/autogen/client/models/set_tag_request_dto.py +2 -2
  175. truefoundry/ml/autogen/client/models/signed_url_dto.py +3 -4
  176. truefoundry/ml/autogen/client/models/sklearn_framework.py +4 -4
  177. truefoundry/ml/autogen/client/models/sklearn_model_schema.py +7 -5
  178. truefoundry/ml/autogen/client/models/sklearn_serialization_format.py +3 -3
  179. truefoundry/ml/autogen/client/models/source.py +32 -7
  180. truefoundry/ml/autogen/client/models/source1.py +3 -3
  181. truefoundry/ml/autogen/client/models/{internal_metadata.py → source2.py} +52 -55
  182. truefoundry/ml/autogen/client/models/spa_cy_framework.py +3 -3
  183. truefoundry/ml/autogen/client/models/stats_models_framework.py +3 -3
  184. truefoundry/ml/autogen/client/models/stop.py +2 -2
  185. truefoundry/ml/autogen/client/models/store_run_logs_request_dto.py +2 -2
  186. truefoundry/ml/autogen/client/models/subject.py +2 -2
  187. truefoundry/ml/autogen/client/models/subject_type.py +2 -2
  188. truefoundry/ml/autogen/client/models/system_message.py +7 -4
  189. truefoundry/ml/autogen/client/models/tensor_flow_framework.py +3 -3
  190. truefoundry/ml/autogen/client/models/text.py +3 -3
  191. truefoundry/ml/autogen/client/models/text_content_part.py +6 -4
  192. truefoundry/ml/autogen/client/models/transformers_framework.py +5 -5
  193. truefoundry/ml/autogen/client/models/trigger_job_run_config_request_dto.py +2 -2
  194. truefoundry/ml/autogen/client/models/trigger_job_run_config_response_dto.py +2 -2
  195. truefoundry/ml/autogen/client/models/true_foundry_managed_source.py +3 -3
  196. truefoundry/ml/autogen/client/models/update_artifact_version_request_dto.py +5 -7
  197. truefoundry/ml/autogen/client/models/update_dataset_request_dto.py +13 -3
  198. truefoundry/ml/autogen/client/models/update_experiment_request_dto.py +2 -2
  199. truefoundry/ml/autogen/client/models/update_model_version_request_dto.py +2 -2
  200. truefoundry/ml/autogen/client/models/update_run_request_dto.py +2 -2
  201. truefoundry/ml/autogen/client/models/update_run_response_dto.py +2 -2
  202. truefoundry/ml/autogen/client/models/url.py +3 -3
  203. truefoundry/ml/autogen/client/models/user_message.py +8 -5
  204. truefoundry/ml/autogen/client/models/validate_external_storage_root_request_dto.py +2 -2
  205. truefoundry/ml/autogen/client/models/validate_external_storage_root_response_dto.py +2 -2
  206. truefoundry/ml/autogen/client/models/validation_error.py +2 -2
  207. truefoundry/ml/autogen/client/models/validation_error_loc_inner.py +2 -2
  208. truefoundry/ml/autogen/client/models/xg_boost_framework.py +4 -4
  209. truefoundry/ml/autogen/client/models/xg_boost_model_schema.py +7 -5
  210. truefoundry/ml/autogen/client/models/xg_boost_serialization_format.py +3 -3
  211. truefoundry/ml/autogen/client/rest.py +2 -2
  212. truefoundry/ml/autogen/client_README.md +17 -19
  213. truefoundry/ml/autogen/entities/artifacts.py +266 -141
  214. truefoundry/ml/clients/servicefoundry_client.py +9 -8
  215. truefoundry/ml/log_types/artifacts/artifact.py +15 -78
  216. truefoundry/ml/log_types/artifacts/model.py +4 -5
  217. truefoundry/ml/log_types/artifacts/utils.py +18 -2
  218. truefoundry/ml/mlfoundry_api.py +0 -40
  219. truefoundry/ml/session.py +6 -3
  220. {truefoundry-0.5.9.dist-info → truefoundry-0.5.10.dist-info}/METADATA +3 -6
  221. {truefoundry-0.5.9.dist-info → truefoundry-0.5.10.dist-info}/RECORD +223 -217
  222. {truefoundry-0.5.9.dist-info → truefoundry-0.5.10.dist-info}/WHEEL +0 -0
  223. {truefoundry-0.5.9.dist-info → truefoundry-0.5.10.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.
@@ -22,7 +22,7 @@ from truefoundry.ml.autogen.client.models.framework import Framework
22
22
  from truefoundry.ml.autogen.client.models.model_version_environment import (
23
23
  ModelVersionEnvironment,
24
24
  )
25
- from truefoundry.ml.autogen.client.models.source1 import Source1
25
+ from truefoundry.ml.autogen.client.models.source2 import Source2
26
26
  from truefoundry.pydantic_v1 import (
27
27
  BaseModel,
28
28
  Field,
@@ -38,66 +38,77 @@ class ModelVersionManifest(BaseModel):
38
38
  Model Version artifact. # noqa: E501
39
39
  """
40
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
+ )
41
45
  description: Optional[constr(strict=True, max_length=512)] = Field(
42
46
  default=None, description="+sort=2 +label=Description"
43
47
  )
44
48
  metadata: Dict[str, Any] = Field(
45
49
  default=...,
46
- 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}',
47
51
  )
48
52
  version_alias: Optional[constr(strict=True, max_length=128)] = Field(
49
53
  default=None,
50
- 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',
51
55
  )
52
- name: Optional[constr(strict=True, max_length=256)] = Field(
56
+ ml_repo: Optional[constr(strict=True)] = Field(
53
57
  default=None,
54
- 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",
55
59
  )
56
- ml_repo: Optional[constr(strict=True)] = Field(
60
+ version: Optional[conint(strict=True, ge=1)] = Field(
57
61
  default=None,
58
- 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",
59
63
  )
60
64
  type: Optional[StrictStr] = "model-version"
61
- source: Source1 = Field(...)
65
+ source: Source2 = Field(...)
62
66
  framework: Optional[Framework] = None
63
67
  environment: Optional[ModelVersionEnvironment] = None
64
68
  step: Optional[conint(strict=True, ge=0)] = Field(
65
- default=0, description="+label=Step"
69
+ default=0,
70
+ description='+label=Step +usage=Step/Epoch number in an iterative training loop the model version was created. Generally useful when logging a model version from a MLRepo Run +uiProps={"descriptionInline":true}',
71
+ )
72
+ run_id: Optional[StrictStr] = Field(
73
+ default=None,
74
+ description="+label=Run ID +usage=ID of the MLRepo Run that generated the model version +uiType=Hidden",
66
75
  )
67
76
  __properties = [
77
+ "name",
68
78
  "description",
69
79
  "metadata",
70
80
  "version_alias",
71
- "name",
72
81
  "ml_repo",
82
+ "version",
73
83
  "type",
74
84
  "source",
75
85
  "framework",
76
86
  "environment",
77
87
  "step",
88
+ "run_id",
78
89
  ]
79
90
 
80
- @validator("version_alias")
81
- def version_alias_validate_regular_expression(cls, value):
91
+ @validator("name")
92
+ def name_validate_regular_expression(cls, value):
82
93
  """Validates the regular expression"""
83
94
  if value is None:
84
95
  return value
85
96
 
86
- if not re.match(r"^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$", value):
97
+ if not re.match(r"^[A-Za-z0-9_\-]+$", value):
87
98
  raise ValueError(
88
- r"must validate the regular expression /^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$/"
99
+ r"must validate the regular expression /^[A-Za-z0-9_\-]+$/"
89
100
  )
90
101
  return value
91
102
 
92
- @validator("name")
93
- def name_validate_regular_expression(cls, value):
103
+ @validator("version_alias")
104
+ def version_alias_validate_regular_expression(cls, value):
94
105
  """Validates the regular expression"""
95
106
  if value is None:
96
107
  return value
97
108
 
98
- if not re.match(r"^[A-Za-z0-9_\-]+$", value):
109
+ if not re.match(r"^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$", value):
99
110
  raise ValueError(
100
- r"must validate the regular expression /^[A-Za-z0-9_\-]+$/"
111
+ r"must validate the regular expression /^v[a-zA-Z0-9.-]*([a-zA-Z0-9]+)$/"
101
112
  )
102
113
  return value
103
114
 
@@ -167,15 +178,16 @@ class ModelVersionManifest(BaseModel):
167
178
 
168
179
  _obj = ModelVersionManifest.parse_obj(
169
180
  {
181
+ "name": obj.get("name"),
170
182
  "description": obj.get("description"),
171
183
  "metadata": obj.get("metadata"),
172
184
  "version_alias": obj.get("version_alias"),
173
- "name": obj.get("name"),
174
185
  "ml_repo": obj.get("ml_repo"),
186
+ "version": obj.get("version"),
175
187
  "type": obj.get("type")
176
188
  if obj.get("type") is not None
177
189
  else "model-version",
178
- "source": Source1.from_dict(obj.get("source"))
190
+ "source": Source2.from_dict(obj.get("source"))
179
191
  if obj.get("source") is not None
180
192
  else None,
181
193
  "framework": Framework.from_dict(obj.get("framework"))
@@ -185,6 +197,7 @@ class ModelVersionManifest(BaseModel):
185
197
  if obj.get("environment") is not None
186
198
  else None,
187
199
  "step": obj.get("step") if obj.get("step") is not None else 0,
200
+ "run_id": obj.get("run_id"),
188
201
  }
189
202
  )
190
203
  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.
@@ -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.
@@ -22,7 +22,7 @@ from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
22
22
 
23
23
  class ONNXFramework(BaseModel):
24
24
  """
25
- +docs=ONNX framework for the model version +label=ONNX # noqa: E501
25
+ +label=ONNX +icon=onnx # noqa: E501
26
26
  """
27
27
 
28
28
  type: StrictStr = Field(
@@ -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.
@@ -33,7 +33,7 @@ OPENAPISPEC_ANY_OF_SCHEMAS = ["BlobStorageReference", "object"]
33
33
 
34
34
  class OpenapiSpec(BaseModel):
35
35
  """
36
- +sort=20 +uiType=OpenapiSchema
36
+ +sort=20 +uiType=OpenapiSchema +label=OpenAPI Spec +usage=OpenAPI Spec for the tool describing the API, endpoints and parameters. [Sample OpenAPI Spec Link](https://assets.production.truefoundry.com/sample-openapi.json) +uiProps={\"descriptionInline\":true} +placeholder={\"openapi\":\"3.0.0\",\"info\":{\"title\":\"LLM Agent Tools API\",\"version\":\"1.0.0\"},\"servers\":[{\"url\":\"https://api.example.com/v1\"}],\"paths\":{\"/weather\":{\"get\":{\"summary\":\"Get current weather\",\"description\":\"Fetches the current weather for a given location.\",\"parameters\":[{\"name\":\"location\",\"in\":\"query\",\"required\":true,\"schema\":{\"type\":\"string\"}}],\"responses\":{\"200\":{\"description\":\"Successful response\",\"content\":{\"application/json\":{\"schema\":{\"type\":\"object\",\"properties\":{\"location\":{\"type\":\"string\"},\"temperature\":{\"type\":\"number\"},\"condition\":{\"type\":\"string\"}}}}}}}}}}}
37
37
  """
38
38
 
39
39
  # data type: BlobStorageReference
@@ -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.
@@ -22,7 +22,7 @@ from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
22
22
 
23
23
  class PaddleFramework(BaseModel):
24
24
  """
25
- +docs=PaddlePaddle framework for the model version +label=Paddle # noqa: E501
25
+ +label=Paddle +icon=paddle # noqa: E501
26
26
  """
27
27
 
28
28
  type: StrictStr = Field(
@@ -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,7 +24,7 @@ from truefoundry.pydantic_v1 import BaseModel, StrictFloat, StrictInt
24
24
 
25
25
  class Parameters(BaseModel):
26
26
  """
27
- +usage=Parameters for the provider +label=Parameters +docs=Key-value pairs to store additional parameters for the provider # noqa: E501
27
+ +label=Parameters +usage=Parameters to pass to the model when generating # noqa: E501
28
28
  """
29
29
 
30
30
  max_tokens: Optional[StrictInt] = 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.
@@ -22,7 +22,7 @@ from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
22
22
 
23
23
  class PyTorchFramework(BaseModel):
24
24
  """
25
- +docs=PyTorch framework for the model version +label=PyTorch # noqa: E501
25
+ +label=PyTorch +icon=pytorch # noqa: E501
26
26
  """
27
27
 
28
28
  type: StrictStr = Field(
@@ -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.
@@ -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.
@@ -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.
@@ -16,7 +16,6 @@ from __future__ import annotations
16
16
  import json
17
17
  import pprint
18
18
  import re # noqa: F401
19
- from typing import Optional
20
19
 
21
20
  from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr
22
21
 
@@ -27,7 +26,7 @@ class SignedURLDto(BaseModel):
27
26
  """
28
27
 
29
28
  signed_url: StrictStr = Field(...)
30
- path: Optional[StrictStr] = None
29
+ path: StrictStr = Field(...)
31
30
  __properties = ["signed_url", "path"]
32
31
 
33
32
  class Config:
@@ -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.
@@ -27,7 +27,7 @@ from truefoundry.pydantic_v1 import BaseModel, Field, StrictStr, validator
27
27
 
28
28
  class SklearnFramework(BaseModel):
29
29
  """
30
- +docs=Scikit-learn framework for the model version +label=Sklearn # noqa: E501
30
+ +label=Sklearn +icon=sklearn # noqa: E501
31
31
  """
32
32
 
33
33
  type: StrictStr = Field(
@@ -36,7 +36,7 @@ class SklearnFramework(BaseModel):
36
36
  )
37
37
  model_filepath: Optional[StrictStr] = Field(
38
38
  default=None,
39
- description="+label=Model file path +usage=Relative path to the model file",
39
+ description='+label=Model file path +usage=Relative path to the model file in the model version contents +uiProps={"descriptionInline":true}',
40
40
  )
41
41
  serialization_format: Optional[SklearnSerializationFormat] = None
42
42
  model_schema: Optional[SklearnModelSchema] = 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.
@@ -24,7 +24,7 @@ from truefoundry.pydantic_v1 import BaseModel, Field, conlist
24
24
 
25
25
  class SklearnModelSchema(BaseModel):
26
26
  """
27
- +label=Sklearn Model Schema # noqa: E501
27
+ +label=Scikit Learn Model Schema # noqa: E501
28
28
  """
29
29
 
30
30
  infer_method_name: InferMethodName = Field(
@@ -32,10 +32,12 @@ class SklearnModelSchema(BaseModel):
32
32
  description="+label=Inference Method Name +usage=Name of the method used for inference",
33
33
  )
34
34
  inputs: conlist(Dict[str, Any]) = Field(
35
- default=..., description="+label= Input Schema +usage=Schema of the input"
35
+ default=...,
36
+ description='+label= Input Schema +usage=Schema of the input +uiProps={"descriptionInline":true}',
36
37
  )
37
38
  outputs: conlist(Dict[str, Any]) = Field(
38
- default=..., description="+label= Output Schema +usage=Schema of the output"
39
+ default=...,
40
+ description='+label= Output Schema +usage=Schema of the output +uiProps={"descriptionInline":true}',
39
41
  )
40
42
  __properties = ["infer_method_name", "inputs", "outputs"]
41
43