🤖 Merge PR DefinitelyTyped#74234 [dom-chromium-ai] disallow prefix in initialPrompts by @christianliebel

Author: christianliebel

types/dom-chromium-ai/dom-chromium-ai-tests.ts

@@ -2,10 +2,15 @@ async function topLevel() {
     // Language Model
 
     await LanguageModel.create({
-        // @ts-expect-error - System prompt must be first element of the initialPrompt array.
+        // @ts-expect-error - System prompt must be first element of the initialPrompts array.
         initialPrompts: [{ role: "user", content: "foo" }, { role: "system", content: "foo" }],
     });
 
+    await LanguageModel.create({
+        // @ts-expect-error - Prefixes are not allowed in initialPrompts.
+        initialPrompts: [{ role: "assistant", content: "foo", prefix: true }],
+    });
+
     const languageModel = await LanguageModel.create({
         topK: 1,
         temperature: 0,

types/dom-chromium-ai/index.d.ts

@@ -146,11 +146,17 @@ interface LanguageModelToolFunction {
     (...args: any[]): Promise<string>;
 }
 
-type LanguageModelPrompt = LanguageModelMessage[] | string;
+type LanguageModelPrompt = (LanguageModelMessage | LanguageModelAssistantMessage)[] | string;
 
 interface LanguageModelMessage {
     role: LanguageModelMessageRole;
     content: LanguageModelMessageContent[] | string;
+}
+
+// Not in IDL, split up here for allowing assistant messages with prefix in prompt() and promptStreaming() only
+interface LanguageModelAssistantMessage {
+    role: LanguageModelAssistantMessageRole;
+    content: LanguageModelMessageContent[] | string;
     prefix?: boolean;
 }
 
@@ -165,7 +171,9 @@ interface LanguageModelMessageContent {
     value: LanguageModelMessageValue;
 }
 
-type LanguageModelMessageRole = "user" | "assistant";
+type LanguageModelMessageRole = "user" | LanguageModelAssistantMessageRole;
+// Not in IDL, split up here for allowing assistant messages with prefix in prompt() and promptStreaming() only
+type LanguageModelAssistantMessageRole = "assistant";
 // Not in IDL, split up here for enforcing the system message as the first element
 type LanguageModelSystemMessageRole = "system";