Compare commits
8 Commits
v0.8.2
...
feature/lo
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
cc7a3661b7 | ||
|
|
c026de5d62 | ||
|
|
f99bd2528d | ||
|
|
40783818b9 | ||
|
|
ed0d8600df | ||
|
|
03939ef562 | ||
|
|
456f70b9e1 | ||
|
|
5b542aca1a |
132
.github/workflows/deploy.yml
vendored
Normal file
132
.github/workflows/deploy.yml
vendored
Normal file
@@ -0,0 +1,132 @@
|
|||||||
|
name: Deploy
|
||||||
|
run-name: Deploy Application Latest
|
||||||
|
on:
|
||||||
|
push:
|
||||||
|
tags:
|
||||||
|
- 'v*'
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
Deploy-Application:
|
||||||
|
runs-on: self-hosted
|
||||||
|
environment: deploy
|
||||||
|
timeout-minutes: 5
|
||||||
|
steps:
|
||||||
|
- name: Checkout Repo
|
||||||
|
uses: actions/checkout@v4
|
||||||
|
|
||||||
|
# Generate Secret File for Compose case
|
||||||
|
- name: Create Environment Variables
|
||||||
|
run: |
|
||||||
|
touch .env
|
||||||
|
echo CLIENT_TOKEN = ${{ secrets.CLIENT }} >> .env
|
||||||
|
echo OLLAMA_IP = ${{ secrets.OLLAMA_IP }} >> .env
|
||||||
|
echo OLLAMA_PORT = ${{ secrets.OLLAMA_PORT }} >> .env
|
||||||
|
echo DISCORD_IP = ${{ secrets.DISCORD_IP }} >> .env
|
||||||
|
echo SUBNET_ADDRESS = ${{ secrets.SUBNET_ADDRESS }} >> .env
|
||||||
|
echo REDIS_IP = ${{ secrets.REDIS_IP }} >> .env
|
||||||
|
echo REDIS_PORT = ${{ secrets.REDIS_PORT }} >> .env
|
||||||
|
|
||||||
|
- name: Check if directory exists and delete it
|
||||||
|
run: |
|
||||||
|
if [ -d "${{ secrets.PATH }}" ]; then
|
||||||
|
echo "Directory exists, deleting old version..."
|
||||||
|
rm -rf ${{ secrets.PATH }}
|
||||||
|
else
|
||||||
|
echo "Directory does not exist."
|
||||||
|
fi
|
||||||
|
|
||||||
|
- name: Clone Repo onto Server
|
||||||
|
run: |
|
||||||
|
git clone https://github.com/kevinthedang/discord-ollama.git ${{ secrets.PATH }}
|
||||||
|
cd ${{ secrets.PATH }}
|
||||||
|
|
||||||
|
- name: Install nvm and Node.js lts/jod
|
||||||
|
run: |
|
||||||
|
curl -o- https://raw.githubusercontent.com/nvm-sh/nvm/v0.40.1/install.sh | bash
|
||||||
|
export NVM_DIR="$HOME/.nvm"
|
||||||
|
[ -s "$NVM_DIR/nvm.sh" ] && \. "$NVM_DIR/nvm.sh"
|
||||||
|
echo "NVM installed successfully."
|
||||||
|
nvm install lts/jod
|
||||||
|
nvm alias default lts/jod
|
||||||
|
node -v
|
||||||
|
npm -v
|
||||||
|
|
||||||
|
- name: Build Application
|
||||||
|
run: |
|
||||||
|
export NVM_DIR="$HOME/.nvm"
|
||||||
|
[ -s "$NVM_DIR/nvm.sh" ] && \. "$NVM_DIR/nvm.sh"
|
||||||
|
npm install
|
||||||
|
|
||||||
|
IMAGE="kevinthedang/discord-ollama"
|
||||||
|
REDIS="redis"
|
||||||
|
OLLAMA="ollama/ollama"
|
||||||
|
|
||||||
|
if docker images | grep -q $IMAGE; then
|
||||||
|
IMAGE_ID=$(docker images -q $IMAGE)
|
||||||
|
CONTAINER_IDS=$(docker ps -q --filter "ancestor=$IMAGE_ID")
|
||||||
|
|
||||||
|
if [ ! -z "$CONTAINER_IDS" ]; then
|
||||||
|
# Stop and remove the running containers
|
||||||
|
docker stop $CONTAINER_IDS
|
||||||
|
echo "Stopped and removed the containers using the image $IMAGE"
|
||||||
|
fi
|
||||||
|
docker rmi $IMAGE_ID
|
||||||
|
echo "Old $IMAGE Image Removed"
|
||||||
|
fi
|
||||||
|
|
||||||
|
if docker images | grep -q $REDIS; then
|
||||||
|
IMAGE_ID=$(docker images -q $REDIS)
|
||||||
|
CONTAINER_IDS=$(docker ps -q --filter "ancestor=$IMAGE_ID")
|
||||||
|
|
||||||
|
if [ ! -z "$CONTAINER_IDS" ]; then
|
||||||
|
# Stop and remove the running containers
|
||||||
|
docker stop $CONTAINER_IDS
|
||||||
|
echo "Stopped and removed the containers using the image $REDIS"
|
||||||
|
fi
|
||||||
|
docker rmi $IMAGE_ID
|
||||||
|
echo "Old $REDIS Image Removed"
|
||||||
|
fi
|
||||||
|
|
||||||
|
if docker images | grep -q $OLLAMA; then
|
||||||
|
IMAGE_ID=$(docker images -q $OLLAMA)
|
||||||
|
CONTAINER_IDS=$(docker ps -q --filter "ancestor=$IMAGE_ID")
|
||||||
|
|
||||||
|
if [ ! -z "$CONTAINER_IDS" ]; then
|
||||||
|
# Stop and remove the running containers
|
||||||
|
docker stop $CONTAINER_IDS
|
||||||
|
echo "Stopped and removed the containers using the image $OLLAMA"
|
||||||
|
fi
|
||||||
|
docker rmi $IMAGE_ID
|
||||||
|
echo "Old $OLLAMA Image Removed"
|
||||||
|
fi
|
||||||
|
|
||||||
|
docker network prune -f
|
||||||
|
docker system prune -a -f
|
||||||
|
|
||||||
|
npm run docker:build-latest
|
||||||
|
|
||||||
|
- name: Start Application
|
||||||
|
run: |
|
||||||
|
docker network create --subnet=${{ secrets.SUBNET_ADDRESS }}/16 ollama-net || true
|
||||||
|
docker run --rm -d \
|
||||||
|
-v ollama:/root/.ollama \
|
||||||
|
-p ${{ secrets.OLLAMA_PORT }}:${{ secrets.OLLAMA_PORT }} \
|
||||||
|
--name ollama \
|
||||||
|
--network ollama-net \
|
||||||
|
--ip ${{ secrets.OLLAMA_IP }} \
|
||||||
|
ollama/ollama:latest
|
||||||
|
|
||||||
|
docker run --rm -d \
|
||||||
|
-v redis:/root/.redis \
|
||||||
|
-p ${{ secrets.REDIS_PORT }}:${{ secrets.REDIS_PORT }} \
|
||||||
|
--name redis \
|
||||||
|
--network ollama-net \
|
||||||
|
--ip ${{ secrets.REDIS_IP }} \
|
||||||
|
redis:latest
|
||||||
|
|
||||||
|
docker run --rm -d \
|
||||||
|
-v discord:/src/app \
|
||||||
|
--name discord \
|
||||||
|
--network ollama-net \
|
||||||
|
--ip ${{ secrets.DISCORD_IP }} \
|
||||||
|
kevinthedang/discord-ollama
|
||||||
50
.github/workflows/release.yml
vendored
50
.github/workflows/release.yml
vendored
@@ -1,50 +0,0 @@
|
|||||||
name: Deploy
|
|
||||||
run-name: Release Docker Image
|
|
||||||
on:
|
|
||||||
push:
|
|
||||||
tags:
|
|
||||||
- 'v*'
|
|
||||||
|
|
||||||
jobs:
|
|
||||||
Release-Docker-Image:
|
|
||||||
runs-on: ubuntu-latest
|
|
||||||
environment: release
|
|
||||||
timeout-minutes: 3
|
|
||||||
steps:
|
|
||||||
- name: Checkout Repository
|
|
||||||
uses: actions/checkout@v4
|
|
||||||
|
|
||||||
- name: Set up Node Environment lts/jod
|
|
||||||
uses: actions/setup-node@v4
|
|
||||||
with:
|
|
||||||
node-version: lts/jod
|
|
||||||
cache: "npm"
|
|
||||||
|
|
||||||
- name: Create Environment Variables
|
|
||||||
run: |
|
|
||||||
touch .env
|
|
||||||
echo CLIENT_TOKEN = NOT_REAL_TOKEN >> .env
|
|
||||||
echo OLLAMA_IP = ${{ secrets.OLLAMA_IP }} >> .env
|
|
||||||
echo OLLAMA_PORT = ${{ secrets.OLLAMA_PORT }} >> .env
|
|
||||||
echo REDIS_IP = ${{ secrets.REDIS_IP }} >> .env
|
|
||||||
echo REDIS_PORT = ${{ secrets.REDIS_PORT }} >> .env
|
|
||||||
|
|
||||||
- name: Get Version from package.json
|
|
||||||
run: echo "VERSION=$(jq -r '.version' package.json)" >> $GITHUB_ENV
|
|
||||||
|
|
||||||
- name: Build Image
|
|
||||||
run: |
|
|
||||||
npm run docker:build
|
|
||||||
|
|
||||||
- name: Build Image as Latest
|
|
||||||
run: |
|
|
||||||
npm run docker:build-latest
|
|
||||||
|
|
||||||
- name: Log into Docker
|
|
||||||
run: |
|
|
||||||
docker login --username ${{ vars.DOCKER_USER }} --password ${{ secrets.DOCKER_PASS }}
|
|
||||||
|
|
||||||
- name: Release Docker Image
|
|
||||||
run: |
|
|
||||||
docker push ${{ vars.DOCKER_USER }}/discord-ollama:${{ env.VERSION }}
|
|
||||||
docker push ${{ vars.DOCKER_USER }}/discord-ollama:latest
|
|
||||||
@@ -5,7 +5,7 @@
|
|||||||
<p><a href="#"></a><a href="https://creativecommons.org/licenses/by/4.0/"><img alt="License" src="https://img.shields.io/badge/License-CC_BY_4.0-darkgreen.svg" /></a>
|
<p><a href="#"></a><a href="https://creativecommons.org/licenses/by/4.0/"><img alt="License" src="https://img.shields.io/badge/License-CC_BY_4.0-darkgreen.svg" /></a>
|
||||||
<a href="#"></a><a href="https://github.com/kevinthedang/discord-ollama/releases/latest"><img alt="Release" src="https://img.shields.io/github/v/release/kevinthedang/discord-ollama?logo=github" /></a>
|
<a href="#"></a><a href="https://github.com/kevinthedang/discord-ollama/releases/latest"><img alt="Release" src="https://img.shields.io/github/v/release/kevinthedang/discord-ollama?logo=github" /></a>
|
||||||
<a href="#"></a><a href="https://github.com/kevinthedang/discord-ollama/actions/workflows/build.yml"><img alt="Build Status" src="https://github.com/kevinthedang/discord-ollama/actions/workflows/build.yml/badge.svg" /></a>
|
<a href="#"></a><a href="https://github.com/kevinthedang/discord-ollama/actions/workflows/build.yml"><img alt="Build Status" src="https://github.com/kevinthedang/discord-ollama/actions/workflows/build.yml/badge.svg" /></a>
|
||||||
<a href="#"></a><a href="https://github.com/kevinthedang/discord-ollama/actions/workflows/release.yml"><img alt="Release Status" src="https://github.com/kevinthedang/discord-ollama/actions/workflows/release.yml/badge.svg" /></a>
|
<a href="#"></a><a href="https://github.com/kevinthedang/discord-ollama/actions/workflows/deploy.yml"><img alt="Deploy Status" src="https://github.com/kevinthedang/discord-ollama/actions/workflows/deploy.yml/badge.svg" /></a>
|
||||||
<a href="#"></a><a href="https://github.com/kevinthedang/discord-ollama/actions/workflows/test.yml"><img alt="Testing Status" src="https://github.com/kevinthedang/discord-ollama/actions/workflows/test.yml/badge.svg" /></a>
|
<a href="#"></a><a href="https://github.com/kevinthedang/discord-ollama/actions/workflows/test.yml"><img alt="Testing Status" src="https://github.com/kevinthedang/discord-ollama/actions/workflows/test.yml/badge.svg" /></a>
|
||||||
<a href="#"></a><a href="https://github.com/kevinthedang/discord-ollama/actions/workflows/coverage.yml"><img alt="Code Coverage" src="https://img.shields.io/endpoint?url=https://gist.githubusercontent.com/kevinthedang/bc7b5dcfa16561ab02bb3df67a99b22d/raw/coverage.json"></a>
|
<a href="#"></a><a href="https://github.com/kevinthedang/discord-ollama/actions/workflows/coverage.yml"><img alt="Code Coverage" src="https://img.shields.io/endpoint?url=https://gist.githubusercontent.com/kevinthedang/bc7b5dcfa16561ab02bb3df67a99b22d/raw/coverage.json"></a>
|
||||||
</div>
|
</div>
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ services:
|
|||||||
build: ./ # find docker file in designated path
|
build: ./ # find docker file in designated path
|
||||||
container_name: discord
|
container_name: discord
|
||||||
restart: always # rebuild container always
|
restart: always # rebuild container always
|
||||||
image: kevinthedang/discord-ollama:0.8.2
|
image: kevinthedang/discord-ollama:0.8.3
|
||||||
environment:
|
environment:
|
||||||
CLIENT_TOKEN: ${CLIENT_TOKEN}
|
CLIENT_TOKEN: ${CLIENT_TOKEN}
|
||||||
OLLAMA_IP: ${OLLAMA_IP}
|
OLLAMA_IP: ${OLLAMA_IP}
|
||||||
@@ -28,7 +28,6 @@ services:
|
|||||||
networks:
|
networks:
|
||||||
ollama-net:
|
ollama-net:
|
||||||
ipv4_address: ${OLLAMA_IP}
|
ipv4_address: ${OLLAMA_IP}
|
||||||
|
|
||||||
runtime: nvidia # use Nvidia Container Toolkit for GPU support
|
runtime: nvidia # use Nvidia Container Toolkit for GPU support
|
||||||
devices:
|
devices:
|
||||||
- /dev/nvidia0
|
- /dev/nvidia0
|
||||||
|
|||||||
1071
package-lock.json
generated
1071
package-lock.json
generated
File diff suppressed because it is too large
Load Diff
16
package.json
16
package.json
@@ -1,6 +1,6 @@
|
|||||||
{
|
{
|
||||||
"name": "discord-ollama",
|
"name": "discord-ollama",
|
||||||
"version": "0.8.2",
|
"version": "0.8.3",
|
||||||
"description": "Ollama Integration into discord",
|
"description": "Ollama Integration into discord",
|
||||||
"main": "build/index.js",
|
"main": "build/index.js",
|
||||||
"exports": "./build/index.js",
|
"exports": "./build/index.js",
|
||||||
@@ -27,18 +27,18 @@
|
|||||||
"author": "Kevin Dang",
|
"author": "Kevin Dang",
|
||||||
"license": "ISC",
|
"license": "ISC",
|
||||||
"dependencies": {
|
"dependencies": {
|
||||||
"discord.js": "^14.16.3",
|
"discord.js": "^14.18.0",
|
||||||
"dotenv": "^16.4.7",
|
"dotenv": "^16.4.7",
|
||||||
"ollama": "^0.5.11",
|
"ollama": "^0.5.13",
|
||||||
"redis": "^4.7.0"
|
"redis": "^4.7.0"
|
||||||
},
|
},
|
||||||
"devDependencies": {
|
"devDependencies": {
|
||||||
"@types/node": "^22.10.2",
|
"@types/node": "^22.13.5",
|
||||||
"@vitest/coverage-v8": "^2.1.8",
|
"@vitest/coverage-v8": "^3.0.6",
|
||||||
"ts-node": "^10.9.2",
|
"ts-node": "^10.9.2",
|
||||||
"tsx": "^4.19.2",
|
"tsx": "^4.19.3",
|
||||||
"typescript": "^5.7.2",
|
"typescript": "^5.7.3",
|
||||||
"vitest": "^2.1.4"
|
"vitest": "^3.0.4"
|
||||||
},
|
},
|
||||||
"type": "module",
|
"type": "module",
|
||||||
"engines": {
|
"engines": {
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
import { Client, GatewayIntentBits } from 'discord.js'
|
import { Client, GatewayIntentBits } from 'discord.js'
|
||||||
import { Ollama } from 'ollama'
|
import { Ollama } from 'ollama'
|
||||||
import { createClient } from 'redis'
|
import { createClient } from 'redis'
|
||||||
import { Queue } from './queues/queue.js'
|
import { Queue } from './components/index.js'
|
||||||
import { UserMessage, registerEvents } from './utils/index.js'
|
import { UserMessage, registerEvents } from './utils/index.js'
|
||||||
import Events from './events/index.js'
|
import Events from './events/index.js'
|
||||||
import Keys from './keys.js'
|
import Keys from './keys.js'
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
import { Client, CommandInteraction, ApplicationCommandOptionType } from 'discord.js'
|
import { Client, CommandInteraction, ApplicationCommandOptionType, MessageFlags } from 'discord.js'
|
||||||
import { openConfig, SlashCommand, UserCommand } from '../utils/index.js'
|
import { openConfig, SlashCommand, UserCommand } from '../utils/index.js'
|
||||||
|
|
||||||
export const Capacity: SlashCommand = {
|
export const Capacity: SlashCommand = {
|
||||||
@@ -22,11 +22,13 @@ export const Capacity: SlashCommand = {
|
|||||||
if (!channel || !UserCommand.includes(channel.type)) return
|
if (!channel || !UserCommand.includes(channel.type)) return
|
||||||
|
|
||||||
// set state of bot chat features
|
// set state of bot chat features
|
||||||
openConfig(`${interaction.user.username}-config.json`, interaction.commandName, interaction.options.get('context-capacity')?.value)
|
openConfig(`${interaction.user.username}-config.json`, interaction.commandName,
|
||||||
|
interaction.options.get('context-capacity')?.value
|
||||||
|
)
|
||||||
|
|
||||||
interaction.reply({
|
interaction.reply({
|
||||||
content: `Max message history is now set to \`${interaction.options.get('context-capacity')?.value}\``,
|
content: `Max message history is now set to \`${interaction.options.get('context-capacity')?.value}\``,
|
||||||
ephemeral: true
|
flags: MessageFlags.Ephemeral
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -1,4 +1,4 @@
|
|||||||
import { Channel, Client, CommandInteraction, TextChannel } from 'discord.js'
|
import { Channel, Client, CommandInteraction, MessageFlags, TextChannel } from 'discord.js'
|
||||||
import { clearChannelInfo, SlashCommand, UserCommand } from '../utils/index.js'
|
import { clearChannelInfo, SlashCommand, UserCommand } from '../utils/index.js'
|
||||||
|
|
||||||
export const ClearUserChannelHistory: SlashCommand = {
|
export const ClearUserChannelHistory: SlashCommand = {
|
||||||
@@ -14,20 +14,22 @@ export const ClearUserChannelHistory: SlashCommand = {
|
|||||||
if (!channel || !UserCommand.includes(channel.type)) return
|
if (!channel || !UserCommand.includes(channel.type)) return
|
||||||
|
|
||||||
// clear channel info for user
|
// clear channel info for user
|
||||||
const successfulWipe = await clearChannelInfo(interaction.channelId,
|
const successfulWipe = await clearChannelInfo(
|
||||||
|
interaction.channelId,
|
||||||
interaction.channel as TextChannel,
|
interaction.channel as TextChannel,
|
||||||
interaction.user.username)
|
interaction.user.username
|
||||||
|
)
|
||||||
|
|
||||||
// check result of clearing history
|
// check result of clearing history
|
||||||
if (successfulWipe)
|
if (successfulWipe)
|
||||||
interaction.reply({
|
interaction.reply({
|
||||||
content: `History cleared in **this channel** cleared for **${interaction.user.username}**.`,
|
content: `History cleared in **this channel** cleared for **${interaction.user.username}**.`,
|
||||||
ephemeral: true
|
flags: MessageFlags.Ephemeral
|
||||||
})
|
})
|
||||||
else
|
else
|
||||||
interaction.reply({
|
interaction.reply({
|
||||||
content: `History was not be found for **${interaction.user.username}** in **this channel**.\n\nPlease chat with **${client.user?.username}** to start a chat history.`,
|
content: `History was not be found for **${interaction.user.username}** in **this channel**.\n\nPlease chat with **${client.user?.username}** to start a chat history.`,
|
||||||
ephemeral: true
|
flags: MessageFlags.Ephemeral
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -1,4 +1,4 @@
|
|||||||
import { ApplicationCommandOptionType, Client, CommandInteraction } from 'discord.js'
|
import { ApplicationCommandOptionType, Client, CommandInteraction, MessageFlags } from 'discord.js'
|
||||||
import { UserCommand, SlashCommand } from '../utils/index.js'
|
import { UserCommand, SlashCommand } from '../utils/index.js'
|
||||||
import { ollama } from '../client.js'
|
import { ollama } from '../client.js'
|
||||||
import { ModelResponse } from 'ollama'
|
import { ModelResponse } from 'ollama'
|
||||||
@@ -31,7 +31,7 @@ export const DeleteModel: SlashCommand = {
|
|||||||
if (!interaction.memberPermissions?.has('Administrator')) {
|
if (!interaction.memberPermissions?.has('Administrator')) {
|
||||||
interaction.reply({
|
interaction.reply({
|
||||||
content: `${interaction.commandName} is an admin command.\n\nPlease contact a server admin to pull the model you want.`,
|
content: `${interaction.commandName} is an admin command.\n\nPlease contact a server admin to pull the model you want.`,
|
||||||
ephemeral: true
|
flags: MessageFlags.Ephemeral
|
||||||
})
|
})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -53,7 +53,7 @@ export const DeleteModel: SlashCommand = {
|
|||||||
// could not delete the model
|
// could not delete the model
|
||||||
interaction.reply({
|
interaction.reply({
|
||||||
content: `Could not delete the **${modelInput}** model. It probably doesn't exist or you spelled it incorrectly.\n\nPlease try again if this is a mistake.`,
|
content: `Could not delete the **${modelInput}** model. It probably doesn't exist or you spelled it incorrectly.\n\nPlease try again if this is a mistake.`,
|
||||||
ephemeral: true
|
flags: MessageFlags.Ephemeral
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
import { Client, CommandInteraction, ApplicationCommandOptionType } from 'discord.js'
|
import { Client, CommandInteraction, ApplicationCommandOptionType, MessageFlags } from 'discord.js'
|
||||||
import { AdminCommand, openConfig, SlashCommand } from '../utils/index.js'
|
import { AdminCommand, openConfig, SlashCommand } from '../utils/index.js'
|
||||||
|
|
||||||
export const Disable: SlashCommand = {
|
export const Disable: SlashCommand = {
|
||||||
@@ -25,17 +25,19 @@ export const Disable: SlashCommand = {
|
|||||||
if (!interaction.memberPermissions?.has('Administrator')) {
|
if (!interaction.memberPermissions?.has('Administrator')) {
|
||||||
interaction.reply({
|
interaction.reply({
|
||||||
content: `${interaction.commandName} is an admin command.\n\nPlease contact an admin to use this command for you.`,
|
content: `${interaction.commandName} is an admin command.\n\nPlease contact an admin to use this command for you.`,
|
||||||
ephemeral: true
|
flags: MessageFlags.Ephemeral
|
||||||
})
|
})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// set state of bot chat features
|
// set state of bot chat features
|
||||||
openConfig(`${interaction.guildId}-config.json`, interaction.commandName, interaction.options.get('enabled')?.value)
|
openConfig(`${interaction.guildId}-config.json`, interaction.commandName,
|
||||||
|
interaction.options.get('enabled')?.value
|
||||||
|
)
|
||||||
|
|
||||||
interaction.reply({
|
interaction.reply({
|
||||||
content: `${client.user?.username} is now **${interaction.options.get('enabled')?.value ? "enabled" : "disabled" }**.`,
|
content: `${client.user?.username} is now **${interaction.options.get('enabled')?.value ? "enabled" : "disabled"}**.`,
|
||||||
ephemeral: true
|
flags: MessageFlags.Ephemeral
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -1,4 +1,4 @@
|
|||||||
import { ApplicationCommandOptionType, Client, CommandInteraction } from 'discord.js'
|
import { ApplicationCommandOptionType, Client, CommandInteraction, MessageFlags } from 'discord.js'
|
||||||
import { openConfig, SlashCommand, UserCommand } from '../utils/index.js'
|
import { openConfig, SlashCommand, UserCommand } from '../utils/index.js'
|
||||||
|
|
||||||
export const MessageStream: SlashCommand = {
|
export const MessageStream: SlashCommand = {
|
||||||
@@ -22,11 +22,13 @@ export const MessageStream: SlashCommand = {
|
|||||||
if (!channel || !UserCommand.includes(channel.type)) return
|
if (!channel || !UserCommand.includes(channel.type)) return
|
||||||
|
|
||||||
// save value to json and write to it
|
// save value to json and write to it
|
||||||
openConfig(`${interaction.user.username}-config.json`, interaction.commandName, interaction.options.get('stream')?.value)
|
openConfig(`${interaction.user.username}-config.json`, interaction.commandName,
|
||||||
|
interaction.options.get('stream')?.value
|
||||||
|
)
|
||||||
|
|
||||||
interaction.reply({
|
interaction.reply({
|
||||||
content: `Message streaming is now set to: \`${interaction.options.get('stream')?.value}\``,
|
content: `Message streaming is now set to: \`${interaction.options.get('stream')?.value}\``,
|
||||||
ephemeral: true
|
flags: MessageFlags.Ephemeral
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -1,4 +1,4 @@
|
|||||||
import { ApplicationCommandOptionType, Client, CommandInteraction } from "discord.js"
|
import { ApplicationCommandOptionType, Client, CommandInteraction, MessageFlags } from "discord.js"
|
||||||
import { ollama } from "../client.js"
|
import { ollama } from "../client.js"
|
||||||
import { ModelResponse } from "ollama"
|
import { ModelResponse } from "ollama"
|
||||||
import { UserCommand, SlashCommand } from "../utils/index.js"
|
import { UserCommand, SlashCommand } from "../utils/index.js"
|
||||||
@@ -31,7 +31,7 @@ export const PullModel: SlashCommand = {
|
|||||||
if (!interaction.memberPermissions?.has('Administrator')) {
|
if (!interaction.memberPermissions?.has('Administrator')) {
|
||||||
interaction.reply({
|
interaction.reply({
|
||||||
content: `${interaction.commandName} is an admin command.\n\nPlease contact a server admin to pull the model you want.`,
|
content: `${interaction.commandName} is an admin command.\n\nPlease contact a server admin to pull the model you want.`,
|
||||||
ephemeral: true
|
flags: MessageFlags.Ephemeral
|
||||||
})
|
})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
import { Client, CommandInteraction } from 'discord.js'
|
import { Client, CommandInteraction, MessageFlags } from 'discord.js'
|
||||||
import { AdminCommand, SlashCommand } from '../utils/index.js'
|
import { AdminCommand, SlashCommand } from '../utils/index.js'
|
||||||
|
|
||||||
export const Shutoff: SlashCommand = {
|
export const Shutoff: SlashCommand = {
|
||||||
@@ -18,7 +18,7 @@ export const Shutoff: SlashCommand = {
|
|||||||
if (!interaction.memberPermissions?.has('Administrator')) {
|
if (!interaction.memberPermissions?.has('Administrator')) {
|
||||||
interaction.reply({
|
interaction.reply({
|
||||||
content: `**Shutdown Aborted:**\n\n${interaction.user.tag}, You do not have permission to shutoff **${client.user?.tag}**.`,
|
content: `**Shutdown Aborted:**\n\n${interaction.user.tag}, You do not have permission to shutoff **${client.user?.tag}**.`,
|
||||||
ephemeral: true
|
flags: MessageFlags.Ephemeral
|
||||||
})
|
})
|
||||||
return // stop from shutting down
|
return // stop from shutting down
|
||||||
}
|
}
|
||||||
@@ -26,8 +26,9 @@ export const Shutoff: SlashCommand = {
|
|||||||
// Shutoff cleared, do it
|
// Shutoff cleared, do it
|
||||||
interaction.reply({
|
interaction.reply({
|
||||||
content: `${client.user?.tag} is shutting down.`,
|
content: `${client.user?.tag} is shutting down.`,
|
||||||
ephemeral: true
|
flags: MessageFlags.Ephemeral
|
||||||
})
|
})
|
||||||
|
|
||||||
console.log(`[Command: shutoff] ${client.user?.tag} is shutting down.`)
|
console.log(`[Command: shutoff] ${client.user?.tag} is shutting down.`)
|
||||||
|
|
||||||
// clean up client instance and stop
|
// clean up client instance and stop
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
import { ChannelType, Client, CommandInteraction, TextChannel, ThreadChannel } from 'discord.js'
|
import { ChannelType, Client, CommandInteraction, MessageFlags, TextChannel, ThreadChannel } from 'discord.js'
|
||||||
import { AdminCommand, openChannelInfo, SlashCommand } from '../utils/index.js'
|
import { AdminCommand, openChannelInfo, SlashCommand } from '../utils/index.js'
|
||||||
|
|
||||||
export const ThreadCreate: SlashCommand = {
|
export const ThreadCreate: SlashCommand = {
|
||||||
@@ -21,14 +21,12 @@ export const ThreadCreate: SlashCommand = {
|
|||||||
thread.send(`Hello ${interaction.user} and others! \n\nIt's nice to meet you. Please talk to me by typing **@${client.user?.username}** with your message.`)
|
thread.send(`Hello ${interaction.user} and others! \n\nIt's nice to meet you. Please talk to me by typing **@${client.user?.username}** with your message.`)
|
||||||
|
|
||||||
// handle storing this chat channel
|
// handle storing this chat channel
|
||||||
openChannelInfo(thread.id,
|
openChannelInfo(thread.id, thread as ThreadChannel, interaction.user.tag)
|
||||||
thread as ThreadChannel,
|
|
||||||
interaction.user.tag)
|
|
||||||
|
|
||||||
// user only reply
|
// user only reply
|
||||||
return interaction.reply({
|
return interaction.reply({
|
||||||
content: `I can help you in <#${thread.id}> below.`,
|
content: `I can help you in <#${thread.id}> below.`,
|
||||||
ephemeral: true
|
flags: MessageFlags.Ephemeral
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -1,4 +1,4 @@
|
|||||||
import { ChannelType, Client, CommandInteraction, TextChannel, ThreadChannel } from 'discord.js'
|
import { ChannelType, Client, CommandInteraction, MessageFlags, TextChannel, ThreadChannel } from 'discord.js'
|
||||||
import { AdminCommand, openChannelInfo, SlashCommand } from '../utils/index.js'
|
import { AdminCommand, openChannelInfo, SlashCommand } from '../utils/index.js'
|
||||||
|
|
||||||
export const PrivateThreadCreate: SlashCommand = {
|
export const PrivateThreadCreate: SlashCommand = {
|
||||||
@@ -22,15 +22,12 @@ export const PrivateThreadCreate: SlashCommand = {
|
|||||||
|
|
||||||
// handle storing this chat channel
|
// handle storing this chat channel
|
||||||
// store: thread.id, thread.name
|
// store: thread.id, thread.name
|
||||||
openChannelInfo(thread.id,
|
openChannelInfo(thread.id, thread as ThreadChannel, interaction.user.tag)
|
||||||
thread as ThreadChannel,
|
|
||||||
interaction.user.tag
|
|
||||||
)
|
|
||||||
|
|
||||||
// user only reply
|
// user only reply
|
||||||
return interaction.reply({
|
return interaction.reply({
|
||||||
content: `I can help you in <#${thread.id}>.`,
|
content: `I can help you in <#${thread.id}>.`,
|
||||||
ephemeral: true
|
flags: MessageFlags.Ephemeral
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
46
src/components/binder.ts
Normal file
46
src/components/binder.ts
Normal file
@@ -0,0 +1,46 @@
|
|||||||
|
|
||||||
|
/**
|
||||||
|
* @class Logger
|
||||||
|
* @description A class to handle logging messages
|
||||||
|
* @method log
|
||||||
|
*/
|
||||||
|
export class Logger {
|
||||||
|
private logPrefix: string = ''
|
||||||
|
private type: string = 'log'
|
||||||
|
|
||||||
|
private constructPrefix(component?: string, method?: string): string {
|
||||||
|
let prefix = this.type.toUpperCase()
|
||||||
|
|
||||||
|
if (component) {
|
||||||
|
prefix += ` [${component}`
|
||||||
|
if (method) prefix += `: ${method}`
|
||||||
|
prefix += ']'
|
||||||
|
}
|
||||||
|
|
||||||
|
return prefix
|
||||||
|
}
|
||||||
|
|
||||||
|
public bind(component?: string, method?: string): CallableFunction {
|
||||||
|
let tempPrefix = this.constructPrefix(component, method)
|
||||||
|
|
||||||
|
if (tempPrefix !== this.logPrefix) this.logPrefix = tempPrefix
|
||||||
|
|
||||||
|
switch (this.type) {
|
||||||
|
case 'warn':
|
||||||
|
return console.warn.bind(console, this.logPrefix)
|
||||||
|
case 'error':
|
||||||
|
return console.error.bind(console, this.logPrefix)
|
||||||
|
case 'log':
|
||||||
|
default:
|
||||||
|
return console.log.bind(console, this.logPrefix)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public log(type: string, message: unknown, component?: string, method?: string): void {
|
||||||
|
if (type && type !== this.type) this.type = type
|
||||||
|
|
||||||
|
let log = this.bind(component, method)
|
||||||
|
|
||||||
|
log(message)
|
||||||
|
}
|
||||||
|
}
|
||||||
2
src/components/index.ts
Normal file
2
src/components/index.ts
Normal file
@@ -0,0 +1,2 @@
|
|||||||
|
export * from './queue.js'
|
||||||
|
export * from './binder.js'
|
||||||
@@ -17,7 +17,7 @@ export class Queue<T> implements IQueue<T> {
|
|||||||
* Set up Queue
|
* Set up Queue
|
||||||
* @param capacity max length of queue
|
* @param capacity max length of queue
|
||||||
*/
|
*/
|
||||||
constructor(public capacity: number = 5) {}
|
constructor(public capacity: number = 5) { }
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Put item in front of queue
|
* Put item in front of queue
|
||||||
@@ -1,6 +1,9 @@
|
|||||||
import { TextChannel } from 'discord.js'
|
import { TextChannel } from 'discord.js'
|
||||||
import { event, Events, normalMessage, UserMessage, clean, getTextFileAttachmentData } from '../utils/index.js'
|
import {
|
||||||
import { getChannelInfo, getServerConfig, getUserConfig, openChannelInfo, openConfig, UserConfig, getAttachmentData } from '../utils/index.js'
|
event, Events, normalMessage, UserMessage, clean,
|
||||||
|
getChannelInfo, getServerConfig, getUserConfig, openChannelInfo,
|
||||||
|
openConfig, UserConfig, getAttachmentData, getTextFileAttachmentData
|
||||||
|
} from '../utils/index.js'
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Max Message length for free users is 2000 characters (bot or not).
|
* Max Message length for free users is 2000 characters (bot or not).
|
||||||
|
|||||||
@@ -21,13 +21,12 @@ export function getEnvVar(name: string, fallback?: string): string {
|
|||||||
throw new Error(`Environment variable ${name} is not set.`)
|
throw new Error(`Environment variable ${name} is not set.`)
|
||||||
|
|
||||||
// validate User-Generated Discord Application Tokens
|
// validate User-Generated Discord Application Tokens
|
||||||
if (name === "CLIENT_TOKEN")
|
if (name === "CLIENT_TOKEN" && value.length > 72)
|
||||||
if (value.length < 72) throw new Error(`The "CLIENT_TOKEN" provided is not of at least length 72.
|
throw new Error(`The "CLIENT_TOKEN" provided is not of at least length 72.
|
||||||
This is probably an invalid token unless Discord updated their token policy. Please provide a valid token.`)
|
This is probably an invalid token unless Discord updated their token policy. Please provide a valid token.`)
|
||||||
|
|
||||||
// validate IPv4 address found in environment variables
|
// validate IPv4 address found in environment variables
|
||||||
if (name.endsWith("_IP") || name.endsWith("_ADDRESS"))
|
if ((name.endsWith("_IP") || name.endsWith("_ADDRESS")) && !ipValidate.test(value))
|
||||||
if (!ipValidate.test(value))
|
|
||||||
throw new Error(`Environment variable ${name} does not follow IPv4 formatting.`)
|
throw new Error(`Environment variable ${name} does not follow IPv4 formatting.`)
|
||||||
|
|
||||||
// return env variable
|
// return env variable
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
import type { ClientEvents, Awaitable, Client } from 'discord.js'
|
import type { ClientEvents, Awaitable, Client } from 'discord.js'
|
||||||
import { Ollama } from 'ollama'
|
import { Ollama } from 'ollama'
|
||||||
import { Queue } from '../queues/queue.js'
|
import { Queue } from '../components/index.js'
|
||||||
|
|
||||||
// Export events through here to reduce amount of imports
|
// Export events through here to reduce amount of imports
|
||||||
export { Events } from 'discord.js'
|
export { Events } from 'discord.js'
|
||||||
@@ -24,6 +24,7 @@ export type ChatParams = {
|
|||||||
* Format for the messages to be stored when communicating when the bot
|
* Format for the messages to be stored when communicating when the bot
|
||||||
* @param role either assistant, user, or system
|
* @param role either assistant, user, or system
|
||||||
* @param content string of the message the user or assistant provided
|
* @param content string of the message the user or assistant provided
|
||||||
|
* @param images array of images that the user or assistant provided
|
||||||
*/
|
*/
|
||||||
export type UserMessage = {
|
export type UserMessage = {
|
||||||
role: string,
|
role: string,
|
||||||
@@ -33,12 +34,18 @@ export type UserMessage = {
|
|||||||
|
|
||||||
// Event properties
|
// Event properties
|
||||||
export interface EventProps {
|
export interface EventProps {
|
||||||
client: Client
|
client: Client,
|
||||||
log: LogMethod
|
log: LogMethod,
|
||||||
msgHist: Queue<UserMessage>
|
msgHist: Queue<UserMessage>,
|
||||||
ollama: Ollama,
|
ollama: Ollama,
|
||||||
defaultModel: String
|
defaultModel: String
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Format for the callback function tied to an event
|
||||||
|
* @param props the properties of the event
|
||||||
|
* @param args the arguments of the event
|
||||||
|
*/
|
||||||
export type EventCallback<T extends EventKeys> = (
|
export type EventCallback<T extends EventKeys> = (
|
||||||
props: EventProps,
|
props: EventProps,
|
||||||
...args: ClientEvents[T]
|
...args: ClientEvents[T]
|
||||||
@@ -50,6 +57,12 @@ export interface Event<T extends EventKeys = EventKeys> {
|
|||||||
callback: EventCallback<T>
|
callback: EventCallback<T>
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Method to create an event object
|
||||||
|
* @param key type of event
|
||||||
|
* @param callback function to run when event is triggered
|
||||||
|
* @returns event object
|
||||||
|
*/
|
||||||
export function event<T extends EventKeys>(key: T, callback: EventCallback<T>): Event<T> {
|
export function event<T extends EventKeys>(key: T, callback: EventCallback<T>): Event<T> {
|
||||||
return { key, callback }
|
return { key, callback }
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -64,7 +64,7 @@ export async function clearChannelInfo(filename: string, channel: TextChannel, u
|
|||||||
* @param user the user's name
|
* @param user the user's name
|
||||||
* @param messages their messages
|
* @param messages their messages
|
||||||
*/
|
*/
|
||||||
export async function openChannelInfo(filename: string, channel: TextChannel | ThreadChannel, user: string, messages: UserMessage[] = []): Promise<void> {
|
export async function openChannelInfo(this: any, filename: string, channel: TextChannel | ThreadChannel, user: string, messages: UserMessage[] = []): Promise<void> {
|
||||||
const fullFileName = `data/${filename}-${user}.json`
|
const fullFileName = `data/${filename}-${user}.json`
|
||||||
if (fs.existsSync(fullFileName)) {
|
if (fs.existsSync(fullFileName)) {
|
||||||
fs.readFile(fullFileName, 'utf8', (error, data) => {
|
fs.readFile(fullFileName, 'utf8', (error, data) => {
|
||||||
@@ -80,7 +80,14 @@ export async function openChannelInfo(filename: string, channel: TextChannel | T
|
|||||||
}
|
}
|
||||||
})
|
})
|
||||||
} else { // file doesn't exist, create it
|
} else { // file doesn't exist, create it
|
||||||
const object: Configuration = JSON.parse(`{ \"id\": \"${channel?.id}\", \"name\": \"${channel?.name}\", \"user\": \"${user}\", \"messages\": []}`)
|
const object: Configuration = JSON.parse(
|
||||||
|
`{
|
||||||
|
\"id\": \"${channel?.id}\",
|
||||||
|
\"name\": \"${channel?.name}\",
|
||||||
|
\"user\": \"${user}\",
|
||||||
|
\"messages\": []
|
||||||
|
}`
|
||||||
|
)
|
||||||
|
|
||||||
const directory = path.dirname(fullFileName)
|
const directory = path.dirname(fullFileName)
|
||||||
if (!fs.existsSync(directory))
|
if (!fs.existsSync(directory))
|
||||||
@@ -88,7 +95,7 @@ export async function openChannelInfo(filename: string, channel: TextChannel | T
|
|||||||
|
|
||||||
// only creating it, no need to add anything
|
// only creating it, no need to add anything
|
||||||
fs.writeFileSync(fullFileName, JSON.stringify(object, null, 2))
|
fs.writeFileSync(fullFileName, JSON.stringify(object, null, 2))
|
||||||
console.log(`[Util: openChannelInfo] Created '${fullFileName}' in working directory`)
|
console.log(`[Util: ${this.name}] Created '${fullFileName}' in working directory`)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -10,7 +10,7 @@ import path from 'path'
|
|||||||
* @param value new value to assign
|
* @param value new value to assign
|
||||||
*/
|
*/
|
||||||
// add type of change (server, user)
|
// add type of change (server, user)
|
||||||
export function openConfig(filename: string, key: string, value: any) {
|
export function openConfig(this: any, filename: string, key: string, value: any) {
|
||||||
const fullFileName = `data/${filename}`
|
const fullFileName = `data/${filename}`
|
||||||
|
|
||||||
// check if the file exists, if not then make the config file
|
// check if the file exists, if not then make the config file
|
||||||
@@ -41,7 +41,7 @@ export function openConfig(filename: string, key: string, value: any) {
|
|||||||
fs.mkdirSync(directory, { recursive: true })
|
fs.mkdirSync(directory, { recursive: true })
|
||||||
|
|
||||||
fs.writeFileSync(`data/${filename}`, JSON.stringify(object, null, 2))
|
fs.writeFileSync(`data/${filename}`, JSON.stringify(object, null, 2))
|
||||||
console.log(`[Util: openConfig] Created '${filename}' in working directory`)
|
console.log(`[Util: ${this.name}] Created '${filename}' in working directory`)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -1,11 +1,10 @@
|
|||||||
import { ChatResponse } from "ollama"
|
import { ChatResponse, AbortableAsyncIterator } from "ollama"
|
||||||
import { ChatParams } from "../index.js"
|
import { ChatParams } from "../index.js"
|
||||||
import { AbortableAsyncIterator } from "ollama/src/utils.js"
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Method to query the Ollama client for async generation
|
* Method to query the Ollama client for async generation
|
||||||
* @param params
|
* @param params
|
||||||
* @returns Asyn
|
* @returns AsyncIterator<ChatResponse> generated by the Ollama client
|
||||||
*/
|
*/
|
||||||
export async function streamResponse(params: ChatParams): Promise<AbortableAsyncIterator<ChatResponse>> {
|
export async function streamResponse(params: ChatParams): Promise<AbortableAsyncIterator<ChatResponse>> {
|
||||||
return await params.ollama.chat({
|
return await params.ollama.chat({
|
||||||
|
|||||||
@@ -1,8 +1,7 @@
|
|||||||
import { Message, SendableChannels } from 'discord.js'
|
import { Message, SendableChannels } from 'discord.js'
|
||||||
import { ChatResponse, Ollama } from 'ollama'
|
import { ChatResponse, Ollama, AbortableAsyncIterator } from 'ollama'
|
||||||
import { ChatParams, UserMessage, streamResponse, blockResponse } from './index.js'
|
import { ChatParams, UserMessage, streamResponse, blockResponse } from './index.js'
|
||||||
import { Queue } from '../queues/queue.js'
|
import { Queue } from '../components/index.js'
|
||||||
import { AbortableAsyncIterator } from 'ollama/src/utils.js'
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Method to send replies as normal text on discord like any other user
|
* Method to send replies as normal text on discord like any other user
|
||||||
@@ -11,6 +10,7 @@ import { AbortableAsyncIterator } from 'ollama/src/utils.js'
|
|||||||
* @param msgHist message history between user and model
|
* @param msgHist message history between user and model
|
||||||
*/
|
*/
|
||||||
export async function normalMessage(
|
export async function normalMessage(
|
||||||
|
this: any,
|
||||||
message: Message,
|
message: Message,
|
||||||
ollama: Ollama,
|
ollama: Ollama,
|
||||||
model: string,
|
model: string,
|
||||||
@@ -40,7 +40,8 @@ export async function normalMessage(
|
|||||||
result = portion.message.content
|
result = portion.message.content
|
||||||
|
|
||||||
// new message block, wait for it to send and assign new block to respond.
|
// new message block, wait for it to send and assign new block to respond.
|
||||||
await channel.send("Creating new stream block...").then(sentMessage => { messageBlock = sentMessage })
|
await channel.send("Creating new stream block...")
|
||||||
|
.then(sentMessage => { messageBlock = sentMessage })
|
||||||
} else {
|
} else {
|
||||||
result += portion.message.content
|
result += portion.message.content
|
||||||
|
|
||||||
@@ -71,8 +72,8 @@ export async function normalMessage(
|
|||||||
} else // edit the 'generic' response to new message since <2000
|
} else // edit the 'generic' response to new message since <2000
|
||||||
sentMessage.edit(result)
|
sentMessage.edit(result)
|
||||||
}
|
}
|
||||||
} catch(error: any) {
|
} catch (error: any) {
|
||||||
console.log(`[Util: messageNormal] Error creating message: ${error.message}`)
|
console.log(`[Util: ${this.name}] Error creating message: ${error.message}`)
|
||||||
if (error.message.includes('try pulling it first'))
|
if (error.message.includes('try pulling it first'))
|
||||||
sentMessage.edit(`**Response generation failed.**\n\nReason: You do not have the ${model} downloaded. Ask an admin to pull it using the \`pull-model\` command.`)
|
sentMessage.edit(`**Response generation failed.**\n\nReason: You do not have the ${model} downloaded. Ask an admin to pull it using the \`pull-model\` command.`)
|
||||||
else
|
else
|
||||||
|
|||||||
@@ -34,7 +34,8 @@ describe('Commands Existence', () => {
|
|||||||
// test specific commands in the object
|
// test specific commands in the object
|
||||||
it('references specific commands', () => {
|
it('references specific commands', () => {
|
||||||
const commandsString = commands.map(e => e.name).join(', ')
|
const commandsString = commands.map(e => e.name).join(', ')
|
||||||
expect(commandsString).toBe('thread, private-thread, message-stream, toggle-chat, shutoff, modify-capacity, clear-user-channel-history, pull-model, switch-model, delete-model')
|
const expectedCommands = ['thread', 'private-thread', 'message-stream', 'toggle-chat', 'shutoff', 'modify-capacity', 'clear-user-channel-history', 'pull-model', 'switch-model', 'delete-model']
|
||||||
|
expect(commandsString).toBe(expectedCommands.join(', '))
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
|
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
import { describe, expect, it } from 'vitest'
|
import { describe, expect, it } from 'vitest'
|
||||||
import { Queue } from '../src/queues/queue.js'
|
import { Queue } from '../src/components/index.js'
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Queue test suite, tests the Queue class
|
* Queue test suite, tests the Queue class
|
||||||
@@ -8,7 +8,7 @@ import { Queue } from '../src/queues/queue.js'
|
|||||||
* @param fn function holding tests to run
|
* @param fn function holding tests to run
|
||||||
*/
|
*/
|
||||||
describe('Queue Structure', () => {
|
describe('Queue Structure', () => {
|
||||||
let queue= new Queue<string>()
|
let queue = new Queue<string>()
|
||||||
|
|
||||||
// test for queue creation
|
// test for queue creation
|
||||||
it('creates a new queue', () => {
|
it('creates a new queue', () => {
|
||||||
|
|||||||
Reference in New Issue
Block a user