fix(types): add InteropMessageRole (#75)

copilot-extensions · Sep 10, 2024 · c593ba6 · c593ba6
1 parent 9c9fb26
commit c593ba6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/index.d.ts b/index.d.ts
@@ -100,7 +100,7 @@ export interface OpenAICompatibilityPayload {
 }
 
 export interface CopilotMessage {
-  role: string;
+  role: MessageRole;
   content: string;
   copilot_references?: MessageCopilotReference[];
   copilot_confirmations?: MessageCopilotConfirmation[];
@@ -116,7 +116,8 @@ export interface CopilotMessage {
   [key: string]: unknown;
 }
 
-export interface InteropMessage<TRole extends string = string> {
+export type MessageRole = 'system' | 'user' | 'assistant';
+export interface InteropMessage<TRole extends MessageRole = MessageRole> {
   role: TRole;
   content: string;
   name?: string;

diff --git a/index.test-d.ts b/index.test-d.ts
@@ -168,7 +168,7 @@ export function transformPayloadForOpenAICompatibilityTest(
   expectType<{
     messages: {
       content: string;
-      role: string;
+      role: 'system' | 'user' | 'assistant';
       name?: string;
       [key: string]: unknown;
     }[];