kindrobot
/
ircgpt
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

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

22
index.js
View File

@ -3,13 +3,19 @@
const irc = require('irc'); const irc = require('irc');
const axios = require('axios'); const axios = require('axios');
const config = require('./config.json'); 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 // context is a list of strings that are used to seed the chatgpt api and it's responses
class Context { class Context {
messages = [];
currentLine = ''; currentLine = '';
currentResponse = ''; currentResponse = '';
constructor(messages = default_messages) {
this.messages = messages;
}
add_user_prompt(message) { add_user_prompt(message) {
this.messages.push({ role: 'user', content: message }); this.messages.push({ role: 'user', content: message });
} }
@ -34,9 +40,16 @@ class Context {
const theLine = this.currentLine; const theLine = this.currentLine;
this.currentResponse = ''; this.currentResponse = '';
this.currentLine = ''; this.currentLine = '';
this.save_history();
return theLine; return theLine;
} }
save_history() {
const prettyData = JSON.stringify(this.messages, null, 2);
fs.writeFileSync('./messages.json', prettyData);
}
is_response_in_progress() { is_response_in_progress() {
return this.currentResponse !== '' || this.currentLine !== ''; 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, { const client = new irc.Client(config.server, config.nick, {
channels: config.channels, 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 // 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) => { client.addListener('message', async (from, to, message) => {
is_chat_cmd = message.startsWith('!chat'); is_chat_cmd = message.startsWith('!chat');
is_cont_cmd = message.startsWith('!cont'); is_cont_cmd = message.startsWith('!cont');
if (is_chat_cmd || is_cont_cmd) { if (is_chat_cmd || is_cont_cmd) {
if(context.is_response_in_progress()) { return; } if(context.is_response_in_progress()) { return; }
if(is_chat_cmd) {
context.clear();
}
const query = message.slice(6); const query = message.slice(6);
chatgpt(query, (line) => { chatgpt(query, (line) => {
client.say(to, line); client.say(to, line);