Merge pull request #53 from microsoft/joh/mammoth-meadowlark

add `FunctionMessage` and `ChatRole.Function`

package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@vscode/prompt-tsx",
-	"version": "0.1.12-alpha",
+	"version": "0.1.13-alpha",
 	"description": "Declare LLM prompts with TSX",
 	"main": "./dist/base/index.js",
 	"types": "./dist/base/index.d.ts",

src/base/openai.ts

@@ -28,7 +28,8 @@ export declare interface ChatMessage {
 export enum ChatRole {
 	System = 'system',
 	User = 'user',
-	Assistant = 'assistant'
+	Assistant = 'assistant',
+	Function = 'function'
 }
 
 /**

src/base/promptElements.tsx

@@ -68,6 +68,18 @@ export class AssistantMessage extends BaseChatMessage {
 	}
 }
 
+/**
+ * A {@link PromptElement} which can be rendered to an OpenAI function chat message.
+ *
+ * See {@link https://platform.openai.com/docs/api-reference/chat/create#chat-create-messages}
+ */
+export class FunctionMessage extends BaseChatMessage {
+	constructor(props: ChatMessageProps & { name: string }) {
+		props.role = ChatRole.Function;
+		super(props);
+	}
+}
+
 /**
  * A chunk of single-line or multi-line text that is a direct child of a {@link ChatMessagePromptElement}.
  *