VSCODE-640: Adapt message content access to latest vscode API

.vscode/launch.json

@@ -40,7 +40,21 @@
       "outFiles": [
         "${workspaceFolder}/dist/**/*.js"
       ]
-    }
+    },
+    {
+			"name": "Run Tests",
+			"type": "extensionHost",
+			"request": "launch",
+			"runtimeExecutable": "${execPath}",
+			"args": [
+        "${workspaceFolder}/out/test/suite", // TODO: remove suite
+				"--disable-extensions",
+				"--extensionDevelopmentPath=${workspaceFolder}",
+				"--extensionTestsPath=${workspaceFolder}/out/test/suite"
+			],
+			"outFiles": ["${workspaceFolder}/out/**/*.js"],
+			"preLaunchTask": "npm: compile:extension",
+		}
   ],
   "compounds": [
     {

src/participant/participant.ts

@@ -189,7 +189,7 @@ export default class ParticipantController {
         (message: vscode.LanguageModelChatMessage) =>
           util.inspect({
             role: message.role,
-            contentLength: message.content.length,
+            contentLength: Prompts.getContentLength(message),
           })
       ),
     });
@@ -790,15 +790,18 @@ export default class ParticipantController {
       // it currently errors (not on insiders, only main VSCode).
       // Here we're defaulting to have some content as a workaround.
       // TODO: Remove this when the issue is fixed.
-      messagesWithNamespace.messages[
-        messagesWithNamespace.messages.length - 1
-        // eslint-disable-next-line new-cap
-      ] = vscode.LanguageModelChatMessage.User(
+      if (
+        !Prompts.doMessagesContainUserInput([
+          messagesWithNamespace.messages[
+            messagesWithNamespace.messages.length - 1
+          ],
+        ])
+      ) {
         messagesWithNamespace.messages[
           messagesWithNamespace.messages.length - 1
-        ].content.trim() || 'see previous messages'
-      );
-
+          // eslint-disable-next-line new-cap
+        ] = vscode.LanguageModelChatMessage.User('see previous messages');
+      }
       const responseContentWithNamespace = await this.getChatResponseContent({
         modelInput: messagesWithNamespace,
         token,

src/participant/prompts/index.ts

@@ -5,6 +5,7 @@ import { IntentPrompt } from './intent';
 import { NamespacePrompt } from './namespace';
 import { QueryPrompt } from './query';
 import { SchemaPrompt } from './schema';
+import { isContentEmpty, getContentLength } from './promptBase';
 
 export class Prompts {
   public static generic = new GenericPrompt();
@@ -26,12 +27,18 @@ export class Prompts {
     for (const message of messages) {
       if (
         message.role === vscode.LanguageModelChatMessageRole.User &&
-        message.content.trim().length > 0
+        !isContentEmpty(message)
       ) {
         return true;
       }
     }
 
     return false;
   }
+
+  public static getContentLength(
+    message: vscode.LanguageModelChatMessage
+  ): number {
+    return getContentLength(message);
+  }
 }

src/participant/prompts/promptBase.ts

@@ -24,6 +24,52 @@ export interface ModelInput {
   stats: ParticipantPromptProperties;
 }
 
+export function getContentLength(
+  message: vscode.LanguageModelChatMessage
+): number {
+  const content = message.content as any;
+  if (typeof content === 'string') {
+    return content.trim().length;
+  }
+
+  // TODO: https://github.com/microsoft/vscode/pull/231788 made it so message.content is no longer a string,
+  // but an array of things that a message can contain. This will eventually be reflected in the type definitions
+  // but until then, we're manually checking the array contents to ensure we don't break when this PR gets released
+  // in the stable channel.
+  if (Array.isArray(content)) {
+    return content.reduce((acc: number, element) => {
+      const value = element?.value ?? element?.content?.value;
+      if (typeof value === 'string') {
+        return acc + value.length;
+      }
+
+      return acc;
+    }, 0);
+  }
+
+  return 0;
+}
+
+export function isContentEmpty(
+  message: vscode.LanguageModelChatMessage
+): boolean {
+  const content = message.content as any;
+  if (typeof content === 'string') {
+    return content.trim().length === 0;
+  }
+
+  if (Array.isArray(content)) {
+    for (const element of content) {
+      const value = element?.value ?? element?.content?.value;
+      if (typeof value === 'string' && value.trim().length > 0) {
+        return false;
+      }
+    }
+  }
+
+  return true;
+}
+
 export abstract class PromptBase<TArgs extends PromptArgsBase> {
   protected abstract getAssistantPrompt(args: TArgs): string;
 
@@ -92,7 +138,7 @@ export abstract class PromptBase<TArgs extends PromptArgsBase> {
   ): ParticipantPromptProperties {
     return {
       total_message_length: messages.reduce(
-        (acc, message) => acc + message.content.length,
+        (acc, message) => acc + getContentLength(message),
         0
       ),
       user_input_length: request.prompt.length,