ollama responds to discord msgs
This commit is contained in:
@@ -1,7 +1,9 @@
|
||||
import { Event } from '../utils/index.js'
|
||||
import messageCreate from './messageCreate.js'
|
||||
import ready from './ready.js'
|
||||
|
||||
// Centralized export for all events
|
||||
export default [
|
||||
ready
|
||||
ready,
|
||||
messageCreate
|
||||
] as Event[] // staticly is better ts practice, dynamic exporting is possible
|
||||
42
src/events/messageCreate.ts
Normal file
42
src/events/messageCreate.ts
Normal file
@@ -0,0 +1,42 @@
|
||||
import { event, Events } from '../utils/index.js'
|
||||
import Axios from 'axios'
|
||||
|
||||
/*
|
||||
* Max Message length for free users is 2000 characters (bot or not).
|
||||
*/
|
||||
export default event(Events.MessageCreate, ({ log }, message) => {
|
||||
log(`Message created \"${message.content}\" from ${message.author.tag}.`)
|
||||
|
||||
// Hard-coded channel to test output there only, in our case "ollama-endpoint"
|
||||
if (message.channelId != '1188262786497785896') {
|
||||
log(`Unauthorized Channel input, Aborting...`)
|
||||
return
|
||||
}
|
||||
log(`Channel id OK!`)
|
||||
|
||||
// Do not respond if bot talks in the chat
|
||||
if (message.author.tag === message.client.user.tag) {
|
||||
log(`Found Bot message reply, Aborting...`)
|
||||
return
|
||||
}
|
||||
log(`Sender Checked!`)
|
||||
|
||||
// Request made to API
|
||||
const request = async () => {
|
||||
try {
|
||||
const response = await Axios.post('http://127.0.0.1:11434/api/generate', {
|
||||
model: 'llama2',
|
||||
prompt: message.content,
|
||||
stream: false
|
||||
})
|
||||
log(response.data)
|
||||
//
|
||||
message.reply(response.data['response'])
|
||||
} catch (error) {
|
||||
log(error)
|
||||
}
|
||||
}
|
||||
|
||||
// Attempt to call ollama's endpoint
|
||||
request()
|
||||
})
|
||||
@@ -1,5 +1,6 @@
|
||||
import { event, Events } from '../utils/index.js'
|
||||
|
||||
// Log when the bot successfully logs in and export it
|
||||
export default event(Events.ClientReady, ({ log }, client) => {
|
||||
return log(`Logged in as ${client.user.username}.`)
|
||||
})
|
||||
Reference in New Issue
Block a user