Merge pull request #887 from deepmodeling/zjgemi

Add HTTPOPTemplate
deepmodeling · Jan 15, 2025 · 4cecac0 · 4cecac0
2 parents c930088 + 93370fc
commit 4cecac0
Show file tree

Hide file tree

Showing 4 changed files with 1,152 additions and 6 deletions.
diff --git a/src/dflow/__init__.py b/src/dflow/__init__.py
@@ -12,8 +12,8 @@
 from .io import (AutonamedDict, IfExpression, InputArtifact, InputParameter,
                  Inputs, OutputArtifact, OutputParameter, Outputs,
                  if_expression)
-from .op_template import (OPTemplate, PythonScriptOPTemplate, Secret,
-                          ShellOPTemplate)
+from .op_template import (HTTPOPTemplate, OPTemplate, PythonScriptOPTemplate,
+                          Secret, ShellOPTemplate)
 from .resource import Resource
 from .slurm import SlurmJob, SlurmJobTemplate, SlurmRemoteExecutor
 from .step import (HookStep, Step, argo_concat, argo_enumerate, argo_len,
@@ -45,7 +45,7 @@
            "query_archived_workflows", "ContainerExecutor", "ArgoStep",
            "ArgoWorkflow", "argo_enumerate", "path_object_of_artifact",
            "CustomArtifact", "gen_code", "jsonpickle", "HTTPArtifact",
-           "HookStep"]
+           "HookStep", "HTTPOPTemplate"]
 
 
 if os.environ.get("DFLOW_LINEAGE"):

diff --git a/src/dflow/client/__init__.py b/src/dflow/client/__init__.py
@@ -4,9 +4,10 @@
 from .v1alpha1_parameter import V1alpha1Parameter
 from .v1alpha1_retry_strategy import V1alpha1RetryStrategy
 from .v1alpha1_sequence import V1alpha1Sequence
+from .v1alpha1_template import V1alpha1Template
 from .v1alpha1_value_from import V1alpha1ValueFrom
 from .v1alpha1_workflow_step import V1alpha1WorkflowStep
 
 __all__ = ["V1alpha1Artifact", "V1alpha1LifecycleHook", "V1alpha1Parameter",
            "V1alpha1RetryStrategy", "V1alpha1Sequence", "V1alpha1ValueFrom",
-           "V1alpha1WorkflowStep", "V1alpha1DAGTask"]
+           "V1alpha1WorkflowStep", "V1alpha1DAGTask", "V1alpha1Template"]