Update types.ts

babaohuang · Dec 14, 2023 · b3f70a5 · b3f70a5
1 parent 318bfea
commit b3f70a5
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/src/types.ts b/src/types.ts
@@ -1,9 +1,13 @@
+export interface ChatMessagePart {
+  text: string;
+}
+
 export interface ChatMessage {
-  role: 'system' | 'user' | 'assistant'
-  content: string
+  role: 'model' | 'user';
+  parts: ChatMessagePart[];
 }
 
 export interface ErrorMessage {
-  code: string
-  message: string
-}
+  code: string;
+  message: string;
+}