Upload 13 files

.dockerignore

@@ -0,0 +1,5 @@
+package-lock.json
+pnpm-lock.yaml
+pnpm-lock.yml
+node_modules/
+.env

.env

@@ -0,0 +1,57 @@
+# Discord bot token
+TOKEN=MTIzNTE0MjgyODE5MTkxMTk2Ng.GTigmW.AkCl8B8TWqsrbRrgvuu1QWO7vgr-GLWN51qMW8
+
+# What language model to use, orca is one of the lower-end models that doesn't require as much computer power as llama2
+MODEL=tinydolphin
+
+# Ollama URL (if you want to use multiple, separate them by commas)
+OLLAMA=https://mahiatlinux-ollama-server.hf.space
+
+# What Discord channels to enable it in (by ID)
+CHANNELS=1235145472470159391
+
+# System message that the language model can understand
+# Feel free to change this
+#SYSTEM="The current date and time is <date>.
+
+Basic markdown is supported.
+Bold: **bold text here**
+Italics: _italic text here_
+Underlined: __underlined text here__
+Strikethrough: ~~strikethrough text here~~
+Spoiler: ||spoiler text here||
+Block quotes: Start the line with a > followed by a space, e.g
+> Hello there
+
+Inline code blocks are supported by surrounding text in backticks, e.g `print('Hello');`, block code is supported by surrounding text in three backticks, e.g ```print('Hello');```.
+Surround code that is produced in code blocks. Use a code block with three backticks if the code has multiple lines, otherwise use an inline code block with one backtick.
+
+Links are supported by wrapping the text in square brackets and the link in parenthesis, e.g [Example](https://example.com)
+
+Lists are supported by starting the line with a dash followed by a space, e.g - List
+Numbered lists are supported by starting the line with a number followed by a dot and a space, e.g 1. List.
+Images, links, tables, LaTeX, and anything else is not supported.
+
+If you need to use the symbols >, |, _, *, ~, @, #, :, `, put a backslash before them to escape them.
+
+If the user is chatting casually, your responses should be only a few sentences, unless they are asking for help or a question.
+Don't use unicode emoji unless needed."
+
+# Use the system message above? (true/false)
+USE_SYSTEM=true
+
+# Use the model's system message? (true/false) If both are specified, model system message will be first
+USE_MODEL_SYSTEM=true
+
+# Require users to mention the bot to interact with it? (true/false)
+REQUIRES_MENTION=true
+
+# Whether to show a message at the start of a conversation
+SHOW_START_OF_CONVERSATION=true
+
+# Whether to use a random Ollama server or use the first available one
+RANDOM_SERVER=false
+
+# Whether to add a message before the first prompt of the conversation
+INITIAL_PROMPT=""
+USE_INITIAL_PROMPT=false

.eslintrc.json

@@ -0,0 +1,53 @@
+{
+	"extends": "eslint:recommended",
+	"env": {
+		"node": true,
+		"es6": true
+	},
+	"parserOptions": {
+		"ecmaVersion": 2021,
+		"sourceType": "module"
+	},
+	"rules": {
+		"arrow-spacing": ["warn", { "before": true, "after": true }],
+		"brace-style": ["error", "1tbs", { "allowSingleLine": true }],
+		"comma-dangle": ["error", "never"],
+		"comma-spacing": "error",
+		"comma-style": "error",
+		"curly": ["error", "multi-line", "consistent"],
+		"dot-location": ["error", "property"],
+		"handle-callback-err": "off",
+		"indent": ["error", "tab", { "SwitchCase": 1 }],
+		"keyword-spacing": "error",
+		"max-nested-callbacks": ["error", { "max": 4 }],
+		"max-statements-per-line": ["error", { "max": 2 }],
+		"no-console": "off",
+		"no-empty": "warn",
+		"no-empty-function": "error",
+		"no-floating-decimal": "error",
+		"no-inline-comments": "error",
+		"no-lonely-if": "error",
+		"no-multi-spaces": "error",
+		"no-multiple-empty-lines": ["error", { "max": 2, "maxEOF": 1, "maxBOF": 0 }],
+		"no-shadow": ["error", { "allow": ["err", "resolve", "reject"] }],
+		"no-trailing-spaces": ["error"],
+		"no-var": "error",
+		"object-curly-spacing": ["error", "always"],
+		"prefer-const": "error",
+		"quotes": ["error", "double"],
+		"semi": ["error", "always"],
+		"space-before-blocks": "error",
+		"space-before-function-paren": ["error", {
+			"anonymous": "never",
+			"named": "never",
+			"asyncArrow": "always"
+		}],
+		"space-in-parens": "error",
+		"space-infix-ops": "error",
+		"space-unary-ops": "error",
+		"spaced-comment": "error",
+		"yoda": "error",
+		"default-case-last": "error",
+		"switch-colon-spacing": ["error", {"after": true, "before": false}]
+	}
+}

.gitignore

@@ -0,0 +1,5 @@
+package-lock.json
+pnpm-lock.yaml
+pnpm-lock.yml
+node_modules/
+.env

Dockerfile

@@ -0,0 +1,7 @@
+FROM node:20
+
+COPY . .
+RUN npm i --omit=dev --no-package-lock
+USER node
+
+CMD ["node","./src/index.js"]

Makefile

@@ -0,0 +1,27 @@
+# As long as you have Make running on your machine you should be able to use this file.
+# make <command-name> runs a given command (e.g. make compose-up)
+# Command-names are given by starting a line without a tab and followed by a colon (i.e.':').
+# what the command runs is the line below the colon and that line must start with a tab of size 4.
+# Running make without a command after it will run the first command in the file.
+
+# starts the discord-ai-bot
+compose-up:
+	$(MAKE) setup_env && docker compose -p discord-ai up
+
+# Stops docker compose without removing the containers from the system.
+compose-stop:
+	docker compose  -p discord-ai stop
+
+# Stops docker compose and removes the containers from the system
+compose-down:
+	docker compose  -p discord-ai down
+
+#  Run the local node project with make and without docker
+local:
+	$(MAKE) setup_env && npm i && node ./src/index.js
+
+# This copies the .env.example (source) file to the .env (destination) file location
+# The -n or no clobber means it will not overwrite the .env file if it already exists.
+# The || : basically ignores the error code of the previous command and always succeeds.
+setup_env:
+	cp -n ./.env.example ./.env 2>/dev/null || :

README.md

@@ -1,10 +1,44 @@
----
-title: Test1234
-emoji: 📉
-colorFrom: pink
-colorTo: blue
-sdk: docker
-pinned: false
----
+<div align="center">
+    <h1><a href="#"></a>Discord AI Bot</h1>
+    <h3 align="center"><a href="#"></a>Discord bot to interact with <a href="https://github.com/jmorganca/ollama">Ollama</a> as a chatbot</h3>
+    <h3><a href="#"></a><img alt="Stars" src="https://img.shields.io/github/stars/mekb-turtle/discord-ai-bot?display_name=tag&style=for-the-badge" /></h3>
+    <h3><a href="#"></a><img alt="Discord chat with the bot" src="assets/screenshot.png" /></h3>
+</div>
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+### Archived
+I have decided to archive this project as I no longer have the time to maintain it. If you would like to take over the project, [please let me know](https://github.com/mekb-turtle).
+
+### Set-up instructions
+1. Install [Node.js](https://nodejs.org) (if you have a package manager, use that instead to install this)
+    - Make sure to install at least v14 of Node.js
+2. Install [Ollama](https://github.com/jmorganca/ollama) (ditto)
+3. Pull (download) a model, e.g `ollama pull orca` or `ollama pull llama2`
+4. Start Ollama by running `ollama serve`
+5. [Create a Discord bot](https://discord.com/developers/applications)
+    - Under Application » Bot
+        - Enable Message Content Intent
+        - Enable Server Members Intent (for replacing user mentions with the username)
+6. Invite the bot to a server
+    1. Go to Application » OAuth2 » URL Generator
+    2. Enable `bot`
+    3. Enable Send Messages, Read Messages/View Channels, and Read Message History
+    4. Under Generated URL, click Copy and paste the URL in your browser
+7. Rename `.env.example` to `.env` and edit the `.env` file
+    - You can get the token from Application » Bot » Token, **never share this with anyone**
+    - Make sure to change the model if you aren't using `orca`
+    - Ollama URL can be kept the same unless you have changed the port
+    - You can use multiple Ollama servers at the same time by separating the URLs with commas
+    - Set the channels to the channel ID, comma separated
+        1. In Discord, go to User Settings » Advanced, and enable Developer Mode
+        2. Right click on a channel you want to use, and click Copy Channel ID
+    - You can edit the system message the bot uses, or disable it entirely
+8. Start the bot with `npm start`
+9. You can interact with the bot by @mentioning it with your message
+
+### Set-up instructions with Docker
+1. Install [Docker](https://docs.docker.com/get-docker/)
+   - Should be atleast compatible with version 3 of compose (docker engine 1.13.0+)
+2. Repeat steps 2—7 from the other setup instructions
+3. Start the bot with `make compose-up` if you have Make installed
+   - Otherwise, try `docker compose -p discord-ai up` instead
+4. You can interact with the bot by @mentioning it with your message

docker-compose.yml

@@ -0,0 +1,9 @@
+version: "3"
+
+services:
+    bot:
+        build: .
+        env_file: .env
+        environment:
+            - OLLAMA=http://host.docker.internal:11434
+        restart: unless-stopped

package-lock.json

@@ -0,0 +1,378 @@
+{
+	"name": "discord-ai-bot",
+	"lockfileVersion": 3,
+	"requires": true,
+	"packages": {
+		"": {
+			"name": "discord-ai-bot",
+			"dependencies": {
+				"axios": "^1.6.3",
+				"discord.js": "^14.14.1",
+				"dotenv": "^16.3.1",
+				"meklog": "^1.0.2"
+			}
+		},
+		"node_modules/@discordjs/builders": {
+			"version": "1.7.0",
+			"resolved": "https://registry.npmjs.org/@discordjs/builders/-/builders-1.7.0.tgz",
+			"integrity": "sha512-GDtbKMkg433cOZur8Dv6c25EHxduNIBsxeHrsRoIM8+AwmEZ8r0tEpckx/sHwTLwQPOF3e2JWloZh9ofCaMfAw==",
+			"dependencies": {
+				"@discordjs/formatters": "^0.3.3",
+				"@discordjs/util": "^1.0.2",
+				"@sapphire/shapeshift": "^3.9.3",
+				"discord-api-types": "0.37.61",
+				"fast-deep-equal": "^3.1.3",
+				"ts-mixer": "^6.0.3",
+				"tslib": "^2.6.2"
+			},
+			"engines": {
+				"node": ">=16.11.0"
+			}
+		},
+		"node_modules/@discordjs/collection": {
+			"version": "1.5.3",
+			"resolved": "https://registry.npmjs.org/@discordjs/collection/-/collection-1.5.3.tgz",
+			"integrity": "sha512-SVb428OMd3WO1paV3rm6tSjM4wC+Kecaa1EUGX7vc6/fddvw/6lg90z4QtCqm21zvVe92vMMDt9+DkIvjXImQQ==",
+			"engines": {
+				"node": ">=16.11.0"
+			}
+		},
+		"node_modules/@discordjs/formatters": {
+			"version": "0.3.3",
+			"resolved": "https://registry.npmjs.org/@discordjs/formatters/-/formatters-0.3.3.tgz",
+			"integrity": "sha512-wTcI1Q5cps1eSGhl6+6AzzZkBBlVrBdc9IUhJbijRgVjCNIIIZPgqnUj3ntFODsHrdbGU8BEG9XmDQmgEEYn3w==",
+			"dependencies": {
+				"discord-api-types": "0.37.61"
+			},
+			"engines": {
+				"node": ">=16.11.0"
+			}
+		},
+		"node_modules/@discordjs/rest": {
+			"version": "2.2.0",
+			"resolved": "https://registry.npmjs.org/@discordjs/rest/-/rest-2.2.0.tgz",
+			"integrity": "sha512-nXm9wT8oqrYFRMEqTXQx9DUTeEtXUDMmnUKIhZn6O2EeDY9VCdwj23XCPq7fkqMPKdF7ldAfeVKyxxFdbZl59A==",
+			"dependencies": {
+				"@discordjs/collection": "^2.0.0",
+				"@discordjs/util": "^1.0.2",
+				"@sapphire/async-queue": "^1.5.0",
+				"@sapphire/snowflake": "^3.5.1",
+				"@vladfrangu/async_event_emitter": "^2.2.2",
+				"discord-api-types": "0.37.61",
+				"magic-bytes.js": "^1.5.0",
+				"tslib": "^2.6.2",
+				"undici": "5.27.2"
+			},
+			"engines": {
+				"node": ">=16.11.0"
+			}
+		},
+		"node_modules/@discordjs/rest/node_modules/@discordjs/collection": {
+			"version": "2.0.0",
+			"resolved": "https://registry.npmjs.org/@discordjs/collection/-/collection-2.0.0.tgz",
+			"integrity": "sha512-YTWIXLrf5FsrLMycpMM9Q6vnZoR/lN2AWX23/Cuo8uOOtS8eHB2dyQaaGnaF8aZPYnttf2bkLMcXn/j6JUOi3w==",
+			"engines": {
+				"node": ">=18"
+			}
+		},
+		"node_modules/@discordjs/util": {
+			"version": "1.0.2",
+			"resolved": "https://registry.npmjs.org/@discordjs/util/-/util-1.0.2.tgz",
+			"integrity": "sha512-IRNbimrmfb75GMNEjyznqM1tkI7HrZOf14njX7tCAAUetyZM1Pr8hX/EK2lxBCOgWDRmigbp24fD1hdMfQK5lw==",
+			"engines": {
+				"node": ">=16.11.0"
+			}
+		},
+		"node_modules/@discordjs/ws": {
+			"version": "1.0.2",
+			"resolved": "https://registry.npmjs.org/@discordjs/ws/-/ws-1.0.2.tgz",
+			"integrity": "sha512-+XI82Rm2hKnFwAySXEep4A7Kfoowt6weO6381jgW+wVdTpMS/56qCvoXyFRY0slcv7c/U8My2PwIB2/wEaAh7Q==",
+			"dependencies": {
+				"@discordjs/collection": "^2.0.0",
+				"@discordjs/rest": "^2.1.0",
+				"@discordjs/util": "^1.0.2",
+				"@sapphire/async-queue": "^1.5.0",
+				"@types/ws": "^8.5.9",
+				"@vladfrangu/async_event_emitter": "^2.2.2",
+				"discord-api-types": "0.37.61",
+				"tslib": "^2.6.2",
+				"ws": "^8.14.2"
+			},
+			"engines": {
+				"node": ">=16.11.0"
+			}
+		},
+		"node_modules/@discordjs/ws/node_modules/@discordjs/collection": {
+			"version": "2.0.0",
+			"resolved": "https://registry.npmjs.org/@discordjs/collection/-/collection-2.0.0.tgz",
+			"integrity": "sha512-YTWIXLrf5FsrLMycpMM9Q6vnZoR/lN2AWX23/Cuo8uOOtS8eHB2dyQaaGnaF8aZPYnttf2bkLMcXn/j6JUOi3w==",
+			"engines": {
+				"node": ">=18"
+			}
+		},
+		"node_modules/@fastify/busboy": {
+			"version": "2.1.1",
+			"resolved": "https://registry.npmjs.org/@fastify/busboy/-/busboy-2.1.1.tgz",
+			"integrity": "sha512-vBZP4NlzfOlerQTnba4aqZoMhE/a9HY7HRqoOPaETQcSQuWEIyZMHGfVu6w9wGtGK5fED5qRs2DteVCjOH60sA==",
+			"engines": {
+				"node": ">=14"
+			}
+		},
+		"node_modules/@sapphire/async-queue": {
+			"version": "1.5.2",
+			"resolved": "https://registry.npmjs.org/@sapphire/async-queue/-/async-queue-1.5.2.tgz",
+			"integrity": "sha512-7X7FFAA4DngXUl95+hYbUF19bp1LGiffjJtu7ygrZrbdCSsdDDBaSjB7Akw0ZbOu6k0xpXyljnJ6/RZUvLfRdg==",
+			"engines": {
+				"node": ">=v14.0.0",
+				"npm": ">=7.0.0"
+			}
+		},
+		"node_modules/@sapphire/shapeshift": {
+			"version": "3.9.7",
+			"resolved": "https://registry.npmjs.org/@sapphire/shapeshift/-/shapeshift-3.9.7.tgz",
+			"integrity": "sha512-4It2mxPSr4OGn4HSQWGmhFMsNFGfFVhWeRPCRwbH972Ek2pzfGRZtb0pJ4Ze6oIzcyh2jw7nUDa6qGlWofgd9g==",
+			"dependencies": {
+				"fast-deep-equal": "^3.1.3",
+				"lodash": "^4.17.21"
+			},
+			"engines": {
+				"node": ">=v16"
+			}
+		},
+		"node_modules/@sapphire/snowflake": {
+			"version": "3.5.1",
+			"resolved": "https://registry.npmjs.org/@sapphire/snowflake/-/snowflake-3.5.1.tgz",
+			"integrity": "sha512-BxcYGzgEsdlG0dKAyOm0ehLGm2CafIrfQTZGWgkfKYbj+pNNsorZ7EotuZukc2MT70E0UbppVbtpBrqpzVzjNA==",
+			"engines": {
+				"node": ">=v14.0.0",
+				"npm": ">=7.0.0"
+			}
+		},
+		"node_modules/@types/node": {
+			"version": "20.12.7",
+			"resolved": "https://registry.npmjs.org/@types/node/-/node-20.12.7.tgz",
+			"integrity": "sha512-wq0cICSkRLVaf3UGLMGItu/PtdY7oaXaI/RVU+xliKVOtRna3PRY57ZDfztpDL0n11vfymMUnXv8QwYCO7L1wg==",
+			"dependencies": {
+				"undici-types": "~5.26.4"
+			}
+		},
+		"node_modules/@types/ws": {
+			"version": "8.5.9",
+			"resolved": "https://registry.npmjs.org/@types/ws/-/ws-8.5.9.tgz",
+			"integrity": "sha512-jbdrY0a8lxfdTp/+r7Z4CkycbOFN8WX+IOchLJr3juT/xzbJ8URyTVSJ/hvNdadTgM1mnedb47n+Y31GsFnQlg==",
+			"dependencies": {
+				"@types/node": "*"
+			}
+		},
+		"node_modules/@vladfrangu/async_event_emitter": {
+			"version": "2.2.4",
+			"resolved": "https://registry.npmjs.org/@vladfrangu/async_event_emitter/-/async_event_emitter-2.2.4.tgz",
+			"integrity": "sha512-ButUPz9E9cXMLgvAW8aLAKKJJsPu1dY1/l/E8xzLFuysowXygs6GBcyunK9rnGC4zTsnIc2mQo71rGw9U+Ykug==",
+			"engines": {
+				"node": ">=v14.0.0",
+				"npm": ">=7.0.0"
+			}
+		},
+		"node_modules/asynckit": {
+			"version": "0.4.0",
+			"resolved": "https://registry.npmjs.org/asynckit/-/asynckit-0.4.0.tgz",
+			"integrity": "sha512-Oei9OH4tRh0YqU3GxhX79dM/mwVgvbZJaSNaRk+bshkj0S5cfHcgYakreBjrHwatXKbz+IoIdYLxrKim2MjW0Q=="
+		},
+		"node_modules/axios": {
+			"version": "1.6.8",
+			"resolved": "https://registry.npmjs.org/axios/-/axios-1.6.8.tgz",
+			"integrity": "sha512-v/ZHtJDU39mDpyBoFVkETcd/uNdxrWRrg3bKpOKzXFA6Bvqopts6ALSMU3y6ijYxbw2B+wPrIv46egTzJXCLGQ==",
+			"dependencies": {
+				"follow-redirects": "^1.15.6",
+				"form-data": "^4.0.0",
+				"proxy-from-env": "^1.1.0"
+			}
+		},
+		"node_modules/combined-stream": {
+			"version": "1.0.8",
+			"resolved": "https://registry.npmjs.org/combined-stream/-/combined-stream-1.0.8.tgz",
+			"integrity": "sha512-FQN4MRfuJeHf7cBbBMJFXhKSDq+2kAArBlmRBvcvFE5BB1HZKXtSFASDhdlz9zOYwxh8lDdnvmMOe/+5cdoEdg==",
+			"dependencies": {
+				"delayed-stream": "~1.0.0"
+			},
+			"engines": {
+				"node": ">= 0.8"
+			}
+		},
+		"node_modules/delayed-stream": {
+			"version": "1.0.0",
+			"resolved": "https://registry.npmjs.org/delayed-stream/-/delayed-stream-1.0.0.tgz",
+			"integrity": "sha512-ZySD7Nf91aLB0RxL4KGrKHBXl7Eds1DAmEdcoVawXnLD7SDhpNgtuII2aAkg7a7QS41jxPSZ17p4VdGnMHk3MQ==",
+			"engines": {
+				"node": ">=0.4.0"
+			}
+		},
+		"node_modules/discord-api-types": {
+			"version": "0.37.61",
+			"resolved": "https://registry.npmjs.org/discord-api-types/-/discord-api-types-0.37.61.tgz",
+			"integrity": "sha512-o/dXNFfhBpYHpQFdT6FWzeO7pKc838QeeZ9d91CfVAtpr5XLK4B/zYxQbYgPdoMiTDvJfzcsLW5naXgmHGDNXw=="
+		},
+		"node_modules/discord.js": {
+			"version": "14.14.1",
+			"resolved": "https://registry.npmjs.org/discord.js/-/discord.js-14.14.1.tgz",
+			"integrity": "sha512-/hUVzkIerxKHyRKopJy5xejp4MYKDPTszAnpYxzVVv4qJYf+Tkt+jnT2N29PIPschicaEEpXwF2ARrTYHYwQ5w==",
+			"dependencies": {
+				"@discordjs/builders": "^1.7.0",
+				"@discordjs/collection": "1.5.3",
+				"@discordjs/formatters": "^0.3.3",
+				"@discordjs/rest": "^2.1.0",
+				"@discordjs/util": "^1.0.2",
+				"@discordjs/ws": "^1.0.2",
+				"@sapphire/snowflake": "3.5.1",
+				"@types/ws": "8.5.9",
+				"discord-api-types": "0.37.61",
+				"fast-deep-equal": "3.1.3",
+				"lodash.snakecase": "4.1.1",
+				"tslib": "2.6.2",
+				"undici": "5.27.2",
+				"ws": "8.14.2"
+			},
+			"engines": {
+				"node": ">=16.11.0"
+			}
+		},
+		"node_modules/dotenv": {
+			"version": "16.4.5",
+			"resolved": "https://registry.npmjs.org/dotenv/-/dotenv-16.4.5.tgz",
+			"integrity": "sha512-ZmdL2rui+eB2YwhsWzjInR8LldtZHGDoQ1ugH85ppHKwpUHL7j7rN0Ti9NCnGiQbhaZ11FpR+7ao1dNsmduNUg==",
+			"engines": {
+				"node": ">=12"
+			},
+			"funding": {
+				"url": "https://dotenvx.com"
+			}
+		},
+		"node_modules/fast-deep-equal": {
+			"version": "3.1.3",
+			"resolved": "https://registry.npmjs.org/fast-deep-equal/-/fast-deep-equal-3.1.3.tgz",
+			"integrity": "sha512-f3qQ9oQy9j2AhBe/H9VC91wLmKBCCU/gDOnKNAYG5hswO7BLKj09Hc5HYNz9cGI++xlpDCIgDaitVs03ATR84Q=="
+		},
+		"node_modules/follow-redirects": {
+			"version": "1.15.6",
+			"resolved": "https://registry.npmjs.org/follow-redirects/-/follow-redirects-1.15.6.tgz",
+			"integrity": "sha512-wWN62YITEaOpSK584EZXJafH1AGpO8RVgElfkuXbTOrPX4fIfOyEpW/CsiNd8JdYrAoOvafRTOEnvsO++qCqFA==",
+			"funding": [
+				{
+					"type": "individual",
+					"url": "https://github.com/sponsors/RubenVerborgh"
+				}
+			],
+			"engines": {
+				"node": ">=4.0"
+			},
+			"peerDependenciesMeta": {
+				"debug": {
+					"optional": true
+				}
+			}
+		},
+		"node_modules/form-data": {
+			"version": "4.0.0",
+			"resolved": "https://registry.npmjs.org/form-data/-/form-data-4.0.0.tgz",
+			"integrity": "sha512-ETEklSGi5t0QMZuiXoA/Q6vcnxcLQP5vdugSpuAyi6SVGi2clPPp+xgEhuMaHC+zGgn31Kd235W35f7Hykkaww==",
+			"dependencies": {
+				"asynckit": "^0.4.0",
+				"combined-stream": "^1.0.8",
+				"mime-types": "^2.1.12"
+			},
+			"engines": {
+				"node": ">= 6"
+			}
+		},
+		"node_modules/lodash": {
+			"version": "4.17.21",
+			"resolved": "https://registry.npmjs.org/lodash/-/lodash-4.17.21.tgz",
+			"integrity": "sha512-v2kDEe57lecTulaDIuNTPy3Ry4gLGJ6Z1O3vE1krgXZNrsQ+LFTGHVxVjcXPs17LhbZVGedAJv8XZ1tvj5FvSg=="
+		},
+		"node_modules/lodash.snakecase": {
+			"version": "4.1.1",
+			"resolved": "https://registry.npmjs.org/lodash.snakecase/-/lodash.snakecase-4.1.1.tgz",
+			"integrity": "sha512-QZ1d4xoBHYUeuouhEq3lk3Uq7ldgyFXGBhg04+oRLnIz8o9T65Eh+8YdroUwn846zchkA9yDsDl5CVVaV2nqYw=="
+		},
+		"node_modules/magic-bytes.js": {
+			"version": "1.10.0",
+			"resolved": "https://registry.npmjs.org/magic-bytes.js/-/magic-bytes.js-1.10.0.tgz",
+			"integrity": "sha512-/k20Lg2q8LE5xiaaSkMXk4sfvI+9EGEykFS4b0CHHGWqDYU0bGUFSwchNOMA56D7TCs9GwVTkqe9als1/ns8UQ=="
+		},
+		"node_modules/meklog": {
+			"version": "1.0.2",
+			"resolved": "https://registry.npmjs.org/meklog/-/meklog-1.0.2.tgz",
+			"integrity": "sha512-9jkTaZzWEpO0tiWQl0xoj/DPktcHELg74nSzsEaLaN7IYGmcczMXgQXQmZT1cXWykj2FAhv0BBRUGpnFCZn/dg=="
+		},
+		"node_modules/mime-db": {
+			"version": "1.52.0",
+			"resolved": "https://registry.npmjs.org/mime-db/-/mime-db-1.52.0.tgz",
+			"integrity": "sha512-sPU4uV7dYlvtWJxwwxHD0PuihVNiE7TyAbQ5SWxDCB9mUYvOgroQOwYQQOKPJ8CIbE+1ETVlOoK1UC2nU3gYvg==",
+			"engines": {
+				"node": ">= 0.6"
+			}
+		},
+		"node_modules/mime-types": {
+			"version": "2.1.35",
+			"resolved": "https://registry.npmjs.org/mime-types/-/mime-types-2.1.35.tgz",
+			"integrity": "sha512-ZDY+bPm5zTTF+YpCrAU9nK0UgICYPT0QtT1NZWFv4s++TNkcgVaT0g6+4R2uI4MjQjzysHB1zxuWL50hzaeXiw==",
+			"dependencies": {
+				"mime-db": "1.52.0"
+			},
+			"engines": {
+				"node": ">= 0.6"
+			}
+		},
+		"node_modules/proxy-from-env": {
+			"version": "1.1.0",
+			"resolved": "https://registry.npmjs.org/proxy-from-env/-/proxy-from-env-1.1.0.tgz",
+			"integrity": "sha512-D+zkORCbA9f1tdWRK0RaCR3GPv50cMxcrz4X8k5LTSUD1Dkw47mKJEZQNunItRTkWwgtaUSo1RVFRIG9ZXiFYg=="
+		},
+		"node_modules/ts-mixer": {
+			"version": "6.0.4",
+			"resolved": "https://registry.npmjs.org/ts-mixer/-/ts-mixer-6.0.4.tgz",
+			"integrity": "sha512-ufKpbmrugz5Aou4wcr5Wc1UUFWOLhq+Fm6qa6P0w0K5Qw2yhaUoiWszhCVuNQyNwrlGiscHOmqYoAox1PtvgjA=="
+		},
+		"node_modules/tslib": {
+			"version": "2.6.2",
+			"resolved": "https://registry.npmjs.org/tslib/-/tslib-2.6.2.tgz",
+			"integrity": "sha512-AEYxH93jGFPn/a2iVAwW87VuUIkR1FVUKB77NwMF7nBTDkDrrT/Hpt/IrCJ0QXhW27jTBDcf5ZY7w6RiqTMw2Q=="
+		},
+		"node_modules/undici": {
+			"version": "5.27.2",
+			"resolved": "https://registry.npmjs.org/undici/-/undici-5.27.2.tgz",
+			"integrity": "sha512-iS857PdOEy/y3wlM3yRp+6SNQQ6xU0mmZcwRSriqk+et/cwWAtwmIGf6WkoDN2EK/AMdCO/dfXzIwi+rFMrjjQ==",
+			"dependencies": {
+				"@fastify/busboy": "^2.0.0"
+			},
+			"engines": {
+				"node": ">=14.0"
+			}
+		},
+		"node_modules/undici-types": {
+			"version": "5.26.5",
+			"resolved": "https://registry.npmjs.org/undici-types/-/undici-types-5.26.5.tgz",
+			"integrity": "sha512-JlCMO+ehdEIKqlFxk6IfVoAUVmgz7cU7zD/h9XZ0qzeosSHmUJVOzSQvvYSYWXkFXC+IfLKSIffhv0sVZup6pA=="
+		},
+		"node_modules/ws": {
+			"version": "8.14.2",
+			"resolved": "https://registry.npmjs.org/ws/-/ws-8.14.2.tgz",
+			"integrity": "sha512-wEBG1ftX4jcglPxgFCMJmZ2PLtSbJ2Peg6TmpJFTbe9GZYOQCDPdMYu/Tm0/bGZkw8paZnJY45J4K2PZrLYq8g==",
+			"engines": {
+				"node": ">=10.0.0"
+			},
+			"peerDependencies": {
+				"bufferutil": "^4.0.1",
+				"utf-8-validate": ">=5.0.2"
+			},
+			"peerDependenciesMeta": {
+				"bufferutil": {
+					"optional": true
+				},
+				"utf-8-validate": {
+					"optional": true
+				}
+			}
+		}
+	}
+}

package.json

@@ -0,0 +1,14 @@
+{
+	"name": "discord-ai-bot",
+	"main": "src/index.js",
+	"scripts": {
+		"start": "node src/index.js"
+	},
+	"dependencies": {
+		"axios": "^1.6.3",
+		"discord.js": "^14.14.1",
+		"dotenv": "^16.3.1",
+		"meklog": "^1.0.2"
+	},
+	"type": "module"
+}

src/bot.js

@@ -0,0 +1,434 @@
+import { Client, Events, GatewayIntentBits, MessageType, Partials } from "discord.js";
+import { Logger, LogLevel } from "meklog";
+import dotenv from "dotenv";
+import axios from "axios";
+
+dotenv.config();
+
+const model = process.env.MODEL;
+const servers = process.env.OLLAMA.split(",").map(url => ({ url: new URL(url), available: true }));
+const channels = process.env.CHANNELS.split(",");
+
+if (servers.length == 0) {
+	throw new Error("No servers available");
+}
+
+let log;
+process.on("message", data => {
+	if (data.shardID) client.shardID = data.shardID;
+	if (data.logger) log = new Logger(data.logger);
+});
+
+const logError = (error) => {
+	if (error.response) {
+		let str = `Error ${error.response.status} ${error.response.statusText}: ${error.request.method} ${error.request.path}`;
+		if (error.response.data?.error) {
+			str += ": " + error.response.data.error;
+		}
+		log(LogLevel.Error, str);
+	} else {
+		log(LogLevel.Error, error);
+	}
+};
+
+function shuffleArray(array) {
+	for (let i = array.length - 1; i > 0; i--) {
+		const j = Math.floor(Math.random() * (i + 1));
+		[array[i], array[j]] = [array[j], array[i]];
+	}
+	return array;
+}
+
+async function makeRequest(path, method, data) {
+	while (servers.filter(server => server.available).length == 0) {
+		// wait until a server is available
+		await new Promise(res => setTimeout(res, 1000));
+	}
+
+	let error = null;
+	// randomly loop through the servers available, don't shuffle the actual array because we want to be notified of any updates
+	let order = new Array(servers.length).fill().map((_, i) => i);
+	if (randomServer) order = shuffleArray(order);
+	for (const j in order) {
+		if (!order.hasOwnProperty(j)) continue;
+		const i = order[j];
+		// try one until it succeeds
+		try {
+			// make a request to ollama
+			if (!servers[i].available) continue;
+			const url = new URL(servers[i].url); // don't modify the original URL
+
+			servers[i].available = false;
+
+			if (path.startsWith("/")) path = path.substring(1);
+			if (!url.pathname.endsWith("/")) url.pathname += "/"; // safety
+			url.pathname += path;
+			log(LogLevel.Debug, `Making request to ${url}`);
+			const result = await axios({
+				method, url, data,
+				responseType: "text"
+			});
+			servers[i].available = true;
+			return result.data;
+		} catch (err) {
+			servers[i].available = true;
+			error = err;
+			logError(error);
+		}
+	}
+	if (!error) {
+		throw new Error("No servers available");
+	}
+	throw error;
+}
+
+const client = new Client({
+	intents: [
+		GatewayIntentBits.Guilds,
+		GatewayIntentBits.GuildMessages,
+		GatewayIntentBits.GuildMembers,
+		GatewayIntentBits.DirectMessages,
+		GatewayIntentBits.MessageContent
+	],
+	allowedMentions: { users: [], roles: [], repliedUser: false },
+	partials: [
+		Partials.Channel
+	]
+});
+
+client.once(Events.ClientReady, async () => {
+	await client.guilds.fetch();
+	client.user.setPresence({ activities: [], status: "online" });
+});
+
+const messages = {};
+
+// split text so it fits in a Discord message
+function splitText(str, length) {
+	// trim matches different characters to \s
+	str = str
+		.replace(/\r\n/g, "\n").replace(/\r/g, "\n")
+		.replace(/^\s+|\s+$/g, "");
+	const segments = [];
+	let segment = "";
+	let word, suffix;
+	function appendSegment() {
+		segment = segment.replace(/^\s+|\s+$/g, "");
+		if (segment.length > 0) {
+			segments.push(segment);
+			segment = "";
+		}
+	}
+	// match a word
+	while ((word = str.match(/^[^\s]*(?:\s+|$)/)) != null) {
+		suffix = "";
+		word = word[0];
+		if (word.length == 0) break;
+		if (segment.length + word.length > length) {
+			// prioritise splitting by newlines over other whitespaces
+			if (segment.includes("\n")) {
+				// append up all but last paragraph
+				const beforeParagraph = segment.match(/^.*\n/s);
+				if (beforeParagraph != null) {
+					const lastParagraph = segment.substring(beforeParagraph[0].length, segment.length);
+					segment = beforeParagraph[0];
+					appendSegment();
+					segment = lastParagraph;
+					continue;
+				}
+			}
+			appendSegment();
+			// if word is larger than the split length
+			if (word.length > length) {
+				word = word.substring(0, length);
+				if (length > 1 && word.match(/^[^\s]+$/)) {
+					// try to hyphenate word
+					word = word.substring(0, word.length - 1);
+					suffix = "-";
+				}
+			}
+		}
+		str = str.substring(word.length, str.length);
+		segment += word + suffix;
+	}
+	appendSegment();
+	return segments;
+}
+
+function getBoolean(str) {
+	return !!str && str != "false" && str != "no" && str != "off" && str != "0";
+}
+
+function parseJSONMessage(str) {
+	return str.split(/[\r\n]+/g).map(function(line) {
+		const result = JSON.parse(`"${line}"`);
+		if (typeof result !== "string") throw new "Invalid syntax in .env file";
+		return result;
+	}).join("\n");
+}
+
+function parseEnvString(str) {
+	return typeof str === "string" ?
+		parseJSONMessage(str).replace(/<date>/gi, new Date().toUTCString()) : null;
+}
+
+const customSystemMessage = parseEnvString(process.env.SYSTEM);
+const useCustomSystemMessage = getBoolean(process.env.USE_SYSTEM) && !!customSystemMessage;
+const useModelSystemMessage = getBoolean(process.env.USE_MODEL_SYSTEM);
+const showStartOfConversation = getBoolean(process.env.SHOW_START_OF_CONVERSATION);
+const randomServer = getBoolean(process.env.RANDOM_SERVER);
+let modelInfo = null;
+const initialPrompt = parseEnvString(process.env.INITIAL_PROMPT);
+const useInitialPrompt = getBoolean(process.env.USE_INITIAL_PROMPT) && !!initialPrompt;
+
+const requiresMention = getBoolean(process.env.REQUIRES_MENTION);
+
+async function replySplitMessage(replyMessage, content) {
+	const responseMessages = splitText(content, 2000).map(content => ({ content }));
+
+	const replyMessages = [];
+	for (let i = 0; i < responseMessages.length; ++i) {
+		if (i == 0) {
+			replyMessages.push(await replyMessage.reply(responseMessages[i]));
+		} else {
+			replyMessages.push(await replyMessage.channel.send(responseMessages[i]));
+		}
+	}
+	return replyMessages;
+}
+
+client.on(Events.MessageCreate, async message => {
+	let typing = false;
+	try {
+		await message.fetch();
+
+		// return if not in the right channel
+		const channelID = message.channel.id;
+		if (message.guild && !channels.includes(channelID)) return;
+
+		// return if user is a bot, or non-default message
+		if (!message.author.id) return;
+		if (message.author.bot || message.author.id == client.user.id) return;
+
+		const botRole = message.guild?.members?.me?.roles?.botRole;
+		const myMention = new RegExp(`<@((!?${client.user.id}${botRole ? `)|(&${botRole.id}` : ""}))>`, "g"); // RegExp to match a mention for the bot
+
+		if (typeof message.content !== "string" || message.content.length == 0) {
+			return;
+		}
+
+		let context = null;
+		if (message.type == MessageType.Reply) {
+			const reply = await message.fetchReference();
+			if (!reply) return;
+			if (reply.author.id != client.user.id) return;
+			if (messages[channelID] == null) return;
+			if ((context = messages[channelID][reply.id]) == null) return;
+		} else if (message.type != MessageType.Default) {
+			return;
+		}
+
+		// fetch info about the model like the template and system message
+		if (modelInfo == null) {
+			modelInfo = (await makeRequest("/api/show", "post", {
+				name: model
+			}));
+			if (typeof modelInfo === "string") modelInfo = JSON.parse(modelInfo);
+			if (typeof modelInfo !== "object") throw "failed to fetch model information";
+		}
+
+		const systemMessages = [];
+
+		if (useModelSystemMessage && modelInfo.system) {
+			systemMessages.push(modelInfo.system);
+		}
+
+		if (useCustomSystemMessage) {
+			systemMessages.push(customSystemMessage);
+		}
+
+		// join them together
+		const systemMessage = systemMessages.join("\n\n");
+
+		// deal with commands first before passing to LLM
+		let userInput = message.content
+			.replace(new RegExp("^\s*" + myMention.source, ""), "").trim();
+
+		// may change this to slash commands in the future
+		// i'm using regular text commands currently because the bot interacts with text content anyway
+		if (userInput.startsWith(".")) {
+			const args = userInput.substring(1).split(/\s+/g);
+			const cmd = args.shift();
+			switch (cmd) {
+				case "reset":
+				case "clear":
+					if (messages[channelID] != null) {
+						// reset conversation
+						const cleared = messages[channelID].amount;
+
+						// clear
+						delete messages[channelID];
+
+						if (cleared > 0) {
+							await message.reply({ content: `Cleared conversation of ${cleared} messages` });
+							break;
+						}
+					}
+					await message.reply({ content: "No messages to clear" });
+					break;
+				case "help":
+				case "?":
+				case "h":
+					await message.reply({ content: "Commands:\n- `.reset` `.clear`\n- `.help` `.?` `.h`\n- `.ping`\n- `.model`\n- `.system`" });
+					break;
+				case "model":
+					await message.reply({
+						content: `Current model: ${model}`
+					});
+					break;
+				case "system":
+					await replySplitMessage(message, `System message:\n\n${systemMessage}`);
+					break;
+				case "ping":
+					// get ms difference
+					const beforeTime = Date.now();
+					const reply = await message.reply({ content: "Ping" });
+					const afterTime = Date.now();
+					const difference = afterTime - beforeTime;
+					await reply.edit({ content: `Ping: ${difference}ms` });
+					break;
+				case "":
+					break;
+				default:
+					await message.reply({ content: "Unknown command, type `.help` for a list of commands" });
+					break;
+			}
+			return;
+		}
+
+		if (message.type == MessageType.Default && (requiresMention && message.guild && !message.content.match(myMention))) return;
+
+		if (message.guild) {
+			await message.guild.channels.fetch();
+			await message.guild.members.fetch();
+		}
+
+		userInput = userInput
+			.replace(myMention, "")
+			.replace(/<#([0-9]+)>/g, (_, id) => {
+				if (message.guild) {
+					const chn = message.guild.channels.cache.get(id);
+					if (chn) return `#${chn.name}`;
+				}
+				return "#unknown-channel";
+			})
+			.replace(/<@!?([0-9]+)>/g, (_, id) => {
+				if (id == message.author.id) return message.author.username;
+				if (message.guild) {
+					const mem = message.guild.members.cache.get(id);
+					if (mem) return `@${mem.user.username}`;
+				}
+				return "@unknown-user";
+			})
+			.replace(/<:([a-zA-Z0-9_]+):([0-9]+)>/g, (_, name) => {
+				return `emoji:${name}:`;
+			})
+			.trim();
+
+		if (userInput.length == 0) return;
+
+		// create conversation
+		if (messages[channelID] == null) {
+			messages[channelID] = { amount: 0, last: null };
+		}
+
+		// log user's message
+		log(LogLevel.Debug, `${message.guild ? `#${message.channel.name}` : "DMs"} - ${message.author.username}: ${userInput}`);
+
+		// start typing
+		typing = true;
+		await message.channel.sendTyping();
+		let typingInterval = setInterval(async () => {
+			try {
+				await message.channel.sendTyping();
+			} catch (error) {
+				if (typingInterval != null) {
+					clearInterval(typingInterval);
+				}
+				typingInterval = null;
+			}
+		}, 7000);
+
+		let response;
+		try {
+			// context if the message is not a reply
+			if (context == null) {
+				context = messages[channelID].last;
+			}
+
+			if (useInitialPrompt && messages[channelID].amount == 0) {
+				userInput = `${initialPrompt}\n\n${userInput}`;
+				log(LogLevel.Debug, "Adding initial prompt to message");
+			}
+
+			// make request to model
+			response = (await makeRequest("/api/generate", "post", {
+				model: model,
+				prompt: userInput,
+				system: systemMessage,
+				context
+			}));
+
+			if (typeof response != "string") {
+				log(LogLevel.Debug, response);
+				throw new TypeError("response is not a string, this may be an error with ollama");
+			}
+
+			response = response.split("\n").filter(e => !!e).map(e => {
+				return JSON.parse(e);
+			});
+		} catch (error) {
+			if (typingInterval != null) {
+				clearInterval(typingInterval);
+			}
+			typingInterval = null;
+			throw error;
+		}
+
+		if (typingInterval != null) {
+			clearInterval(typingInterval);
+		}
+		typingInterval = null;
+
+		let responseText = response.map(e => e.response).filter(e => e != null).join("").trim();
+		if (responseText.length == 0) {
+			responseText = "(No response)";
+		}
+
+		log(LogLevel.Debug, `Response: ${responseText}`);
+
+		const prefix = showStartOfConversation && messages[channelID].amount == 0 ?
+			"> This is the beginning of the conversation, type `.help` for help.\n\n" : "";
+
+		// reply (will automatically stop typing)
+		const replyMessageIDs = (await replySplitMessage(message, `${prefix}${responseText}`)).map(msg => msg.id);
+
+		// add response to conversation
+		context = response.filter(e => e.done && e.context)[0].context;
+		for (let i = 0; i < replyMessageIDs.length; ++i) {
+			messages[channelID][replyMessageIDs[i]] = context;
+		}
+		messages[channelID].last = context;
+		++messages[channelID].amount;
+	} catch (error) {
+		if (typing) {
+			try {
+				// return error
+				await message.reply({ content: "Error, please check the console" });
+			} catch (ignored) {}
+		}
+		logError(error);
+	}
+});
+
+client.login(process.env.TOKEN);

src/index.js

@@ -0,0 +1,30 @@
+import { ShardingManager, Events } from "discord.js";
+import path from "node:path";
+import { fileURLToPath } from "node:url";
+import { Logger, LogLevel } from "meklog";
+import dotenv from "dotenv";
+
+dotenv.config();
+
+const production = process.env.NODE_ENV == "prod" || process.env.NODE_ENV == "production";
+const log = new Logger(production, "Shard Manager");
+
+log(LogLevel.Info, "Loading");
+
+const filePath = path.join(path.dirname(fileURLToPath(import.meta.url)), "bot.js");
+const manager = new ShardingManager(filePath, { token: process.env.TOKEN });
+
+manager.on("shardCreate", async shard => {
+	const shardLog = new Logger(production, `Shard #${shard.id}`);
+
+	shardLog(LogLevel.Info, "Created shard");
+
+	shard.once(Events.ClientReady, async () => {
+		shard.send({ shardID: shard.id, logger: shardLog.data });
+
+		shardLog(LogLevel.Info, "Shard ready");
+	});
+});
+
+manager.spawn();
+