fix: simplify model creation

upstash · Jan 14, 2025 · d933232 · d933232
1 parent 0e1fc16
commit d933232
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/agents/index.ts b/src/agents/index.ts
@@ -35,7 +35,8 @@ export class WorkflowAgents {
     return new Task({ context: this.context, taskParameters });
   }
 
-  public getOpenai() {
-    return createWorkflowOpenAI(this.context);
+  public openai(...params: Parameters<ReturnType<typeof createWorkflowOpenAI>>) {
+    const openai = createWorkflowOpenAI(this.context);
+    return openai(...params);
   }
 }