mirror of
https://github.com/matrixgpt/matrix-chatgpt-bot.git
synced 2024-07-01 17:11:25 +00:00
95 lines
3.6 KiB
TypeScript
95 lines
3.6 KiB
TypeScript
import {
|
|
MatrixAuth, MatrixClient, SimpleFsStorageProvider, AutojoinRoomsMixin,
|
|
LogService, LogLevel,
|
|
RichConsoleLogger,
|
|
ICryptoStorageProvider,
|
|
RustSdkCryptoStorageProvider,
|
|
} from "matrix-bot-sdk";
|
|
|
|
import * as path from "path";
|
|
import { DATA_PATH, OPENAI_API_KEY, OPENAI_PRO, MATRIX_HOMESERVER_URL, MATRIX_ACCESS_TOKEN, MATRIX_AUTOJOIN, MATRIX_BOT_PASSWORD, MATRIX_BOT_USERNAME, MATRIX_ENCRYPTION, MATRIX_THREADS, CHATGPT_CONTEXT, CHATGPT_MODEL } from './env.js'
|
|
import { parseMatrixUsernamePretty } from './utils.js';
|
|
import CommandHandler from "./handlers.js"
|
|
import { ChatGPTAPI } from 'chatgpt'
|
|
|
|
LogService.setLogger(new RichConsoleLogger());
|
|
|
|
// Shows the Matrix sync loop details - not needed most of the time
|
|
// LogService.setLevel(LogLevel.DEBUG);
|
|
|
|
LogService.setLevel(LogLevel.INFO);
|
|
|
|
// LogService.muteModule("Metrics");
|
|
LogService.trace = LogService.debug;
|
|
|
|
const storage = new SimpleFsStorageProvider(path.join(DATA_PATH, "bot.json")); // /storage/bot.json
|
|
|
|
// Prepare a crypto store if we need that
|
|
let cryptoStore: ICryptoStorageProvider;
|
|
if (MATRIX_ENCRYPTION) {
|
|
cryptoStore = new RustSdkCryptoStorageProvider(path.join(DATA_PATH, "encrypted")); // /storage/encrypted
|
|
}
|
|
|
|
async function main() {
|
|
if (!MATRIX_ACCESS_TOKEN){
|
|
const botUsernameWithoutDomain = parseMatrixUsernamePretty(MATRIX_BOT_USERNAME);
|
|
const authedClient = await (new MatrixAuth(MATRIX_HOMESERVER_URL)).passwordLogin(botUsernameWithoutDomain, MATRIX_BOT_PASSWORD);
|
|
console.log(authedClient.homeserverUrl + " token: \n" + authedClient.accessToken)
|
|
console.log("Set MATRIX_ACCESS_TOKEN to above token, MATRIX_BOT_PASSWORD can now be blank")
|
|
return;
|
|
}
|
|
if (!MATRIX_THREADS && CHATGPT_CONTEXT !== "room") throw Error("You must set CHATGPT_CONTEXT to 'room' if you set MATRIX_THREADS to false")
|
|
const client: MatrixClient = new MatrixClient(MATRIX_HOMESERVER_URL, MATRIX_ACCESS_TOKEN, storage, cryptoStore);
|
|
|
|
// use puppeteer to bypass cloudflare (headful because of captchas)
|
|
const chatGPT: ChatGPTAPI = new ChatGPTAPI({
|
|
apiKey: OPENAI_API_KEY,
|
|
completionParams: {
|
|
model: CHATGPT_MODEL
|
|
}
|
|
})
|
|
|
|
// // call `api.refreshSession()` every hour to refresh the session
|
|
// setInterval(() => {
|
|
// chatGPT.refreshSession().then(() => {
|
|
// LogService.info('ChatGPT session reset');
|
|
// });
|
|
// }, 60 * 60 * 1000);
|
|
|
|
// // call `api.resetSession()` every 24 hours to reset the session
|
|
// setInterval(() => {
|
|
// chatGPT.resetSession().then(() => {
|
|
// LogService.info('ChatGPT session reset');
|
|
// });
|
|
// }, 24 * 60 * 60 * 1000);
|
|
|
|
// Automatically join rooms the bot is invited to
|
|
if (MATRIX_AUTOJOIN) {
|
|
AutojoinRoomsMixin.setupOnClient(client);
|
|
}
|
|
|
|
client.on("room.failed_decryption", async (roomId, event, error) => {
|
|
// handle `m.room.encrypted` event that could not be decrypted
|
|
LogService.error("index", `Failed decryption event!\n${{ roomId, event, error }}`);
|
|
await client.sendText(roomId, `I couldn't decrypt the message :( Please add me to an unencrypted room.`);
|
|
});
|
|
|
|
client.on("room.join", async (roomId: string, _event: any) => {
|
|
LogService.info("index", `Bot joined room ${roomId}`);
|
|
await client.sendMessage(roomId, {
|
|
"msgtype": "m.notice",
|
|
"body": `👋 Hello, I'm the ChatGPT bot! Encrypted message support: ${MATRIX_ENCRYPTION}`,
|
|
});
|
|
});
|
|
|
|
// Prepare the command handler
|
|
const commands = new CommandHandler(client, chatGPT);
|
|
await commands.start();
|
|
|
|
LogService.info("index", `Starting bot using ChatGPT model ${CHATGPT_MODEL}`);
|
|
await client.start()
|
|
LogService.info("index", "Bot started!");
|
|
}
|
|
|
|
main();
|