38 lines
864 B
TypeScript
38 lines
864 B
TypeScript
import fs from "node:fs/promises";
|
|
import os from "node:os";
|
|
import type { ModelMessage } from "ai";
|
|
|
|
const home = os.homedir();
|
|
const convoPath = `${home}/.lmc_convo.json`;
|
|
|
|
export interface IConversation {
|
|
model: string;
|
|
messages: ModelMessage[];
|
|
}
|
|
|
|
export class Conversation implements IConversation {
|
|
messages: ModelMessage[] = [];
|
|
|
|
constructor(public model: string) {}
|
|
|
|
async load() {
|
|
try {
|
|
const text = await fs.readFile(convoPath, "utf8");
|
|
const convo: IConversation = JSON.parse(text);
|
|
this.model = convo.model;
|
|
this.messages = convo.messages;
|
|
} catch {
|
|
console.error("No conversation found, starting one.");
|
|
}
|
|
}
|
|
|
|
async save() {
|
|
await fs.writeFile(
|
|
convoPath,
|
|
JSON.stringify({
|
|
model: this.model,
|
|
messages: this.messages,
|
|
} satisfies IConversation),
|
|
);
|
|
}
|
|
}
|