Merge pull request #26 from WorkflowAI/yann/wor-2997-add-labels-support-to-the-sdk

guillaq · web-flow · commit c0da51caa03c · 2024-12-18T10:01:54.000-05:00
Add label support for the run endpoint
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "workflowai"
-version = "0.5.2"
+version = "0.5.3"
 description = ""
 authors = ["Guillaume Aquilina <guillaume@workflowai.com>"]
 readme = "README.md"
diff --git a/workflowai/core/client/_models.py b/workflowai/core/client/_models.py
@@ -20,6 +20,8 @@ class RunRequest(BaseModel):
 
     metadata: Optional[dict[str, Any]] = None
 
+    labels: Optional[set[str]] = None  # deprecated, to be included in metadata
+
     private_fields: Optional[set[str]] = None
 
     stream: Optional[bool] = None
diff --git a/workflowai/core/client/client.py b/workflowai/core/client/client.py
@@ -95,6 +95,7 @@ async def run(
             stream=stream,
             use_cache=kwargs.get("use_cache"),
             metadata=kwargs.get("metadata"),
+            labels=kwargs.get("labels"),
         )
 
         route = f"/v1/_/tasks/{task.id}/schemas/{task.schema_id}/run"

Original file line number	Diff line number	Diff line change
`@@ -95,6 +95,7 @@ async def run(`
`95`	`95`	`stream=stream,`
`96`	`96`	`use_cache=kwargs.get("use_cache"),`
`97`	`97`	`metadata=kwargs.get("metadata"),`
	`98`	`+ labels=kwargs.get("labels"),`
`98`	`99`	`)`
`99`	`100`
`100`	`101`	`route = f"/v1/_/tasks/{task.id}/schemas/{task.schema_id}/run"`