truefoundry 0.3.3__py3-none-any.whl → 0.4.0.dev1__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 (225) hide show
  1. truefoundry/cli/__main__.py +3 -17
  2. truefoundry/common/__init__.py +0 -0
  3. truefoundry/common/request_utils.py +56 -0
  4. truefoundry/deploy/cli/cli.py +1 -1
  5. truefoundry/deploy/cli/util.py +3 -1
  6. truefoundry/deploy/lib/auth/credential_provider.py +2 -12
  7. truefoundry/deploy/lib/clients/servicefoundry_client.py +0 -9
  8. truefoundry/deploy/lib/exceptions.py +1 -6
  9. truefoundry/deploy/lib/session.py +1 -16
  10. truefoundry/langchain/truefoundry_chat.py +1 -1
  11. truefoundry/langchain/truefoundry_embeddings.py +1 -1
  12. truefoundry/langchain/truefoundry_llm.py +1 -1
  13. truefoundry/langchain/utils.py +0 -41
  14. truefoundry/ml/__init__.py +46 -6
  15. truefoundry/ml/artifact/__init__.py +0 -0
  16. truefoundry/ml/artifact/truefoundry_artifact_repo.py +1120 -0
  17. truefoundry/ml/autogen/__init__.py +0 -0
  18. truefoundry/ml/autogen/client/__init__.py +373 -0
  19. truefoundry/ml/autogen/client/api/__init__.py +16 -0
  20. truefoundry/ml/autogen/client/api/auth_api.py +184 -0
  21. truefoundry/ml/autogen/client/api/deprecated_api.py +605 -0
  22. truefoundry/ml/autogen/client/api/experiments_api.py +2109 -0
  23. truefoundry/ml/autogen/client/api/health_api.py +299 -0
  24. truefoundry/ml/autogen/client/api/metrics_api.py +371 -0
  25. truefoundry/ml/autogen/client/api/mlfoundry_artifacts_api.py +7213 -0
  26. truefoundry/ml/autogen/client/api/python_deployment_config_api.py +201 -0
  27. truefoundry/ml/autogen/client/api/run_artifacts_api.py +231 -0
  28. truefoundry/ml/autogen/client/api/runs_api.py +2919 -0
  29. truefoundry/ml/autogen/client/api_client.py +822 -0
  30. truefoundry/ml/autogen/client/api_response.py +30 -0
  31. truefoundry/ml/autogen/client/configuration.py +489 -0
  32. truefoundry/ml/autogen/client/exceptions.py +161 -0
  33. truefoundry/ml/autogen/client/models/__init__.py +344 -0
  34. truefoundry/ml/autogen/client/models/add_custom_metrics_to_model_version_request_dto.py +69 -0
  35. truefoundry/ml/autogen/client/models/add_features_to_model_version_request_dto.py +83 -0
  36. truefoundry/ml/autogen/client/models/agent.py +125 -0
  37. truefoundry/ml/autogen/client/models/agent_app.py +118 -0
  38. truefoundry/ml/autogen/client/models/agent_open_api_tool.py +143 -0
  39. truefoundry/ml/autogen/client/models/agent_open_api_tool_with_fqn.py +144 -0
  40. truefoundry/ml/autogen/client/models/agent_with_fqn.py +127 -0
  41. truefoundry/ml/autogen/client/models/artifact_dto.py +115 -0
  42. truefoundry/ml/autogen/client/models/artifact_response_dto.py +75 -0
  43. truefoundry/ml/autogen/client/models/artifact_type.py +39 -0
  44. truefoundry/ml/autogen/client/models/artifact_version_dto.py +141 -0
  45. truefoundry/ml/autogen/client/models/artifact_version_response_dto.py +77 -0
  46. truefoundry/ml/autogen/client/models/artifact_version_status.py +35 -0
  47. truefoundry/ml/autogen/client/models/assistant_message.py +89 -0
  48. truefoundry/ml/autogen/client/models/authorize_user_for_model_request_dto.py +69 -0
  49. truefoundry/ml/autogen/client/models/authorize_user_for_model_version_request_dto.py +69 -0
  50. truefoundry/ml/autogen/client/models/backfill_default_storage_integration_id_request_dto.py +67 -0
  51. truefoundry/ml/autogen/client/models/blob_storage_reference.py +93 -0
  52. truefoundry/ml/autogen/client/models/body_get_search_runs_get.py +72 -0
  53. truefoundry/ml/autogen/client/models/chat_prompt.py +156 -0
  54. truefoundry/ml/autogen/client/models/chat_prompt_messages_inner.py +171 -0
  55. truefoundry/ml/autogen/client/models/columns_dto.py +73 -0
  56. truefoundry/ml/autogen/client/models/content.py +153 -0
  57. truefoundry/ml/autogen/client/models/content1.py +153 -0
  58. truefoundry/ml/autogen/client/models/content2.py +174 -0
  59. truefoundry/ml/autogen/client/models/content2_any_of_inner.py +150 -0
  60. truefoundry/ml/autogen/client/models/create_artifact_request_dto.py +74 -0
  61. truefoundry/ml/autogen/client/models/create_artifact_response_dto.py +66 -0
  62. truefoundry/ml/autogen/client/models/create_artifact_version_request_dto.py +74 -0
  63. truefoundry/ml/autogen/client/models/create_artifact_version_response_dto.py +66 -0
  64. truefoundry/ml/autogen/client/models/create_dataset_request_dto.py +76 -0
  65. truefoundry/ml/autogen/client/models/create_experiment_request_dto.py +94 -0
  66. truefoundry/ml/autogen/client/models/create_experiment_response_dto.py +67 -0
  67. truefoundry/ml/autogen/client/models/create_model_version_request_dto.py +95 -0
  68. truefoundry/ml/autogen/client/models/create_multi_part_upload_for_dataset_request_dto.py +73 -0
  69. truefoundry/ml/autogen/client/models/create_multi_part_upload_for_dataset_response_dto.py +79 -0
  70. truefoundry/ml/autogen/client/models/create_multi_part_upload_request_dto.py +73 -0
  71. truefoundry/ml/autogen/client/models/create_python_deployment_config_request_dto.py +72 -0
  72. truefoundry/ml/autogen/client/models/create_python_deployment_config_response_dto.py +68 -0
  73. truefoundry/ml/autogen/client/models/create_run_request_dto.py +97 -0
  74. truefoundry/ml/autogen/client/models/create_run_response_dto.py +76 -0
  75. truefoundry/ml/autogen/client/models/dataset_dto.py +112 -0
  76. truefoundry/ml/autogen/client/models/dataset_response_dto.py +75 -0
  77. truefoundry/ml/autogen/client/models/delete_artifact_versions_request_dto.py +65 -0
  78. truefoundry/ml/autogen/client/models/delete_dataset_request_dto.py +74 -0
  79. truefoundry/ml/autogen/client/models/delete_model_version_request_dto.py +65 -0
  80. truefoundry/ml/autogen/client/models/delete_run_request.py +65 -0
  81. truefoundry/ml/autogen/client/models/delete_tag_request_dto.py +68 -0
  82. truefoundry/ml/autogen/client/models/experiment_dto.py +127 -0
  83. truefoundry/ml/autogen/client/models/experiment_id_request_dto.py +67 -0
  84. truefoundry/ml/autogen/client/models/experiment_response_dto.py +75 -0
  85. truefoundry/ml/autogen/client/models/experiment_tag_dto.py +69 -0
  86. truefoundry/ml/autogen/client/models/feature_dto.py +68 -0
  87. truefoundry/ml/autogen/client/models/feature_value_type.py +35 -0
  88. truefoundry/ml/autogen/client/models/file_info_dto.py +76 -0
  89. truefoundry/ml/autogen/client/models/finalize_artifact_version_request_dto.py +101 -0
  90. truefoundry/ml/autogen/client/models/get_experiment_response_dto.py +88 -0
  91. truefoundry/ml/autogen/client/models/get_latest_run_log_response_dto.py +76 -0
  92. truefoundry/ml/autogen/client/models/get_metric_history_response.py +79 -0
  93. truefoundry/ml/autogen/client/models/get_signed_url_for_dataset_write_request_dto.py +68 -0
  94. truefoundry/ml/autogen/client/models/get_signed_urls_for_artifact_version_read_request_dto.py +68 -0
  95. truefoundry/ml/autogen/client/models/get_signed_urls_for_artifact_version_read_response_dto.py +81 -0
  96. truefoundry/ml/autogen/client/models/get_signed_urls_for_artifact_version_write_request_dto.py +69 -0
  97. truefoundry/ml/autogen/client/models/get_signed_urls_for_artifact_version_write_response_dto.py +83 -0
  98. truefoundry/ml/autogen/client/models/get_signed_urls_for_dataset_read_request_dto.py +68 -0
  99. truefoundry/ml/autogen/client/models/get_signed_urls_for_dataset_read_response_dto.py +81 -0
  100. truefoundry/ml/autogen/client/models/get_signed_urls_for_dataset_write_response_dto.py +81 -0
  101. truefoundry/ml/autogen/client/models/get_tenant_id_response_dto.py +74 -0
  102. truefoundry/ml/autogen/client/models/http_validation_error.py +82 -0
  103. truefoundry/ml/autogen/client/models/image_content_part.py +87 -0
  104. truefoundry/ml/autogen/client/models/image_url.py +75 -0
  105. truefoundry/ml/autogen/client/models/internal_metadata.py +180 -0
  106. truefoundry/ml/autogen/client/models/latest_run_log_dto.py +78 -0
  107. truefoundry/ml/autogen/client/models/list_artifact_versions_request_dto.py +107 -0
  108. truefoundry/ml/autogen/client/models/list_artifact_versions_response_dto.py +87 -0
  109. truefoundry/ml/autogen/client/models/list_artifacts_request_dto.py +96 -0
  110. truefoundry/ml/autogen/client/models/list_artifacts_response_dto.py +86 -0
  111. truefoundry/ml/autogen/client/models/list_colums_response_dto.py +75 -0
  112. truefoundry/ml/autogen/client/models/list_datasets_request_dto.py +78 -0
  113. truefoundry/ml/autogen/client/models/list_datasets_response_dto.py +86 -0
  114. truefoundry/ml/autogen/client/models/list_experiments_response_dto.py +86 -0
  115. truefoundry/ml/autogen/client/models/list_files_for_artifact_version_request_dto.py +76 -0
  116. truefoundry/ml/autogen/client/models/list_files_for_artifact_versions_response_dto.py +82 -0
  117. truefoundry/ml/autogen/client/models/list_files_for_dataset_request_dto.py +76 -0
  118. truefoundry/ml/autogen/client/models/list_files_for_dataset_response_dto.py +82 -0
  119. truefoundry/ml/autogen/client/models/list_latest_run_logs_response_dto.py +82 -0
  120. truefoundry/ml/autogen/client/models/list_metric_history_request_dto.py +69 -0
  121. truefoundry/ml/autogen/client/models/list_metric_history_response_dto.py +84 -0
  122. truefoundry/ml/autogen/client/models/list_model_version_response_dto.py +87 -0
  123. truefoundry/ml/autogen/client/models/list_model_versions_request_dto.py +93 -0
  124. truefoundry/ml/autogen/client/models/list_models_request_dto.py +89 -0
  125. truefoundry/ml/autogen/client/models/list_models_response_dto.py +84 -0
  126. truefoundry/ml/autogen/client/models/list_run_artifacts_response_dto.py +84 -0
  127. truefoundry/ml/autogen/client/models/list_run_logs_response_dto.py +82 -0
  128. truefoundry/ml/autogen/client/models/list_seed_experiments_response_dto.py +81 -0
  129. truefoundry/ml/autogen/client/models/log_batch_request_dto.py +106 -0
  130. truefoundry/ml/autogen/client/models/log_metric_request_dto.py +80 -0
  131. truefoundry/ml/autogen/client/models/log_param_request_dto.py +76 -0
  132. truefoundry/ml/autogen/client/models/method.py +37 -0
  133. truefoundry/ml/autogen/client/models/metric_collection_dto.py +82 -0
  134. truefoundry/ml/autogen/client/models/metric_dto.py +76 -0
  135. truefoundry/ml/autogen/client/models/mime_type.py +37 -0
  136. truefoundry/ml/autogen/client/models/model_configuration.py +103 -0
  137. truefoundry/ml/autogen/client/models/model_dto.py +122 -0
  138. truefoundry/ml/autogen/client/models/model_response_dto.py +75 -0
  139. truefoundry/ml/autogen/client/models/model_schema_dto.py +85 -0
  140. truefoundry/ml/autogen/client/models/model_version_dto.py +163 -0
  141. truefoundry/ml/autogen/client/models/model_version_response_dto.py +75 -0
  142. truefoundry/ml/autogen/client/models/multi_part_upload_dto.py +107 -0
  143. truefoundry/ml/autogen/client/models/multi_part_upload_response_dto.py +79 -0
  144. truefoundry/ml/autogen/client/models/multi_part_upload_storage_provider.py +34 -0
  145. truefoundry/ml/autogen/client/models/notify_artifact_version_failure_dto.py +65 -0
  146. truefoundry/ml/autogen/client/models/openapi_spec.py +152 -0
  147. truefoundry/ml/autogen/client/models/param_dto.py +66 -0
  148. truefoundry/ml/autogen/client/models/parameters.py +84 -0
  149. truefoundry/ml/autogen/client/models/prediction_type.py +34 -0
  150. truefoundry/ml/autogen/client/models/resolve_agent_app_response_dto.py +75 -0
  151. truefoundry/ml/autogen/client/models/restore_run_request_dto.py +65 -0
  152. truefoundry/ml/autogen/client/models/run_data_dto.py +104 -0
  153. truefoundry/ml/autogen/client/models/run_dto.py +84 -0
  154. truefoundry/ml/autogen/client/models/run_info_dto.py +105 -0
  155. truefoundry/ml/autogen/client/models/run_log_dto.py +90 -0
  156. truefoundry/ml/autogen/client/models/run_log_input_dto.py +80 -0
  157. truefoundry/ml/autogen/client/models/run_response_dto.py +75 -0
  158. truefoundry/ml/autogen/client/models/run_tag_dto.py +66 -0
  159. truefoundry/ml/autogen/client/models/search_runs_request_dto.py +94 -0
  160. truefoundry/ml/autogen/client/models/search_runs_response_dto.py +84 -0
  161. truefoundry/ml/autogen/client/models/set_experiment_tag_request_dto.py +73 -0
  162. truefoundry/ml/autogen/client/models/set_tag_request_dto.py +76 -0
  163. truefoundry/ml/autogen/client/models/signed_url_dto.py +69 -0
  164. truefoundry/ml/autogen/client/models/stop.py +152 -0
  165. truefoundry/ml/autogen/client/models/store_run_logs_request_dto.py +83 -0
  166. truefoundry/ml/autogen/client/models/system_message.py +89 -0
  167. truefoundry/ml/autogen/client/models/text.py +153 -0
  168. truefoundry/ml/autogen/client/models/text_content_part.py +84 -0
  169. truefoundry/ml/autogen/client/models/update_artifact_version_request_dto.py +74 -0
  170. truefoundry/ml/autogen/client/models/update_dataset_request_dto.py +74 -0
  171. truefoundry/ml/autogen/client/models/update_experiment_request_dto.py +74 -0
  172. truefoundry/ml/autogen/client/models/update_model_version_request_dto.py +93 -0
  173. truefoundry/ml/autogen/client/models/update_run_request_dto.py +78 -0
  174. truefoundry/ml/autogen/client/models/update_run_response_dto.py +76 -0
  175. truefoundry/ml/autogen/client/models/url.py +153 -0
  176. truefoundry/ml/autogen/client/models/user_message.py +89 -0
  177. truefoundry/ml/autogen/client/models/validation_error.py +87 -0
  178. truefoundry/ml/autogen/client/models/validation_error_loc_inner.py +154 -0
  179. truefoundry/ml/autogen/client/rest.py +426 -0
  180. truefoundry/ml/autogen/client_README.md +322 -0
  181. truefoundry/ml/cli/__init__.py +0 -0
  182. truefoundry/ml/cli/cli.py +18 -0
  183. truefoundry/ml/cli/commands/__init__.py +3 -0
  184. truefoundry/ml/cli/commands/download.py +87 -0
  185. truefoundry/ml/constants.py +84 -0
  186. truefoundry/ml/enums.py +70 -0
  187. truefoundry/ml/env_vars.py +13 -0
  188. truefoundry/ml/exceptions.py +8 -0
  189. truefoundry/ml/git_info.py +60 -0
  190. truefoundry/ml/internal_namespace.py +52 -0
  191. truefoundry/ml/log_types/__init__.py +4 -0
  192. truefoundry/ml/log_types/artifacts/artifact.py +427 -0
  193. truefoundry/ml/log_types/artifacts/constants.py +33 -0
  194. truefoundry/ml/log_types/artifacts/dataset.py +383 -0
  195. truefoundry/ml/log_types/artifacts/general_artifact.py +110 -0
  196. truefoundry/ml/log_types/artifacts/model.py +628 -0
  197. truefoundry/ml/log_types/artifacts/model_extras.py +48 -0
  198. truefoundry/ml/log_types/artifacts/utils.py +161 -0
  199. truefoundry/ml/log_types/image/__init__.py +3 -0
  200. truefoundry/ml/log_types/image/constants.py +8 -0
  201. truefoundry/ml/log_types/image/image.py +358 -0
  202. truefoundry/ml/log_types/image/image_normalizer.py +101 -0
  203. truefoundry/ml/log_types/image/types.py +68 -0
  204. truefoundry/ml/log_types/plot.py +281 -0
  205. truefoundry/ml/log_types/pydantic_base.py +10 -0
  206. truefoundry/ml/log_types/utils.py +12 -0
  207. truefoundry/ml/logger.py +17 -0
  208. truefoundry/ml/login.py +241 -0
  209. truefoundry/ml/mlfoundry_api.py +1620 -0
  210. truefoundry/ml/mlfoundry_run.py +1238 -0
  211. truefoundry/ml/run_utils.py +102 -0
  212. truefoundry/ml/services/__init__.py +0 -0
  213. truefoundry/ml/services/auth_service.py +109 -0
  214. truefoundry/ml/services/entities.py +108 -0
  215. truefoundry/ml/services/servicefoundry_service.py +35 -0
  216. truefoundry/ml/services/utils.py +122 -0
  217. truefoundry/ml/session.py +271 -0
  218. truefoundry/ml/validation_utils.py +346 -0
  219. truefoundry/pydantic_v1.py +5 -1
  220. {truefoundry-0.3.3.dist-info → truefoundry-0.4.0.dev1.dist-info}/METADATA +18 -11
  221. truefoundry-0.4.0.dev1.dist-info/RECORD +342 -0
  222. truefoundry-0.3.3.dist-info/RECORD +0 -136
  223. /truefoundry/{python_deploy_codegen.py → deploy/python_deploy_codegen.py} +0 -0
  224. {truefoundry-0.3.3.dist-info → truefoundry-0.4.0.dev1.dist-info}/WHEEL +0 -0
  225. {truefoundry-0.3.3.dist-info → truefoundry-0.4.0.dev1.dist-info}/entry_points.txt +0 -0
File without changes
@@ -0,0 +1,373 @@
1
+ # coding: utf-8
2
+
3
+ # flake8: noqa
4
+
5
+ """
6
+ FastAPI
7
+
8
+ No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
9
+
10
+ The version of the OpenAPI document: 0.1.0
11
+ Generated by OpenAPI Generator (https://openapi-generator.tech)
12
+
13
+ Do not edit the class manually.
14
+ """ # noqa: E501
15
+
16
+ __version__ = "0.1.0"
17
+
18
+ # import apis into sdk package
19
+ from truefoundry.ml.autogen.client.api.auth_api import AuthApi
20
+ from truefoundry.ml.autogen.client.api.deprecated_api import DeprecatedApi
21
+ from truefoundry.ml.autogen.client.api.experiments_api import ExperimentsApi
22
+ from truefoundry.ml.autogen.client.api.health_api import HealthApi
23
+ from truefoundry.ml.autogen.client.api.metrics_api import MetricsApi
24
+ from truefoundry.ml.autogen.client.api.mlfoundry_artifacts_api import (
25
+ MlfoundryArtifactsApi,
26
+ )
27
+ from truefoundry.ml.autogen.client.api.python_deployment_config_api import (
28
+ PythonDeploymentConfigApi,
29
+ )
30
+ from truefoundry.ml.autogen.client.api.run_artifacts_api import RunArtifactsApi
31
+ from truefoundry.ml.autogen.client.api.runs_api import RunsApi
32
+
33
+ # import ApiClient
34
+ from truefoundry.ml.autogen.client.api_response import ApiResponse
35
+ from truefoundry.ml.autogen.client.api_client import ApiClient
36
+ from truefoundry.ml.autogen.client.configuration import Configuration
37
+ from truefoundry.ml.autogen.client.exceptions import OpenApiException
38
+ from truefoundry.ml.autogen.client.exceptions import ApiTypeError
39
+ from truefoundry.ml.autogen.client.exceptions import ApiValueError
40
+ from truefoundry.ml.autogen.client.exceptions import ApiKeyError
41
+ from truefoundry.ml.autogen.client.exceptions import ApiAttributeError
42
+ from truefoundry.ml.autogen.client.exceptions import ApiException
43
+
44
+ # import models into sdk package
45
+ from truefoundry.ml.autogen.client.models.add_custom_metrics_to_model_version_request_dto import (
46
+ AddCustomMetricsToModelVersionRequestDto,
47
+ )
48
+ from truefoundry.ml.autogen.client.models.add_features_to_model_version_request_dto import (
49
+ AddFeaturesToModelVersionRequestDto,
50
+ )
51
+ from truefoundry.ml.autogen.client.models.agent import Agent
52
+ from truefoundry.ml.autogen.client.models.agent_app import AgentApp
53
+ from truefoundry.ml.autogen.client.models.agent_open_api_tool import AgentOpenAPITool
54
+ from truefoundry.ml.autogen.client.models.agent_open_api_tool_with_fqn import (
55
+ AgentOpenAPIToolWithFQN,
56
+ )
57
+ from truefoundry.ml.autogen.client.models.agent_with_fqn import AgentWithFQN
58
+ from truefoundry.ml.autogen.client.models.artifact_dto import ArtifactDto
59
+ from truefoundry.ml.autogen.client.models.artifact_response_dto import (
60
+ ArtifactResponseDto,
61
+ )
62
+ from truefoundry.ml.autogen.client.models.artifact_type import ArtifactType
63
+ from truefoundry.ml.autogen.client.models.artifact_version_dto import ArtifactVersionDto
64
+ from truefoundry.ml.autogen.client.models.artifact_version_response_dto import (
65
+ ArtifactVersionResponseDto,
66
+ )
67
+ from truefoundry.ml.autogen.client.models.artifact_version_status import (
68
+ ArtifactVersionStatus,
69
+ )
70
+ from truefoundry.ml.autogen.client.models.assistant_message import AssistantMessage
71
+ from truefoundry.ml.autogen.client.models.authorize_user_for_model_request_dto import (
72
+ AuthorizeUserForModelRequestDto,
73
+ )
74
+ from truefoundry.ml.autogen.client.models.authorize_user_for_model_version_request_dto import (
75
+ AuthorizeUserForModelVersionRequestDto,
76
+ )
77
+ from truefoundry.ml.autogen.client.models.backfill_default_storage_integration_id_request_dto import (
78
+ BackfillDefaultStorageIntegrationIdRequestDto,
79
+ )
80
+ from truefoundry.ml.autogen.client.models.blob_storage_reference import (
81
+ BlobStorageReference,
82
+ )
83
+ from truefoundry.ml.autogen.client.models.body_get_search_runs_get import (
84
+ BodyGetSearchRunsGet,
85
+ )
86
+ from truefoundry.ml.autogen.client.models.chat_prompt import ChatPrompt
87
+ from truefoundry.ml.autogen.client.models.chat_prompt_messages_inner import (
88
+ ChatPromptMessagesInner,
89
+ )
90
+ from truefoundry.ml.autogen.client.models.columns_dto import ColumnsDto
91
+ from truefoundry.ml.autogen.client.models.content import Content
92
+ from truefoundry.ml.autogen.client.models.content1 import Content1
93
+ from truefoundry.ml.autogen.client.models.content2 import Content2
94
+ from truefoundry.ml.autogen.client.models.content2_any_of_inner import (
95
+ Content2AnyOfInner,
96
+ )
97
+ from truefoundry.ml.autogen.client.models.create_artifact_request_dto import (
98
+ CreateArtifactRequestDto,
99
+ )
100
+ from truefoundry.ml.autogen.client.models.create_artifact_response_dto import (
101
+ CreateArtifactResponseDto,
102
+ )
103
+ from truefoundry.ml.autogen.client.models.create_artifact_version_request_dto import (
104
+ CreateArtifactVersionRequestDto,
105
+ )
106
+ from truefoundry.ml.autogen.client.models.create_artifact_version_response_dto import (
107
+ CreateArtifactVersionResponseDto,
108
+ )
109
+ from truefoundry.ml.autogen.client.models.create_dataset_request_dto import (
110
+ CreateDatasetRequestDto,
111
+ )
112
+ from truefoundry.ml.autogen.client.models.create_experiment_request_dto import (
113
+ CreateExperimentRequestDto,
114
+ )
115
+ from truefoundry.ml.autogen.client.models.create_experiment_response_dto import (
116
+ CreateExperimentResponseDto,
117
+ )
118
+ from truefoundry.ml.autogen.client.models.create_model_version_request_dto import (
119
+ CreateModelVersionRequestDto,
120
+ )
121
+ from truefoundry.ml.autogen.client.models.create_multi_part_upload_for_dataset_request_dto import (
122
+ CreateMultiPartUploadForDatasetRequestDto,
123
+ )
124
+ from truefoundry.ml.autogen.client.models.create_multi_part_upload_for_dataset_response_dto import (
125
+ CreateMultiPartUploadForDatasetResponseDto,
126
+ )
127
+ from truefoundry.ml.autogen.client.models.create_multi_part_upload_request_dto import (
128
+ CreateMultiPartUploadRequestDto,
129
+ )
130
+ from truefoundry.ml.autogen.client.models.create_python_deployment_config_request_dto import (
131
+ CreatePythonDeploymentConfigRequestDto,
132
+ )
133
+ from truefoundry.ml.autogen.client.models.create_python_deployment_config_response_dto import (
134
+ CreatePythonDeploymentConfigResponseDto,
135
+ )
136
+ from truefoundry.ml.autogen.client.models.create_run_request_dto import (
137
+ CreateRunRequestDto,
138
+ )
139
+ from truefoundry.ml.autogen.client.models.create_run_response_dto import (
140
+ CreateRunResponseDto,
141
+ )
142
+ from truefoundry.ml.autogen.client.models.dataset_dto import DatasetDto
143
+ from truefoundry.ml.autogen.client.models.dataset_response_dto import DatasetResponseDto
144
+ from truefoundry.ml.autogen.client.models.delete_artifact_versions_request_dto import (
145
+ DeleteArtifactVersionsRequestDto,
146
+ )
147
+ from truefoundry.ml.autogen.client.models.delete_dataset_request_dto import (
148
+ DeleteDatasetRequestDto,
149
+ )
150
+ from truefoundry.ml.autogen.client.models.delete_model_version_request_dto import (
151
+ DeleteModelVersionRequestDto,
152
+ )
153
+ from truefoundry.ml.autogen.client.models.delete_run_request import DeleteRunRequest
154
+ from truefoundry.ml.autogen.client.models.delete_tag_request_dto import (
155
+ DeleteTagRequestDto,
156
+ )
157
+ from truefoundry.ml.autogen.client.models.experiment_dto import ExperimentDto
158
+ from truefoundry.ml.autogen.client.models.experiment_id_request_dto import (
159
+ ExperimentIdRequestDto,
160
+ )
161
+ from truefoundry.ml.autogen.client.models.experiment_response_dto import (
162
+ ExperimentResponseDto,
163
+ )
164
+ from truefoundry.ml.autogen.client.models.experiment_tag_dto import ExperimentTagDto
165
+ from truefoundry.ml.autogen.client.models.feature_dto import FeatureDto
166
+ from truefoundry.ml.autogen.client.models.feature_value_type import FeatureValueType
167
+ from truefoundry.ml.autogen.client.models.file_info_dto import FileInfoDto
168
+ from truefoundry.ml.autogen.client.models.finalize_artifact_version_request_dto import (
169
+ FinalizeArtifactVersionRequestDto,
170
+ )
171
+ from truefoundry.ml.autogen.client.models.get_experiment_response_dto import (
172
+ GetExperimentResponseDto,
173
+ )
174
+ from truefoundry.ml.autogen.client.models.get_latest_run_log_response_dto import (
175
+ GetLatestRunLogResponseDto,
176
+ )
177
+ from truefoundry.ml.autogen.client.models.get_metric_history_response import (
178
+ GetMetricHistoryResponse,
179
+ )
180
+ from truefoundry.ml.autogen.client.models.get_signed_url_for_dataset_write_request_dto import (
181
+ GetSignedURLForDatasetWriteRequestDto,
182
+ )
183
+ from truefoundry.ml.autogen.client.models.get_signed_urls_for_artifact_version_read_request_dto import (
184
+ GetSignedURLsForArtifactVersionReadRequestDto,
185
+ )
186
+ from truefoundry.ml.autogen.client.models.get_signed_urls_for_artifact_version_read_response_dto import (
187
+ GetSignedURLsForArtifactVersionReadResponseDto,
188
+ )
189
+ from truefoundry.ml.autogen.client.models.get_signed_urls_for_artifact_version_write_request_dto import (
190
+ GetSignedURLsForArtifactVersionWriteRequestDto,
191
+ )
192
+ from truefoundry.ml.autogen.client.models.get_signed_urls_for_artifact_version_write_response_dto import (
193
+ GetSignedURLsForArtifactVersionWriteResponseDto,
194
+ )
195
+ from truefoundry.ml.autogen.client.models.get_signed_urls_for_dataset_read_request_dto import (
196
+ GetSignedURLsForDatasetReadRequestDto,
197
+ )
198
+ from truefoundry.ml.autogen.client.models.get_signed_urls_for_dataset_read_response_dto import (
199
+ GetSignedURLsForDatasetReadResponseDto,
200
+ )
201
+ from truefoundry.ml.autogen.client.models.get_signed_urls_for_dataset_write_response_dto import (
202
+ GetSignedURLsForDatasetWriteResponseDto,
203
+ )
204
+ from truefoundry.ml.autogen.client.models.get_tenant_id_response_dto import (
205
+ GetTenantIdResponseDto,
206
+ )
207
+ from truefoundry.ml.autogen.client.models.http_validation_error import (
208
+ HTTPValidationError,
209
+ )
210
+ from truefoundry.ml.autogen.client.models.image_content_part import ImageContentPart
211
+ from truefoundry.ml.autogen.client.models.image_url import ImageUrl
212
+ from truefoundry.ml.autogen.client.models.internal_metadata import InternalMetadata
213
+ from truefoundry.ml.autogen.client.models.latest_run_log_dto import LatestRunLogDto
214
+ from truefoundry.ml.autogen.client.models.list_artifact_versions_request_dto import (
215
+ ListArtifactVersionsRequestDto,
216
+ )
217
+ from truefoundry.ml.autogen.client.models.list_artifact_versions_response_dto import (
218
+ ListArtifactVersionsResponseDto,
219
+ )
220
+ from truefoundry.ml.autogen.client.models.list_artifacts_request_dto import (
221
+ ListArtifactsRequestDto,
222
+ )
223
+ from truefoundry.ml.autogen.client.models.list_artifacts_response_dto import (
224
+ ListArtifactsResponseDto,
225
+ )
226
+ from truefoundry.ml.autogen.client.models.list_colums_response_dto import (
227
+ ListColumsResponseDto,
228
+ )
229
+ from truefoundry.ml.autogen.client.models.list_datasets_request_dto import (
230
+ ListDatasetsRequestDto,
231
+ )
232
+ from truefoundry.ml.autogen.client.models.list_datasets_response_dto import (
233
+ ListDatasetsResponseDto,
234
+ )
235
+ from truefoundry.ml.autogen.client.models.list_experiments_response_dto import (
236
+ ListExperimentsResponseDto,
237
+ )
238
+ from truefoundry.ml.autogen.client.models.list_files_for_artifact_version_request_dto import (
239
+ ListFilesForArtifactVersionRequestDto,
240
+ )
241
+ from truefoundry.ml.autogen.client.models.list_files_for_artifact_versions_response_dto import (
242
+ ListFilesForArtifactVersionsResponseDto,
243
+ )
244
+ from truefoundry.ml.autogen.client.models.list_files_for_dataset_request_dto import (
245
+ ListFilesForDatasetRequestDto,
246
+ )
247
+ from truefoundry.ml.autogen.client.models.list_files_for_dataset_response_dto import (
248
+ ListFilesForDatasetResponseDto,
249
+ )
250
+ from truefoundry.ml.autogen.client.models.list_latest_run_logs_response_dto import (
251
+ ListLatestRunLogsResponseDto,
252
+ )
253
+ from truefoundry.ml.autogen.client.models.list_metric_history_request_dto import (
254
+ ListMetricHistoryRequestDto,
255
+ )
256
+ from truefoundry.ml.autogen.client.models.list_metric_history_response_dto import (
257
+ ListMetricHistoryResponseDto,
258
+ )
259
+ from truefoundry.ml.autogen.client.models.list_model_version_response_dto import (
260
+ ListModelVersionResponseDto,
261
+ )
262
+ from truefoundry.ml.autogen.client.models.list_model_versions_request_dto import (
263
+ ListModelVersionsRequestDto,
264
+ )
265
+ from truefoundry.ml.autogen.client.models.list_models_request_dto import (
266
+ ListModelsRequestDto,
267
+ )
268
+ from truefoundry.ml.autogen.client.models.list_models_response_dto import (
269
+ ListModelsResponseDto,
270
+ )
271
+ from truefoundry.ml.autogen.client.models.list_run_artifacts_response_dto import (
272
+ ListRunArtifactsResponseDto,
273
+ )
274
+ from truefoundry.ml.autogen.client.models.list_run_logs_response_dto import (
275
+ ListRunLogsResponseDto,
276
+ )
277
+ from truefoundry.ml.autogen.client.models.list_seed_experiments_response_dto import (
278
+ ListSeedExperimentsResponseDto,
279
+ )
280
+ from truefoundry.ml.autogen.client.models.log_batch_request_dto import (
281
+ LogBatchRequestDto,
282
+ )
283
+ from truefoundry.ml.autogen.client.models.log_metric_request_dto import (
284
+ LogMetricRequestDto,
285
+ )
286
+ from truefoundry.ml.autogen.client.models.log_param_request_dto import (
287
+ LogParamRequestDto,
288
+ )
289
+ from truefoundry.ml.autogen.client.models.method import Method
290
+ from truefoundry.ml.autogen.client.models.metric_collection_dto import (
291
+ MetricCollectionDto,
292
+ )
293
+ from truefoundry.ml.autogen.client.models.metric_dto import MetricDto
294
+ from truefoundry.ml.autogen.client.models.mime_type import MimeType
295
+ from truefoundry.ml.autogen.client.models.model_configuration import ModelConfiguration
296
+ from truefoundry.ml.autogen.client.models.model_dto import ModelDto
297
+ from truefoundry.ml.autogen.client.models.model_response_dto import ModelResponseDto
298
+ from truefoundry.ml.autogen.client.models.model_schema_dto import ModelSchemaDto
299
+ from truefoundry.ml.autogen.client.models.model_version_dto import ModelVersionDto
300
+ from truefoundry.ml.autogen.client.models.model_version_response_dto import (
301
+ ModelVersionResponseDto,
302
+ )
303
+ from truefoundry.ml.autogen.client.models.multi_part_upload_dto import (
304
+ MultiPartUploadDto,
305
+ )
306
+ from truefoundry.ml.autogen.client.models.multi_part_upload_response_dto import (
307
+ MultiPartUploadResponseDto,
308
+ )
309
+ from truefoundry.ml.autogen.client.models.multi_part_upload_storage_provider import (
310
+ MultiPartUploadStorageProvider,
311
+ )
312
+ from truefoundry.ml.autogen.client.models.notify_artifact_version_failure_dto import (
313
+ NotifyArtifactVersionFailureDto,
314
+ )
315
+ from truefoundry.ml.autogen.client.models.openapi_spec import OpenapiSpec
316
+ from truefoundry.ml.autogen.client.models.param_dto import ParamDto
317
+ from truefoundry.ml.autogen.client.models.parameters import Parameters
318
+ from truefoundry.ml.autogen.client.models.prediction_type import PredictionType
319
+ from truefoundry.ml.autogen.client.models.resolve_agent_app_response_dto import (
320
+ ResolveAgentAppResponseDto,
321
+ )
322
+ from truefoundry.ml.autogen.client.models.restore_run_request_dto import (
323
+ RestoreRunRequestDto,
324
+ )
325
+ from truefoundry.ml.autogen.client.models.run_data_dto import RunDataDto
326
+ from truefoundry.ml.autogen.client.models.run_dto import RunDto
327
+ from truefoundry.ml.autogen.client.models.run_info_dto import RunInfoDto
328
+ from truefoundry.ml.autogen.client.models.run_log_dto import RunLogDto
329
+ from truefoundry.ml.autogen.client.models.run_log_input_dto import RunLogInputDto
330
+ from truefoundry.ml.autogen.client.models.run_response_dto import RunResponseDto
331
+ from truefoundry.ml.autogen.client.models.run_tag_dto import RunTagDto
332
+ from truefoundry.ml.autogen.client.models.search_runs_request_dto import (
333
+ SearchRunsRequestDto,
334
+ )
335
+ from truefoundry.ml.autogen.client.models.search_runs_response_dto import (
336
+ SearchRunsResponseDto,
337
+ )
338
+ from truefoundry.ml.autogen.client.models.set_experiment_tag_request_dto import (
339
+ SetExperimentTagRequestDto,
340
+ )
341
+ from truefoundry.ml.autogen.client.models.set_tag_request_dto import SetTagRequestDto
342
+ from truefoundry.ml.autogen.client.models.signed_url_dto import SignedURLDto
343
+ from truefoundry.ml.autogen.client.models.stop import Stop
344
+ from truefoundry.ml.autogen.client.models.store_run_logs_request_dto import (
345
+ StoreRunLogsRequestDto,
346
+ )
347
+ from truefoundry.ml.autogen.client.models.system_message import SystemMessage
348
+ from truefoundry.ml.autogen.client.models.text import Text
349
+ from truefoundry.ml.autogen.client.models.text_content_part import TextContentPart
350
+ from truefoundry.ml.autogen.client.models.update_artifact_version_request_dto import (
351
+ UpdateArtifactVersionRequestDto,
352
+ )
353
+ from truefoundry.ml.autogen.client.models.update_dataset_request_dto import (
354
+ UpdateDatasetRequestDto,
355
+ )
356
+ from truefoundry.ml.autogen.client.models.update_experiment_request_dto import (
357
+ UpdateExperimentRequestDto,
358
+ )
359
+ from truefoundry.ml.autogen.client.models.update_model_version_request_dto import (
360
+ UpdateModelVersionRequestDto,
361
+ )
362
+ from truefoundry.ml.autogen.client.models.update_run_request_dto import (
363
+ UpdateRunRequestDto,
364
+ )
365
+ from truefoundry.ml.autogen.client.models.update_run_response_dto import (
366
+ UpdateRunResponseDto,
367
+ )
368
+ from truefoundry.ml.autogen.client.models.url import Url
369
+ from truefoundry.ml.autogen.client.models.user_message import UserMessage
370
+ from truefoundry.ml.autogen.client.models.validation_error import ValidationError
371
+ from truefoundry.ml.autogen.client.models.validation_error_loc_inner import (
372
+ ValidationErrorLocInner,
373
+ )
@@ -0,0 +1,16 @@
1
+ # flake8: noqa
2
+
3
+ # import apis into api package
4
+ from truefoundry.ml.autogen.client.api.auth_api import AuthApi
5
+ from truefoundry.ml.autogen.client.api.deprecated_api import DeprecatedApi
6
+ from truefoundry.ml.autogen.client.api.experiments_api import ExperimentsApi
7
+ from truefoundry.ml.autogen.client.api.health_api import HealthApi
8
+ from truefoundry.ml.autogen.client.api.metrics_api import MetricsApi
9
+ from truefoundry.ml.autogen.client.api.mlfoundry_artifacts_api import (
10
+ MlfoundryArtifactsApi,
11
+ )
12
+ from truefoundry.ml.autogen.client.api.python_deployment_config_api import (
13
+ PythonDeploymentConfigApi,
14
+ )
15
+ from truefoundry.ml.autogen.client.api.run_artifacts_api import RunArtifactsApi
16
+ from truefoundry.ml.autogen.client.api.runs_api import RunsApi
@@ -0,0 +1,184 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ FastAPI
5
+
6
+ No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
+
8
+ The version of the OpenAPI document: 0.1.0
9
+ Generated by OpenAPI Generator (https://openapi-generator.tech)
10
+
11
+ Do not edit the class manually.
12
+ """ # noqa: E501
13
+
14
+ import re # noqa: F401
15
+
16
+ from truefoundry.ml.autogen.client.api_client import ApiClient
17
+ from truefoundry.ml.autogen.client.api_response import ApiResponse
18
+ from truefoundry.ml.autogen.client.exceptions import ( # noqa: F401
19
+ ApiTypeError,
20
+ ApiValueError,
21
+ )
22
+ from truefoundry.ml.autogen.client.models.get_tenant_id_response_dto import (
23
+ GetTenantIdResponseDto,
24
+ )
25
+ from truefoundry.pydantic_v1 import StrictStr, validate_arguments
26
+
27
+
28
+ class AuthApi:
29
+ """NOTE: This class is auto generated by OpenAPI Generator
30
+ Ref: https://openapi-generator.tech
31
+
32
+ Do not edit the class manually.
33
+ """
34
+
35
+ def __init__(self, api_client=None) -> None:
36
+ if api_client is None:
37
+ api_client = ApiClient.get_default()
38
+ self.api_client = api_client
39
+
40
+ @validate_arguments
41
+ def get_tenant_id_get(
42
+ self, host_name: StrictStr, **kwargs
43
+ ) -> GetTenantIdResponseDto: # noqa: E501
44
+ """Get Tenant Id # noqa: E501
45
+
46
+ Get Tenant Id # noqa: E501
47
+ This method makes a synchronous HTTP request by default. To make an
48
+ asynchronous HTTP request, please pass async_req=True
49
+
50
+ >>> thread = api.get_tenant_id_get(host_name, async_req=True)
51
+ >>> result = thread.get()
52
+
53
+ :param host_name: (required)
54
+ :type host_name: str
55
+ :param async_req: Whether to execute the request asynchronously.
56
+ :type async_req: bool, optional
57
+ :param _request_timeout: timeout setting for this request.
58
+ If one number provided, it will be total request
59
+ timeout. It can also be a pair (tuple) of
60
+ (connection, read) timeouts.
61
+ :return: Returns the result object.
62
+ If the method is called asynchronously,
63
+ returns the request thread.
64
+ :rtype: GetTenantIdResponseDto
65
+ """
66
+ kwargs["_return_http_data_only"] = True
67
+ if "_preload_content" in kwargs:
68
+ message = "Error! Please call the get_tenant_id_get_with_http_info method with `_preload_content` instead and obtain raw data from ApiResponse.raw_data" # noqa: E501
69
+ raise ValueError(message)
70
+ return self.get_tenant_id_get_with_http_info(host_name, **kwargs) # noqa: E501
71
+
72
+ @validate_arguments
73
+ def get_tenant_id_get_with_http_info(
74
+ self, host_name: StrictStr, **kwargs
75
+ ) -> ApiResponse: # noqa: E501
76
+ """Get Tenant Id # noqa: E501
77
+
78
+ Get Tenant Id # noqa: E501
79
+ This method makes a synchronous HTTP request by default. To make an
80
+ asynchronous HTTP request, please pass async_req=True
81
+
82
+ >>> thread = api.get_tenant_id_get_with_http_info(host_name, async_req=True)
83
+ >>> result = thread.get()
84
+
85
+ :param host_name: (required)
86
+ :type host_name: str
87
+ :param async_req: Whether to execute the request asynchronously.
88
+ :type async_req: bool, optional
89
+ :param _preload_content: if False, the ApiResponse.data will
90
+ be set to none and raw_data will store the
91
+ HTTP response body without reading/decoding.
92
+ Default is True.
93
+ :type _preload_content: bool, optional
94
+ :param _return_http_data_only: response data instead of ApiResponse
95
+ object with status code, headers, etc
96
+ :type _return_http_data_only: bool, optional
97
+ :param _request_timeout: timeout setting for this request. If one
98
+ number provided, it will be total request
99
+ timeout. It can also be a pair (tuple) of
100
+ (connection, read) timeouts.
101
+ :param _request_auth: set to override the auth_settings for an a single
102
+ request; this effectively ignores the authentication
103
+ in the spec for a single request.
104
+ :type _request_auth: dict, optional
105
+ :type _content_type: string, optional: force content-type for the request
106
+ :return: Returns the result object.
107
+ If the method is called asynchronously,
108
+ returns the request thread.
109
+ :rtype: tuple(GetTenantIdResponseDto, status_code(int), headers(HTTPHeaderDict))
110
+ """
111
+
112
+ _params = locals()
113
+
114
+ _all_params = ["host_name"]
115
+ _all_params.extend(
116
+ [
117
+ "async_req",
118
+ "_return_http_data_only",
119
+ "_preload_content",
120
+ "_request_timeout",
121
+ "_request_auth",
122
+ "_content_type",
123
+ "_headers",
124
+ ]
125
+ )
126
+
127
+ # validate the arguments
128
+ for _key, _val in _params["kwargs"].items():
129
+ if _key not in _all_params:
130
+ raise ApiTypeError(
131
+ "Got an unexpected keyword argument '%s'"
132
+ " to method get_tenant_id_get" % _key
133
+ )
134
+ _params[_key] = _val
135
+ del _params["kwargs"]
136
+
137
+ _collection_formats = {}
138
+
139
+ # process the path parameters
140
+ _path_params = {}
141
+
142
+ # process the query parameters
143
+ _query_params = []
144
+ if _params.get("host_name") is not None: # noqa: E501
145
+ _query_params.append(("host_name", _params["host_name"]))
146
+
147
+ # process the header parameters
148
+ _header_params = dict(_params.get("_headers", {}))
149
+ # process the form parameters
150
+ _form_params = []
151
+ _files = {}
152
+ # process the body parameter
153
+ _body_params = None
154
+ # set the HTTP header `Accept`
155
+ _header_params["Accept"] = self.api_client.select_header_accept(
156
+ ["application/json"]
157
+ ) # noqa: E501
158
+
159
+ # authentication setting
160
+ _auth_settings = [] # noqa: E501
161
+
162
+ _response_types_map = {
163
+ "200": "GetTenantIdResponseDto",
164
+ "422": "HTTPValidationError",
165
+ }
166
+
167
+ return self.api_client.call_api(
168
+ "/api/2.0/mlflow/tenant-id",
169
+ "GET",
170
+ _path_params,
171
+ _query_params,
172
+ _header_params,
173
+ body=_body_params,
174
+ post_params=_form_params,
175
+ files=_files,
176
+ response_types_map=_response_types_map,
177
+ auth_settings=_auth_settings,
178
+ async_req=_params.get("async_req"),
179
+ _return_http_data_only=_params.get("_return_http_data_only"), # noqa: E501
180
+ _preload_content=_params.get("_preload_content", True),
181
+ _request_timeout=_params.get("_request_timeout"),
182
+ collection_formats=_collection_formats,
183
+ _request_auth=_params.get("_request_auth"),
184
+ )