From 77dd08bd6e290bc3f54cc058bdc018ff404e2b5c Mon Sep 17 00:00:00 2001
From: Mitchell Kotler <mitch@mitchellkotler.net>
Date: Wed, 29 Nov 2023 12:50:23 -0500
Subject: [PATCH] use credit cost for premium addons if it is set to more than
 1

---
 src/addons/dispatch/Premium.svelte                 | 9 +++++----
 src/addons/dispatch/stories/Premium.stories.svelte | 8 ++++++++
 src/addons/fixtures/addons.json                    | 7 ++++---
 src/langs/json/en.json                             | 4 +++-
 4 files changed, 20 insertions(+), 8 deletions(-)

diff --git a/src/addons/dispatch/Premium.svelte b/src/addons/dispatch/Premium.svelte
index b289fd01d..a8e03d702 100644
--- a/src/addons/dispatch/Premium.svelte
+++ b/src/addons/dispatch/Premium.svelte
@@ -28,8 +28,7 @@
   $: isIndividualOrg =
     typeof user?.organization !== "string" && user?.organization?.individual;
   $: isPremium = addon?.parameters?.categories?.includes("premium") ?? false;
-  const { amount, unit } = addon?.parameters?.cost ?? {};
-  $: prettyCost = amount && unit ? handlePlural(amount, unit) : null;
+  const { amount, unit, cost } = addon?.parameters?.cost ?? {};
 
   function toggleSpendingLimit(
     event: Event & { currentTarget: HTMLInputElement },
@@ -151,9 +150,11 @@
       <legend>{$_("addonDispatchDialog.premium")}</legend>
       <div class="row">
         <div class="column">
-          {#if prettyCost}
+          {#if amount}
             <h3 class="prettyCost">
-              {prettyCost} per credit
+              {$_("addonDispatchDialog.cost", {
+                values: { amount: amount, unit: unit, cost: cost || 1},
+              })}
             </h3>
           {/if}
           <!-- TODO: Support spend limit (#343)
diff --git a/src/addons/dispatch/stories/Premium.stories.svelte b/src/addons/dispatch/stories/Premium.stories.svelte
index 8a06e455a..2e3c9a06b 100644
--- a/src/addons/dispatch/stories/Premium.stories.svelte
+++ b/src/addons/dispatch/stories/Premium.stories.svelte
@@ -107,3 +107,11 @@
     user: { ...user, organization: { ...groupOrg, monthly_credits: 0 } },
   }}
 />
+<Story
+  name="Azure"
+  args={{ addon: addons[8], user: { ...user, organization: individualOrg } }}
+/>
+<Story
+  name="GPT"
+  args={{ addon: addons[9], user: { ...user, organization: individualOrg } }}
+/>
diff --git a/src/addons/fixtures/addons.json b/src/addons/fixtures/addons.json
index ba7304a40..adf4ab2ea 100644
--- a/src/addons/fixtures/addons.json
+++ b/src/addons/fixtures/addons.json
@@ -252,7 +252,7 @@
       "documents": ["selected"],
       "cost": {
         "amount": 150,
-        "unit": "character"
+        "unit": "characters"
       },
       "categories": ["ai", "premium", "file"],
       "properties": {
@@ -601,8 +601,9 @@
       "title": "Diakopoulos GPT-3 Example: Classifying Comments Based on Subject",
       "documents": ["query", "selected"],
       "cost": {
-        "unit": "token",
-        "amount": 150
+        "unit": "document",
+        "amount": 1,
+        "cost": 14
       },
       "categories": ["ai", "premium"],
       "properties": {
diff --git a/src/langs/json/en.json b/src/langs/json/en.json
index da44d4894..5446839df 100644
--- a/src/langs/json/en.json
+++ b/src/langs/json/en.json
@@ -763,6 +763,7 @@
     "runSuccess": "Add-on is now running",
     "selectionHelp": "From the main list, select individual documents or run a search for the documents you want, for example “+project:mueller-docs-200005”.",
     "selectionLearnMore": "Learn more about how Add-Ons work",
+    "cost": "{amount} {unit} per {cost, plural, one {credit} other {# credits}}",
     "premiumUpgrade": {
       "message": "This Premium Add-On uses AI to perform advanced analysis. Upgrade to a {plan} account to utilize this and other powerful Add-Ons.",
       "callToAction": "Upgrade Plan",
@@ -828,5 +829,6 @@
     "p3": "Want more fascinating documents, open data and original reporting to your inbox? Subscribe to MuckRock's newsletter:",
     "p4": "DocumentCloud is part of a suite of transparency tools from the MuckRock Foundation, a 501c3 registered non-profit. This archive is open to the public and advertisement free thanks to support from readers like you — you can <a href=\"https://www.muckrock.com/about/\">learn more about our work</a> or <a href=\"https://www.muckrock.com/donate/\">make a donation</a>.",
     "subscribe": "Subscribe"
-  }
+  },
+  "premium": {}
 }