Add actionId to PredictedDoCommand interface

microsoft · Aug 5, 2024 · 52d9c47 · 52d9c47
1 parent f3344c4
commit 52d9c47
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/js/packages/teams-ai/src/AI.ts b/js/packages/teams-ai/src/AI.ts
@@ -409,7 +409,7 @@ export class AI<TState extends TurnState = TurnState> {
                 const cmd = plan.commands[i];
                 switch (cmd.type) {
                     case 'DO': {
-                        const { action } = cmd as PredictedDoCommand;
+                        const { action, actionId } = cmd as PredictedDoCommand;
                         if (this._actions.has(action)) {
                             // Call action handler
                             const handler = this._actions.get(action)!.handler;
@@ -418,6 +418,9 @@ export class AI<TState extends TurnState = TurnState> {
                                 .handler(context, state, { handler, ...(cmd as PredictedDoCommand) }, action);
                             should_loop = output.length > 0;
                             state.temp.actionOutputs[action] = output;
+                            if (actionId) {
+                                should_loop = true;
+                            }
                         } else {
                             // Redirect to UnknownAction handler
                             output = await this._actions

diff --git a/js/packages/teams-ai/src/planners/Planner.ts b/js/packages/teams-ai/src/planners/Planner.ts
@@ -94,6 +94,12 @@ export interface PredictedDoCommand extends PredictedCommand {
      * Any parameters that the AI system should use to perform the action.
      */
     parameters: Record<string, any>;
+
+    /**
+     * The id mapped to the name action that the AI system should perform.
+     * In OpenAI, this is associated with the action_id tool calls.
+     */
+    actionId?: string;
 }
 
 /**