diff --git a/bun.lockb b/bun.lockb
index 0ac6552..ae58aaf 100755
Binary files a/bun.lockb and b/bun.lockb differ
diff --git a/jsr.json b/jsr.json
index 6b4a582..2f212b1 100644
--- a/jsr.json
+++ b/jsr.json
@@ -1,6 +1,6 @@
 {
 	"name": "@kunkun/kunkun-ext-rag",
-	"version": "0.0.5",
+	"version": "0.0.6",
 	"license": "MIT",
 	"exports": "./mod.ts",
 	"publish": {
diff --git a/package.json b/package.json
index 6a2679a..0319995 100644
--- a/package.json
+++ b/package.json
@@ -3,7 +3,7 @@
 	"license": "MIT",
 	"name": "kunkun-ext-rag",
 	"draft": true,
-	"version": "0.0.5",
+	"version": "0.0.6",
 	"private": true,
 	"kunkun": {
 		"name": "RAG",
@@ -97,7 +97,7 @@
 	},
 	"dependencies": {
 		"@iconify/svelte": "^4.2.0",
-		"@kksh/api": "^0.0.55",
+		"@kksh/api": "^0.1.1",
 		"@kksh/svelte5": "0.1.15",
 		"@langchain/openai": "^0.4.2",
 		"clsx": "^2.1.1",
diff --git a/src/lib/components/TauriLink.svelte b/src/lib/components/TauriLink.svelte
index cea6197..94df785 100644
--- a/src/lib/components/TauriLink.svelte
+++ b/src/lib/components/TauriLink.svelte
@@ -1,5 +1,5 @@
 <script lang="ts">
-	import { open } from '@kksh/api/ui/iframe';
+	import { open } from '@kksh/api/ui/custom';
 
 	let { href, children } = $props();
 </script>
diff --git a/src/lib/components/ThemeCustomizer.svelte b/src/lib/components/ThemeCustomizer.svelte
index d7b86b2..f36ebbf 100644
--- a/src/lib/components/ThemeCustomizer.svelte
+++ b/src/lib/components/ThemeCustomizer.svelte
@@ -1,6 +1,6 @@
 <script lang="ts">
 	import { ThemeCustomizerButton, type ThemeConfig, updateTheme } from '@kksh/svelte5';
-	import { ui } from '@kksh/api/ui/iframe';
+	import { ui } from '@kksh/api/ui/custom';
 	import { onMount } from 'svelte';
 
 	let config: ThemeConfig = {
diff --git a/src/lib/components/app-sidebar.svelte b/src/lib/components/app-sidebar.svelte
index 61cd550..fee0fed 100644
--- a/src/lib/components/app-sidebar.svelte
+++ b/src/lib/components/app-sidebar.svelte
@@ -1,5 +1,5 @@
 <script lang="ts">
-	import { ui } from '@kksh/api/ui/iframe';
+	import { ui } from '@kksh/api/ui/custom';
 	import { Sidebar } from '@kksh/svelte5';
 	import { DatabaseIcon, InfoIcon } from 'lucide-svelte';
 	import { onMount } from 'svelte';
diff --git a/src/lib/deno.ts b/src/lib/deno.ts
index 2720108..a822212 100644
--- a/src/lib/deno.ts
+++ b/src/lib/deno.ts
@@ -1,4 +1,4 @@
-import { fs, shell, path, toast } from '@kksh/api/ui/iframe';
+import { fs, shell, path, toast } from '@kksh/api/ui/custom';
 import type { DenoAPI } from '../api.types.ts';
 
 export async function getRpcAPI(env: { OPENAI_API_KEY: string }) {
diff --git a/src/lib/stores/db.ts b/src/lib/stores/db.ts
index 3322f85..c2e01a4 100644
--- a/src/lib/stores/db.ts
+++ b/src/lib/stores/db.ts
@@ -1,6 +1,6 @@
 import { DBInfo } from '@/models';
 import { get, writable } from 'svelte/store';
-import { db as dbAPI } from '@kksh/api/ui/iframe';
+import { db as dbAPI } from '@kksh/api/ui/custom';
 import { DB_DATATYPE_DATABASE } from '@/constants';
 import * as v from 'valibot';
 import { SearchModeEnum } from '@kksh/api/models';
diff --git a/src/routes/+layout.svelte b/src/routes/+layout.svelte
index 5de8c42..f5e3f35 100644
--- a/src/routes/+layout.svelte
+++ b/src/routes/+layout.svelte
@@ -3,7 +3,7 @@
 	import { ModeWatcher } from 'mode-watcher';
 	import { ThemeWrapper, updateTheme } from '@kksh/svelte5';
 	import { onMount } from 'svelte';
-	import { ui } from '@kksh/api/ui/iframe';
+	import { ui } from '@kksh/api/ui/custom';
 	import { Sidebar } from '@kksh/svelte5';
 	import AppSidebar from '$lib/components/app-sidebar.svelte';
 	import { DB_DATATYPE_DATABASE } from '@/constants';
diff --git a/src/routes/+page.svelte b/src/routes/+page.svelte
index 9517ca7..a33b53c 100644
--- a/src/routes/+page.svelte
+++ b/src/routes/+page.svelte
@@ -1,5 +1,5 @@
 <script lang="ts">
-	import { db, ui, kv, toast } from '@kksh/api/ui/iframe';
+	import { db, ui, kv, toast } from '@kksh/api/ui/custom';
 	import AddBucket from '$lib/components/forms/AddBucket.svelte';
 	import type { DBInfo } from '@/models';
 	import { onMount } from 'svelte';
diff --git a/src/routes/database/[id]/+page.svelte b/src/routes/database/[id]/+page.svelte
index 939fa43..bd356dd 100644
--- a/src/routes/database/[id]/+page.svelte
+++ b/src/routes/database/[id]/+page.svelte
@@ -1,5 +1,5 @@
 <script lang="ts">
-	import { dialog, path, toast } from '@kksh/api/ui/iframe';
+	import { dialog, path, toast } from '@kksh/api/ui/custom';
 	import { dbStore } from '@/stores/db';
 	import { Button, Input, Popover } from '@kksh/svelte5';
 	import { getRpcAPI } from '@/deno';