generated from nhcarrigan/template
Naomi Carrigan
bb1c327160
All checks were successful
Node.js CI / Lint and Test (push) Successful in 38s
### Explanation _No response_ ### Issue _No response_ ### Attestations - [x] I have read and agree to the [Code of Conduct](https://docs.nhcarrigan.com/community/coc/) - [x] I have read and agree to the [Community Guidelines](https://docs.nhcarrigan.com/community/guide/). - [x] My contribution complies with the [Contributor Covenant](https://docs.nhcarrigan.com/dev/covenant/). ### Dependencies - [ ] I have pinned the dependencies to a specific patch version. ### Style - [x] I have run the linter and resolved any errors. - [x] My pull request uses an appropriate title, matching the conventional commit standards. - [x] My scope of feat/fix/chore/etc. correctly matches the nature of changes in my pull request. ### Tests - [ ] My contribution adds new code, and I have added tests to cover it. - [ ] My contribution modifies existing code, and I have updated the tests to reflect these changes. - [ ] All new and existing tests pass locally with my changes. - [ ] Code coverage remains at or above the configured threshold. ### Documentation _No response_ ### Versioning _No response_ Reviewed-on: #3 Co-authored-by: Naomi Carrigan <commits@nhcarrigan.com> Co-committed-by: Naomi Carrigan <commits@nhcarrigan.com>
58 lines
1.9 KiB
TypeScript
58 lines
1.9 KiB
TypeScript
/**
|
|
* @copyright nhcarrigan
|
|
* @license Naomi's Public License
|
|
* @author Naomi Carrigan
|
|
*/
|
|
import { MessageFlags, type ChatInputCommandInteraction } from "discord.js";
|
|
import { personality } from "../config/personality.js";
|
|
import { ai } from "../utils/ai.js";
|
|
import { calculateCost } from "../utils/calculateCost.js";
|
|
import { isSubscribed } from "../utils/isSubscribed.js";
|
|
import { logger } from "../utils/logger.js";
|
|
import { replyToError } from "../utils/replyToError.js";
|
|
|
|
/**
|
|
* Accepts an arbitrary question from the user, then sends it to Anthropic
|
|
* to be answered.
|
|
* @param interaction -- The interaction payload from Discord.
|
|
*/
|
|
export const query = async(
|
|
interaction: ChatInputCommandInteraction,
|
|
): Promise<void> => {
|
|
try {
|
|
await interaction.deferReply({ flags: [ MessageFlags.Ephemeral ] });
|
|
const sub = await isSubscribed(interaction);
|
|
if (!sub) {
|
|
return;
|
|
}
|
|
|
|
const prompt = interaction.options.getString("prompt", true);
|
|
|
|
const messages = await ai.messages.create({
|
|
// eslint-disable-next-line @typescript-eslint/naming-convention -- Required key format for SDK.
|
|
max_tokens: 2000,
|
|
messages: [ { content: prompt, role: "user" } ],
|
|
model: "claude-3-5-sonnet-latest",
|
|
system: `${personality} Your role in this conversation is to answer the user's question to the best of your abilities. When possible, include links to relevant sources.`,
|
|
temperature: 1,
|
|
});
|
|
|
|
const response = messages.content.find((message) => {
|
|
return message.type === "text";
|
|
});
|
|
|
|
await interaction.editReply(
|
|
response?.text
|
|
?? "I'm sorry, I don't have an answer for that. Please try again later.",
|
|
);
|
|
|
|
const { usage } = messages;
|
|
await calculateCost(usage, interaction.user.username, "query");
|
|
} catch (error) {
|
|
await replyToError(interaction);
|
|
if (error instanceof Error) {
|
|
await logger.error("query command", error);
|
|
}
|
|
}
|
|
};
|