Add saving

Stef Dunlap 2023-03-08 22:37:05 -05:00
parent 112e960135
commit 90d457c790
2 changed files with 18 additions and 5 deletions

1
.gitignore vendored
View File

@ -1,2 +1,3 @@
config.json
node_modules
messages.json

View File

@ -3,13 +3,19 @@
const irc = require('irc');
const axios = require('axios');
const config = require('./config.json');
const fs = require('fs');
const seed_messages = [{role: 'system', content: 'You are a dungeon master, DMing a game for your friends. The setting is Left World, a dreamscape where not everything is as it seems. Let user introduce their characters and give them scenarios to interact with. Feel free to introduce NPCs, but do not make moves for the characters. If user introduces another character, keep the story going where you left off as if that new character just arrived.'}];
// context is a list of strings that are used to seed the chatgpt api and it's responses
class Context {
messages = [];
currentLine = '';
currentResponse = '';
constructor(messages = default_messages) {
this.messages = messages;
}
add_user_prompt(message) {
this.messages.push({ role: 'user', content: message });
}
@ -34,9 +40,16 @@ class Context {
const theLine = this.currentLine;
this.currentResponse = '';
this.currentLine = '';
this.save_history();
return theLine;
}
save_history() {
const prettyData = JSON.stringify(this.messages, null, 2);
fs.writeFileSync('./messages.json', prettyData);
}
is_response_in_progress() {
return this.currentResponse !== '' || this.currentLine !== '';
}
@ -52,7 +65,8 @@ class Context {
}
}
const context = new Context();
const savedMessages = require('./messages.json');
const context = new Context(savedMessages);
const client = new irc.Client(config.server, config.nick, {
channels: config.channels,
@ -60,13 +74,11 @@ const client = new irc.Client(config.server, config.nick, {
// listen for messages that start with !chat and call the chatgpt api with a callback that prints the response line by line
client.addListener('message', async (from, to, message) => {
is_chat_cmd = message.startsWith('!chat');
is_cont_cmd = message.startsWith('!cont');
if (is_chat_cmd || is_cont_cmd) {
if(context.is_response_in_progress()) { return; }
if(is_chat_cmd) {
context.clear();
}
const query = message.slice(6);
chatgpt(query, (line) => {
client.say(to, line);