ob-metaflow-stubs 6.0.3.174__py2.py3-none-any.whl → 6.0.3.175__py2.py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (213) hide show
  1. metaflow-stubs/__init__.pyi +643 -637
  2. metaflow-stubs/cards.pyi +1 -1
  3. metaflow-stubs/cli.pyi +1 -1
  4. metaflow-stubs/cli_components/__init__.pyi +1 -1
  5. metaflow-stubs/cli_components/utils.pyi +1 -1
  6. metaflow-stubs/client/__init__.pyi +1 -1
  7. metaflow-stubs/client/core.pyi +4 -4
  8. metaflow-stubs/client/filecache.pyi +2 -2
  9. metaflow-stubs/events.pyi +2 -2
  10. metaflow-stubs/exception.pyi +1 -1
  11. metaflow-stubs/flowspec.pyi +3 -3
  12. metaflow-stubs/generated_for.txt +1 -1
  13. metaflow-stubs/includefile.pyi +3 -3
  14. metaflow-stubs/info_file.pyi +1 -1
  15. metaflow-stubs/metadata_provider/__init__.pyi +1 -1
  16. metaflow-stubs/metadata_provider/heartbeat.pyi +1 -1
  17. metaflow-stubs/metadata_provider/metadata.pyi +1 -1
  18. metaflow-stubs/metadata_provider/util.pyi +1 -1
  19. metaflow-stubs/metaflow_config.pyi +1 -1
  20. metaflow-stubs/metaflow_current.pyi +114 -114
  21. metaflow-stubs/metaflow_git.pyi +1 -1
  22. metaflow-stubs/mf_extensions/__init__.pyi +1 -1
  23. metaflow-stubs/mf_extensions/obcheckpoint/__init__.pyi +1 -1
  24. metaflow-stubs/mf_extensions/obcheckpoint/plugins/__init__.pyi +1 -1
  25. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/__init__.pyi +1 -1
  26. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/__init__.pyi +1 -1
  27. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/async_cards.pyi +2 -2
  28. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/deco_injection_mixin.pyi +1 -1
  29. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/extra_components.pyi +2 -2
  30. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/__init__.pyi +1 -1
  31. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/__init__.pyi +1 -1
  32. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/checkpoint_lister.pyi +4 -4
  33. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/lineage_card.pyi +1 -1
  34. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/checkpoint_storage.pyi +3 -3
  35. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/constructors.pyi +1 -1
  36. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/core.pyi +3 -3
  37. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/decorator.pyi +3 -3
  38. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/exceptions.pyi +1 -1
  39. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/final_api.pyi +2 -2
  40. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/lineage.pyi +1 -1
  41. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/__init__.pyi +1 -1
  42. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/context.pyi +2 -2
  43. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/core.pyi +2 -2
  44. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/decorator.pyi +1 -1
  45. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/exceptions.pyi +1 -1
  46. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/task_utils.pyi +2 -2
  47. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/utils.pyi +1 -1
  48. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastructures.pyi +1 -1
  49. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/exceptions.pyi +1 -1
  50. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/hf_hub/__init__.pyi +1 -1
  51. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/hf_hub/decorator.pyi +2 -2
  52. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/__init__.pyi +1 -1
  53. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/core.pyi +2 -2
  54. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/exceptions.pyi +1 -1
  55. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/model_storage.pyi +3 -3
  56. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/__init__.pyi +1 -1
  57. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/flowspec_utils.pyi +1 -1
  58. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/general.pyi +1 -1
  59. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/identity_utils.pyi +2 -2
  60. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/__init__.pyi +1 -1
  61. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/base.pyi +1 -1
  62. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/tar.pyi +1 -1
  63. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/tar_utils.pyi +1 -1
  64. metaflow-stubs/mf_extensions/outerbounds/__init__.pyi +1 -1
  65. metaflow-stubs/mf_extensions/outerbounds/plugins/__init__.pyi +1 -1
  66. metaflow-stubs/mf_extensions/outerbounds/plugins/card_utilities/__init__.pyi +1 -1
  67. metaflow-stubs/mf_extensions/outerbounds/plugins/card_utilities/async_cards.pyi +60 -0
  68. metaflow-stubs/mf_extensions/outerbounds/plugins/card_utilities/injector.pyi +1 -1
  69. metaflow-stubs/mf_extensions/outerbounds/plugins/checkpoint_datastores/__init__.pyi +1 -1
  70. metaflow-stubs/mf_extensions/outerbounds/plugins/checkpoint_datastores/coreweave.pyi +1 -1
  71. metaflow-stubs/mf_extensions/outerbounds/plugins/checkpoint_datastores/nebius.pyi +1 -1
  72. metaflow-stubs/mf_extensions/outerbounds/plugins/kubernetes/__init__.pyi +1 -1
  73. metaflow-stubs/mf_extensions/outerbounds/plugins/kubernetes/pod_killer.pyi +1 -1
  74. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/__init__.pyi +1 -1
  75. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/constants.pyi +1 -1
  76. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/exceptions.pyi +1 -1
  77. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/ollama.pyi +60 -2
  78. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/status_card.pyi +73 -0
  79. metaflow-stubs/mf_extensions/outerbounds/plugins/snowflake/__init__.pyi +1 -1
  80. metaflow-stubs/mf_extensions/outerbounds/plugins/snowflake/snowflake.pyi +1 -1
  81. metaflow-stubs/mf_extensions/outerbounds/profilers/__init__.pyi +1 -1
  82. metaflow-stubs/mf_extensions/outerbounds/profilers/gpu.pyi +1 -1
  83. metaflow-stubs/mf_extensions/outerbounds/remote_config.pyi +1 -1
  84. metaflow-stubs/mf_extensions/outerbounds/toplevel/__init__.pyi +1 -1
  85. metaflow-stubs/mf_extensions/outerbounds/toplevel/global_aliases_for_metaflow_package.pyi +1 -1
  86. metaflow-stubs/multicore_utils.pyi +1 -1
  87. metaflow-stubs/parameters.pyi +3 -3
  88. metaflow-stubs/plugins/__init__.pyi +9 -9
  89. metaflow-stubs/plugins/airflow/__init__.pyi +1 -1
  90. metaflow-stubs/plugins/airflow/airflow_utils.pyi +1 -1
  91. metaflow-stubs/plugins/airflow/exception.pyi +1 -1
  92. metaflow-stubs/plugins/airflow/sensors/__init__.pyi +1 -1
  93. metaflow-stubs/plugins/airflow/sensors/base_sensor.pyi +1 -1
  94. metaflow-stubs/plugins/airflow/sensors/external_task_sensor.pyi +1 -1
  95. metaflow-stubs/plugins/airflow/sensors/s3_sensor.pyi +1 -1
  96. metaflow-stubs/plugins/argo/__init__.pyi +1 -1
  97. metaflow-stubs/plugins/argo/argo_client.pyi +1 -1
  98. metaflow-stubs/plugins/argo/argo_events.pyi +1 -1
  99. metaflow-stubs/plugins/argo/argo_workflows.pyi +2 -2
  100. metaflow-stubs/plugins/argo/argo_workflows_decorator.pyi +3 -3
  101. metaflow-stubs/plugins/argo/argo_workflows_deployer.pyi +1 -1
  102. metaflow-stubs/plugins/argo/argo_workflows_deployer_objects.pyi +2 -2
  103. metaflow-stubs/plugins/aws/__init__.pyi +1 -1
  104. metaflow-stubs/plugins/aws/aws_client.pyi +1 -1
  105. metaflow-stubs/plugins/aws/aws_utils.pyi +1 -1
  106. metaflow-stubs/plugins/aws/batch/__init__.pyi +1 -1
  107. metaflow-stubs/plugins/aws/batch/batch.pyi +1 -1
  108. metaflow-stubs/plugins/aws/batch/batch_client.pyi +1 -1
  109. metaflow-stubs/plugins/aws/batch/batch_decorator.pyi +1 -1
  110. metaflow-stubs/plugins/aws/secrets_manager/__init__.pyi +1 -1
  111. metaflow-stubs/plugins/aws/secrets_manager/aws_secrets_manager_secrets_provider.pyi +4 -4
  112. metaflow-stubs/plugins/aws/step_functions/__init__.pyi +1 -1
  113. metaflow-stubs/plugins/aws/step_functions/event_bridge_client.pyi +1 -1
  114. metaflow-stubs/plugins/aws/step_functions/schedule_decorator.pyi +1 -1
  115. metaflow-stubs/plugins/aws/step_functions/step_functions.pyi +1 -1
  116. metaflow-stubs/plugins/aws/step_functions/step_functions_client.pyi +1 -1
  117. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer.pyi +1 -1
  118. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer_objects.pyi +1 -1
  119. metaflow-stubs/plugins/azure/__init__.pyi +1 -1
  120. metaflow-stubs/plugins/azure/azure_credential.pyi +1 -1
  121. metaflow-stubs/plugins/azure/azure_exceptions.pyi +1 -1
  122. metaflow-stubs/plugins/azure/azure_secret_manager_secrets_provider.pyi +4 -4
  123. metaflow-stubs/plugins/azure/azure_utils.pyi +1 -1
  124. metaflow-stubs/plugins/azure/blob_service_client_factory.pyi +1 -1
  125. metaflow-stubs/plugins/azure/includefile_support.pyi +1 -1
  126. metaflow-stubs/plugins/cards/__init__.pyi +1 -1
  127. metaflow-stubs/plugins/cards/card_client.pyi +1 -1
  128. metaflow-stubs/plugins/cards/card_creator.pyi +1 -1
  129. metaflow-stubs/plugins/cards/card_datastore.pyi +1 -1
  130. metaflow-stubs/plugins/cards/card_decorator.pyi +1 -1
  131. metaflow-stubs/plugins/cards/card_modules/__init__.pyi +1 -1
  132. metaflow-stubs/plugins/cards/card_modules/basic.pyi +2 -2
  133. metaflow-stubs/plugins/cards/card_modules/card.pyi +1 -1
  134. metaflow-stubs/plugins/cards/card_modules/components.pyi +2 -2
  135. metaflow-stubs/plugins/cards/card_modules/convert_to_native_type.pyi +1 -1
  136. metaflow-stubs/plugins/cards/card_modules/renderer_tools.pyi +1 -1
  137. metaflow-stubs/plugins/cards/card_modules/test_cards.pyi +1 -1
  138. metaflow-stubs/plugins/cards/card_resolver.pyi +1 -1
  139. metaflow-stubs/plugins/cards/component_serializer.pyi +1 -1
  140. metaflow-stubs/plugins/cards/exception.pyi +1 -1
  141. metaflow-stubs/plugins/catch_decorator.pyi +2 -2
  142. metaflow-stubs/plugins/datatools/__init__.pyi +1 -1
  143. metaflow-stubs/plugins/datatools/local.pyi +1 -1
  144. metaflow-stubs/plugins/datatools/s3/__init__.pyi +1 -1
  145. metaflow-stubs/plugins/datatools/s3/s3.pyi +3 -3
  146. metaflow-stubs/plugins/datatools/s3/s3tail.pyi +1 -1
  147. metaflow-stubs/plugins/datatools/s3/s3util.pyi +1 -1
  148. metaflow-stubs/plugins/debug_logger.pyi +1 -1
  149. metaflow-stubs/plugins/debug_monitor.pyi +1 -1
  150. metaflow-stubs/plugins/environment_decorator.pyi +1 -1
  151. metaflow-stubs/plugins/events_decorator.pyi +1 -1
  152. metaflow-stubs/plugins/frameworks/__init__.pyi +1 -1
  153. metaflow-stubs/plugins/frameworks/pytorch.pyi +1 -1
  154. metaflow-stubs/plugins/gcp/__init__.pyi +1 -1
  155. metaflow-stubs/plugins/gcp/gcp_secret_manager_secrets_provider.pyi +4 -4
  156. metaflow-stubs/plugins/gcp/gs_exceptions.pyi +1 -1
  157. metaflow-stubs/plugins/gcp/gs_storage_client_factory.pyi +1 -1
  158. metaflow-stubs/plugins/gcp/gs_utils.pyi +1 -1
  159. metaflow-stubs/plugins/gcp/includefile_support.pyi +1 -1
  160. metaflow-stubs/plugins/kubernetes/__init__.pyi +1 -1
  161. metaflow-stubs/plugins/kubernetes/kube_utils.pyi +1 -1
  162. metaflow-stubs/plugins/kubernetes/kubernetes.pyi +1 -1
  163. metaflow-stubs/plugins/kubernetes/kubernetes_client.pyi +1 -1
  164. metaflow-stubs/plugins/kubernetes/kubernetes_decorator.pyi +1 -1
  165. metaflow-stubs/plugins/kubernetes/kubernetes_jobsets.pyi +1 -1
  166. metaflow-stubs/plugins/kubernetes/spot_monitor_sidecar.pyi +1 -1
  167. metaflow-stubs/plugins/ollama/__init__.pyi +13 -4
  168. metaflow-stubs/plugins/parallel_decorator.pyi +1 -1
  169. metaflow-stubs/plugins/perimeters.pyi +1 -1
  170. metaflow-stubs/plugins/project_decorator.pyi +1 -1
  171. metaflow-stubs/plugins/pypi/__init__.pyi +2 -2
  172. metaflow-stubs/plugins/pypi/conda_decorator.pyi +1 -1
  173. metaflow-stubs/plugins/pypi/conda_environment.pyi +5 -5
  174. metaflow-stubs/plugins/pypi/parsers.pyi +1 -1
  175. metaflow-stubs/plugins/pypi/pypi_decorator.pyi +1 -1
  176. metaflow-stubs/plugins/pypi/pypi_environment.pyi +1 -1
  177. metaflow-stubs/plugins/pypi/utils.pyi +1 -1
  178. metaflow-stubs/plugins/resources_decorator.pyi +1 -1
  179. metaflow-stubs/plugins/retry_decorator.pyi +1 -1
  180. metaflow-stubs/plugins/secrets/__init__.pyi +2 -2
  181. metaflow-stubs/plugins/secrets/inline_secrets_provider.pyi +3 -3
  182. metaflow-stubs/plugins/secrets/secrets_decorator.pyi +1 -1
  183. metaflow-stubs/plugins/snowflake/__init__.pyi +1 -1
  184. metaflow-stubs/plugins/storage_executor.pyi +1 -1
  185. metaflow-stubs/plugins/test_unbounded_foreach_decorator.pyi +2 -2
  186. metaflow-stubs/plugins/timeout_decorator.pyi +2 -2
  187. metaflow-stubs/plugins/torchtune/__init__.pyi +1 -1
  188. metaflow-stubs/plugins/uv/__init__.pyi +1 -1
  189. metaflow-stubs/plugins/uv/uv_environment.pyi +2 -2
  190. metaflow-stubs/profilers/__init__.pyi +1 -1
  191. metaflow-stubs/pylint_wrapper.pyi +1 -1
  192. metaflow-stubs/runner/__init__.pyi +1 -1
  193. metaflow-stubs/runner/deployer.pyi +4 -4
  194. metaflow-stubs/runner/deployer_impl.pyi +2 -2
  195. metaflow-stubs/runner/metaflow_runner.pyi +2 -2
  196. metaflow-stubs/runner/nbdeploy.pyi +1 -1
  197. metaflow-stubs/runner/nbrun.pyi +1 -1
  198. metaflow-stubs/runner/subprocess_manager.pyi +1 -1
  199. metaflow-stubs/runner/utils.pyi +2 -2
  200. metaflow-stubs/system/__init__.pyi +1 -1
  201. metaflow-stubs/system/system_logger.pyi +1 -1
  202. metaflow-stubs/system/system_monitor.pyi +1 -1
  203. metaflow-stubs/tagging_util.pyi +1 -1
  204. metaflow-stubs/tuple_util.pyi +1 -1
  205. metaflow-stubs/user_configs/__init__.pyi +1 -1
  206. metaflow-stubs/user_configs/config_decorators.pyi +5 -5
  207. metaflow-stubs/user_configs/config_options.pyi +2 -2
  208. metaflow-stubs/user_configs/config_parameters.pyi +6 -6
  209. {ob_metaflow_stubs-6.0.3.174.dist-info → ob_metaflow_stubs-6.0.3.175.dist-info}/METADATA +1 -1
  210. ob_metaflow_stubs-6.0.3.175.dist-info/RECORD +213 -0
  211. ob_metaflow_stubs-6.0.3.174.dist-info/RECORD +0 -211
  212. {ob_metaflow_stubs-6.0.3.174.dist-info → ob_metaflow_stubs-6.0.3.175.dist-info}/WHEEL +0 -0
  213. {ob_metaflow_stubs-6.0.3.174.dist-info → ob_metaflow_stubs-6.0.3.175.dist-info}/top_level.txt +0 -0
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:58.922366 #
4
+ # Generated on 2025-06-02T15:33:51.305382 #
5
5
  ######################################################################################################
6
6
 
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:58.922413 #
4
+ # Generated on 2025-06-02T15:33:51.305426 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:58.922648 #
4
+ # Generated on 2025-06-02T15:33:51.305663 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,11 +1,15 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:58.922987 #
4
+ # Generated on 2025-06-02T15:33:51.306397 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
+ import enum
10
+ import typing
11
+ if typing.TYPE_CHECKING:
12
+ import enum
9
13
 
10
14
  from .exceptions import EmptyOllamaManifestCacheException as EmptyOllamaManifestCacheException
11
15
  from .exceptions import EmptyOllamaBlobCacheException as EmptyOllamaBlobCacheException
@@ -16,12 +20,66 @@ OLLAMA_SUFFIX: str
16
20
  class ProcessStatus(object, metaclass=type):
17
21
  ...
18
22
 
23
+ class CircuitBreakerState(enum.Enum, metaclass=enum.EnumType):
24
+ def __new__(cls, value):
25
+ ...
26
+ ...
27
+
28
+ class CircuitBreaker(object, metaclass=type):
29
+ def __init__(self, failure_threshold, recovery_timeout, reset_timeout, debug = False, status_card = None):
30
+ ...
31
+ def record_success(self):
32
+ ...
33
+ def record_failure(self):
34
+ ...
35
+ def should_attempt_reset(self):
36
+ """
37
+ Check if we should attempt to reset/restart Ollama based on reset_timeout
38
+ """
39
+ ...
40
+ def is_request_allowed(self):
41
+ ...
42
+ def get_status(self):
43
+ ...
44
+ ...
45
+
46
+ class TimeoutCommand(object, metaclass=type):
47
+ def __init__(self, command, timeout, debug = False, **kwargs):
48
+ ...
49
+ def run(self):
50
+ ...
51
+ ...
52
+
53
+ class OllamaHealthChecker(object, metaclass=type):
54
+ def __init__(self, ollama_url, circuit_breaker, ollama_manager, debug = False):
55
+ ...
56
+ def start(self):
57
+ ...
58
+ def stop(self):
59
+ ...
60
+ ...
61
+
62
+ class OllamaRequestInterceptor(object, metaclass=type):
63
+ def __init__(self, circuit_breaker, debug = False):
64
+ ...
65
+ def install_protection(self):
66
+ """
67
+ Install request protection by monkey-patching the ollama package
68
+ """
69
+ ...
70
+ def remove_protection(self):
71
+ """
72
+ Remove request protection by restoring original methods
73
+ """
74
+ ...
75
+ ...
76
+
19
77
  class OllamaManager(object, metaclass=type):
20
78
  """
21
79
  A process manager for Ollama runtimes.
22
80
  Implements interface @ollama([models=...], ...) has a local, remote, or managed backend.
23
81
  """
24
- def __init__(self, models, backend = 'local', flow_datastore_backend = None, remote_storage_root = None, force_pull = False, skip_push_check = False, debug = False):
82
+ def __init__(self, models, backend = 'local', flow_datastore_backend = None, remote_storage_root = None, force_pull = False, cache_update_policy = 'auto', force_cache_update = False, debug = False, circuit_breaker_config = None, timeout_config = None, status_card = None):
25
83
  ...
26
84
  def terminate_models(self, skip_push_check = None):
27
85
  """
@@ -0,0 +1,73 @@
1
+ ######################################################################################################
2
+ # Auto-generated Metaflow stub file #
3
+ # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
+ # Generated on 2025-06-02T15:33:51.306842 #
5
+ ######################################################################################################
6
+
7
+ from __future__ import annotations
8
+
9
+ import metaflow
10
+ import typing
11
+ if typing.TYPE_CHECKING:
12
+ import metaflow.mf_extensions.outerbounds.plugins.card_utilities.async_cards
13
+
14
+ from .....plugins.cards.card_modules.components import Markdown as Markdown
15
+ from .....plugins.cards.card_modules.components import Table as Table
16
+ from .....plugins.cards.card_modules.components import VegaChart as VegaChart
17
+ from .....metaflow_current import current as current
18
+ from ..card_utilities.async_cards import CardRefresher as CardRefresher
19
+
20
+ class OllamaStatusCard(metaflow.mf_extensions.outerbounds.plugins.card_utilities.async_cards.CardRefresher, metaclass=type):
21
+ """
22
+ Real-time status card for Ollama system monitoring.
23
+ Shows circuit breaker state, server health, model status, and recent events.
24
+ """
25
+ def __init__(self, refresh_interval = 10):
26
+ ...
27
+ def update_status(self, category, data):
28
+ """
29
+ Thread-safe method to update status data
30
+ """
31
+ ...
32
+ def add_event(self, event_type, message, timestamp = None):
33
+ """
34
+ Add an event to the timeline
35
+ """
36
+ ...
37
+ def get_circuit_breaker_emoji(self, state):
38
+ """
39
+ Get status emoji for circuit breaker state
40
+ """
41
+ ...
42
+ def get_uptime_string(self, start_time):
43
+ """
44
+ Calculate uptime string
45
+ """
46
+ ...
47
+ def on_startup(self, current_card):
48
+ """
49
+ Initialize the card when monitoring starts
50
+ """
51
+ ...
52
+ def render_card_fresh(self, current_card, data):
53
+ """
54
+ Render the complete card with all status information
55
+ """
56
+ ...
57
+ def on_error(self, current_card, error_message):
58
+ """
59
+ Handle errors in card rendering
60
+ """
61
+ ...
62
+ def on_update(self, current_card, data_object):
63
+ """
64
+ Update the card with new data
65
+ """
66
+ ...
67
+ def sqlite_fetch_func(self, conn):
68
+ """
69
+ Required by CardRefresher (which needs a refactor), but we use in-memory data instead
70
+ """
71
+ ...
72
+ ...
73
+
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:58.919843 #
4
+ # Generated on 2025-06-02T15:33:51.302787 #
5
5
  ######################################################################################################
6
6
 
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:58.919886 #
4
+ # Generated on 2025-06-02T15:33:51.302831 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:58.924205 #
4
+ # Generated on 2025-06-02T15:33:51.308121 #
5
5
  ######################################################################################################
6
6
 
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:58.924252 #
4
+ # Generated on 2025-06-02T15:33:51.308175 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:58.947259 #
4
+ # Generated on 2025-06-02T15:33:51.331845 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:58.908488 #
4
+ # Generated on 2025-06-02T15:33:51.291352 #
5
5
  ######################################################################################################
6
6
 
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:58.908534 #
4
+ # Generated on 2025-06-02T15:33:51.291396 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:58.900307 #
4
+ # Generated on 2025-06-02T15:33:51.283171 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:58.901489 #
4
+ # Generated on 2025-06-02T15:33:51.284332 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,9 +9,9 @@ from __future__ import annotations
9
9
  import metaflow
10
10
  import typing
11
11
  if typing.TYPE_CHECKING:
12
- import metaflow._vendor.click.types
13
- import metaflow.parameters
14
12
  import typing
13
+ import metaflow.parameters
14
+ import metaflow._vendor.click.types
15
15
 
16
16
  from ._vendor import click as click
17
17
  from .exception import ParameterFieldFailed as ParameterFieldFailed
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:58.907994 #
4
+ # Generated on 2025-06-02T15:33:51.290831 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,25 +9,25 @@ from __future__ import annotations
9
9
 
10
10
  from . import test_unbounded_foreach_decorator as test_unbounded_foreach_decorator
11
11
  from .test_unbounded_foreach_decorator import InternalTestUnboundedForeachInput as InternalTestUnboundedForeachInput
12
- from . import resources_decorator as resources_decorator
13
- from . import frameworks as frameworks
14
- from . import parallel_decorator as parallel_decorator
15
- from . import timeout_decorator as timeout_decorator
16
12
  from . import datatools as datatools
17
13
  from . import gcp as gcp
18
14
  from . import storage_executor as storage_executor
19
15
  from . import cards as cards
16
+ from . import pypi as pypi
17
+ from . import catch_decorator as catch_decorator
18
+ from . import parallel_decorator as parallel_decorator
20
19
  from . import retry_decorator as retry_decorator
20
+ from . import timeout_decorator as timeout_decorator
21
+ from . import argo as argo
21
22
  from . import environment_decorator as environment_decorator
22
23
  from . import secrets as secrets
23
- from . import pypi as pypi
24
- from . import catch_decorator as catch_decorator
25
24
  from . import kubernetes as kubernetes
25
+ from . import resources_decorator as resources_decorator
26
26
  from . import aws as aws
27
- from . import argo as argo
27
+ from . import frameworks as frameworks
28
+ from . import airflow as airflow
28
29
  from . import events_decorator as events_decorator
29
30
  from . import project_decorator as project_decorator
30
- from . import airflow as airflow
31
31
  from . import uv as uv
32
32
  from . import azure as azure
33
33
  from . import debug_logger as debug_logger
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:58.940071 #
4
+ # Generated on 2025-06-02T15:33:51.323782 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:58.965437 #
4
+ # Generated on 2025-06-02T15:33:51.351450 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:58.964566 #
4
+ # Generated on 2025-06-02T15:33:51.350554 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:58.965672 #
4
+ # Generated on 2025-06-02T15:33:51.351698 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:59.011166 #
4
+ # Generated on 2025-06-02T15:33:51.398207 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:59.011435 #
4
+ # Generated on 2025-06-02T15:33:51.398468 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:59.011655 #
4
+ # Generated on 2025-06-02T15:33:51.398685 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:58.939041 #
4
+ # Generated on 2025-06-02T15:33:51.322239 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:59.023571 #
4
+ # Generated on 2025-06-02T15:33:51.401674 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:58.963699 #
4
+ # Generated on 2025-06-02T15:33:51.345414 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:59.027604 #
4
+ # Generated on 2025-06-02T15:33:51.405662 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,8 +9,8 @@ from __future__ import annotations
9
9
  import metaflow
10
10
  import typing
11
11
  if typing.TYPE_CHECKING:
12
- import metaflow.graph
13
12
  import metaflow.exception
13
+ import metaflow.graph
14
14
 
15
15
  from ...parameters import JSONType as JSONType
16
16
  from ...metaflow_current import current as current
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:58.964000 #
4
+ # Generated on 2025-06-02T15:33:51.345693 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,9 +9,9 @@ from __future__ import annotations
9
9
  import metaflow
10
10
  import typing
11
11
  if typing.TYPE_CHECKING:
12
- import metaflow.graph
13
- import metaflow.decorators
14
12
  import metaflow.flowspec
13
+ import metaflow.decorators
14
+ import metaflow.graph
15
15
 
16
16
  from ...metaflow_current import current as current
17
17
  from ...events import Trigger as Trigger
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:58.964384 #
4
+ # Generated on 2025-06-02T15:33:51.346103 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:59.010810 #
4
+ # Generated on 2025-06-02T15:33:51.394324 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,8 +9,8 @@ from __future__ import annotations
9
9
  import metaflow
10
10
  import typing
11
11
  if typing.TYPE_CHECKING:
12
- import metaflow.plugins.argo.argo_workflows_deployer_objects
13
12
  import metaflow.runner.deployer
13
+ import metaflow.plugins.argo.argo_workflows_deployer_objects
14
14
 
15
15
  from ...client.core import get_metadata as get_metadata
16
16
  from ...exception import MetaflowException as MetaflowException
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:58.938779 #
4
+ # Generated on 2025-06-02T15:33:51.323368 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:58.963068 #
4
+ # Generated on 2025-06-02T15:33:51.349911 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:58.962412 #
4
+ # Generated on 2025-06-02T15:33:51.349242 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:58.962838 #
4
+ # Generated on 2025-06-02T15:33:51.349658 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:58.980857 #
4
+ # Generated on 2025-06-02T15:33:51.396937 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:58.980181 #
4
+ # Generated on 2025-06-02T15:33:51.396277 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:58.981394 #
4
+ # Generated on 2025-06-02T15:33:51.397515 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:58.963287 #
4
+ # Generated on 2025-06-02T15:33:51.350119 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,18 +1,18 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-29T18:56:58.981684 #
4
+ # Generated on 2025-06-02T15:33:51.397818 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import abc
10
9
  import metaflow
10
+ import abc
11
11
  import typing
12
12
  if typing.TYPE_CHECKING:
13
- import abc
14
- import metaflow.plugins.secrets
15
13
  import metaflow.exception
14
+ import metaflow.plugins.secrets
15
+ import abc
16
16
 
17
17
  from ....exception import MetaflowException as MetaflowException
18
18
  from ...secrets import SecretsProvider as SecretsProvider