Add saving
parent
112e960135
commit
90d457c790
|
@ -1,2 +1,3 @@
|
|||
config.json
|
||||
node_modules
|
||||
messages.json
|
||||
|
|
22
index.js
22
index.js
|
@ -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);
|
||||
|
|
Loading…
Reference in New Issue