feat: ✨ add system message handling to promptdom

microsoft · Oct 18, 2024 · a731dfe · a731dfe
1 parent 9104f0d
commit a731dfe
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/packages/core/src/promptdom.ts b/packages/core/src/promptdom.ts
@@ -23,6 +23,7 @@ import { dedent } from "./indent"
 import {
     ChatCompletionAssistantMessageParam,
     ChatCompletionMessageParam,
+    ChatCompletionSystemMessageParam,
 } from "./chattypes"
 import { resolveTokenEncoder } from "./encoders"
 import { expandFiles } from "./fs"
@@ -1078,10 +1079,15 @@ ${fods.map((fo) => `   ${fo.pattern}: ${fo.description}`)}
         toChatCompletionUserMessage(userPrompt, images),
     ]
     if (assistantPrompt)
-        messages.push(<ChatCompletionAssistantMessageParam>{
+        messages.push({
             role: "assistant",
             content: assistantPrompt,
-        })
+        } as ChatCompletionAssistantMessageParam)
+    if (systemPrompt)
+        messages.unshift({
+            role: "system",
+            content: systemPrompt,
+        } as ChatCompletionSystemMessageParam)
     const res = Object.freeze<PromptNodeRender>({
         userPrompt,
         assistantPrompt,

diff --git a/packages/sample/genaisrc/import-prompty.genai.mjs b/packages/sample/genaisrc/import-prompty.genai.mjs
@@ -5,6 +5,14 @@ script({
     },
 })
 
+const res = await runPrompt((ctx) => {
+    ctx.importTemplate("src/basic.prompty", {
+        question: "what is the capital of france?",
+        hint: "starts with p",
+    })
+})
+console.log(`inline: ${res.text}`)
+
 importTemplate("src/basic.prompty", {
     question: "what is the capital of france?",
     hint: "starts with p",