Compare commits
6 Commits
v0.8.3
...
snyk-upgra
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
e07f10506a | ||
|
|
a5faca87aa | ||
|
|
4c96b3863a | ||
|
|
40783818b9 | ||
|
|
ed0d8600df | ||
|
|
03939ef562 |
2
.github/workflows/build.yml
vendored
2
.github/workflows/build.yml
vendored
@@ -33,6 +33,7 @@ jobs:
|
|||||||
echo CLIENT_TOKEN = ${{ secrets.BOT_TOKEN }} >> .env
|
echo CLIENT_TOKEN = ${{ secrets.BOT_TOKEN }} >> .env
|
||||||
echo OLLAMA_IP = ${{ secrets.OLLAMA_IP }} >> .env
|
echo OLLAMA_IP = ${{ secrets.OLLAMA_IP }} >> .env
|
||||||
echo OLLAMA_PORT = ${{ secrets.OLLAMA_PORT }} >> .env
|
echo OLLAMA_PORT = ${{ secrets.OLLAMA_PORT }} >> .env
|
||||||
|
echo MODEL = ${{ secrets.MODEL }} >> .env
|
||||||
echo REDIS_IP = ${{ secrets.REDIS_IP }} >> .env
|
echo REDIS_IP = ${{ secrets.REDIS_IP }} >> .env
|
||||||
echo REDIS_PORT = ${{ secrets.REDIS_PORT }} >> .env
|
echo REDIS_PORT = ${{ secrets.REDIS_PORT }} >> .env
|
||||||
|
|
||||||
@@ -61,6 +62,7 @@ jobs:
|
|||||||
echo CLIENT_TOKEN = ${{ secrets.BOT_TOKEN }} >> .env
|
echo CLIENT_TOKEN = ${{ secrets.BOT_TOKEN }} >> .env
|
||||||
echo OLLAMA_IP = ${{ secrets.OLLAMA_IP }} >> .env
|
echo OLLAMA_IP = ${{ secrets.OLLAMA_IP }} >> .env
|
||||||
echo OLLAMA_PORT = ${{ secrets.OLLAMA_PORT }} >> .env
|
echo OLLAMA_PORT = ${{ secrets.OLLAMA_PORT }} >> .env
|
||||||
|
echo MODEL = ${{ secrets.MODEL }} >> .env
|
||||||
echo REDIS_IP = ${{ secrets.REDIS_IP }} >> .env
|
echo REDIS_IP = ${{ secrets.REDIS_IP }} >> .env
|
||||||
echo REDIS_PORT = ${{ secrets.REDIS_PORT }} >> .env
|
echo REDIS_PORT = ${{ secrets.REDIS_PORT }} >> .env
|
||||||
|
|
||||||
|
|||||||
1
.github/workflows/coverage.yml
vendored
1
.github/workflows/coverage.yml
vendored
@@ -30,6 +30,7 @@ jobs:
|
|||||||
echo CLIENT_TOKEN = ${{ secrets.BOT_TOKEN }} >> .env
|
echo CLIENT_TOKEN = ${{ secrets.BOT_TOKEN }} >> .env
|
||||||
echo OLLAMA_IP = ${{ secrets.OLLAMA_IP }} >> .env
|
echo OLLAMA_IP = ${{ secrets.OLLAMA_IP }} >> .env
|
||||||
echo OLLAMA_PORT = ${{ secrets.OLLAMA_PORT }} >> .env
|
echo OLLAMA_PORT = ${{ secrets.OLLAMA_PORT }} >> .env
|
||||||
|
echo MODEL = ${{ secrets.MODEL }} >> .env
|
||||||
echo REDIS_IP = ${{ secrets.REDIS_IP }} >> .env
|
echo REDIS_IP = ${{ secrets.REDIS_IP }} >> .env
|
||||||
echo REDIS_PORT = ${{ secrets.REDIS_PORT }} >> .env
|
echo REDIS_PORT = ${{ secrets.REDIS_PORT }} >> .env
|
||||||
|
|
||||||
|
|||||||
133
.github/workflows/deploy.yml
vendored
Normal file
133
.github/workflows/deploy.yml
vendored
Normal file
@@ -0,0 +1,133 @@
|
|||||||
|
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 MODEL = ${{ secrets.MODEL }} >> .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
|
|
||||||
1
.github/workflows/test.yml
vendored
1
.github/workflows/test.yml
vendored
@@ -41,6 +41,7 @@ jobs:
|
|||||||
echo CLIENT_TOKEN = ${{ secrets.BOT_TOKEN }} >> .env
|
echo CLIENT_TOKEN = ${{ secrets.BOT_TOKEN }} >> .env
|
||||||
echo OLLAMA_IP = ${{ secrets.OLLAMA_IP }} >> .env
|
echo OLLAMA_IP = ${{ secrets.OLLAMA_IP }} >> .env
|
||||||
echo OLLAMA_PORT = ${{ secrets.OLLAMA_PORT }} >> .env
|
echo OLLAMA_PORT = ${{ secrets.OLLAMA_PORT }} >> .env
|
||||||
|
echo MODEL = ${{ secrets.MODEL }} >> .env
|
||||||
echo REDIS_IP = ${{ secrets.REDIS_IP }} >> .env
|
echo REDIS_IP = ${{ secrets.REDIS_IP }} >> .env
|
||||||
echo REDIS_PORT = ${{ secrets.REDIS_PORT }} >> .env
|
echo REDIS_PORT = ${{ secrets.REDIS_PORT }} >> .env
|
||||||
|
|
||||||
|
|||||||
@@ -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,11 +7,12 @@ 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.3
|
image: kevinthedang/discord-ollama:0.8.4
|
||||||
environment:
|
environment:
|
||||||
CLIENT_TOKEN: ${CLIENT_TOKEN}
|
CLIENT_TOKEN: ${CLIENT_TOKEN}
|
||||||
OLLAMA_IP: ${OLLAMA_IP}
|
OLLAMA_IP: ${OLLAMA_IP}
|
||||||
OLLAMA_PORT: ${OLLAMA_PORT}
|
OLLAMA_PORT: ${OLLAMA_PORT}
|
||||||
|
MODEL: ${MODEL}
|
||||||
REDIS_IP: ${REDIS_IP}
|
REDIS_IP: ${REDIS_IP}
|
||||||
REDIS_PORT: ${REDIS_PORT}
|
REDIS_PORT: ${REDIS_PORT}
|
||||||
networks:
|
networks:
|
||||||
|
|||||||
1253
package-lock.json
generated
1253
package-lock.json
generated
File diff suppressed because it is too large
Load Diff
18
package.json
18
package.json
@@ -1,6 +1,6 @@
|
|||||||
{
|
{
|
||||||
"name": "discord-ollama",
|
"name": "discord-ollama",
|
||||||
"version": "0.8.3",
|
"version": "0.8.4",
|
||||||
"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.17.3",
|
"discord.js": "^14.18.0",
|
||||||
"dotenv": "^16.4.7",
|
"dotenv": "^16.5.0",
|
||||||
"ollama": "^0.5.11",
|
"ollama": "^0.5.14",
|
||||||
"redis": "^4.7.0"
|
"redis": "^4.7.0"
|
||||||
},
|
},
|
||||||
"devDependencies": {
|
"devDependencies": {
|
||||||
"@types/node": "^22.10.2",
|
"@types/node": "^22.13.14",
|
||||||
"@vitest/coverage-v8": "^2.1.8",
|
"@vitest/coverage-v8": "^3.0.9",
|
||||||
"ts-node": "^10.9.2",
|
"ts-node": "^10.9.2",
|
||||||
"tsx": "^4.19.2",
|
"tsx": "^4.19.3",
|
||||||
"typescript": "^5.7.2",
|
"typescript": "^5.8.2",
|
||||||
"vitest": "^2.1.4"
|
"vitest": "^3.0.4"
|
||||||
},
|
},
|
||||||
"type": "module",
|
"type": "module",
|
||||||
"engines": {
|
"engines": {
|
||||||
|
|||||||
@@ -6,15 +6,14 @@ import commands from '../commands/index.js'
|
|||||||
* @param interaction the interaction received from the server
|
* @param interaction the interaction received from the server
|
||||||
*/
|
*/
|
||||||
export default event(Events.InteractionCreate, async ({ log, client }, interaction) => {
|
export default event(Events.InteractionCreate, async ({ log, client }, interaction) => {
|
||||||
if (!interaction.isCommand() || !interaction.isChatInputCommand()) return
|
if (!interaction.isCommand() || !interaction.isChatInputCommand()) return
|
||||||
|
|
||||||
log(`Interaction called \'${interaction.commandName}\' from ${interaction.user.tag}.`)
|
log(`Interaction called \'${interaction.commandName}\' from ${interaction.user.tag}.`)
|
||||||
|
|
||||||
// ensure command exists, otherwise kill event
|
// ensure command exists, otherwise kill event
|
||||||
const command = commands.find(command => command.name === interaction.commandName)
|
const command = commands.find(command => command.name === interaction.commandName)
|
||||||
if (!command) return
|
if (!command) return
|
||||||
|
|
||||||
// the command exists, execute it
|
// the command exists, execute it
|
||||||
command.run(client, interaction)
|
command.run(client, interaction)
|
||||||
}
|
})
|
||||||
)
|
|
||||||
@@ -16,129 +16,129 @@ export default event(Events.MessageCreate, async ({ log, msgHist, ollama, client
|
|||||||
let cleanedMessage = clean(message.content, clientId)
|
let cleanedMessage = clean(message.content, clientId)
|
||||||
log(`Message \"${cleanedMessage}\" from ${message.author.tag} in channel/thread ${message.channelId}.`)
|
log(`Message \"${cleanedMessage}\" from ${message.author.tag} in channel/thread ${message.channelId}.`)
|
||||||
|
|
||||||
// Do not respond if bot talks in the chat
|
// Do not respond if bot talks in the chat
|
||||||
if (message.author.username === message.client.user.username) return
|
if (message.author.username === message.client.user.username) return
|
||||||
|
|
||||||
// Only respond if message mentions the bot
|
// Only respond if message mentions the bot
|
||||||
if (!message.mentions.has(clientId)) return
|
if (!message.mentions.has(clientId)) return
|
||||||
|
|
||||||
// default stream to false
|
// default stream to false
|
||||||
let shouldStream = false
|
let shouldStream = false
|
||||||
|
|
||||||
// Params for Preferences Fetching
|
// Params for Preferences Fetching
|
||||||
const maxRetries = 3
|
const maxRetries = 3
|
||||||
const delay = 1000 // in millisecons
|
const delay = 1000 // in millisecons
|
||||||
|
|
||||||
try {
|
try {
|
||||||
// Retrieve Server/Guild Preferences
|
// Retrieve Server/Guild Preferences
|
||||||
let attempt = 0
|
let attempt = 0
|
||||||
while (attempt < maxRetries) {
|
while (attempt < maxRetries) {
|
||||||
try {
|
try {
|
||||||
await new Promise((resolve, reject) => {
|
await new Promise((resolve, reject) => {
|
||||||
getServerConfig(`${message.guildId}-config.json`, (config) => {
|
getServerConfig(`${message.guildId}-config.json`, (config) => {
|
||||||
// check if config.json exists
|
// check if config.json exists
|
||||||
if (config === undefined) {
|
if (config === undefined) {
|
||||||
// Allowing chat options to be available
|
// Allowing chat options to be available
|
||||||
openConfig(`${message.guildId}-config.json`, 'toggle-chat', true)
|
openConfig(`${message.guildId}-config.json`, 'toggle-chat', true)
|
||||||
reject(new Error('Failed to locate or create Server Preferences\n\nPlease try chatting again...'))
|
reject(new Error('Failed to locate or create Server Preferences\n\nPlease try chatting again...'))
|
||||||
}
|
}
|
||||||
|
|
||||||
// check if chat is disabled
|
|
||||||
else if (!config.options['toggle-chat'])
|
|
||||||
reject(new Error('Admin(s) have disabled chat features.\n\n Please contact your server\'s admin(s).'))
|
|
||||||
else
|
|
||||||
resolve(config)
|
|
||||||
})
|
|
||||||
})
|
|
||||||
break // successful
|
|
||||||
} catch (error) {
|
|
||||||
++attempt
|
|
||||||
if (attempt < maxRetries) {
|
|
||||||
log(`Attempt ${attempt} failed for Server Preferences. Retrying in ${delay}ms...`)
|
|
||||||
await new Promise(ret => setTimeout(ret, delay))
|
|
||||||
} else
|
|
||||||
throw new Error(`Could not retrieve Server Preferences, please try chatting again...`)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Reset attempts for User preferences
|
|
||||||
attempt = 0
|
|
||||||
let userConfig: UserConfig | undefined
|
|
||||||
|
|
||||||
while (attempt < maxRetries) {
|
|
||||||
try {
|
|
||||||
// Retrieve User Preferences
|
|
||||||
userConfig = await new Promise((resolve, reject) => {
|
|
||||||
getUserConfig(`${message.author.username}-config.json`, (config) => {
|
|
||||||
if (config === undefined) {
|
|
||||||
openConfig(`${message.author.username}-config.json`, 'message-style', false)
|
|
||||||
openConfig(`${message.author.username}-config.json`, 'switch-model', defaultModel)
|
|
||||||
reject(new Error('No User Preferences is set up.\n\nCreating preferences file with \`message-style\` set as \`false\` for regular message style.\nPlease try chatting again.'))
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// check if there is a set capacity in config
|
|
||||||
else if (typeof config.options['modify-capacity'] !== 'number')
|
|
||||||
log(`Capacity is undefined, using default capacity of ${msgHist.capacity}.`)
|
|
||||||
else if (config.options['modify-capacity'] === msgHist.capacity)
|
|
||||||
log(`Capacity matches config as ${msgHist.capacity}, no changes made.`)
|
|
||||||
else {
|
|
||||||
log(`New Capacity found. Setting Context Capacity to ${config.options['modify-capacity']}.`)
|
|
||||||
msgHist.capacity = config.options['modify-capacity']
|
|
||||||
}
|
|
||||||
|
|
||||||
// set stream state
|
|
||||||
shouldStream = config.options['message-stream'] as boolean || false
|
|
||||||
|
|
||||||
if (typeof config.options['switch-model'] !== 'string')
|
|
||||||
reject(new Error(`No Model was set. Please set a model by running \`/switch-model <model of choice>\`.\n\nIf you do not have any models. Run \`/pull-model <model name>\`.`))
|
|
||||||
|
|
||||||
|
// check if chat is disabled
|
||||||
|
else if (!config.options['toggle-chat'])
|
||||||
|
reject(new Error('Admin(s) have disabled chat features.\n\n Please contact your server\'s admin(s).'))
|
||||||
|
else
|
||||||
resolve(config)
|
resolve(config)
|
||||||
})
|
|
||||||
})
|
})
|
||||||
break // successful
|
})
|
||||||
} catch (error) {
|
break // successful
|
||||||
++attempt
|
} catch (error) {
|
||||||
if (attempt < maxRetries) {
|
++attempt
|
||||||
log(`Attempt ${attempt} failed for User Preferences. Retrying in ${delay}ms...`)
|
if (attempt < maxRetries) {
|
||||||
await new Promise(ret => setTimeout(ret, delay))
|
log(`Attempt ${attempt} failed for Server Preferences. Retrying in ${delay}ms...`)
|
||||||
} else
|
await new Promise(ret => setTimeout(ret, delay))
|
||||||
throw new Error(`Could not retrieve User Preferences, please try chatting again...`)
|
} else
|
||||||
}
|
throw new Error(`Could not retrieve Server Preferences, please try chatting again...`)
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// need new check for "open/active" threads/channels here!
|
// Reset attempts for User preferences
|
||||||
let chatMessages: UserMessage[] = await new Promise((resolve) => {
|
attempt = 0
|
||||||
// set new queue to modify
|
let userConfig: UserConfig | undefined
|
||||||
|
|
||||||
|
while (attempt < maxRetries) {
|
||||||
|
try {
|
||||||
|
// Retrieve User Preferences
|
||||||
|
userConfig = await new Promise((resolve, reject) => {
|
||||||
|
getUserConfig(`${message.author.username}-config.json`, (config) => {
|
||||||
|
if (config === undefined) {
|
||||||
|
openConfig(`${message.author.username}-config.json`, 'message-style', false)
|
||||||
|
openConfig(`${message.author.username}-config.json`, 'switch-model', defaultModel)
|
||||||
|
reject(new Error('No User Preferences is set up.\n\nCreating preferences file with \`message-style\` set as \`false\` for regular message style.\nPlease try chatting again.'))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// check if there is a set capacity in config
|
||||||
|
else if (typeof config.options['modify-capacity'] !== 'number')
|
||||||
|
log(`Capacity is undefined, using default capacity of ${msgHist.capacity}.`)
|
||||||
|
else if (config.options['modify-capacity'] === msgHist.capacity)
|
||||||
|
log(`Capacity matches config as ${msgHist.capacity}, no changes made.`)
|
||||||
|
else {
|
||||||
|
log(`New Capacity found. Setting Context Capacity to ${config.options['modify-capacity']}.`)
|
||||||
|
msgHist.capacity = config.options['modify-capacity']
|
||||||
|
}
|
||||||
|
|
||||||
|
// set stream state
|
||||||
|
shouldStream = config.options['message-stream'] as boolean || false
|
||||||
|
|
||||||
|
if (typeof config.options['switch-model'] !== 'string')
|
||||||
|
reject(new Error(`No Model was set. Please set a model by running \`/switch-model <model of choice>\`.\n\nIf you do not have any models. Run \`/pull-model <model name>\`.`))
|
||||||
|
|
||||||
|
resolve(config)
|
||||||
|
})
|
||||||
|
})
|
||||||
|
break // successful
|
||||||
|
} catch (error) {
|
||||||
|
++attempt
|
||||||
|
if (attempt < maxRetries) {
|
||||||
|
log(`Attempt ${attempt} failed for User Preferences. Retrying in ${delay}ms...`)
|
||||||
|
await new Promise(ret => setTimeout(ret, delay))
|
||||||
|
} else
|
||||||
|
throw new Error(`Could not retrieve User Preferences, please try chatting again...`)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// need new check for "open/active" threads/channels here!
|
||||||
|
let chatMessages: UserMessage[] = await new Promise((resolve) => {
|
||||||
|
// set new queue to modify
|
||||||
|
getChannelInfo(`${message.channelId}-${message.author.username}.json`, (channelInfo) => {
|
||||||
|
if (channelInfo?.messages)
|
||||||
|
resolve(channelInfo.messages)
|
||||||
|
else {
|
||||||
|
log(`Channel/Thread ${message.channel}-${message.author.username} does not exist. File will be created shortly...`)
|
||||||
|
resolve([])
|
||||||
|
}
|
||||||
|
})
|
||||||
|
})
|
||||||
|
|
||||||
|
if (chatMessages.length === 0) {
|
||||||
|
chatMessages = await new Promise((resolve, reject) => {
|
||||||
|
openChannelInfo(message.channelId,
|
||||||
|
message.channel as TextChannel,
|
||||||
|
message.author.tag
|
||||||
|
)
|
||||||
getChannelInfo(`${message.channelId}-${message.author.username}.json`, (channelInfo) => {
|
getChannelInfo(`${message.channelId}-${message.author.username}.json`, (channelInfo) => {
|
||||||
if (channelInfo?.messages)
|
if (channelInfo?.messages)
|
||||||
resolve(channelInfo.messages)
|
resolve(channelInfo.messages)
|
||||||
else {
|
else {
|
||||||
log(`Channel/Thread ${message.channel}-${message.author.username} does not exist. File will be created shortly...`)
|
log(`Channel/Thread ${message.channel}-${message.author.username} does not exist. File will be created shortly...`)
|
||||||
resolve([])
|
reject(new Error(`Failed to find ${message.author.username}'s history. Try chatting again.`))
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
|
}
|
||||||
|
|
||||||
if (chatMessages.length === 0) {
|
if (!userConfig)
|
||||||
chatMessages = await new Promise((resolve, reject) => {
|
throw new Error(`Failed to initialize User Preference for **${message.author.username}**.\n\nIt's likely you do not have a model set. Please use the \`switch-model\` command to do that.`)
|
||||||
openChannelInfo(message.channelId,
|
|
||||||
message.channel as TextChannel,
|
|
||||||
message.author.tag
|
|
||||||
)
|
|
||||||
getChannelInfo(`${message.channelId}-${message.author.username}.json`, (channelInfo) => {
|
|
||||||
if (channelInfo?.messages)
|
|
||||||
resolve(channelInfo.messages)
|
|
||||||
else {
|
|
||||||
log(`Channel/Thread ${message.channel}-${message.author.username} does not exist. File will be created shortly...`)
|
|
||||||
reject(new Error(`Failed to find ${message.author.username}'s history. Try chatting again.`))
|
|
||||||
}
|
|
||||||
})
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!userConfig)
|
|
||||||
throw new Error(`Failed to initialize User Preference for **${message.author.username}**.\n\nIt's likely you do not have a model set. Please use the \`switch-model\` command to do that.`)
|
|
||||||
|
|
||||||
// get message attachment if exists
|
// get message attachment if exists
|
||||||
const attachment = message.attachments.first()
|
const attachment = message.attachments.first()
|
||||||
@@ -148,47 +148,46 @@ export default event(Events.MessageCreate, async ({ log, msgHist, ollama, client
|
|||||||
cleanedMessage += await getTextFileAttachmentData(attachment)
|
cleanedMessage += await getTextFileAttachmentData(attachment)
|
||||||
else if (attachment)
|
else if (attachment)
|
||||||
messageAttachment = await getAttachmentData(attachment)
|
messageAttachment = await getAttachmentData(attachment)
|
||||||
|
|
||||||
const model: string = userConfig.options['switch-model']
|
const model: string = userConfig.options['switch-model']
|
||||||
|
|
||||||
// set up new queue
|
// set up new queue
|
||||||
msgHist.setQueue(chatMessages)
|
msgHist.setQueue(chatMessages)
|
||||||
|
|
||||||
// check if we can push, if not, remove oldest
|
// check if we can push, if not, remove oldest
|
||||||
while (msgHist.size() >= msgHist.capacity) msgHist.dequeue()
|
while (msgHist.size() >= msgHist.capacity) msgHist.dequeue()
|
||||||
|
|
||||||
// push user response before ollama query
|
// push user response before ollama query
|
||||||
msgHist.enqueue({
|
msgHist.enqueue({
|
||||||
role: 'user',
|
role: 'user',
|
||||||
content: cleanedMessage,
|
content: cleanedMessage,
|
||||||
images: messageAttachment || []
|
images: messageAttachment || []
|
||||||
})
|
})
|
||||||
|
|
||||||
// response string for ollama to put its response
|
// response string for ollama to put its response
|
||||||
const response: string = await normalMessage(message, ollama, model, msgHist, shouldStream)
|
const response: string = await normalMessage(message, ollama, model, msgHist, shouldStream)
|
||||||
|
|
||||||
// If something bad happened, remove user query and stop
|
// If something bad happened, remove user query and stop
|
||||||
if (response == undefined) { msgHist.pop(); return }
|
if (response == undefined) { msgHist.pop(); return }
|
||||||
|
|
||||||
// if queue is full, remove the oldest message
|
// if queue is full, remove the oldest message
|
||||||
while (msgHist.size() >= msgHist.capacity) msgHist.dequeue()
|
while (msgHist.size() >= msgHist.capacity) msgHist.dequeue()
|
||||||
|
|
||||||
// successful query, save it in context history
|
// successful query, save it in context history
|
||||||
msgHist.enqueue({
|
msgHist.enqueue({
|
||||||
role: 'assistant',
|
role: 'assistant',
|
||||||
content: response,
|
content: response,
|
||||||
images: messageAttachment || []
|
images: messageAttachment || []
|
||||||
})
|
})
|
||||||
|
|
||||||
// only update the json on success
|
// only update the json on success
|
||||||
openChannelInfo(message.channelId,
|
openChannelInfo(message.channelId,
|
||||||
message.channel as TextChannel,
|
message.channel as TextChannel,
|
||||||
message.author.tag,
|
message.author.tag,
|
||||||
msgHist.getItems()
|
msgHist.getItems()
|
||||||
)
|
)
|
||||||
} catch (error: any) {
|
} catch (error: any) {
|
||||||
msgHist.pop() // remove message because of failure
|
msgHist.pop() // remove message because of failure
|
||||||
message.reply(`**Error Occurred:**\n\n**Reason:** *${error.message}*`)
|
message.reply(`**Error Occurred:**\n\n**Reason:** *${error.message}*`)
|
||||||
}
|
|
||||||
}
|
}
|
||||||
)
|
})
|
||||||
@@ -4,15 +4,14 @@ import commands from '../commands/index.js'
|
|||||||
|
|
||||||
// Log when the bot successfully logs in and export it
|
// Log when the bot successfully logs in and export it
|
||||||
export default event(Events.ClientReady, ({ log }, client) => {
|
export default event(Events.ClientReady, ({ log }, client) => {
|
||||||
// Register the commands associated with the bot upon loggin in
|
// Register the commands associated with the bot upon loggin in
|
||||||
registerCommands(client, commands)
|
registerCommands(client, commands)
|
||||||
|
|
||||||
// set status of the bot
|
// set status of the bot
|
||||||
client.user.setActivity({
|
client.user.setActivity({
|
||||||
name: 'Powered by Ollama',
|
name: 'Powered by Ollama',
|
||||||
type: ActivityType.Custom
|
type: ActivityType.Custom
|
||||||
})
|
})
|
||||||
|
|
||||||
log(`Logged in as ${client.user.username}.`)
|
log(`Logged in as ${client.user.username}.`)
|
||||||
}
|
})
|
||||||
)
|
|
||||||
@@ -6,36 +6,35 @@ import fs from 'fs'
|
|||||||
* Event to remove the associated .json file for a thread once deleted
|
* Event to remove the associated .json file for a thread once deleted
|
||||||
*/
|
*/
|
||||||
export default event(Events.ThreadDelete, async ({ log }, thread: ThreadChannel) => {
|
export default event(Events.ThreadDelete, async ({ log }, thread: ThreadChannel) => {
|
||||||
// iterate through every guild member in the thread and delete their history, except the bot
|
// iterate through every guild member in the thread and delete their history, except the bot
|
||||||
try {
|
try {
|
||||||
log(`Number of User Guild Members in Thread being deleted: ${thread.memberCount!! - 1}`)
|
log(`Number of User Guild Members in Thread being deleted: ${thread.memberCount!! - 1}`)
|
||||||
const dirPath = 'data/'
|
const dirPath = 'data/'
|
||||||
|
|
||||||
// read all files in data/
|
// read all files in data/
|
||||||
fs.readdir(dirPath, (error, files) => {
|
fs.readdir(dirPath, (error, files) => {
|
||||||
if (error) {
|
if (error) {
|
||||||
log(`Error reading directory ${dirPath}`, error)
|
log(`Error reading directory ${dirPath}`, error)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// filter files by thread id being deleted
|
// filter files by thread id being deleted
|
||||||
const filesToDiscard = files.filter(
|
const filesToDiscard = files.filter(
|
||||||
file => file.startsWith(`${thread.id}-`) &&
|
file => file.startsWith(`${thread.id}-`) &&
|
||||||
file.endsWith('.json'))
|
file.endsWith('.json'))
|
||||||
|
|
||||||
// remove files by unlinking
|
// remove files by unlinking
|
||||||
filesToDiscard.forEach(file => {
|
filesToDiscard.forEach(file => {
|
||||||
const filePath = dirPath + file
|
const filePath = dirPath + file
|
||||||
fs.unlink(filePath, error => {
|
fs.unlink(filePath, error => {
|
||||||
if (error)
|
if (error)
|
||||||
log(`Error deleting file ${filePath}`, error)
|
log(`Error deleting file ${filePath}`, error)
|
||||||
else
|
else
|
||||||
log(`Successfully deleted ${filePath} thread information`)
|
log(`Successfully deleted ${filePath} thread information`)
|
||||||
})
|
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
} catch (error) {
|
})
|
||||||
log(`Issue deleting user history files from ${thread.id}`)
|
} catch (error) {
|
||||||
}
|
log(`Issue deleting user history files from ${thread.id}`)
|
||||||
}
|
}
|
||||||
)
|
})
|
||||||
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user