Merge pull request #27 from WorkflowAI/update-RunParams

guillaq · web-flow · commit 5513272c6cda · 2024-12-20T15:18:52.000-05:00
Update run params
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "workflowai"
-version = "0.5.3"
+version = "0.5.4"
 description = ""
 authors = ["Guillaume Aquilina <guillaume@workflowai.com>"]
 readme = "README.md"
diff --git a/workflowai/core/client/_types.py b/workflowai/core/client/_types.py
@@ -29,6 +29,7 @@ class RunParams(TypedDict, Generic[TaskOutput]):
     version: NotRequired[Optional[VersionReference]]
     use_cache: NotRequired[CacheUsage]
     metadata: NotRequired[Optional[dict[str, Any]]]
+    labels: NotRequired[Optional[set[str]]]
     max_retry_delay: NotRequired[float]
     max_retry_count: NotRequired[float]
     validator: NotRequired[OutputValidator[TaskOutput]]
