wip

packages/schema/src/models/O1Message.ts

@@ -0,0 +1,20 @@
+import { types } from 'mobx-state-tree';
+
+export default types
+  .model('O1Message', {
+    role: types.enumeration(['user', 'assistant', 'system']),
+    content: types.array(
+      types.model({
+        type: types.string,
+        text: types.string,
+      }),
+    ),
+  })
+  .actions(self => ({
+    setContent(newContent: string, contentType: string = 'text') {
+      self.content = [{ type: contentType, text: newContent }];
+    },
+    append(newContent: string, contentType: string = 'text') {
+      self.content.push({ type: contentType, text: newContent });
+    },
+  }));