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 });
    },
  }));