fix(frontend): improved dynamic imports

Agenta-AI · Dec 17, 2024 · aad37b8 · aad37b8
1 parent c8a3ace
commit aad37b8
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 5 deletions.
diff --git a/agenta-web/src/hooks/useVaultSecret.ts b/agenta-web/src/hooks/useVaultSecret.ts
@@ -7,14 +7,15 @@ import {
     saveLlmProviderKey,
 } from "@/lib/helpers/llmProviders"
 import {isDemo} from "@/lib/helpers/utils"
+import {dynamicLib, dynamicService} from "@/lib/helpers/dynamic"
 
 export const useVaultSecret = () => {
     const [secrets, setSecrets] = useState<LlmProvider[]>(llmAvailableProviders)
 
     const getVaultSecrets = async () => {
         try {
             if (isDemo()) {
-                const {fetchVaultSecret} = await import("@/services/vault/api"!)
+                const {fetchVaultSecret} = await dynamicService("vault/api")
                 const data = await fetchVaultSecret()
 
                 setSecrets((prevSecret) => {
@@ -46,8 +47,8 @@ export const useVaultSecret = () => {
     const handleModifyVaultSecret = async (provider: LlmProvider) => {
         try {
             if (isDemo()) {
-                const {createVaultSecret, updateVaultSecret} = await import("@/services/vault/api"!)
-                const {SecretDTOKind, SecretDTOProvider} = await import("@/lib/types_ee"!)
+                const {updateVaultSecret, createVaultSecret} = await dynamicService("vault/api")
+                const {SecretDTOProvider, SecretDTOKind} = await dynamicLib("types_ee")
 
                 const envNameMap: Record<string, any> = {
                     OPENAI_API_KEY: SecretDTOProvider.OPENAI,
@@ -98,8 +99,7 @@ export const useVaultSecret = () => {
     const handleDeleteVaultSecret = async (provider: LlmProvider) => {
         try {
             if (isDemo() && provider.id) {
-                const {deleteVaultSecret} = await import("@/services/vault/api"!)
-
+                const {deleteVaultSecret} = await dynamicService("vault/api")
                 await deleteVaultSecret({secret_id: provider.id})
                 await getVaultSecrets()
             } else {

diff --git a/agenta-web/src/lib/helpers/dynamic.ts b/agenta-web/src/lib/helpers/dynamic.ts
@@ -30,3 +30,11 @@ export async function dynamicService(path: string, fallback?: any) {
         return fallback
     }
 }
+
+export async function dynamicLib(path: string, fallback?: any) {
+    try {
+        return await import(`@/lib/${path}`)
+    } catch (error) {
+        return fallback
+    }
+}