From cfb68e48a91cba2ad733bd1a4d44ec86f182631f Mon Sep 17 00:00:00 2001
From: Henry <hzj94@hotmail.com>
Date: Tue, 24 Jun 2025 15:21:00 +0100
Subject: [PATCH] fix circular dependency of updateStorageUsage and
 checkStorage in utils

---
 packages/server/package.json                          |  3 +++
 packages/server/src/services/openai-realtime/index.ts |  5 ++++-
 packages/server/src/utils/buildChatflow.ts            |  4 +++-
 packages/server/src/utils/index.ts                    | 10 ++++++----
 packages/server/src/utils/upsertVector.ts             |  4 +++-
 5 files changed, 19 insertions(+), 7 deletions(-)

diff --git a/packages/server/package.json b/packages/server/package.json
index 69ed77491..57d99011c 100644
--- a/packages/server/package.json
+++ b/packages/server/package.json
@@ -28,6 +28,9 @@
         "start:default": "cd bin && ./run start",
         "start-worker:windows": "cd bin && run worker",
         "start-worker:default": "cd bin && ./run worker",
+        "user": "run-script-os",
+        "user:windows": "cd bin && run user",
+        "user:default": "cd bin && ./run user",
         "dev": "nodemon",
         "oclif-dev": "run-script-os",
         "oclif-dev:windows": "cd bin && dev start",
diff --git a/packages/server/src/services/openai-realtime/index.ts b/packages/server/src/services/openai-realtime/index.ts
index f5126e977..a93a9068c 100644
--- a/packages/server/src/services/openai-realtime/index.ts
+++ b/packages/server/src/services/openai-realtime/index.ts
@@ -10,6 +10,7 @@ import {
     getStartingNodes,
     resolveVariables
 } from '../../utils'
+import { checkStorage, updateStorageUsage } from '../../utils/quotaUsage'
 import { getRunningExpressApp } from '../../utils/getRunningExpressApp'
 import { ChatFlow } from '../../database/entities/ChatFlow'
 import { IDepthQueue, IReactFlowNode } from '../../Interface'
@@ -110,7 +111,9 @@ const buildAndInitTool = async (chatflowid: string, _chatId?: string, _apiMessag
         variableOverrides,
         orgId,
         workspaceId,
-        subscriptionId
+        subscriptionId,
+        updateStorageUsage,
+        checkStorage
     })
 
     const nodeToExecute =
diff --git a/packages/server/src/utils/buildChatflow.ts b/packages/server/src/utils/buildChatflow.ts
index df5b64e64..0805308d1 100644
--- a/packages/server/src/utils/buildChatflow.ts
+++ b/packages/server/src/utils/buildChatflow.ts
@@ -515,7 +515,9 @@ export const executeFlow = async ({
         baseURL,
         orgId,
         workspaceId,
-        subscriptionId
+        subscriptionId,
+        updateStorageUsage,
+        checkStorage
     })
 
     const setVariableNodesOutput = getSetVariableNodesOutput(reactFlowNodes)
diff --git a/packages/server/src/utils/index.ts b/packages/server/src/utils/index.ts
index 1e45ffd3b..6438451ab 100644
--- a/packages/server/src/utils/index.ts
+++ b/packages/server/src/utils/index.ts
@@ -64,8 +64,6 @@ import {
     SecretsManagerClient,
     SecretsManagerClientConfig
 } from '@aws-sdk/client-secrets-manager'
-import { checkStorage, updateStorageUsage } from './quotaUsage'
-import { UsageCacheManager } from '../UsageCacheManager'
 
 export const QUESTION_VAR_PREFIX = 'question'
 export const FILE_ATTACHMENT_PREFIX = 'file_attachment'
@@ -504,8 +502,10 @@ type BuildFlowParams = {
     orgId?: string
     workspaceId?: string
     subscriptionId?: string
-    usageCacheManager?: UsageCacheManager
+    usageCacheManager?: any
     uploadedFilesContent?: string
+    updateStorageUsage?: (orgId: string, workspaceId: string, totalSize: number, usageCacheManager?: any) => void
+    checkStorage?: (orgId: string, subscriptionId: string, usageCacheManager: any) => Promise<any>
 }
 
 /**
@@ -540,7 +540,9 @@ export const buildFlow = async ({
     orgId,
     workspaceId,
     subscriptionId,
-    usageCacheManager
+    usageCacheManager,
+    updateStorageUsage,
+    checkStorage
 }: BuildFlowParams) => {
     const flowNodes = cloneDeep(reactFlowNodes)
 
diff --git a/packages/server/src/utils/upsertVector.ts b/packages/server/src/utils/upsertVector.ts
index d292ec2df..4df5546e0 100644
--- a/packages/server/src/utils/upsertVector.ts
+++ b/packages/server/src/utils/upsertVector.ts
@@ -193,7 +193,9 @@ export const executeUpsert = async ({
         variableOverrides,
         orgId,
         workspaceId,
-        subscriptionId
+        subscriptionId,
+        updateStorageUsage,
+        checkStorage
     })
 
     // Save to DB