diff --git a/common/requests/util.ts b/common/requests/util.ts
index c2461368d..fcf908f52 100644
--- a/common/requests/util.ts
+++ b/common/requests/util.ts
@@ -66,13 +66,14 @@ export function trimResponseV2(
     generated = generated.split(`${member.handle} :`).join(`${member.handle}:`)
   }
 
-  if (bots) {
-    for (const bot of Object.values(bots)) {
-      if (!bot) continue
-      if (bot?._id === char._id) continue
-      endTokens.push(`${bot.name}:`)
-    }
-  }
+  /** Do not always add character names as stop tokens here */
+  // if (bots) {
+  //   for (const bot of Object.values(bots)) {
+  //     if (!bot) continue
+  //     if (bot?._id === char._id) continue
+  //     endTokens.push(`${bot.name}:`)
+  //   }
+  // }
 
   let index = -1
   let trimmed = allEndTokens.concat(...endTokens).reduce((prev, endToken) => {
diff --git a/common/types/ui.ts b/common/types/ui.ts
index 42a98ec2e..f424bb5d8 100644
--- a/common/types/ui.ts
+++ b/common/types/ui.ts
@@ -42,7 +42,7 @@ export type CustomUI = {
   chatQuoteColor: string
 }
 
-export type MessageOption = 'edit' | 'regen' | 'trash' | 'fork' | 'prompt'
+export type MessageOption = 'edit' | 'regen' | 'trash' | 'fork' | 'prompt' | 'schema-regen'
 
 export type UISettings = {
   theme: string
@@ -147,5 +147,6 @@ export const defaultUIsettings: UISettings = {
     fork: { outer: false, pos: 2 },
     regen: { outer: true, pos: 1 },
     trash: { outer: false, pos: 4 },
+    'schema-regen': { outer: false, pos: 5 },
   },
 }
diff --git a/srv/adapter/agnaistic.ts b/srv/adapter/agnaistic.ts
index 97635b695..3979d95e3 100644
--- a/srv/adapter/agnaistic.ts
+++ b/srv/adapter/agnaistic.ts
@@ -251,9 +251,15 @@ export const handleAgnaistic: ModelAdapter = async function* (opts) {
 
   log.debug(`Prompt:\n${prompt}`)
 
-  const [submodel, override] = subPreset.subModel.split(',')
-
-  let params = [`type=text`, `id=${opts.user._id}`, `model=${submodel}`, `level=${level}`]
+  const [submodel, override = ''] = subPreset.subModel.split(',')
+
+  let params = [
+    `type=text`,
+    `id=${opts.user._id}`,
+    `model=${submodel}`,
+    `level=${level}`,
+    `sub_model=${override}`,
+  ]
     .filter((p) => !!p)
     .join('&')
 
diff --git a/srv/adapter/generate.ts b/srv/adapter/generate.ts
index 56789e0c9..b3c0e500a 100644
--- a/srv/adapter/generate.ts
+++ b/srv/adapter/generate.ts
@@ -438,6 +438,7 @@ export async function createChatStream(
     lastMessage: opts.lastMessage,
     imageData: opts.imageData,
     jsonSchema: jsonSchema || opts.jsonSchema,
+    reschemaPrompt: opts.reschemaPrompt,
     subscription,
     encoder,
     jsonValues: opts.jsonValues,
diff --git a/srv/adapter/payloads.ts b/srv/adapter/payloads.ts
index 37b417ca0..5497d8060 100644
--- a/srv/adapter/payloads.ts
+++ b/srv/adapter/payloads.ts
@@ -114,6 +114,7 @@ function getBasePayload(opts: AdapterProps, stops: string[] = []) {
       lists: opts.lists,
       previous: opts.previous,
       json_schema_v2: ensureSafeSchema(json_schema),
+      reschema_prompt: opts.reschemaPrompt,
       json_schema,
       imageData: opts.imageData,
       context_size: opts.contextSize,
diff --git a/srv/adapter/type.ts b/srv/adapter/type.ts
index 7bc46ec07..36d11a0c5 100644
--- a/srv/adapter/type.ts
+++ b/srv/adapter/type.ts
@@ -69,6 +69,7 @@ export type GenerateRequestV2 = {
   impersonate?: AppSchema.Character
 
   jsonSchema?: JsonField[]
+  reschemaPrompt?: string
   jsonValues?: Record<string, any>
 
   /** Base64 */
@@ -122,6 +123,7 @@ export type AdapterProps = {
   encoder?: TokenCounter
 
   jsonSchema?: any
+  reschemaPrompt?: string
   jsonValues: Record<string, any> | undefined
 
   imageData?: string
diff --git a/srv/api/billing/checkout.ts b/srv/api/billing/checkout.ts
index e71d8c13c..8b9dcdedf 100644
--- a/srv/api/billing/checkout.ts
+++ b/srv/api/billing/checkout.ts
@@ -6,6 +6,7 @@ import { v4 } from 'uuid'
 import { config } from '../../config'
 import { billingCmd, domain } from '../../domains'
 import { subsCmd } from '../../domains/subs/cmd'
+import { AppSchema } from '/common/types'
 
 export const startCheckout = handle(async ({ body, userId }) => {
   assertValid({ tierId: 'string', callback: 'string' }, body)
@@ -180,6 +181,10 @@ export const finishCheckout = handle(async ({ body, userId }) => {
       subscriptionId: subscription.id,
     })
 
+    if (user && subscription.id) {
+      await ensureOnlyActiveSubscription(user, subscription.id)
+    }
+
     const config = await store.users.updateUser(userId, {
       sub: {
         tierId: agg.tierId,
@@ -205,3 +210,22 @@ export const finishCheckout = handle(async ({ body, userId }) => {
     await billingCmd.cancel(body.sessionId, { userId })
   }
 })
+
+async function ensureOnlyActiveSubscription(user: AppSchema.User, subscriptionId: string) {
+  // The user isn't an existing customer -- ignore
+  if (!user.billing?.customerId) return
+
+  const subs = await stripe.subscriptions
+    .list({ customer: user.billing.customerId, status: 'active' })
+    .then((res) => res.data)
+    .catch(() => [])
+
+  for (const sub of subs) {
+    if (sub.id !== subscriptionId) {
+      await subsCmd.cancelDuplicate(user._id, { subscriptionId, replacementId: subscriptionId })
+      await stripe.subscriptions.cancel(sub.id, {
+        cancellation_details: { comment: 'duplicate detected during checkout' },
+      })
+    }
+  }
+}
diff --git a/srv/domains/subs/agg.ts b/srv/domains/subs/agg.ts
index c8c617628..6d7a11d4e 100644
--- a/srv/domains/subs/agg.ts
+++ b/srv/domains/subs/agg.ts
@@ -55,6 +55,14 @@ export const subsAgg = createAggregate<SubsEvt, SubsAgg, 'subscriptions'>({
         }
       }
 
+      case 'cancelled-duplicate': {
+        return {
+          state: prev.state,
+          downgrade: prev.downgrade,
+          history,
+        }
+      }
+
       case 'cancelled': {
         const endAt = new Date(prev.periodStart)
         endAt.setFullYear(meta.timestamp.getFullYear())
diff --git a/srv/domains/subs/cmd.ts b/srv/domains/subs/cmd.ts
index ffb11bf54..10aba2c06 100644
--- a/srv/domains/subs/cmd.ts
+++ b/srv/domains/subs/cmd.ts
@@ -96,6 +96,15 @@ export const subsCmd = createCommands<SubsEvt, SubsAgg, SubsCmd>(domain.subscrip
 
     return { type: 'resumed' }
   },
+
+  async cancelDuplicate(cmd, agg) {
+    return {
+      type: 'cancelled-duplicate',
+      subscriptionId: cmd.subscriptionId,
+      replacementId: cmd.replacementId,
+    }
+  },
+
   async cancel(cmd, agg) {
     if (agg.state !== 'active') {
       throw new CommandError('Cannot cancel subscription - Subscription not active', 'NOT_ACTIVE')
diff --git a/srv/domains/subs/types.ts b/srv/domains/subs/types.ts
index 5502ad5b0..2a075935b 100644
--- a/srv/domains/subs/types.ts
+++ b/srv/domains/subs/types.ts
@@ -10,6 +10,7 @@ export type SubsEvt =
       periodStart: string
     }
   | { type: 'cancelled' }
+  | { type: 'cancelled-duplicate'; subscriptionId: string; replacementId?: string }
   | { type: 'upgraded'; tierId: string; priceId: string }
   | { type: 'downgraded'; tierId: string; priceId: string; activeAt: string }
   | { type: 'session-started'; sessionId: string; tierId: string }
@@ -36,6 +37,7 @@ export type SubsCmd =
       productId: string
       subscription: Stripe.Subscription
     }
+  | { type: 'cancelDuplicate'; subscriptionId: string; replacementId?: string }
   | { type: 'cancel' }
   | { type: 'resume' }
   | { type: 'upgrade'; tierId: string; priceId: string }
diff --git a/web/pages/Character/components/CharacterFolderView.tsx b/web/pages/Character/components/CharacterFolderView.tsx
index eb5a7ff1e..353434020 100644
--- a/web/pages/Character/components/CharacterFolderView.tsx
+++ b/web/pages/Character/components/CharacterFolderView.tsx
@@ -442,15 +442,15 @@ function getChildFolders(tree: FolderTree, path: string, sort: SortDirection) {
 }
 
 const ChangeFolder: Component<{ char?: AppSchema.Character; close: () => void }> = (props) => {
-  const [actual, setActual] = createSignal('/')
+  const [name, setName] = createSignal(props.char?.folder || '')
+  const actual = createMemo(() => toFolderSlug(name()))
 
   createEffect(
     on(
       () => props.char,
       () => {
         if (!props.char) return
-
-        setActual(toFolderSlug(props.char.folder || '/'))
+        setName(props.char.folder || '/')
       }
     )
   )
@@ -489,10 +489,10 @@ const ChangeFolder: Component<{ char?: AppSchema.Character; close: () => void }>
         />
 
         <TextInput
-          helperMarkdown={`Folder names are 'normalized'.\nNoramlized name: ${actual()}`}
+          helperMarkdown={`Folder names are 'normalized'.\nNormalized name: ${actual()}`}
           label="New Folder"
-          value={props.char?.folder || '/'}
-          onChange={(ev) => setActual(toFolderSlug(ev.currentTarget.value))}
+          value={name()}
+          onChange={(ev) => setName(ev.currentTarget.value)}
         />
       </div>
     </RootModal>
diff --git a/web/pages/Chat/components/Message.tsx b/web/pages/Chat/components/Message.tsx
index c0fa42733..3ce6312d6 100644
--- a/web/pages/Chat/components/Message.tsx
+++ b/web/pages/Chat/components/Message.tsx
@@ -16,6 +16,7 @@ import {
   Zap,
   Split,
   MoreHorizontal,
+  Braces,
 } from 'lucide-solid'
 import {
   Accessor,
@@ -548,10 +549,6 @@ const MessageOptions: Component<{
   onRemove: () => void
   showMore: Signal<boolean>
 }> = (props) => {
-  const showInner = createMemo(() =>
-    Object.values(props.ui.msgOptsInline || {}).some((v) => !v?.outer)
-  )
-
   const closer = (action: () => void) => {
     return () => {
       action()
@@ -617,6 +614,20 @@ const MessageOptions: Component<{
         icon: RefreshCw,
       },
 
+      'schema-regen': {
+        key: 'schema-regen',
+        class: 'refresh-btn',
+        label: 'Schema Regen',
+        outer: props.ui.msgOptsInline['schema-regen'],
+        show:
+          window.flags.reschema &&
+          ((props.msg.json && props.last) ||
+            (props.msg.adapter === 'image' && !!props.msg.imagePrompt)) &&
+          !!props.msg.characterId,
+        onClick: () => !props.partial && retryJsonSchema(props.msg, props.msg),
+        icon: Braces,
+      },
+
       trash: {
         key: 'trash',
         label: 'Delete',
@@ -632,6 +643,14 @@ const MessageOptions: Component<{
     return items
   })
 
+  const showInner = createMemo(() => {
+    for (const opt of Object.values(logic())) {
+      if (!opt.outer && opt.show) return true
+    }
+
+    return false
+  })
+
   const order = createMemo(() => {
     open()
     logic()
@@ -736,7 +755,7 @@ const MessageOption: Component<{
 
         <Show when={!props.outer}>
           <Button
-            class={`${props.class || ''} w-full`}
+            class={`${props.class || ''} w-full min-w-max`}
             schema={props.schema || 'secondary'}
             onClick={props.onClick}
             size="sm"
@@ -758,6 +777,10 @@ function retryMessage(original: AppSchema.ChatMessage, split: SplitMessage) {
   }
 }
 
+function retryJsonSchema(original: AppSchema.ChatMessage, split: SplitMessage) {
+  msgStore.retrySchema(split.chatId, original._id)
+}
+
 function renderMessage(ctx: ContextState, text: string, isUser: boolean, adapter?: string) {
   // Address unfortunate Showdown bug where spaces in code blocks are replaced with nbsp, except
   // it also encodes the ampersand, which results in them actually being rendered as `&amp;nbsp;`
diff --git a/web/pages/Settings/UISettings.tsx b/web/pages/Settings/UISettings.tsx
index 46ec8a28a..2c2d41e06 100644
--- a/web/pages/Settings/UISettings.tsx
+++ b/web/pages/Settings/UISettings.tsx
@@ -27,6 +27,7 @@ const msgInlineLabels: Record<UI.MessageOption, string> = {
   prompt: 'Prompt View',
   fork: 'Fork',
   trash: 'Delete',
+  'schema-regen': 'Retry Schema',
 }
 
 const UISettings: Component<{}> = () => {
diff --git a/web/shared/CustomSelect.tsx b/web/shared/CustomSelect.tsx
index 44a693e6a..de4692c46 100644
--- a/web/shared/CustomSelect.tsx
+++ b/web/shared/CustomSelect.tsx
@@ -14,7 +14,8 @@ export type CustomOption = {
 export const CustomSelect: Component<{
   buttonLabel: string | JSX.Element | ((opt: CustomOption) => JSX.Element | string)
   onSelect: (opt: CustomOption) => void
-  options: CustomOption[]
+  options?: CustomOption[]
+  categories?: Array<{ name: string; options: CustomOption[] }>
   value: any
 
   header?: JSX.Element
@@ -74,15 +75,33 @@ export const CustomSelect: Component<{
       </div>
       <RootModal show={open()} close={() => setOpen(false)} title={props.modalTitle}>
         <div class="flex flex-col gap-4">
-          <div class="flex flex-wrap gap-2 pr-3">
-            <OptionList
-              header={props.header}
-              options={props.options}
-              onSelect={onSelect}
-              selected={props.selected}
-              search={props.search}
-            />
-          </div>
+          <Show when={props.categories}>
+            <For each={props.categories}>
+              {(category) => (
+                <div class="flex flex-wrap gap-2 pr-3">
+                  <div class="bold text-md">{category.name}</div>
+                  <OptionList
+                    header={props.header}
+                    options={category.options}
+                    onSelect={onSelect}
+                    selected={props.selected}
+                    search={props.search}
+                  />
+                </div>
+              )}
+            </For>
+          </Show>
+          <Show when={props.options}>
+            <div class="flex flex-wrap gap-2 pr-3">
+              <OptionList
+                header={props.header}
+                options={props.options!}
+                onSelect={onSelect}
+                selected={props.selected}
+                search={props.search}
+              />
+            </div>
+          </Show>
         </div>
       </RootModal>
     </div>
diff --git a/web/shared/PersonaAttributes.tsx b/web/shared/PersonaAttributes.tsx
index edfa192f6..b0ae2d826 100644
--- a/web/shared/PersonaAttributes.tsx
+++ b/web/shared/PersonaAttributes.tsx
@@ -70,6 +70,7 @@ const PersonaAttributes: Component<{
             if (key === 'text') {
               squished.push(values)
             } else {
+              if (!values.trim()) continue
               squished.push(`${key}:\n${values}`)
             }
           }
@@ -82,6 +83,11 @@ const PersonaAttributes: Component<{
           const text = props.state.find((s) => s.key === 'text')
           if (!text) return
 
+          if (props.state.length === 1) {
+            props.setter([{ key: 'personality', values: props.state[0].values }])
+            return
+          }
+
           let matching = true
           for (const { values } of props.state) {
             if (!text.values.includes(values)) matching = false
diff --git a/web/shared/PresetSettings/Agnaistic.tsx b/web/shared/PresetSettings/Agnaistic.tsx
index 6a925f25d..97a15db34 100644
--- a/web/shared/PresetSettings/Agnaistic.tsx
+++ b/web/shared/PresetSettings/Agnaistic.tsx
@@ -16,6 +16,8 @@ import { useAppContext } from '/web/store/context'
 
 export const AgnaisticSettings: Field<{ noSave: boolean }> = (props) => {
   const opts = useModelOptions()
+  const categories = useModelCategories()
+
   const [ctx] = useAppContext()
 
   const onSave = (value: string) => {
@@ -92,7 +94,8 @@ export const AgnaisticSettings: Field<{ noSave: boolean }> = (props) => {
             Model <span class="text-500 text-xs italic">(Available: {opts().length})</span>
           </>
         }
-        options={opts()}
+        // options={opts()}
+        categories={categories}
         onSelect={(ev) => onSave(ev.value)}
         value={props.state.registered?.agnaistic?.subscriptionId}
         selected={props.state.registered?.agnaistic?.subscriptionId}
@@ -107,6 +110,7 @@ export const AgnaisticModel: Component = (props) => {
 
   const [selected, setSelected] = createSignal(ctx.preset?.registered?.agnaistic?.subscriptionId)
   const opts = useModelOptions()
+  const categories = useModelCategories()
 
   createEffect(
     on(
@@ -175,7 +179,8 @@ export const AgnaisticModel: Component = (props) => {
             </div>
           </div>
         }
-        options={opts()}
+        // options={opts()}
+        categories={categories}
         onSelect={onSave}
         value={ctx.preset?.registered?.agnaistic?.subscriptionId}
         selected={selected()}
@@ -185,6 +190,88 @@ export const AgnaisticModel: Component = (props) => {
   )
 }
 
+function useModelCategories() {
+  const state = userStore((s) => ({
+    user: s.user,
+    tiers: s.tiers,
+    sub: s.sub,
+    userLevel: s.userLevel,
+  }))
+
+  const settings = settingStore()
+
+  const tierLevel = state.user?.admin ? Infinity : state.userLevel
+  const level = state.user?.admin ? Infinity : tierLevel
+
+  const cats = new Map<
+    string,
+    {
+      options: Array<{
+        label: any
+        value: string
+        level: number
+        sub: any
+        limit?: SubscriptionModelLevel
+        disabled: boolean
+        tierName: string
+        requires: string
+        title: string
+      }>
+      tier: number
+    }
+  >()
+
+  for (const sub of settings.config.subs) {
+    const limit = getSubscriptionModelLimits(sub.preset, level)
+    const disabled = !!sub.preset.allowGuestUsage ? false : sub.level > level
+    const tier =
+      sub.level <= 0
+        ? 'Free'
+        : state.tiers.reduce<SubscriptionTier | null>((prev, curr) => {
+            if (prev?.level === sub.level) return prev
+            if (curr.level === sub.level) return curr
+            if (curr.level < sub.level) return prev
+            if (!prev) return curr
+
+            // Return the lowest tier above the threshold
+            return prev.level > curr.level ? curr : prev
+          }, null)?.name
+
+    const requires = sub.level <= 0 ? 'Registering' : tier || 'Staff Only'
+    const tierName = tier || 'Staff'
+
+    const category = cats.get(tierName) || { tier: sub.level, options: [] }
+    category.options.push({
+      label: (
+        <ModelLabel
+          sub={sub}
+          limit={limit}
+          disabled={disabled}
+          tier={tier || 'Staff'}
+          requires={requires}
+        />
+      ),
+      value: sub._id,
+      level: sub.level,
+      sub,
+      limit,
+      disabled,
+      tierName: tier || 'Staff',
+      requires,
+      title: sub.name,
+    })
+    cats.set(tierName, category)
+  }
+
+  const categories = Array.from(cats.entries())
+    .sort((l, r) => l[1].tier - r[1].tier)
+    .map(([name, item]) => ({
+      name,
+      options: item.options.sort((l, r) => l.title.localeCompare(r.title)),
+    }))
+  return categories
+}
+
 function useModelOptions() {
   const state = userStore((s) => ({
     user: s.user,
@@ -277,8 +364,8 @@ const ModelLabel: Component<{
       </div>
       <Show when={!props.disabled && !props.nodesc}>
         <div class="text-700 text-xs">
-          {props.tier}
-          {props.sub.preset.description ? ', ' : ''}
+          {/* {props.tier}
+          {props.sub.preset.description ? ', ' : ''} */}
           {props.sub.preset.description}
         </div>
       </Show>
diff --git a/web/store/data/bot-generate.ts b/web/store/data/bot-generate.ts
index 42bef1e55..e640a42f9 100644
--- a/web/store/data/bot-generate.ts
+++ b/web/store/data/bot-generate.ts
@@ -49,7 +49,7 @@ export type GenerateOpts =
    * - The last message in the chat is a user message so we are going to generate a new response
    * - The last message in the chat is a bot message so we are going to re-generate a response and update the 'replacingId' chat message
    */
-  | { kind: 'retry'; messageId?: string }
+  | { kind: 'retry'; messageId?: string; reschema_prompt?: string }
   /**
    * The last message in the chat is a bot message and we want to generate more text for this message.
    */
@@ -122,6 +122,7 @@ export async function generateResponse(
     chatEmbeds,
     userEmbeds,
     jsonValues: props.json,
+    reschemaPrompt: props.reschemaPrompt,
   }
 
   if (
@@ -358,6 +359,7 @@ export type GenerateProps = {
   impersonate?: AppSchema.Character
   parent?: AppSchema.ChatMessage
   json: Record<string, any>
+  reschemaPrompt?: string
 }
 
 async function getGenerateProps(
@@ -418,6 +420,8 @@ async function getGenerateProps(
   switch (opts.kind) {
     case 'retry': {
       props.impersonate = entities.impersonating
+      props.reschemaPrompt = opts.reschema_prompt
+
       if (opts.messageId) {
         // Case: When regenerating a response that isn't last. Typically when image messages follow the last text message
         const index = entities.messages.findIndex((msg) => msg._id === opts.messageId)
diff --git a/web/store/flags.ts b/web/store/flags.ts
index 13f16fd68..3165838d6 100644
--- a/web/store/flags.ts
+++ b/web/store/flags.ts
@@ -13,4 +13,5 @@ export const defaultFlags = {
   folders: false,
   sounds: false,
   google: false,
+  reschema: false,
 } satisfies { [key: string]: boolean }
diff --git a/web/store/message.ts b/web/store/message.ts
index bd75dec08..9435a3ffe 100644
--- a/web/store/message.ts
+++ b/web/store/message.ts
@@ -428,12 +428,7 @@ export const msgStore = createStore<MsgState>(
       yield { msgs: page, messageHistory: path }
     },
 
-    async *retry(
-      { msgs, activeCharId },
-      chatId: string,
-      messageId?: string,
-      onSuccess?: () => void
-    ) {
+    async *retry({ msgs, activeCharId }, chatId: string, messageId?: string) {
       if (!chatId) {
         toastStore.error('Could not send message: No active chat')
         yield { partial: undefined }
@@ -441,7 +436,7 @@ export const msgStore = createStore<MsgState>(
       }
 
       if (msgs.length === 0) {
-        msgStore.request(chatId, activeCharId, onSuccess)
+        msgStore.request(chatId, activeCharId)
         return
       }
 
@@ -461,8 +456,43 @@ export const msgStore = createStore<MsgState>(
       if (res.error) {
         toastStore.error(`(Retry) Generation request failed: ${res.error}`)
         yield { partial: undefined, waiting: undefined, retrying: undefined }
-      } else if (res.result) {
-        onSuccess?.()
+      }
+    },
+
+    async *retrySchema({ msgs, activeCharId }, chatId: string, messageId: string) {
+      if (!chatId) {
+        toastStore.error('Could not send message: No active chat')
+        yield { partial: undefined }
+        return
+      }
+
+      if (msgs.length === 0) {
+        msgStore.request(chatId, activeCharId)
+        return
+      }
+
+      const msg = msgs.find((msg) => msg._id === messageId)
+      if (!msg) {
+        toastStore.error(`Could not regenerate: Message not found`)
+        yield { partial: undefined }
+        return
+      }
+
+      const replace = msg?.userId ? undefined : { ...msg, voiceUrl: undefined }
+      const characterId = replace?.characterId || activeCharId
+      yield {
+        partial: '',
+        waiting: { chatId, mode: 'retry', characterId },
+        retrying: replace,
+      }
+
+      const res = await botGen
+        .generate({ kind: 'retry', messageId, reschema_prompt: msg.json?.values.response })
+        .catch((err) => ({ error: err.message, result: undefined }))
+
+      if (res.error) {
+        toastStore.error(`(Retry) Generation request failed: ${res.error}`)
+        yield { partial: undefined, waiting: undefined, retrying: undefined }
       }
     },