mirror of
https://github.com/kevinthedang/discord-ollama.git
synced 2025-12-12 03:46:08 -05:00
* 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>
48 lines
1.5 KiB
TypeScript
48 lines
1.5 KiB
TypeScript
import { Message } from 'discord.js'
|
|
import { ChatResponse, Ollama } from 'ollama'
|
|
import { UserMessage } from './events.js'
|
|
|
|
/**
|
|
* Method to send replies as normal text on discord like any other user
|
|
* @param message message sent by the user
|
|
* @param tokens tokens to run query
|
|
* @param msgHist message history between user and model
|
|
*/
|
|
export async function normalMessage(
|
|
message: Message,
|
|
ollama: Ollama,
|
|
tokens: {
|
|
channel: string,
|
|
model: string
|
|
},
|
|
msgHist: UserMessage[]
|
|
) {
|
|
// bot's respnse
|
|
let response: ChatResponse
|
|
|
|
await message.reply('Generating Response . . .').then(async sentMessage => {
|
|
try {
|
|
// Attempt to query model for message
|
|
response = await ollama.chat({
|
|
model: tokens.model,
|
|
messages: msgHist,
|
|
options: {
|
|
num_thread: 8, // remove if optimization needed further
|
|
mirostat: 1,
|
|
mirostat_tau: 2.0,
|
|
top_k: 70
|
|
},
|
|
stream: false
|
|
})
|
|
|
|
// edit the 'generic' response to new message
|
|
sentMessage.edit(response.message.content)
|
|
} catch(error: any) {
|
|
console.log(`[Util: messageNormal] Error creating message: ${error.message}`)
|
|
sentMessage.edit(`**Response generation failed.**\n\nReason: ${error.message}`)
|
|
}
|
|
})
|
|
|
|
// Hope there is a response, force client to believe
|
|
return response!!
|
|
} |