alex/discord-ollama
1
0
Fork 0
mirror of https://github.com/kevinthedang/discord-ollama.git synced 2025-12-13 03:56:07 -05:00
Code Issues Packages Projects Releases Wiki Activity

Pull and Switch Model Revised (#142)

Browse Source 
* Update: pull-model only runnable by admins now

* Update: switch-model cannot pull models anymore

* Update: less technical responses

* Update: version increment
This commit is contained in:
Kevin Dang
2024-12-04 21:29:01 -08:00
committed by GitHub
parent 1c8449d578
commit d570a50d46
12 changed files with 52 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
src/commands/switchModel.ts
View File

@@ -6,7 +6,7 @@ import { openConfig, UserCommand } from "../utils/index.js";
export const SwitchModel: SlashCommand = {
name: 'switch-model',
description: 'switches current model to preferred model to use.',
description: 'switches current model to use.',
// set available user options to pass to the command
options: [
@@ -29,7 +29,7 @@ export const SwitchModel: SlashCommand = {
if (!channel || !UserCommand.includes(channel.type)) return
try {
// Phase 1: Set the model
// Phase 1: Switch to the model
let switchSuccess = false
await ollama.list()
.then(response => {
@@ -47,28 +47,20 @@ export const SwitchModel: SlashCommand = {
}
})
// todo: problem can be here if async messes up
if (switchSuccess) return
if (switchSuccess) {
// set model now that it exists
openConfig(`${interaction.user.username}-config.json`, interaction.commandName, modelInput)
return
}
// Phase 2: Try to get it regardless
// Phase 2: Notify user of failure to find model.
interaction.editReply({
content: `Could not find **${modelInput}** in local model library, trying to pull it now...\n\nThis could take a few moments... Please be patient!`
})
await ollama.pull({
model: modelInput
})
// set model now that it exists
openConfig(`${interaction.user.username}-config.json`, interaction.commandName, modelInput)
// We got the model!
interaction.editReply({
content: `Successfully added and set **${modelInput}** as your preferred model.`
})
content: `Could not find **${modelInput}** in local model library.\n\nPlease contact an server admin for access to this model.`
})
} catch (error) {
// could not resolve user model switch
interaction.editReply({
content: `Unable to switch user preferred model to **${modelInput}**.\n\n${error}\n\nPossible solution is to run \`/pull-model ${modelInput}\` and try again.`
content: `Unable to switch user preferred model to **${modelInput}**.\n\n${error}\n\nPossible solution is to request an server admin run \`/pull-model ${modelInput}\` and try again.`
})
return
}