nolo-cli 0.1.29 → 0.1.31

package/ai/chat/sendOpenAICompletionsRequest.ts

@@ -12,6 +12,7 @@ import {
 import { handleToolCalls } from "../../chat/messages/toolThunks";
 import {
   MessageContentPart,
+  ImageGenerationState,
   Message,
   OpenAITextContent,
 } from "../../chat/messages/types";
@@ -34,6 +35,7 @@ import {
 } from "./inlineImageUrlsForCustomProvider";
 
 import { getModelInfo } from "../llm/getModelContextWindow";
+import { supportsImageGeneration } from "../agent/utils/imageOutput";
 
 import type { RootState } from "../../app/store";
 
@@ -69,6 +71,13 @@ type AssistantToolCall = {
   function: { name: string; arguments: string };
 };
 
+const EXPLICIT_IMAGE_TOOL_NAMES = new Set([
+  "openAIGptImage",
+  "openAIGptImageGenerate",
+  "openAIGptImageEdit",
+  "geminiProImagePreview",
+]);
+
 function getStreamErrorMessage(data: any): string {
   const message =
     typeof data?.error?.message === "string" && data.error.message.trim()
@@ -150,6 +159,33 @@ type StreamCompletionContext = {
   agentConfig: any;
 };
 
+const withImageGenerationStage = (
+  messageMetadata: Partial<Message> | undefined,
+  stage: ImageGenerationState["stage"]
+): Partial<Message> | undefined => {
+  const previousState = messageMetadata?.imageGenerationState;
+  if (previousState?.kind !== "image_generation") {
+    return messageMetadata;
+  }
+
+  return {
+    ...(messageMetadata ?? {}),
+    imageGenerationState: {
+      ...previousState,
+      stage,
+    },
+  };
+};
+
+const shouldShowImageSavingState = (contentBuffer: MessageContentPart[]) =>
+  contentBuffer.some((part) => part?.type === "image_url") &&
+  !contentBuffer.some(
+    (part) =>
+      part?.type === "text" &&
+      typeof part.text === "string" &&
+      part.text.trim().length > 0
+  );
+
 /** 单轮调用后返回给 Agent Loop 的元信息 */
 export type CompletionMeta = {
   hasToolCalls: boolean;
@@ -212,6 +248,13 @@ function buildRequestBodyWithTools(
 
   const rawTools = agentConfig.tools;
   if (!Array.isArray(rawTools) || rawTools.length === 0) return bodyData;
+  const hasExplicitImageTool = rawTools.some(
+    (toolName: unknown) =>
+      typeof toolName === "string" && EXPLICIT_IMAGE_TOOL_NAMES.has(toolName)
+  );
+  if (supportsImageGeneration(agentConfig) && !hasExplicitImageTool) {
+    return bodyData;
+  }
 
   const tools = prepareTools(rawTools, { provider: agentConfig.provider });
   if (!tools.length) return bodyData;
@@ -232,6 +275,30 @@ async function finalizeStream(
 ): Promise<StreamState> {
   if (state.hasHandedOff || state.alreadyFinalized) return state;
 
+  const savingMetadata = withImageGenerationStage(
+    ctx.messageMetadata,
+    "saving"
+  );
+  if (savingMetadata && shouldShowImageSavingState(state.contentBuffer)) {
+    ctx.dispatch(
+      messageStreaming({
+        id: ctx.messageId,
+        dialogId: ctx.dialogId,
+        dbKey: ctx.msgKey,
+        content: "",
+        thinkContent: state.reasoningBuffer,
+        role: "assistant",
+        agentKey: ctx.agentConfig.dbKey,
+        cybotKey: ctx.agentConfig.dbKey,
+        ...(typeof ctx.agentConfig?.name === "string" && ctx.agentConfig.name.trim()
+          ? { agentName: ctx.agentConfig.name.trim() }
+          : {}),
+        ...(savingMetadata ?? {}),
+      })
+    );
+    ctx.messageMetadata = savingMetadata;
+  }
+
   await ctx.dispatch(
     messageStreamEnd({
       finalContentBuffer: state.contentBuffer,
@@ -556,6 +623,7 @@ export const sendOpenAICompletionsRequest = async ({
   let hasHandedOffOverall = false;
   let hasPendingInteractionOverall = false;
   let lastFinishReason: string | null = null;
+  let activeMessageMetadata = messageMetadata;
 
   const buildMeta = (): CompletionMeta => ({
     hasToolCalls:
@@ -582,7 +650,7 @@ export const sendOpenAICompletionsRequest = async ({
           ...(typeof agentConfig?.name === "string" && agentConfig.name.trim()
             ? { agentName: agentConfig.name.trim() }
             : {}),
-          ...(messageMetadata ?? {}),
+          ...(activeMessageMetadata ?? {}),
           isStreaming: true,
         })
       );
@@ -607,6 +675,29 @@ export const sendOpenAICompletionsRequest = async ({
       status: response.status,
       dialogKey,
     });
+    activeMessageMetadata = withImageGenerationStage(
+      activeMessageMetadata,
+      "generating"
+    );
+    finalizeCtx.messageMetadata = activeMessageMetadata;
+    if (!parentMessageId && activeMessageMetadata?.imageGenerationState) {
+      dispatch(
+        messageStreaming({
+          id: messageId,
+          dialogId,
+          dbKey: msgKey,
+          content: "",
+          role: "assistant",
+          agentKey: agentConfig.dbKey,
+          cybotKey: agentConfig.dbKey,
+          ...(typeof agentConfig?.name === "string" && agentConfig.name.trim()
+            ? { agentName: agentConfig.name.trim() }
+            : {}),
+          ...(activeMessageMetadata ?? {}),
+          isStreaming: true,
+        })
+      );
+    }
 
     if (!response.ok) {
       const errorMessage = await parseApiError(response);
@@ -726,6 +817,7 @@ export const sendOpenAICompletionsRequest = async ({
             messageId,
             msgKey,
             dialogId,
+            messageMetadata: activeMessageMetadata,
           });
 
           const finishReason = choice.finish_reason;

package/ai/chat/sendOpenAIResponseRequest.ts

@@ -8,7 +8,7 @@ import {
   messageStreamEnd,
 } from "../../chat/messages/messageSlice";
 import { handleToolCalls } from "../../chat/messages/toolThunks";
-import type { Message } from "../../chat/messages/types";
+import type { ImageGenerationState, Message } from "../../chat/messages/types";
 import { selectRuntimeCurrentServer } from "../../app/stateViews/runtime";
 import { selectCurrentSpaceId } from "../../create/space/spaceSlice";
 import { getApiEndpoint } from "../llm/providers";
@@ -42,6 +42,34 @@ const shouldEnableBuiltInImageGeneration = (agentConfig: any): boolean =>
   !getModelInfo(String(agentConfig?.model || ""))?.hasImageOutput &&
   !!agentConfig?.imageConfig?.enabled;
 
+const withImageGenerationStage = (
+  messageMetadata: Partial<Message> | undefined,
+  stage: ImageGenerationState["stage"]
+): Partial<Message> | undefined => {
+  const previousState = messageMetadata?.imageGenerationState;
+  if (previousState?.kind !== "image_generation") {
+    return messageMetadata;
+  }
+
+  return {
+    ...(messageMetadata ?? {}),
+    imageGenerationState: {
+      ...previousState,
+      stage,
+    },
+  };
+};
+
+const shouldShowImageSavingState = (
+  contentBuffer: Array<
+    { type: "text"; text: string } | { type: "image_url"; image_url: { url: string } }
+  >
+) =>
+  contentBuffer.some((part) => part?.type === "image_url") &&
+  !contentBuffer.some(
+    (part) => part?.type === "text" && typeof part.text === "string" && part.text.trim()
+  );
+
 const logQuickChatPerfStage = (
   startedAt: number | undefined,
   stage: string,
@@ -193,6 +221,7 @@ export const sendOpenAIResponseRequest = async ({
     assistantToolCalls: [],
     completedResponse: null,
   };
+  let activeMessageMetadata = messageMetadata;
 
   const buildMeta = (
     hasPendingInteraction = false,
@@ -221,7 +250,7 @@ export const sendOpenAIResponseRequest = async ({
       ...(typeof agentConfig?.name === "string" && agentConfig.name.trim()
         ? { agentName: agentConfig.name.trim() }
         : {}),
-      ...(messageMetadata ?? {}),
+      ...(activeMessageMetadata ?? {}),
     })
   );
 
@@ -250,7 +279,33 @@ export const sendOpenAIResponseRequest = async ({
       state.contentBuffer = seg(state.content);
     }
 
-    flush();
+    activeMessageMetadata = withImageGenerationStage(
+      activeMessageMetadata,
+      "saving"
+    );
+    if (
+      activeMessageMetadata?.imageGenerationState &&
+      shouldShowImageSavingState(state.contentBuffer)
+    ) {
+      dispatch(
+        messageStreaming({
+          id: messageId,
+          dialogId,
+          dbKey: msgKey,
+          content: "",
+          thinkContent: state.reasoning,
+          role: "assistant",
+          agentKey: agentConfig.dbKey,
+          cybotKey: agentConfig.dbKey,
+          ...(typeof agentConfig?.name === "string" && agentConfig.name.trim()
+            ? { agentName: agentConfig.name.trim() }
+            : {}),
+          ...(activeMessageMetadata ?? {}),
+        })
+      );
+    } else {
+      flush();
+    }
     await dispatch(
       messageStreamEnd({
         finalContentBuffer: state.contentBuffer,
@@ -261,7 +316,7 @@ export const sendOpenAIResponseRequest = async ({
         dialogKey,
         messageId,
         reasoningBuffer: state.reasoning,
-        messageMetadata,
+        messageMetadata: activeMessageMetadata,
         toolCalls: state.assistantToolCalls,
         spaceId: streamSpaceId,
       })
@@ -294,10 +349,10 @@ export const sendOpenAIResponseRequest = async ({
         messageId,
         dialogId,
         dialogKey,
-        parallelSessionId: messageMetadata?.parallelSessionId,
-        parallelBranchId: messageMetadata?.parallelBranchId,
-        parallelLabel: messageMetadata?.parallelLabel,
-        parallelIndex: messageMetadata?.parallelIndex,
+        parallelSessionId: activeMessageMetadata?.parallelSessionId,
+        parallelBranchId: activeMessageMetadata?.parallelBranchId,
+        parallelLabel: activeMessageMetadata?.parallelLabel,
+        parallelIndex: activeMessageMetadata?.parallelIndex,
       })
     ).unwrap();
 
@@ -331,7 +386,7 @@ export const sendOpenAIResponseRequest = async ({
           ...(typeof agentConfig?.name === "string" && agentConfig.name.trim()
             ? { agentName: agentConfig.name.trim() }
             : {}),
-          ...(messageMetadata ?? {}),
+          ...(activeMessageMetadata ?? {}),
           isStreaming: true,
         })
       );
@@ -388,6 +443,28 @@ export const sendOpenAIResponseRequest = async ({
       status: response.status,
       dialogKey,
     });
+    activeMessageMetadata = withImageGenerationStage(
+      activeMessageMetadata,
+      "generating"
+    );
+    if (!parentMessageId && activeMessageMetadata?.imageGenerationState) {
+      dispatch(
+        messageStreaming({
+          id: messageId,
+          dialogId,
+          dbKey: msgKey,
+          content: "",
+          role: "assistant",
+          agentKey: agentConfig.dbKey,
+          cybotKey: agentConfig.dbKey,
+          ...(typeof agentConfig?.name === "string" && agentConfig.name.trim()
+            ? { agentName: agentConfig.name.trim() }
+            : {}),
+          ...(activeMessageMetadata ?? {}),
+          isStreaming: true,
+        })
+      );
+    }
 
     if (!response.ok) {
       const errorMessage = await parseApiError(response);

package/ai/chat/updateTotalUsage.ts

@@ -85,5 +85,12 @@ export function updateTotalUsage(
     updatedUsage.billing_model = newUsageChunk.billing_model.trim();
   }
 
+  if (
+    typeof newUsageChunk.billing_service_tier === "string" &&
+    newUsageChunk.billing_service_tier.trim()
+  ) {
+    updatedUsage.billing_service_tier = newUsageChunk.billing_service_tier.trim();
+  }
+
   return updatedUsage;
 }

package/ai/llm/deepinfra.ts

@@ -1,15 +1,30 @@
 // ai/llm/deepinfra.ts
+import { scaleProviderPrice } from "../policy/commercialMultiplier";
 import { DEEPINFRA_KIMI_FALLBACK_MODEL } from "./kimi";
 
+const price = (rawProviderPrice: number, model: string) =>
+  scaleProviderPrice(rawProviderPrice, "deepinfra", model);
+
 export const deepinfraModels = [
+  {
+    name: "meta-llama/Meta-Llama-3.1-8B-Instruct-Turbo",
+    displayName: "Meta Llama 3.1 8B Turbo (DeepInfra)",
+    hasVision: false,
+    price: {
+      input: price(0.02, "meta-llama/Meta-Llama-3.1-8B-Instruct-Turbo"),
+      output: price(0.03, "meta-llama/Meta-Llama-3.1-8B-Instruct-Turbo"),
+    },
+    contextWindow: 131072,
+    supportsTool: true,
+  },
   {
     name: DEEPINFRA_KIMI_FALLBACK_MODEL,
     displayName: "MoonshotAI: Kimi K2.6 (DeepInfra)",
     hasVision: true,
     price: {
-      input: 0.6 * 8,
-      output: 2.5 * 8,
-      inputCacheHit: 0.06 * 8,
+      input: price(0.6, DEEPINFRA_KIMI_FALLBACK_MODEL),
+      output: price(2.5, DEEPINFRA_KIMI_FALLBACK_MODEL),
+      inputCacheHit: price(0.06, DEEPINFRA_KIMI_FALLBACK_MODEL),
     },
     maxOutputTokens: 262144,
     contextWindow: 262144,
@@ -20,9 +35,9 @@ export const deepinfraModels = [
     displayName: "MoonshotAI: Kimi K2.5 (DeepInfra)",
     hasVision: true,
     price: {
-      input: 0.45 * 8,
-      output: 2.25 * 8,
-      inputCacheHit: 0.07 * 8,
+      input: price(0.45, "moonshotai/Kimi-K2.5"),
+      output: price(2.25, "moonshotai/Kimi-K2.5"),
+      inputCacheHit: price(0.07, "moonshotai/Kimi-K2.5"),
     },
     maxOutputTokens: 262144,
     contextWindow: 262144,
@@ -33,9 +48,9 @@ export const deepinfraModels = [
     displayName: "Z.AI: GLM 5.1 (DeepInfra)",
     hasVision: false,
     price: {
-      input: 1.26 * 8,
-      output: 3.96 * 8,
-      inputCacheHit: 0.26 * 8,
+      input: price(1.26, "zai-org/GLM-5.1"),
+      output: price(3.96, "zai-org/GLM-5.1"),
+      inputCacheHit: price(0.26, "zai-org/GLM-5.1"),
     },
     contextWindow: 202752,
     supportsTool: true,
@@ -45,8 +60,8 @@ export const deepinfraModels = [
     displayName: "Anthropic: Claude Haiku 4.5 (DeepInfra)",
     hasVision: true,
     price: {
-      input: 1 * 9,
-      output: 5 * 9,
+      input: price(1, "anthropic/claude-haiku-4-5"),
+      output: price(5, "anthropic/claude-haiku-4-5"),
     },
     contextWindow: 195000,
     maxOutputTokens: 4092,
@@ -57,8 +72,8 @@ export const deepinfraModels = [
     displayName: "Anthropic: Claude Sonnet 4.6 (DeepInfra)",
     hasVision: true,
     price: {
-      input: 3 * 9,
-      output: 15 * 9,
+      input: price(3, "anthropic/claude-sonnet-4-6"),
+      output: price(15, "anthropic/claude-sonnet-4-6"),
     },
     contextWindow: 976000,
     maxOutputTokens: 4092,
@@ -69,8 +84,8 @@ export const deepinfraModels = [
     displayName: "Anthropic: Claude Opus 4.7 (DeepInfra)",
     hasVision: true,
     price: {
-      input: 5 * 9,
-      output: 25 * 9,
+      input: price(5, "anthropic/claude-opus-4-7"),
+      output: price(25, "anthropic/claude-opus-4-7"),
     },
     contextWindow: 976000,
     maxOutputTokens: 4092,

package/ai/llm/mistral.ts

@@ -9,9 +9,8 @@ export const mistralModels: Model[] = [
         displayName: "Mistral: Devstral 2",
         hasVision: false,
         price: {
-            // 单位：$/M tokens，按你给的文档：input 0.4，output 2
-            input: 0.1,
-            output: 0.1,
+            input: 0.1 * 8,
+            output: 0.1 * 8,
         },
         maxOutputTokens: 262144,
         contextWindow: 262144, // 256k ≈ 256 * 1024
@@ -19,4 +18,4 @@ export const mistralModels: Model[] = [
         // 如果后面确认支持 reasoning effort，可以加上：
         // supportsReasoningEffort: true,
     },
-];
+];

package/ai/llm/providers.ts

@@ -119,6 +119,8 @@ const toModel = (candidate: ModelLookupCandidate): Model => ({
   imagePricingNote: candidate.imagePricingNote,
   imageTokenPricePerMillion: candidate.imageTokenPricePerMillion,
   imageOutputTokenEstimateBySize: candidate.imageOutputTokenEstimateBySize,
+  imageGenerationWaitTimeSeconds: candidate.imageGenerationWaitTimeSeconds,
+  imageGenerationProfiles: candidate.imageGenerationProfiles,
   supportsReasoningEffort:
     typeof candidate.supportsReasoningEffort === "boolean"
       ? candidate.supportsReasoningEffort

package/ai/llm/types.ts

@@ -25,6 +25,19 @@ export interface ServiceTierPriceMultiplier {
   cache?: number;
 }
 
+export interface ImageGenerationWaitTimeSeconds {
+  min: number;
+  max: number;
+}
+
+export interface ImageGenerationProfile {
+  key: "speed" | "quality";
+  label: string;
+  imageModel: string;
+  description?: string;
+  waitTimeSeconds?: ImageGenerationWaitTimeSeconds;
+}
+
 export interface Model {
   name: string;
   displayName?: string; // 可选的 displayName 字段
@@ -61,6 +74,8 @@ export interface Model {
   imagePricingNote?: string;
   imageTokenPricePerMillion?: number;
   imageOutputTokenEstimateBySize?: Partial<Record<"1K" | "2K" | "4K", number>>;
+  imageGenerationWaitTimeSeconds?: ImageGenerationWaitTimeSeconds;
+  imageGenerationProfiles?: ImageGenerationProfile[];
 
   supportsReasoningEffort?: boolean; // 是否支持推理功能
   endpointKey?: string;

package/ai/memory/capture.ts

@@ -1,4 +1,4 @@
-import serverDb from "../../database/server/db";
+import { getServerAuthorityStore } from "../../database/server/db";
 import { maybeConsolidateExplicitRemember } from "./consolidate";
 import { createMemoryItem, writeMemoryItemWithIndexesToDb } from "./store";
 import type { MemoryOwnerType, MemorySubjectType, MemoryVisibility } from "./types";
@@ -81,7 +81,7 @@ export const captureExplicitMemoryEpisode = async (input: {
   userInput: string;
 }): Promise<void> => {
   if (!shouldCaptureExplicitMemory(input.userInput)) return;
-  const db = input.db ?? serverDb;
+  const db = input.db ?? getServerAuthorityStore();
   const targets = buildExplicitMemoryTargets(input);
 
   for (const target of targets) {
@@ -112,7 +112,7 @@ export const consolidateExplicitMemoryAfterDialog = async (input: {
   userInput: string;
 }): Promise<void> => {
   if (!shouldCaptureExplicitMemory(input.userInput)) return;
-  const db = input.db ?? serverDb;
+  const db = input.db ?? getServerAuthorityStore();
   const targets = buildExplicitMemoryTargets(input);
 
   for (const target of targets) {

package/ai/memory/consolidate.ts

@@ -1,4 +1,4 @@
-import serverDb from "../../database/server/db";
+import { getServerAuthorityStore } from "../../database/server/db";
 import { loadMemoryCandidatesFromDb } from "./query";
 import { createMemoryItem, writeMemoryItemWithIndexesToDb } from "./store";
 import type { MemoryItem, MemoryOwnerRef, MemorySubjectType } from "./types";
@@ -56,7 +56,7 @@ export const maybeConsolidateExplicitRemember = async (input: {
   const normalizedContent = normalizeExplicitRememberContent(input.content);
   if (!shouldKeepAsSemanticCandidate(normalizedContent)) return;
 
-  const db = input.db ?? serverDb;
+  const db = input.db ?? getServerAuthorityStore();
   const items = await loadMemoryCandidatesFromDb(db, {
     owners: [input.owner],
     subjects: [{ subjectType: input.subjectType, subjectId: input.subjectId }],

package/ai/memory/delete.ts

@@ -1,4 +1,5 @@
-import serverDb from "../../database/server/db";
+import { getServerAuthorityStore } from "../../database/server/db";
+import { createStoreBatch } from "../../database/server/storeBatchCompat";
 import {
   createMemoryKey,
   createMemoryOwnerIndexKey,
@@ -127,7 +128,7 @@ export const deleteMemoriesForOwnerFromDb = async (
     return { deletedCount: 0, deletedIds: [], matchedItems: [] };
   }
 
-  const batch = db.batch();
+  const batch = createStoreBatch(db);
   for (const item of matchedItems) {
     deleteMemoryItemWithIndexesInBatch(batch, item);
   }
@@ -144,4 +145,4 @@ export const deleteMemoriesForOwner = async (
   owner: MemoryOwnerRef,
   filters: DeleteMemoryFilters = {}
 ): Promise<DeleteMemoryResult> =>
-  deleteMemoriesForOwnerFromDb(serverDb, owner, filters);
+  deleteMemoriesForOwnerFromDb(getServerAuthorityStore(), owner, filters);

package/ai/memory/query.ts

@@ -1,4 +1,4 @@
-import serverDb from "../../database/server/db";
+import { getServerAuthorityStore } from "../../database/server/db";
 import type {
   MemoryItem,
   MemoryKind,
@@ -25,7 +25,7 @@ export const loadOwnerItems = async (
   owner: MemoryOwnerRef,
   limit: number
 ): Promise<MemoryItem[]> => {
-  return loadOwnerItemsFromDb(serverDb, owner, limit);
+  return loadOwnerItemsFromDb(getServerAuthorityStore(), owner, limit);
 };
 
 export const loadMemoryCandidates = async (input: {
@@ -34,5 +34,5 @@ export const loadMemoryCandidates = async (input: {
   kinds?: MemoryKind[];
   ownerLimit?: number;
 }): Promise<MemoryItem[]> => {
-  return loadMemoryCandidatesFromDb(serverDb, input);
+  return loadMemoryCandidatesFromDb(getServerAuthorityStore(), input);
 };

package/ai/memory/remember.ts

@@ -1,4 +1,4 @@
-import serverDb from "../../database/server/db";
+import { getServerAuthorityStore } from "../../database/server/db";
 import { createMemoryItem, writeMemoryItemWithIndexesToDb } from "./store";
 import type { MemoryKind, MemoryOwnerType, MemorySubjectType, MemoryVisibility } from "./types";
 
@@ -126,7 +126,7 @@ export const rememberMemory = async (
     throw new Error("rememberMemory: no valid owner scope found");
   }
 
-  const db = input.db ?? serverDb;
+  const db = input.db ?? getServerAuthorityStore();
   const kind = input.kind ?? "episodic";
   if (!MEMORY_KINDS.has(kind)) {
     throw new Error("rememberMemory: kind must be episodic, semantic, or procedural");

package/ai/memory/store.ts

@@ -1,4 +1,4 @@
-import serverDb from "../../database/server/db";
+import { getServerAuthorityStore } from "../../database/server/db";
 import type { MemoryItem } from "./types";
 import {
   createMemoryItem,
@@ -9,12 +9,12 @@ import {
 export { createMemoryItem, writeMemoryItemWithIndexesToDb, touchMemoryItemsInDb };
 
 export const writeMemoryItemWithIndexes = async (item: MemoryItem): Promise<void> => {
-  return writeMemoryItemWithIndexesToDb(serverDb, item);
+  return writeMemoryItemWithIndexesToDb(getServerAuthorityStore(), item);
 };
 
 export const touchMemoryItems = async (
   items: MemoryItem[],
   now = new Date().toISOString()
 ): Promise<void> => {
-  return touchMemoryItemsInDb(serverDb, items, now);
+  return touchMemoryItemsInDb(getServerAuthorityStore(), items, now);
 };

package/ai/memory/storeShared.ts

@@ -4,6 +4,7 @@ import {
   createMemoryOwnerIndexKey,
   createMemorySubjectKindIndexKey,
 } from "../../database/keys";
+import { createStoreBatch } from "../../database/server/storeBatchCompat";
 import type { MemoryItem } from "./types";
 
 export const createMemoryItem = (
@@ -23,7 +24,7 @@ export const writeMemoryItemWithIndexesToDb = async (
   db: any,
   item: MemoryItem
 ): Promise<void> => {
-  const batch = db.batch();
+  const batch = createStoreBatch(db);
   batch.put(createMemoryKey(item.ownerType, item.ownerId, item.id), item);
   batch.put(
     createMemoryOwnerIndexKey(
@@ -64,7 +65,7 @@ export const touchMemoryItemsInDb = async (
   now = new Date().toISOString()
 ): Promise<void> => {
   if (items.length === 0) return;
-  const batch = db.batch();
+  const batch = createStoreBatch(db);
   for (const item of items) {
     batch.put(createMemoryKey(item.ownerType, item.ownerId, item.id), {
       ...item,

package/ai/policy/commercialMultiplier.ts

@@ -0,0 +1,16 @@
+export const DEFAULT_COMMERCIAL_MULTIPLIER = 8;
+export const DEEPINFRA_CLAUDE_COMMERCIAL_MULTIPLIER = 9;
+
+export const isDeepInfraHostedClaude = (provider: string, model: string) =>
+  provider === "deepinfra" && model.startsWith("anthropic/");
+
+export const resolveCommercialMultiplier = (provider: string, model: string) =>
+  isDeepInfraHostedClaude(provider, model)
+    ? DEEPINFRA_CLAUDE_COMMERCIAL_MULTIPLIER
+    : DEFAULT_COMMERCIAL_MULTIPLIER;
+
+export const scaleProviderPrice = (
+  rawProviderPrice: number,
+  provider: string,
+  model: string
+) => rawProviderPrice * resolveCommercialMultiplier(provider, model);