feat: Add max-value of 100 to parallelism

Signed-off-by: Diwank Singh Tomer <[email protected]>
julep-ai · Sep 4, 2024 · c54f172 · c54f172
1 parent 50c8e66
commit c54f172
Show file tree

Hide file tree

Showing 6 changed files with 7 additions and 2 deletions.
diff --git a/agents-api/agents_api/autogen/Tasks.py b/agents-api/agents_api/autogen/Tasks.py
@@ -350,7 +350,7 @@ class Main(BaseModel):
     """
     The initial value of the reduce expression
     """
-    parallelism: Annotated[int | None, Field(None, ge=1)]
+    parallelism: Annotated[int | None, Field(None, ge=1, le=100)]
     """
     Whether to run the reduce expression in parallel and how many items to run in each batch
     """
@@ -391,7 +391,7 @@ class MainModel(BaseModel):
     """
     The initial value of the reduce expression
     """
-    parallelism: Annotated[int | None, Field(None, ge=1)]
+    parallelism: Annotated[int | None, Field(None, ge=1, le=100)]
     """
     Whether to run the reduce expression in parallel and how many items to run in each batch
     """

diff --git a/agents-api/tests/test_execution_workflow.py b/agents-api/tests/test_execution_workflow.py
@@ -700,6 +700,7 @@ async def _(
 
 
 for p in [1, 3, 5]:
+
     @test(f"workflow: map reduce step parallel (parallelism={p})")
     async def _(
         client=cozo_client,

diff --git a/sdks/ts/src/api/schemas/$Tasks_CreateTaskRequest.ts b/sdks/ts/src/api/schemas/$Tasks_CreateTaskRequest.ts
@@ -139,6 +139,7 @@ export const $Tasks_CreateTaskRequest = {
                   type: "number",
                   description: `Whether to run the reduce expression in parallel and how many items to run in each batch`,
                   format: "uint16",
+                  maximum: 100,
                   minimum: 1,
                 },
               },

diff --git a/sdks/ts/src/api/schemas/$Tasks_PatchTaskRequest.ts b/sdks/ts/src/api/schemas/$Tasks_PatchTaskRequest.ts
@@ -133,6 +133,7 @@ export const $Tasks_PatchTaskRequest = {
                   type: "number",
                   description: `Whether to run the reduce expression in parallel and how many items to run in each batch`,
                   format: "uint16",
+                  maximum: 100,
                   minimum: 1,
                 },
               },

diff --git a/sdks/ts/src/api/schemas/$Tasks_Task.ts b/sdks/ts/src/api/schemas/$Tasks_Task.ts
@@ -139,6 +139,7 @@ export const $Tasks_Task = {
                   type: "number",
                   description: `Whether to run the reduce expression in parallel and how many items to run in each batch`,
                   format: "uint16",
+                  maximum: 100,
                   minimum: 1,
                 },
               },

diff --git a/sdks/ts/src/api/schemas/$Tasks_UpdateTaskRequest.ts b/sdks/ts/src/api/schemas/$Tasks_UpdateTaskRequest.ts
@@ -139,6 +139,7 @@ export const $Tasks_UpdateTaskRequest = {
                   type: "number",
                   description: `Whether to run the reduce expression in parallel and how many items to run in each batch`,
                   format: "uint16",
+                  maximum: 100,
                   minimum: 1,
                 },
               },