truefoundry 0.5.5rc2__tar.gz → 0.5.6__tar.gz

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 (373) hide show
  1. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/PKG-INFO +2 -2
  2. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/pyproject.toml +2 -2
  3. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/auto_gen/models.py +25 -25
  4. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/cli/commands/__init__.py +0 -1
  5. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/lib/clients/servicefoundry_client.py +20 -16
  6. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/v2/lib/deploy.py +23 -15
  7. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/v2/lib/deploy_workflow.py +9 -4
  8. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/v2/lib/patched_models.py +3 -3
  9. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/__init__.py +4 -4
  10. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/api/experiments_api.py +0 -156
  11. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/api/mlfoundry_artifacts_api.py +49 -305
  12. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/__init__.py +4 -4
  13. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/agent.py +64 -13
  14. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/agent_open_api_tool.py +63 -17
  15. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/agent_open_api_tool_with_fqn.py +63 -17
  16. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/agent_with_fqn.py +64 -12
  17. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/artifact_version_manifest.py +39 -4
  18. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/chat_prompt.py +60 -22
  19. truefoundry-0.5.5rc2/truefoundry/ml/autogen/client/models/external_artifact_source.py → truefoundry-0.5.6/truefoundry/ml/autogen/client/models/external_blob_storage_source.py +10 -11
  20. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/model_version_manifest.py +39 -4
  21. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/source.py +22 -22
  22. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/source1.py +22 -22
  23. truefoundry-0.5.5rc2/truefoundry/ml/autogen/client/models/true_foundry_artifact_source.py → truefoundry-0.5.6/truefoundry/ml/autogen/client/models/true_foundry_managed_source.py +12 -11
  24. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client_README.md +2 -3
  25. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/entities/artifacts.py +87 -69
  26. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/log_types/artifacts/artifact.py +26 -18
  27. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/log_types/artifacts/general_artifact.py +4 -13
  28. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/log_types/artifacts/model.py +28 -21
  29. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/mlfoundry_api.py +2 -6
  30. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/workflow/task.py +1 -1
  31. truefoundry-0.5.5rc2/truefoundry/deploy/cli/commands/build_logs_command.py +0 -89
  32. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/README.md +0 -0
  33. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/__init__.py +0 -0
  34. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/autodeploy/__init__.py +0 -0
  35. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/autodeploy/agents/__init__.py +0 -0
  36. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/autodeploy/agents/base.py +0 -0
  37. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/autodeploy/agents/developer.py +0 -0
  38. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/autodeploy/agents/project_identifier.py +0 -0
  39. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/autodeploy/agents/tester.py +0 -0
  40. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/autodeploy/cli.py +0 -0
  41. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/autodeploy/constants.py +0 -0
  42. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/autodeploy/exception.py +0 -0
  43. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/autodeploy/logger.py +0 -0
  44. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/autodeploy/tools/__init__.py +0 -0
  45. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/autodeploy/tools/ask.py +0 -0
  46. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/autodeploy/tools/base.py +0 -0
  47. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/autodeploy/tools/commit.py +0 -0
  48. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/autodeploy/tools/docker_build.py +0 -0
  49. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/autodeploy/tools/docker_run.py +0 -0
  50. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/autodeploy/tools/file_type_counts.py +0 -0
  51. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/autodeploy/tools/list_files.py +0 -0
  52. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/autodeploy/tools/read_file.py +0 -0
  53. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/autodeploy/tools/send_request.py +0 -0
  54. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/autodeploy/tools/write_file.py +0 -0
  55. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/autodeploy/utils/client.py +0 -0
  56. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/autodeploy/utils/diff.py +0 -0
  57. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/autodeploy/utils/pydantic_compat.py +0 -0
  58. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/cli/__init__.py +0 -0
  59. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/cli/__main__.py +0 -0
  60. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/cli/config.py +0 -0
  61. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/cli/console.py +0 -0
  62. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/cli/const.py +0 -0
  63. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/cli/display_util.py +0 -0
  64. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/cli/util.py +0 -0
  65. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/common/__init__.py +0 -0
  66. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/common/auth_service_client.py +0 -0
  67. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/common/constants.py +0 -0
  68. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/common/credential_file_manager.py +0 -0
  69. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/common/credential_provider.py +0 -0
  70. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/common/entities.py +0 -0
  71. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/common/exceptions.py +0 -0
  72. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/common/request_utils.py +0 -0
  73. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/common/servicefoundry_client.py +0 -0
  74. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/common/session.py +0 -0
  75. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/common/storage_provider_utils.py +0 -0
  76. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/common/utils.py +0 -0
  77. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/common/warnings.py +0 -0
  78. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/__init__.py +0 -0
  79. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/builder/__init__.py +0 -0
  80. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/builder/builders/__init__.py +0 -0
  81. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/builder/builders/dockerfile.py +0 -0
  82. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/builder/builders/tfy_notebook_buildpack/__init__.py +0 -0
  83. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/builder/builders/tfy_notebook_buildpack/dockerfile_template.py +0 -0
  84. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/builder/builders/tfy_python_buildpack/__init__.py +0 -0
  85. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/builder/builders/tfy_python_buildpack/dockerfile_template.py +0 -0
  86. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/builder/constants.py +0 -0
  87. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/builder/docker_service.py +0 -0
  88. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/builder/utils.py +0 -0
  89. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/cli/__init__.py +0 -0
  90. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/cli/commands/apply_command.py +0 -0
  91. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/cli/commands/build_command.py +0 -0
  92. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/cli/commands/create_command.py +0 -0
  93. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/cli/commands/delete_command.py +0 -0
  94. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/cli/commands/deploy_command.py +0 -0
  95. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/cli/commands/deploy_init_command.py +0 -0
  96. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/cli/commands/get_command.py +0 -0
  97. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/cli/commands/list_command.py +0 -0
  98. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/cli/commands/login_command.py +0 -0
  99. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/cli/commands/logout_command.py +0 -0
  100. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/cli/commands/logs_command.py +0 -0
  101. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/cli/commands/patch_application_command.py +0 -0
  102. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/cli/commands/patch_command.py +0 -0
  103. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/cli/commands/redeploy_command.py +0 -0
  104. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/cli/commands/terminate_comand.py +0 -0
  105. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/cli/commands/trigger_command.py +0 -0
  106. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/core/__init__.py +0 -0
  107. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/core/login.py +0 -0
  108. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/core/logout.py +0 -0
  109. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/io/__init__.py +0 -0
  110. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/io/output_callback.py +0 -0
  111. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/io/rich_output_callback.py +0 -0
  112. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/lib/__init__.py +0 -0
  113. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/lib/clients/__init__.py +0 -0
  114. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/lib/const.py +0 -0
  115. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/lib/dao/__init__.py +0 -0
  116. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/lib/dao/application.py +0 -0
  117. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/lib/dao/apply.py +0 -0
  118. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/lib/dao/version.py +0 -0
  119. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/lib/dao/workspace.py +0 -0
  120. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/lib/logs_utils.py +0 -0
  121. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/lib/messages.py +0 -0
  122. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/lib/model/__init__.py +0 -0
  123. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/lib/model/entity.py +0 -0
  124. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/lib/session.py +0 -0
  125. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/lib/util.py +0 -0
  126. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/lib/win32.py +0 -0
  127. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/python_deploy_codegen.py +0 -0
  128. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/v2/__init__.py +0 -0
  129. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/v2/lib/__init__.py +0 -0
  130. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/v2/lib/deployable_patched_models.py +0 -0
  131. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/v2/lib/models.py +0 -0
  132. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/deploy/v2/lib/source.py +0 -0
  133. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/logger.py +0 -0
  134. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/__init__.py +0 -0
  135. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/artifact/__init__.py +0 -0
  136. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/artifact/truefoundry_artifact_repo.py +0 -0
  137. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/__init__.py +0 -0
  138. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/api/__init__.py +0 -0
  139. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/api/auth_api.py +0 -0
  140. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/api/deprecated_api.py +0 -0
  141. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/api/generate_code_snippet_api.py +0 -0
  142. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/api/health_api.py +0 -0
  143. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/api/metrics_api.py +0 -0
  144. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/api/run_artifacts_api.py +0 -0
  145. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/api/runs_api.py +0 -0
  146. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/api_client.py +0 -0
  147. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/api_response.py +0 -0
  148. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/configuration.py +0 -0
  149. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/exceptions.py +0 -0
  150. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/add_custom_metrics_to_model_version_request_dto.py +0 -0
  151. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/add_features_to_model_version_request_dto.py +0 -0
  152. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/agent_app.py +0 -0
  153. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/artifact_dto.py +0 -0
  154. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/artifact_response_dto.py +0 -0
  155. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/artifact_type.py +0 -0
  156. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/artifact_version_dto.py +0 -0
  157. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/artifact_version_response_dto.py +0 -0
  158. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/artifact_version_serialization_format.py +0 -0
  159. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/artifact_version_status.py +0 -0
  160. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/assistant_message.py +0 -0
  161. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/authorize_user_for_model_request_dto.py +0 -0
  162. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/authorize_user_for_model_version_request_dto.py +0 -0
  163. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/blob_storage_reference.py +0 -0
  164. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/body_get_search_runs_get.py +0 -0
  165. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/chat_prompt_messages_inner.py +0 -0
  166. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/columns_dto.py +0 -0
  167. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/command.py +0 -0
  168. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/content.py +0 -0
  169. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/content1.py +0 -0
  170. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/content2.py +0 -0
  171. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/content2_any_of_inner.py +0 -0
  172. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/create_artifact_request_dto.py +0 -0
  173. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/create_artifact_response_dto.py +0 -0
  174. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/create_artifact_version_request_dto.py +0 -0
  175. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/create_artifact_version_response_dto.py +0 -0
  176. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/create_dataset_request_dto.py +0 -0
  177. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/create_experiment_request_dto.py +0 -0
  178. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/create_experiment_response_dto.py +0 -0
  179. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/create_model_version_request_dto.py +0 -0
  180. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/create_multi_part_upload_for_dataset_request_dto.py +0 -0
  181. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/create_multi_part_upload_for_dataset_response_dto.py +0 -0
  182. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/create_multi_part_upload_request_dto.py +0 -0
  183. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/create_python_deployment_config_request_dto.py +0 -0
  184. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/create_python_deployment_config_response_dto.py +0 -0
  185. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/create_run_request_dto.py +0 -0
  186. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/create_run_response_dto.py +0 -0
  187. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/create_workflow_task_config_request_dto.py +0 -0
  188. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/dataset_dto.py +0 -0
  189. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/dataset_response_dto.py +0 -0
  190. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/delete_artifact_versions_request_dto.py +0 -0
  191. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/delete_dataset_request_dto.py +0 -0
  192. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/delete_files_for_dataset_request_dto.py +0 -0
  193. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/delete_model_version_request_dto.py +0 -0
  194. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/delete_run_request.py +0 -0
  195. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/delete_tag_request_dto.py +0 -0
  196. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/experiment_dto.py +0 -0
  197. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/experiment_id_request_dto.py +0 -0
  198. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/experiment_response_dto.py +0 -0
  199. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/experiment_tag_dto.py +0 -0
  200. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/export_deployment_files_request_dto.py +0 -0
  201. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/fast_ai_framework.py +0 -0
  202. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/file_info_dto.py +0 -0
  203. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/finalize_artifact_version_request_dto.py +0 -0
  204. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/framework.py +0 -0
  205. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/get_artifact_version_aliases_response_dto.py +0 -0
  206. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/get_experiment_response_dto.py +0 -0
  207. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/get_latest_run_log_response_dto.py +0 -0
  208. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/get_metric_history_response.py +0 -0
  209. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/get_signed_url_for_dataset_write_request_dto.py +0 -0
  210. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/get_signed_urls_for_artifact_version_read_request_dto.py +0 -0
  211. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/get_signed_urls_for_artifact_version_read_response_dto.py +0 -0
  212. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/get_signed_urls_for_artifact_version_write_request_dto.py +0 -0
  213. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/get_signed_urls_for_artifact_version_write_response_dto.py +0 -0
  214. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/get_signed_urls_for_dataset_read_request_dto.py +0 -0
  215. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/get_signed_urls_for_dataset_read_response_dto.py +0 -0
  216. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/get_signed_urls_for_dataset_write_response_dto.py +0 -0
  217. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/get_tenant_id_response_dto.py +0 -0
  218. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/gluon_framework.py +0 -0
  219. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/h2_o_framework.py +0 -0
  220. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/http_validation_error.py +0 -0
  221. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/image_content_part.py +0 -0
  222. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/image_url.py +0 -0
  223. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/infer_method_name.py +0 -0
  224. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/internal_metadata.py +0 -0
  225. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/keras_framework.py +0 -0
  226. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/latest_run_log_dto.py +0 -0
  227. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/library_name.py +0 -0
  228. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/light_gbm_framework.py +0 -0
  229. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/list_artifact_versions_request_dto.py +0 -0
  230. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/list_artifact_versions_response_dto.py +0 -0
  231. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/list_artifacts_request_dto.py +0 -0
  232. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/list_artifacts_response_dto.py +0 -0
  233. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/list_colums_response_dto.py +0 -0
  234. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/list_datasets_request_dto.py +0 -0
  235. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/list_datasets_response_dto.py +0 -0
  236. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/list_experiments_response_dto.py +0 -0
  237. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/list_files_for_artifact_version_request_dto.py +0 -0
  238. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/list_files_for_artifact_versions_response_dto.py +0 -0
  239. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/list_files_for_dataset_request_dto.py +0 -0
  240. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/list_files_for_dataset_response_dto.py +0 -0
  241. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/list_latest_run_logs_response_dto.py +0 -0
  242. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/list_metric_history_request_dto.py +0 -0
  243. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/list_metric_history_response_dto.py +0 -0
  244. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/list_model_version_response_dto.py +0 -0
  245. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/list_model_versions_request_dto.py +0 -0
  246. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/list_models_request_dto.py +0 -0
  247. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/list_models_response_dto.py +0 -0
  248. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/list_run_artifacts_response_dto.py +0 -0
  249. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/list_run_logs_response_dto.py +0 -0
  250. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/log_batch_request_dto.py +0 -0
  251. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/log_metric_request_dto.py +0 -0
  252. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/log_param_request_dto.py +0 -0
  253. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/manifest.py +0 -0
  254. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/method.py +0 -0
  255. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/metric_collection_dto.py +0 -0
  256. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/metric_dto.py +0 -0
  257. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/mime_type.py +0 -0
  258. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/model_configuration.py +0 -0
  259. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/model_dto.py +0 -0
  260. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/model_response_dto.py +0 -0
  261. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/model_server.py +0 -0
  262. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/model_version_dto.py +0 -0
  263. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/model_version_environment.py +0 -0
  264. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/model_version_response_dto.py +0 -0
  265. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/multi_part_upload_dto.py +0 -0
  266. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/multi_part_upload_response_dto.py +0 -0
  267. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/multi_part_upload_storage_provider.py +0 -0
  268. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/notify_artifact_version_failure_dto.py +0 -0
  269. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/onnx_framework.py +0 -0
  270. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/openapi_spec.py +0 -0
  271. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/paddle_framework.py +0 -0
  272. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/param_dto.py +0 -0
  273. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/parameters.py +0 -0
  274. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/py_torch_framework.py +0 -0
  275. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/resolve_agent_app_response_dto.py +0 -0
  276. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/restore_run_request_dto.py +0 -0
  277. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/run_data_dto.py +0 -0
  278. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/run_dto.py +0 -0
  279. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/run_info_dto.py +0 -0
  280. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/run_log_dto.py +0 -0
  281. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/run_log_input_dto.py +0 -0
  282. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/run_response_dto.py +0 -0
  283. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/run_tag_dto.py +0 -0
  284. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/search_runs_request_dto.py +0 -0
  285. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/search_runs_response_dto.py +0 -0
  286. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/set_experiment_tag_request_dto.py +0 -0
  287. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/set_tag_request_dto.py +0 -0
  288. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/signed_url_dto.py +0 -0
  289. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/sklearn_framework.py +0 -0
  290. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/sklearn_model_schema.py +0 -0
  291. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/sklearn_serialization_format.py +0 -0
  292. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/spa_cy_framework.py +0 -0
  293. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/stats_models_framework.py +0 -0
  294. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/stop.py +0 -0
  295. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/store_run_logs_request_dto.py +0 -0
  296. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/subject.py +0 -0
  297. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/subject_type.py +0 -0
  298. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/system_message.py +0 -0
  299. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/tensor_flow_framework.py +0 -0
  300. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/text.py +0 -0
  301. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/text_content_part.py +0 -0
  302. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/transformers_framework.py +0 -0
  303. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/trigger_job_run_config_request_dto.py +0 -0
  304. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/trigger_job_run_config_response_dto.py +0 -0
  305. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/update_artifact_version_request_dto.py +0 -0
  306. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/update_dataset_request_dto.py +0 -0
  307. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/update_experiment_request_dto.py +0 -0
  308. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/update_model_version_request_dto.py +0 -0
  309. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/update_run_request_dto.py +0 -0
  310. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/update_run_response_dto.py +0 -0
  311. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/url.py +0 -0
  312. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/user_message.py +0 -0
  313. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/validate_external_storage_root_request_dto.py +0 -0
  314. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/validate_external_storage_root_response_dto.py +0 -0
  315. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/validation_error.py +0 -0
  316. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/validation_error_loc_inner.py +0 -0
  317. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/xg_boost_framework.py +0 -0
  318. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/xg_boost_model_schema.py +0 -0
  319. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/models/xg_boost_serialization_format.py +0 -0
  320. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/client/rest.py +0 -0
  321. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/models/__init__.py +0 -0
  322. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/models/exceptions.py +0 -0
  323. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/models/schema.py +0 -0
  324. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/models/signature.py +0 -0
  325. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/autogen/models/utils.py +0 -0
  326. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/cli/__init__.py +0 -0
  327. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/cli/cli.py +0 -0
  328. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/cli/commands/__init__.py +0 -0
  329. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/cli/commands/download.py +0 -0
  330. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/cli/commands/model_init.py +0 -0
  331. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/cli/utils.py +0 -0
  332. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/clients/__init__.py +0 -0
  333. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/clients/servicefoundry_client.py +0 -0
  334. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/constants.py +0 -0
  335. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/entities.py +0 -0
  336. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/enums.py +0 -0
  337. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/exceptions.py +0 -0
  338. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/git_info.py +0 -0
  339. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/internal_namespace.py +0 -0
  340. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/log_types/__init__.py +0 -0
  341. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/log_types/artifacts/constants.py +0 -0
  342. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/log_types/artifacts/dataset.py +0 -0
  343. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/log_types/artifacts/utils.py +0 -0
  344. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/log_types/image/__init__.py +0 -0
  345. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/log_types/image/constants.py +0 -0
  346. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/log_types/image/image.py +0 -0
  347. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/log_types/image/image_normalizer.py +0 -0
  348. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/log_types/image/types.py +0 -0
  349. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/log_types/plot.py +0 -0
  350. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/log_types/pydantic_base.py +0 -0
  351. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/log_types/utils.py +0 -0
  352. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/logger.py +0 -0
  353. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/mlfoundry_run.py +0 -0
  354. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/model_framework.py +0 -0
  355. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/run_utils.py +0 -0
  356. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/session.py +0 -0
  357. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/ml/validation_utils.py +0 -0
  358. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/pydantic_v1.py +0 -0
  359. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/version.py +0 -0
  360. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/workflow/__init__.py +0 -0
  361. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/workflow/container_task.py +0 -0
  362. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/workflow/example/deploy.sh +0 -0
  363. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/workflow/example/hello_world_package/workflow.py +0 -0
  364. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/workflow/example/package/test_workflow.py +0 -0
  365. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/workflow/example/truefoundry.yaml +0 -0
  366. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/workflow/example/workflow.yaml +0 -0
  367. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/workflow/map_task.py +0 -0
  368. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/workflow/python_task.py +0 -0
  369. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/workflow/remote_filesystem/__init__.py +0 -0
  370. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/workflow/remote_filesystem/logger.py +0 -0
  371. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/workflow/remote_filesystem/tfy_signed_url_client.py +0 -0
  372. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/workflow/remote_filesystem/tfy_signed_url_fs.py +0 -0
  373. {truefoundry-0.5.5rc2 → truefoundry-0.5.6}/truefoundry/workflow/workflow.py +0 -0
@@ -1,7 +1,7 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: truefoundry
3
- Version: 0.5.5rc2
4
- Summary: Truefoundry CLI
3
+ Version: 0.5.6
4
+ Summary: TrueFoundry CLI
5
5
  Author: Abhishek Choudhary
6
6
  Author-email: abhishek@truefoundry.com
7
7
  Requires-Python: >=3.8.1,<3.13
@@ -1,7 +1,7 @@
1
1
  [tool.poetry]
2
2
  name = "truefoundry"
3
- version = "0.5.5rc2"
4
- description = "Truefoundry CLI"
3
+ version = "0.5.6"
4
+ description = "TrueFoundry CLI"
5
5
  authors = ["Abhishek Choudhary <abhishek@truefoundry.com>"]
6
6
  readme = "README.md"
7
7
 
@@ -1,6 +1,6 @@
1
1
  # generated by datamodel-codegen:
2
2
  # filename: application.json
3
- # timestamp: 2025-01-20T07:24:00+00:00
3
+ # timestamp: 2025-01-29T10:40:43+00:00
4
4
 
5
5
  from __future__ import annotations
6
6
 
@@ -149,7 +149,7 @@ class BaseAutoscaling(BaseModel):
149
149
 
150
150
  class BasicAuthCreds(BaseModel):
151
151
  """
152
- +label=Username and password for service auth
152
+ +label=Username and password
153
153
  """
154
154
 
155
155
  type: Literal["basic_auth"] = Field(..., description="+value=basic_auth")
@@ -979,7 +979,7 @@ class SecretMount(BaseModel):
979
979
  )
980
980
  secret_fqn: constr(regex=r"^tfy-secret:\/\/.+:.+:.+$") = Field(
981
981
  ...,
982
- description="+label=Secret\n+usage=The Truefoundry secret whose value will be the file content.",
982
+ description="+label=Secret\n+usage=The TrueFoundry secret whose value will be the file content.",
983
983
  )
984
984
 
985
985
 
@@ -1111,20 +1111,10 @@ class TaskPythonBuild(BaseModel):
1111
1111
  )
1112
1112
 
1113
1113
 
1114
- class TrueFoundryInteractiveLogin(BaseModel):
1115
- """
1116
- +label=Login with truefoundry
1117
- """
1118
-
1119
- type: Literal["truefoundry_oauth"] = Field(
1120
- ..., description="+value=truefoundry_oauth"
1121
- )
1122
-
1123
-
1124
- class TruefoundryArtifactSource(BaseModel):
1114
+ class TrueFoundryArtifactSource(BaseModel):
1125
1115
  """
1126
- +docs=Input for Artifact from Truefoundry Artifact Registry
1127
- +label=Truefoundry Artifact Source
1116
+ +docs=Input for Artifact from TrueFoundry Artifact Registry
1117
+ +label=TrueFoundry Artifact Source
1128
1118
  """
1129
1119
 
1130
1120
  type: Literal["truefoundry-artifact"] = Field(
@@ -1140,6 +1130,16 @@ class TruefoundryArtifactSource(BaseModel):
1140
1130
  )
1141
1131
 
1142
1132
 
1133
+ class TrueFoundryInteractiveLogin(BaseModel):
1134
+ """
1135
+ +label=Login with truefoundry
1136
+ """
1137
+
1138
+ type: Literal["truefoundry_oauth"] = Field(
1139
+ ..., description="+value=truefoundry_oauth"
1140
+ )
1141
+
1142
+
1143
1143
  class VolumeBrowser(BaseModel):
1144
1144
  """
1145
1145
  +label=Volume Browser
@@ -1172,7 +1172,7 @@ class VolumeMount(BaseModel):
1172
1172
  )
1173
1173
  volume_fqn: constr(regex=r"^tfy-volume:\/\/.+:.+:.+$") = Field(
1174
1174
  ...,
1175
- description="+label=Volume\n+usage=The Truefoundry volume that needs to be mounted.",
1175
+ description="+label=Volume\n+usage=The TrueFoundry volume that needs to be mounted.",
1176
1176
  )
1177
1177
 
1178
1178
 
@@ -1203,7 +1203,7 @@ class ArtifactsDownload(BaseModel):
1203
1203
  """
1204
1204
 
1205
1205
  cache_volume: Optional[ArtifactsCacheVolume] = None
1206
- artifacts: List[Union[TruefoundryArtifactSource, HuggingfaceArtifactSource]] = (
1206
+ artifacts: List[Union[TrueFoundryArtifactSource, HuggingfaceArtifactSource]] = (
1207
1207
  Field(
1208
1208
  ..., description="+label=Artifacts\n+usage=List of artifacts to be cached"
1209
1209
  )
@@ -1228,7 +1228,7 @@ class BaseWorkbenchInput(BaseModel):
1228
1228
  +docs=Describes the configuration for the service
1229
1229
  """
1230
1230
 
1231
- name: constr(regex=r"^[a-z][a-z0-9\-]{1,30}[a-z0-9]$") = Field(
1231
+ name: constr(regex=r"^[a-z](?:[a-z0-9]|-(?!-)){1,30}[a-z0-9]$") = Field(
1232
1232
  ...,
1233
1233
  description="+usage=Name of the workbench. This uniquely identifies this workbench in the workspace.\n> Name can only contain alphanumeric characters and '-' and can be atmost 25 characters long\n+sort=1\n+message=3 to 32 lower case characters long alphanumeric word, may contain - in between, cannot start with a number",
1234
1234
  )
@@ -1378,7 +1378,7 @@ class HealthProbe(BaseModel):
1378
1378
 
1379
1379
  class Helm(BaseModel):
1380
1380
  type: Literal["helm"] = Field(..., description="+value=helm")
1381
- name: constr(regex=r"^[a-z][a-z0-9\-]{1,30}[a-z0-9]$") = Field(
1381
+ name: constr(regex=r"^[a-z](?:[a-z0-9]|-(?!-)){1,30}[a-z0-9]$") = Field(
1382
1382
  ...,
1383
1383
  description="+sort=1\n+message=3 to 32 lower case characters long alphanumeric word, may contain - in between, cannot start with a number\n+usage=Name of the Helm deployment. This will be set as the release name of the chart you are deploying.",
1384
1384
  )
@@ -1405,7 +1405,7 @@ class Job(BaseModel):
1405
1405
  """
1406
1406
 
1407
1407
  type: Literal["job"] = Field(..., description="+value=job")
1408
- name: constr(regex=r"^[a-z][a-z0-9\-]{1,30}[a-z0-9]$") = Field(
1408
+ name: constr(regex=r"^[a-z](?:[a-z0-9]|-(?!-)){1,30}[a-z0-9]$") = Field(
1409
1409
  ...,
1410
1410
  description="+usage=Name of the job\n+sort=1\n+message=3 to 32 lower case characters long alphanumeric word, may contain - in between, cannot start with a number",
1411
1411
  )
@@ -1673,7 +1673,7 @@ class SparkExecutorConfig(BaseModel):
1673
1673
 
1674
1674
  class SparkJob(BaseModel):
1675
1675
  type: Literal["spark-job"] = Field(..., description="+value=spark-job\n+sort=1")
1676
- name: constr(regex=r"^[a-z][a-z0-9\-]{1,30}[a-z0-9]$") = Field(
1676
+ name: constr(regex=r"^[a-z](?:[a-z0-9]|-(?!-)){1,30}[a-z0-9]$") = Field(
1677
1677
  ...,
1678
1678
  description="+label=Name\n+usage=Name of the job\n+message=3 to 32 lower case characters long alphanumeric word, may contain - in between, cannot start with a number\n+sort=2",
1679
1679
  )
@@ -1715,7 +1715,7 @@ class SparkJob(BaseModel):
1715
1715
 
1716
1716
  class Volume(BaseModel):
1717
1717
  type: Literal["volume"] = Field(..., description="+value=volume")
1718
- name: constr(regex=r"^[a-z][a-z0-9\-]{1,30}[a-z0-9]$") = Field(
1718
+ name: constr(regex=r"^[a-z](?:[a-z0-9]|-(?!-)){1,30}[a-z0-9]$") = Field(
1719
1719
  ...,
1720
1720
  description="+sort=1\n+message=3 to 32 lower case characters long alphanumeric word, may contain - in between, cannot start with a number\n+usage=Name of the Volume. This will be set as the volume name.",
1721
1721
  )
@@ -1750,7 +1750,7 @@ class WorkerConfig(BaseModel):
1750
1750
 
1751
1751
 
1752
1752
  class BaseService(BaseModel):
1753
- name: constr(regex=r"^[a-z][a-z0-9\-]{1,30}[a-z0-9]$") = Field(
1753
+ name: constr(regex=r"^[a-z](?:[a-z0-9]|-(?!-)){1,30}[a-z0-9]$") = Field(
1754
1754
  ...,
1755
1755
  description="+usage=Name of the service. This uniquely identifies this service in the workspace.\n> Name can only contain alphanumeric characters and '-' and can be atmost 25 characters long\n+sort=1\n+message=3 to 32 lower case characters long alphanumeric word, may contain - in between, cannot start with a number",
1756
1756
  )
@@ -1845,7 +1845,7 @@ class Workflow(BaseModel):
1845
1845
  """
1846
1846
 
1847
1847
  type: Literal["workflow"] = Field(..., description="+value=workflow")
1848
- name: constr(regex=r"^[a-z][a-z0-9\-]{1,30}[a-z0-9]$") = Field(
1848
+ name: constr(regex=r"^[a-z](?:[a-z0-9]|-(?!-)){1,30}[a-z0-9]$") = Field(
1849
1849
  ...,
1850
1850
  description="+usage=Name of the workflow\n+sort=1\n+message=3 to 32 lower case characters long alphanumeric word, may contain - in between, cannot start with a number",
1851
1851
  )
@@ -1,6 +1,5 @@
1
1
  from truefoundry.deploy.cli.commands.apply_command import get_apply_command
2
2
  from truefoundry.deploy.cli.commands.build_command import get_build_command
3
- from truefoundry.deploy.cli.commands.build_logs_command import get_build_logs_command
4
3
  from truefoundry.deploy.cli.commands.create_command import get_create_command
5
4
  from truefoundry.deploy.cli.commands.delete_command import get_delete_command
6
5
  from truefoundry.deploy.cli.commands.deploy_command import get_deploy_command
@@ -273,20 +273,33 @@ class ServiceFoundryServiceClient(BaseServiceFoundryServiceClient):
273
273
  subscribe_message: str,
274
274
  socketio_path: str = "socket.io",
275
275
  callback=None,
276
- wait=True,
277
- ):
276
+ ) -> socketio.Client:
278
277
  callback = callback or OutputCallBack()
279
278
  sio = socketio.Client(request_timeout=60)
280
279
  callback.print_line("Waiting for the task to start...")
280
+ next_log_start_timestamp = query_dict.get("startTs")
281
281
 
282
282
  @sio.on(subscribe_message)
283
283
  def logs(data):
284
284
  try:
285
285
  _log = json.loads(data)
286
- callback.print_line(self._get_log_print_line(_log["body"]))
286
+ body = _log["body"]
287
+ callback.print_line(self._get_log_print_line(body))
288
+ nonlocal next_log_start_timestamp
289
+ next_log_start_timestamp = body["time"]
287
290
  except Exception:
288
291
  logger.exception(f"Error while parsing log line, {data!r}")
289
292
 
293
+ @sio.on("connect")
294
+ def on_connect():
295
+ # TODO: We should have have a timeout here. `emit` does
296
+ # not support timeout. Explore `sio.call`.
297
+ query_dict["startTs"] = next_log_start_timestamp
298
+ sio.emit(
299
+ subscribe_message,
300
+ json.dumps(query_dict),
301
+ )
302
+
290
303
  def sio_disconnect_no_exception():
291
304
  try:
292
305
  sio.disconnect()
@@ -300,14 +313,7 @@ class ServiceFoundryServiceClient(BaseServiceFoundryServiceClient):
300
313
  headers=self._get_header(),
301
314
  socketio_path=socketio_path,
302
315
  )
303
- # TODO: We should have have a timeout here. `emit` does
304
- # not support timeout. Explore `sio.call`.
305
- sio.emit(
306
- subscribe_message,
307
- json.dumps(query_dict),
308
- )
309
- if wait:
310
- sio.wait()
316
+ return sio
311
317
 
312
318
  @check_min_cli_version
313
319
  def get_deployment(self, application_id: str, deployment_id: str) -> Deployment:
@@ -372,11 +378,10 @@ class ServiceFoundryServiceClient(BaseServiceFoundryServiceClient):
372
378
  self,
373
379
  build_response: BuildResponse,
374
380
  callback=None,
375
- wait: bool = True,
376
- ):
381
+ ) -> socketio.Client:
377
382
  callback = callback or OutputCallBack()
378
383
  tail_logs_obj = json.loads(build_response.tailLogsUrl)
379
- self._tail_logs(
384
+ socket = self._tail_logs(
380
385
  tail_logs_url=urljoin(
381
386
  tail_logs_obj["uri"], f"/?type={BUILD_LOGS_SUBSCRIBE_MESSAGE}"
382
387
  ),
@@ -386,9 +391,9 @@ class ServiceFoundryServiceClient(BaseServiceFoundryServiceClient):
386
391
  "startTs": build_response.logsStartTs,
387
392
  },
388
393
  callback=callback,
389
- wait=wait,
390
394
  subscribe_message=BUILD_LOGS_SUBSCRIBE_MESSAGE,
391
395
  )
396
+ return socket
392
397
 
393
398
  @check_min_cli_version
394
399
  def tail_logs_for_deployment(
@@ -416,7 +421,6 @@ class ServiceFoundryServiceClient(BaseServiceFoundryServiceClient):
416
421
  },
417
422
  },
418
423
  callback=callback,
419
- wait=wait,
420
424
  subscribe_message=DEPLOYMENT_LOGS_SUBSCRIBE_MESSAGE,
421
425
  )
422
426
 
@@ -1,7 +1,8 @@
1
1
  import sys
2
2
  import time
3
- from typing import List, Optional, TypeVar
3
+ from typing import Optional, TypeVar
4
4
 
5
+ import socketio
5
6
  from rich.status import Status
6
7
 
7
8
  from truefoundry.common.utils import poll_for_function
@@ -96,18 +97,15 @@ def _log_application_dashboard_url(deployment: Deployment, log_message: str):
96
97
  logger.info(log_message, url)
97
98
 
98
99
 
99
- def _tail_build_logs(build_responses: List[BuildResponse]) -> None:
100
+ def _tail_build_logs(build_response: BuildResponse) -> socketio.Client:
100
101
  client = ServiceFoundryServiceClient()
101
102
 
102
- # TODO: Explore other options like,
103
- # https://rich.readthedocs.io/en/stable/live.html#live-display
104
- # How does docker/compose does multiple build logs?
105
- for build_response in build_responses:
106
- logger.info("Tailing build logs for '%s'", build_response.componentName)
107
- client.tail_build_logs(build_response=build_response, wait=True)
103
+ logger.info("Tailing build logs for '%s'", build_response.componentName)
104
+ socket = client.tail_build_logs(build_response=build_response)
105
+ return socket
108
106
 
109
107
 
110
- def _deploy_wait_handler(
108
+ def _deploy_wait_handler( # noqa: C901
111
109
  deployment: Deployment,
112
110
  ) -> Optional[DeploymentTransitionStatus]:
113
111
  _log_application_dashboard_url(
@@ -122,9 +120,10 @@ def _deploy_wait_handler(
122
120
  last_status_printed = None
123
121
  client = ServiceFoundryServiceClient()
124
122
  start_time = time.monotonic()
125
- total_timeout_time: int = 300
123
+ total_timeout_time: int = 600
126
124
  poll_interval_seconds = 5
127
125
  time_elapsed = 0
126
+ socket: socketio.Client = None
128
127
 
129
128
  for deployment_statuses in poll_for_function(
130
129
  client.get_deployment_statuses,
@@ -150,20 +149,29 @@ def _deploy_wait_handler(
150
149
  last_status_printed = status_to_print
151
150
 
152
151
  if latest_deployment_status.state.isTerminalState:
152
+ if socket and socket.connected:
153
+ socket.disconnect()
153
154
  break
154
155
 
155
156
  if (
156
157
  latest_deployment_status.transition
157
158
  == DeploymentTransitionStatus.BUILDING
158
159
  ):
159
- build_responses = client.get_deployment_build_response(
160
- application_id=deployment.applicationId, deployment_id=deployment.id
161
- )
162
- _tail_build_logs(build_responses)
160
+ if not socket:
161
+ build_responses = client.get_deployment_build_response(
162
+ application_id=deployment.applicationId,
163
+ deployment_id=deployment.id,
164
+ )
165
+ socket = _tail_build_logs(build_responses[0])
163
166
 
164
167
  time_elapsed = time.monotonic() - start_time
165
168
  if time_elapsed > total_timeout_time:
166
- logger.warning("Polled server for %s secs.", int(time_elapsed))
169
+ logger.warning(
170
+ "Polled server for %s secs. Disconnecting from server, the deployment will still continue.",
171
+ int(time_elapsed),
172
+ )
173
+ if socket and socket.connected:
174
+ socket.disconnect()
167
175
  break
168
176
 
169
177
  return last_status_printed
@@ -1,7 +1,7 @@
1
1
  import os
2
2
  import sys
3
3
  from pathlib import Path
4
- from typing import Dict, List, Optional, Union
4
+ from typing import Callable, Dict, List, Optional, Union
5
5
 
6
6
  import requirements
7
7
  from flytekit.configuration import (
@@ -37,13 +37,16 @@ from truefoundry.workflow.workflow import (
37
37
 
38
38
 
39
39
  def _handle_code_upload_for_workflow(
40
- workflow: auto_gen_models.Workflow, workspace_fqn: str
40
+ workflow: auto_gen_models.Workflow,
41
+ workspace_fqn: str,
42
+ upload_code_package: Callable[[str, str, str], str],
41
43
  ) -> auto_gen_models.Workflow:
42
44
  new_workflow = workflow.copy(deep=True)
43
45
  new_workflow.source = local_source_to_remote_source(
44
46
  local_source=workflow.source,
45
47
  workspace_fqn=workspace_fqn,
46
48
  component_name=workflow.name,
49
+ upload_code_package=upload_code_package,
47
50
  )
48
51
  return new_workflow
49
52
 
@@ -283,12 +286,14 @@ def deploy_workflow(
283
286
  logger.info(
284
287
  f"Deploying workflow {workflow.name} to workspace {workspace_fqn} ({workspace_id})"
285
288
  )
289
+ client = ServiceFoundryServiceClient()
286
290
 
287
291
  workflow = _handle_code_upload_for_workflow(
288
- workflow=workflow, workspace_fqn=workspace_fqn
292
+ workflow=workflow,
293
+ workspace_fqn=workspace_fqn,
294
+ upload_code_package=client.upload_code_package,
289
295
  )
290
296
 
291
- client = ServiceFoundryServiceClient()
292
297
  deployment = client.deploy_application(
293
298
  workspace_id=workspace_id, application=workflow
294
299
  )
@@ -504,12 +504,12 @@ class HuggingfaceArtifactSource(models.HuggingfaceArtifactSource, PatchedModelBa
504
504
  type: Literal["huggingface-hub"] = "huggingface-hub"
505
505
 
506
506
 
507
- class TruefoundryArtifactSource(models.TruefoundryArtifactSource, PatchedModelBase):
507
+ class TrueFoundryArtifactSource(models.TrueFoundryArtifactSource, PatchedModelBase):
508
508
  type: Literal["truefoundry-artifact"] = "truefoundry-artifact"
509
509
 
510
510
 
511
- # TODO: When we do the renaming in CUE, we will make TruefoundryArtifactSource the alias and mark it deprecated
512
- TrueFoundryArtifactSource = TruefoundryArtifactSource
511
+ # Deprecated alias, kept for backward compatibility
512
+ TruefoundryArtifactSource = TrueFoundryArtifactSource
513
513
 
514
514
 
515
515
  class ArtifactsDownload(models.ArtifactsDownload, PatchedModelBase):
@@ -175,8 +175,8 @@ from truefoundry.ml.autogen.client.models.experiment_tag_dto import ExperimentTa
175
175
  from truefoundry.ml.autogen.client.models.export_deployment_files_request_dto import (
176
176
  ExportDeploymentFilesRequestDto,
177
177
  )
178
- from truefoundry.ml.autogen.client.models.external_artifact_source import (
179
- ExternalArtifactSource,
178
+ from truefoundry.ml.autogen.client.models.external_blob_storage_source import (
179
+ ExternalBlobStorageSource,
180
180
  )
181
181
  from truefoundry.ml.autogen.client.models.fast_ai_framework import FastAIFramework
182
182
  from truefoundry.ml.autogen.client.models.file_info_dto import FileInfoDto
@@ -403,8 +403,8 @@ from truefoundry.ml.autogen.client.models.trigger_job_run_config_request_dto imp
403
403
  from truefoundry.ml.autogen.client.models.trigger_job_run_config_response_dto import (
404
404
  TriggerJobRunConfigResponseDto,
405
405
  )
406
- from truefoundry.ml.autogen.client.models.true_foundry_artifact_source import (
407
- TrueFoundryArtifactSource,
406
+ from truefoundry.ml.autogen.client.models.true_foundry_managed_source import (
407
+ TrueFoundryManagedSource,
408
408
  )
409
409
  from truefoundry.ml.autogen.client.models.update_artifact_version_request_dto import (
410
410
  UpdateArtifactVersionRequestDto,
@@ -1182,162 +1182,6 @@ class ExperimentsApi:
1182
1182
  _request_auth=_params.get("_request_auth"),
1183
1183
  )
1184
1184
 
1185
- @validate_arguments
1186
- def put_privacy_type_put(
1187
- self, set_experiment_tag_request_dto: SetExperimentTagRequestDto, **kwargs
1188
- ) -> object: # noqa: E501
1189
- """Put Privacy Type # noqa: E501
1190
-
1191
- Set the privacy type of an expermient to Private or Public(publically readable) # noqa: E501
1192
- This method makes a synchronous HTTP request by default. To make an
1193
- asynchronous HTTP request, please pass async_req=True
1194
-
1195
- >>> thread = api.put_privacy_type_put(set_experiment_tag_request_dto, async_req=True)
1196
- >>> result = thread.get()
1197
-
1198
- :param set_experiment_tag_request_dto: (required)
1199
- :type set_experiment_tag_request_dto: SetExperimentTagRequestDto
1200
- :param async_req: Whether to execute the request asynchronously.
1201
- :type async_req: bool, optional
1202
- :param _request_timeout: timeout setting for this request.
1203
- If one number provided, it will be total request
1204
- timeout. It can also be a pair (tuple) of
1205
- (connection, read) timeouts.
1206
- :return: Returns the result object.
1207
- If the method is called asynchronously,
1208
- returns the request thread.
1209
- :rtype: object
1210
- """
1211
- kwargs["_return_http_data_only"] = True
1212
- if "_preload_content" in kwargs:
1213
- message = "Error! Please call the put_privacy_type_put_with_http_info method with `_preload_content` instead and obtain raw data from ApiResponse.raw_data" # noqa: E501
1214
- raise ValueError(message)
1215
- return self.put_privacy_type_put_with_http_info(
1216
- set_experiment_tag_request_dto, **kwargs
1217
- ) # noqa: E501
1218
-
1219
- @validate_arguments
1220
- def put_privacy_type_put_with_http_info(
1221
- self, set_experiment_tag_request_dto: SetExperimentTagRequestDto, **kwargs
1222
- ) -> ApiResponse: # noqa: E501
1223
- """Put Privacy Type # noqa: E501
1224
-
1225
- Set the privacy type of an expermient to Private or Public(publically readable) # noqa: E501
1226
- This method makes a synchronous HTTP request by default. To make an
1227
- asynchronous HTTP request, please pass async_req=True
1228
-
1229
- >>> thread = api.put_privacy_type_put_with_http_info(set_experiment_tag_request_dto, async_req=True)
1230
- >>> result = thread.get()
1231
-
1232
- :param set_experiment_tag_request_dto: (required)
1233
- :type set_experiment_tag_request_dto: SetExperimentTagRequestDto
1234
- :param async_req: Whether to execute the request asynchronously.
1235
- :type async_req: bool, optional
1236
- :param _preload_content: if False, the ApiResponse.data will
1237
- be set to none and raw_data will store the
1238
- HTTP response body without reading/decoding.
1239
- Default is True.
1240
- :type _preload_content: bool, optional
1241
- :param _return_http_data_only: response data instead of ApiResponse
1242
- object with status code, headers, etc
1243
- :type _return_http_data_only: bool, optional
1244
- :param _request_timeout: timeout setting for this request. If one
1245
- number provided, it will be total request
1246
- timeout. It can also be a pair (tuple) of
1247
- (connection, read) timeouts.
1248
- :param _request_auth: set to override the auth_settings for an a single
1249
- request; this effectively ignores the authentication
1250
- in the spec for a single request.
1251
- :type _request_auth: dict, optional
1252
- :type _content_type: string, optional: force content-type for the request
1253
- :return: Returns the result object.
1254
- If the method is called asynchronously,
1255
- returns the request thread.
1256
- :rtype: tuple(object, status_code(int), headers(HTTPHeaderDict))
1257
- """
1258
-
1259
- _params = locals()
1260
-
1261
- _all_params = ["set_experiment_tag_request_dto"]
1262
- _all_params.extend(
1263
- [
1264
- "async_req",
1265
- "_return_http_data_only",
1266
- "_preload_content",
1267
- "_request_timeout",
1268
- "_request_auth",
1269
- "_content_type",
1270
- "_headers",
1271
- ]
1272
- )
1273
-
1274
- # validate the arguments
1275
- for _key, _val in _params["kwargs"].items():
1276
- if _key not in _all_params:
1277
- raise ApiTypeError(
1278
- "Got an unexpected keyword argument '%s'"
1279
- " to method put_privacy_type_put" % _key
1280
- )
1281
- _params[_key] = _val
1282
- del _params["kwargs"]
1283
-
1284
- _collection_formats = {}
1285
-
1286
- # process the path parameters
1287
- _path_params = {}
1288
-
1289
- # process the query parameters
1290
- _query_params = []
1291
- # process the header parameters
1292
- _header_params = dict(_params.get("_headers", {}))
1293
- # process the form parameters
1294
- _form_params = []
1295
- _files = {}
1296
- # process the body parameter
1297
- _body_params = None
1298
- if _params["set_experiment_tag_request_dto"] is not None:
1299
- _body_params = _params["set_experiment_tag_request_dto"]
1300
-
1301
- # set the HTTP header `Accept`
1302
- _header_params["Accept"] = self.api_client.select_header_accept(
1303
- ["application/json"]
1304
- ) # noqa: E501
1305
-
1306
- # set the HTTP header `Content-Type`
1307
- _content_types_list = _params.get(
1308
- "_content_type",
1309
- self.api_client.select_header_content_type(["application/json"]),
1310
- )
1311
- if _content_types_list:
1312
- _header_params["Content-Type"] = _content_types_list
1313
-
1314
- # authentication setting
1315
- _auth_settings = ["HTTPBearer", "APIKeyCookie"] # noqa: E501
1316
-
1317
- _response_types_map = {
1318
- "200": "object",
1319
- "422": "HTTPValidationError",
1320
- }
1321
-
1322
- return self.api_client.call_api(
1323
- "/api/2.0/mlflow/experiments/privacy-type",
1324
- "PUT",
1325
- _path_params,
1326
- _query_params,
1327
- _header_params,
1328
- body=_body_params,
1329
- post_params=_form_params,
1330
- files=_files,
1331
- response_types_map=_response_types_map,
1332
- auth_settings=_auth_settings,
1333
- async_req=_params.get("async_req"),
1334
- _return_http_data_only=_params.get("_return_http_data_only"), # noqa: E501
1335
- _preload_content=_params.get("_preload_content", True),
1336
- _request_timeout=_params.get("_request_timeout"),
1337
- collection_formats=_collection_formats,
1338
- _request_auth=_params.get("_request_auth"),
1339
- )
1340
-
1341
1185
  @validate_arguments
1342
1186
  def restore_experiment_post(
1343
1187
  self, experiment_id_request_dto: ExperimentIdRequestDto, **kwargs