[JS] fix: SequenceAugmentation + Citations (#1650)

## Linked issues #minor ensure response message is copied and mapped properly with citations during sequence augmentation.
microsoft · May 15, 2024 · 58fdeab · 58fdeab
1 parent c976453
commit 58fdeab
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 2 deletions.
diff --git a/js/packages/teams-ai/src/augmentations/SequenceAugmentation.spec.ts b/js/packages/teams-ai/src/augmentations/SequenceAugmentation.spec.ts
@@ -238,7 +238,19 @@ describe('SequenceAugmentation', () => {
                 // Create plan from response
                 const plan = await augmentation.createPlanFromResponse(context, state, {
                     status: 'success',
-                    message: { role: 'assistant', content: validation.value }
+                    message: {
+                        role: 'assistant',
+                        content: validation.value,
+                        context: {
+                            intent: 'to test citations',
+                            citations: [{
+                                url: '',
+                                title: '',
+                                filepath: '',
+                                content: 'my citation content...'
+                            }]
+                        }
+                    }
                 });
                 assert.notEqual(plan, undefined);
                 assert.deepEqual(plan, {
@@ -255,7 +267,16 @@ describe('SequenceAugmentation', () => {
                             type: 'SAY',
                             response: {
                                 role: 'assistant',
-                                content: 'hello world'
+                                content: 'hello world',
+                                context: {
+                                    intent: 'to test citations',
+                                    citations: [{
+                                        url: '',
+                                        title: '',
+                                        filepath: '',
+                                        content: 'my citation content...'
+                                    }]
+                                }
                             }
                         }
                     ]

diff --git a/js/packages/teams-ai/src/augmentations/SequenceAugmentation.ts b/js/packages/teams-ai/src/augmentations/SequenceAugmentation.ts
@@ -190,6 +190,7 @@ export class SequenceAugmentation implements Augmentation<Plan | undefined> {
                 return {
                     ...c,
                     response: {
+                        ...response.message,
                         role: 'assistant',
                         content: (c as PredictedSayCommand).response as unknown as string
                     }