updated types

nalbion · Mar 12, 2024 · a24921e · a24921e
1 parent 51989ec
commit a24921e
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/src/llm/message.ts b/src/llm/message.ts
@@ -1,10 +1,17 @@
 // as per 'openai/resources/index.mjs'
 export type ChatCompletionRole = 'system' | 'user' | 'assistant' | 'tool' | 'function';
 
-export type LlmMessage = {
-  role: ChatCompletionRole;
-  content: string;
-};
+export type LlmMessage =
+  | {
+      role: ChatCompletionRole;
+      content: string;
+    }
+  | {
+      role: 'assistant';
+      content?: string;
+      name?: string;
+      tool_calls: ChatCompletionMessageToolCall[];
+    };
 
 export type LlmRequestMessage =
   | {