feat: start logging token usage and subscriptions (#2)

### 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 - [x] 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. - [x] All new and existing tests pass locally with my changes. - [x] Code coverage remains at or above the configured threshold. ### Documentation _No response_ ### Versioning Minor - My pull request introduces a new non-breaking feature. Reviewed-on: nhcarrigan/cordelia-taryne#2 Co-authored-by: Naomi Carrigan <commits@nhcarrigan.com> Co-committed-by: Naomi Carrigan <commits@nhcarrigan.com>
2025-02-10 21:11:15 -08:00
parent 4218596559
commit 95976b8b01
14 changed files with 96 additions and 238 deletions
@@ -0,0 +1,30 @@
+/**
+ * @copyright nhcarrigan
+ * @license Naomi's Public License
+ * @author Naomi Carrigan
+ */
+import { logger } from "./logger.js";
+import type { Usage } from "@anthropic-ai/sdk/resources/index.js";
+
+/**
+ * Calculates the cost of a command run by a user, and sends to
+ * our logging service.
+ * @param usage -- The usage payload from Anthropic.
+ * @param uuid -- The Discord ID of the user who ran the command.
+ * @param command -- The command that was run.
+ */
+export const calculateCost = async(
+  usage: Usage,
+  uuid: string,
+  command: string,
+): Promise<void> => {
+  const inputCost = usage.input_tokens * (3 / 1_000_000);
+  const outputCost = usage.output_tokens * (15 / 1_000_000);
+  const totalCost = inputCost + outputCost;
+  await logger.log(
+    "info",
+    `User ${uuid} ran \`${command}\` which accepted ${usage.input_tokens.toString()} and generated ${usage.output_tokens.toString()}.
+    
+    Total cost: ${totalCost.toLocaleString("en-GB", { currency: "USD", style: "currency" })}`,
+  );
+};