feat(agents-api): Add parallel flag to map reduce

Signed-off-by: Diwank Singh Tomer <diwank.singh@gmail.com>
julep-ai · Sep 4, 2024 · 1860574 · 1860574
1 parent 95e0df6
commit 1860574
Show file tree

Hide file tree

Showing 10 changed files with 66 additions and 0 deletions.
diff --git a/agents-api/agents_api/autogen/Tasks.py b/agents-api/agents_api/autogen/Tasks.py
@@ -347,6 +347,13 @@ class Main(BaseModel):
     A special parameter named `results` is the accumulator and `_` is the current value.
     """
     initial: Any = []
+    """
+    The initial value of the reduce expression
+    """
+    parallel: StrictBool = False
+    """
+    Whether to run the reduce expression in parallel
+    """
 
 
 class MainModel(BaseModel):
@@ -381,6 +388,13 @@ class MainModel(BaseModel):
     A special parameter named `results` is the accumulator and `_` is the current value.
     """
     initial: Any = []
+    """
+    The initial value of the reduce expression
+    """
+    parallel: StrictBool = False
+    """
+    Whether to run the reduce expression in parallel
+    """
 
 
 class ParallelStep(BaseModel):

diff --git a/sdks/ts/src/api/models/Tasks_CreateTaskRequest.ts b/sdks/ts/src/api/models/Tasks_CreateTaskRequest.ts
@@ -72,7 +72,14 @@ export type Tasks_CreateTaskRequest = Record<
          * A special parameter named `results` is the accumulator and `_` is the current value.
          */
         reduce?: Common_PyExpression;
+        /**
+         * The initial value of the reduce expression
+         */
         initial?: any;
+        /**
+         * Whether to run the reduce expression in parallel
+         */
+        parallel?: boolean;
       })
   >
 >;
diff --git a/sdks/ts/src/api/models/Tasks_PatchTaskRequest.ts b/sdks/ts/src/api/models/Tasks_PatchTaskRequest.ts
@@ -71,7 +71,14 @@ export type Tasks_PatchTaskRequest = Record<
          * A special parameter named `results` is the accumulator and `_` is the current value.
          */
         reduce?: Common_PyExpression;
+        /**
+         * The initial value of the reduce expression
+         */
         initial?: any;
+        /**
+         * Whether to run the reduce expression in parallel
+         */
+        parallel?: boolean;
       })
   >
 >;
diff --git a/sdks/ts/src/api/models/Tasks_Task.ts b/sdks/ts/src/api/models/Tasks_Task.ts
@@ -72,7 +72,14 @@ export type Tasks_Task = Record<
          * A special parameter named `results` is the accumulator and `_` is the current value.
          */
         reduce?: Common_PyExpression;
+        /**
+         * The initial value of the reduce expression
+         */
         initial?: any;
+        /**
+         * Whether to run the reduce expression in parallel
+         */
+        parallel?: boolean;
       })
   >
 >;
diff --git a/sdks/ts/src/api/models/Tasks_UpdateTaskRequest.ts b/sdks/ts/src/api/models/Tasks_UpdateTaskRequest.ts
@@ -72,7 +72,14 @@ export type Tasks_UpdateTaskRequest = Record<
          * A special parameter named `results` is the accumulator and `_` is the current value.
          */
         reduce?: Common_PyExpression;
+        /**
+         * The initial value of the reduce expression
+         */
         initial?: any;
+        /**
+         * Whether to run the reduce expression in parallel
+         */
+        parallel?: boolean;
       })
   >
 >;
diff --git a/sdks/ts/src/api/schemas/$Tasks_CreateTaskRequest.ts b/sdks/ts/src/api/schemas/$Tasks_CreateTaskRequest.ts
@@ -132,8 +132,13 @@ export const $Tasks_CreateTaskRequest = {
                   ],
                 },
                 initial: {
+                  description: `The initial value of the reduce expression`,
                   properties: {},
                 },
+                parallel: {
+                  type: "boolean",
+                  description: `Whether to run the reduce expression in parallel`,
+                },
               },
             },
           ],

diff --git a/sdks/ts/src/api/schemas/$Tasks_PatchTaskRequest.ts b/sdks/ts/src/api/schemas/$Tasks_PatchTaskRequest.ts
@@ -126,8 +126,13 @@ export const $Tasks_PatchTaskRequest = {
                   ],
                 },
                 initial: {
+                  description: `The initial value of the reduce expression`,
                   properties: {},
                 },
+                parallel: {
+                  type: "boolean",
+                  description: `Whether to run the reduce expression in parallel`,
+                },
               },
             },
           ],

diff --git a/sdks/ts/src/api/schemas/$Tasks_Task.ts b/sdks/ts/src/api/schemas/$Tasks_Task.ts
@@ -132,8 +132,13 @@ export const $Tasks_Task = {
                   ],
                 },
                 initial: {
+                  description: `The initial value of the reduce expression`,
                   properties: {},
                 },
+                parallel: {
+                  type: "boolean",
+                  description: `Whether to run the reduce expression in parallel`,
+                },
               },
             },
           ],

diff --git a/sdks/ts/src/api/schemas/$Tasks_UpdateTaskRequest.ts b/sdks/ts/src/api/schemas/$Tasks_UpdateTaskRequest.ts
@@ -132,8 +132,13 @@ export const $Tasks_UpdateTaskRequest = {
                   ],
                 },
                 initial: {
+                  description: `The initial value of the reduce expression`,
                   properties: {},
                 },
+                parallel: {
+                  type: "boolean",
+                  description: `Whether to run the reduce expression in parallel`,
+                },
               },
             },
           ],

diff --git a/typespec/tasks/steps.tsp b/typespec/tasks/steps.tsp
@@ -247,7 +247,11 @@ model MapReduceStep<Value = unknown, Accumulator = Array<Value>> extends BaseWor
      * A special parameter named `results` is the accumulator and `_` is the current value. */
     reduce?: ReduceExpression<Value, Accumulator>;
 
+    /** The initial value of the reduce expression */
     initial?: unknown = #[];
+
+    /** Whether to run the reduce expression in parallel */
+    parallel?: boolean = false;
 }
 
 /////////////////////////