mirror of
https://github.com/kevinthedang/discord-ollama.git
synced 2025-12-12 03:46:08 -05:00
43fb2ea94e
* added message style command * docker setup scripts * reformat messageStyle.ts * fix: register unregister on deploy * add: messageStream preference * add: json config handler * update: messageCreate gets config * update: shifted chat to config callback * fix: naming conventions based on discord * update: setup in docs now * add: static docker ips * version increment * add: bot message for no config * fix: no config case * add: clarification for subnetting * update: version increment in lock file --------- Co-authored-by: JT2M0L3Y <jtsmoley@icloud.com>
59 lines
2.2 KiB
TypeScript
59 lines
2.2 KiB
TypeScript
import { ChatResponse } from 'ollama'
|
|
import { embedMessage, event, Events, normalMessage } from '../utils/index.js'
|
|
import { Configuration, getConfig } from '../utils/jsonHandler.js'
|
|
|
|
/**
|
|
* Max Message length for free users is 2000 characters (bot or not).
|
|
* @param message the message received from the channel
|
|
*/
|
|
export default event(Events.MessageCreate, async ({ log, msgHist, tokens, ollama }, message) => {
|
|
log(`Message \"${message.content}\" from ${message.author.tag} in channel/thread ${message.channelId}.`)
|
|
|
|
// Hard-coded channel to test output there only, in our case "ollama-endpoint"
|
|
if (message.channelId != tokens.channel) return
|
|
|
|
// Do not respond if bot talks in the chat
|
|
if (message.author.tag === message.client.user.tag) return
|
|
|
|
// Only respond if message mentions the bot
|
|
if (!message.mentions.has(tokens.clientUid)) return
|
|
|
|
// push user response
|
|
msgHist.push({
|
|
role: 'user',
|
|
content: message.content
|
|
})
|
|
|
|
// Try to query and send embed
|
|
try {
|
|
const config: Configuration = await new Promise((resolve, reject) => {
|
|
getConfig('config.json', (config) => {
|
|
if (config === undefined) {
|
|
reject(new Error('No Configuration is set up.'))
|
|
return
|
|
}
|
|
resolve(config)
|
|
})
|
|
})
|
|
|
|
let response: ChatResponse
|
|
|
|
// undefined or false, use normal, otherwise use embed
|
|
if (config.options['message-style'])
|
|
response = await embedMessage(message, ollama, tokens, msgHist)
|
|
else
|
|
response = await normalMessage(message, ollama, tokens, msgHist)
|
|
|
|
// If something bad happened, remove user query and stop
|
|
if (response == undefined) { msgHist.pop(); return }
|
|
|
|
// successful query, save it as history
|
|
msgHist.push({
|
|
role: 'assistant',
|
|
content: response.message.content
|
|
})
|
|
} catch (error: any) {
|
|
msgHist.pop() // remove message because of failure
|
|
message.reply(`**Response generation failed.**\n\nReason: ${error.message}\n\nPlease use any config slash command.`)
|
|
}
|
|
}) |