Files
hikari/bot/prod/modules/ai.js
Naomi Carrigan cd5c3761f4
All checks were successful
Node.js CI / Lint and Test (pull_request) Successful in 1m17s
feat: build out discord support agent
2025-07-05 23:06:45 -07:00

90 lines
3.9 KiB
JavaScript

/**
* @copyright nhcarrigan
* @license Naomi's Public License
* @author Naomi Carrigan
*/
/* eslint-disable no-await-in-loop -- Ordinarily I would use Promise.all, but we want these sent in order. */
// eslint-disable-next-line @typescript-eslint/naming-convention -- It is a class, so should be uppercased.
import Anthropic from "@anthropic-ai/sdk";
import { prompt } from "../config/prompt.js";
import { calculateCost } from "../utils/calculateCost.js";
import { errorHandler } from "../utils/errorHandler.js";
const anthropic = new Anthropic({
apiKey: process.env.ANTHROPIC_KEY ?? "",
timeout: 5 * 60 * 1000,
});
/**
* Formats Discord messages into a prompt for the AI,
* sends the prompt to the AI, and returns the AI's response.
* @param hikari - Hikari's Discord instance.
* @param messages - The Discord messages to process.
* @param username - The username of the user who triggered this request - that is, the author of the most recent message.
* @param channel - The channel in which to respond.
* @returns The AI's response as a string.
*/
// eslint-disable-next-line max-lines-per-function -- This is a big function, but it does a lot of things.
export const ai = async (hikari, messages, username, channel) => {
try {
const typingInterval = setInterval(() => {
void channel.sendTyping();
}, 3000);
const parsedPrompt = prompt.replace("{{username}}", username);
const result = await anthropic.beta.messages.create({
betas: ["web-search-2025-03-05"],
// eslint-disable-next-line @typescript-eslint/naming-convention -- API requirement
max_tokens: 20_000,
messages: messages.map((message) => {
return {
content: message.content,
role: message.author.id === hikari.user?.id
? "assistant"
: "user",
};
}),
model: "claude-sonnet-4-20250514",
system: parsedPrompt,
temperature: 1,
tools: [
{
// eslint-disable-next-line @typescript-eslint/naming-convention -- API requirement
allowed_domains: ["nhcarrigan.com"],
name: "web_search",
type: "web_search_20250305",
},
],
});
await calculateCost(result.usage, username);
for (const payload of result.content) {
await channel.sendTyping();
// Sleep for 5 seconds,
await new Promise((resolve) => {
// eslint-disable-next-line no-promise-executor-return -- We want to wait for a bit.
return setTimeout(resolve, 3000);
});
if (payload.type === "text") {
await channel.send({ content: payload.text });
}
if (payload.type === "tool_use") {
await channel.send({ content: `Searching web via: ${String(payload.name)}` });
}
if (payload.type === "web_search_tool_result") {
if (Array.isArray(payload.content)) {
await channel.send({
content: `Checking content on:\n${payload.content.map((item) => {
return `- [${item.title}](<${item.url}>)`;
}).join("\n\n")}`,
});
}
else {
await channel.send({ content: `Web search error: ${payload.content.error_code}` });
}
}
}
clearInterval(typingInterval);
}
catch (error) {
const id = await errorHandler(error, "AI module");
await channel.send(`Something went wrong while processing your request. Please try again later, or [reach out in our support channel](<https://discord.com/channels/1354624415861833870/1385797209706201198>).\n-# ${id}`);
}
};