fix: add telemetry for office agent for unit test - bug fix 7

OfficeDev · Jul 1, 2024 · 59afe81 · 59afe81
1 parent 1b7b992
commit 59afe81
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 8 deletions.
diff --git a/packages/vscode-extension/test/officeChat/common/skills/codeExplainer.test.ts b/packages/vscode-extension/test/officeChat/common/skills/codeExplainer.test.ts
@@ -37,10 +37,13 @@ describe("CodeExplainer", () => {
           relatedSampleName: ["sample1", "sample2"],
           timeToFirstToken: 0,
           chatMessages: [
-            new LanguageModelChatMessage(LanguageModelChatMessageRole.User, "message1"),
-            new LanguageModelChatMessage(LanguageModelChatMessageRole.User, "message2"),
+            new LanguageModelChatMessage(LanguageModelChatMessageRole.User, "requestMessage1"),
+            new LanguageModelChatMessage(LanguageModelChatMessageRole.User, "requestMessage2"),
+          ],
+          responseChatMessages: [
+            new LanguageModelChatMessage(LanguageModelChatMessageRole.User, "responseMessage1"),
+            new LanguageModelChatMessage(LanguageModelChatMessageRole.User, "responseMessage2"),
           ],
-          responseChatMessages: [],
           properties: { property1: "value1", property2: "value2" },
           measurements: { measurement1: 1, measurement2: 2 },
         },
@@ -106,10 +109,13 @@ describe("CodeExplainer", () => {
         relatedSampleName: ["sample1", "sample2"],
         timeToFirstToken: 0,
         chatMessages: [
-          new LanguageModelChatMessage(LanguageModelChatMessageRole.User, "message1"),
-          new LanguageModelChatMessage(LanguageModelChatMessageRole.User, "message2"),
+          new LanguageModelChatMessage(LanguageModelChatMessageRole.User, "requestMessage1"),
+          new LanguageModelChatMessage(LanguageModelChatMessageRole.User, "requestMessage2"),
+        ],
+        responseChatMessages: [
+          new LanguageModelChatMessage(LanguageModelChatMessageRole.User, "responseMessage1"),
+          new LanguageModelChatMessage(LanguageModelChatMessageRole.User, "responseMessage2"),
         ],
-        responseChatMessages: [],
         properties: {
           property1: "value1",
           property2: "value2",

diff --git a/packages/vscode-extension/test/officeChat/common/skills/projectCreator.test.ts b/packages/vscode-extension/test/officeChat/common/skills/projectCreator.test.ts
@@ -109,8 +109,8 @@ describe("projectCreator", () => {
         relatedSampleName: ["sample1", "sample2"],
         timeToFirstToken: 0,
         chatMessages: [
-          new LanguageModelChatMessage(LanguageModelChatMessageRole.User, "message1"),
-          new LanguageModelChatMessage(LanguageModelChatMessageRole.User, "message2"),
+          new LanguageModelChatMessage(LanguageModelChatMessageRole.User, "requestMessage1"),
+          new LanguageModelChatMessage(LanguageModelChatMessageRole.User, "requestMessage2"),
         ],
         responseChatMessages: [
           new LanguageModelChatMessage(LanguageModelChatMessageRole.User, "responseMessage1"),