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

Update: Use built-in catch method for logging

Browse Source
This commit is contained in:
Kevin Dang
2025-06-17 07:03:23 -07:00
parent c7f9f19c88
commit 9a2744f646
3 changed files with 20 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/commands/deleteModel.ts
View File

@@ -22,7 +22,6 @@ export const DeleteModel: SlashCommand = {
// defer reply to avoid timeout // defer reply to avoid timeout
await interaction.deferReply() await interaction.deferReply()
const modelInput: string = interaction.options.get('model-name')!!.value as string const modelInput: string = interaction.options.get('model-name')!!.value as string
let modelExists: boolean
// fetch channel and message // fetch channel and message
const channel = await client.channels.fetch(interaction.channelId) const channel = await client.channels.fetch(interaction.channelId)
@@ -38,17 +37,21 @@ export const DeleteModel: SlashCommand = {
} }
// check if model exists // check if model exists
try { const modelExists = await ollama.list()
modelExists = await ollama.list() .then(response => response.models.some((model: ModelResponse) => model.name.startsWith(modelInput)))
.then(response => response.models.some((model: ModelResponse) => model.name.startsWith(modelInput))) .catch(error => {
} catch (error) { console.error(`[Command: delete-model] Failed to connect with Ollama service. Error: ${error.message}`)
})
// Validate for any issue or if service is running
if (!modelExists) {
interaction.editReply({ interaction.editReply({
content: `The Ollama service is not running. Please turn on/download the [service](https://ollama.com/).` content: `The Ollama service is not running. Please turn on/download the [service](https://ollama.com/).`
}) })
return return
} }
try { try {
// call ollama to delete model // call ollama to delete model
if (modelExists) { if (modelExists) {

13
src/commands/pullModel.ts
View File

@@ -22,7 +22,6 @@ export const PullModel: SlashCommand = {
// defer reply to avoid timeout // defer reply to avoid timeout
await interaction.deferReply() await interaction.deferReply()
const modelInput: string = interaction.options.get('model-to-pull')!!.value as string const modelInput: string = interaction.options.get('model-to-pull')!!.value as string
let modelExists: boolean
// fetch channel and message // fetch channel and message
const channel = await client.channels.fetch(interaction.channelId) const channel = await client.channels.fetch(interaction.channelId)
@@ -38,10 +37,14 @@ export const PullModel: SlashCommand = {
} }
// check if model was already pulled, if the ollama service isn't running throw error // check if model was already pulled, if the ollama service isn't running throw error
try { const modelExists = await ollama.list()
modelExists = await ollama.list() .then(response => response.models.some((model: ModelResponse) => model.name.startsWith(modelInput)))
.then(response => response.models.some((model: ModelResponse) => model.name.startsWith(modelInput))) .catch(error => {
} catch (error) { console.error(`[Command: pull-model] Failed to connect with Ollama service. Error: ${error.message}`)
})
// Validate for any issue or if service is running
if (!modelExists) {
interaction.editReply({ interaction.editReply({
content: `The Ollama service is not running. Please turn on/download the [service](https://ollama.com/).` content: `The Ollama service is not running. Please turn on/download the [service](https://ollama.com/).`
}) })

3
src/commands/switchModel.ts
View File

@@ -45,6 +45,9 @@ export const SwitchModel: SlashCommand = {
} }
} }
}) })
.catch(error => {
console.error(`[Command: switch-model] Failed to connect with Ollama service. Error: ${error.message}`)
})
// todo: problem can be here if async messes up // todo: problem can be here if async messes up
if (switchSuccess) { if (switchSuccess) {
// set model now that it exists // set model now that it exists