diff --git a/src/app.ts b/src/app.ts
index 59419bb..0cf43bc 100644
--- a/src/app.ts
+++ b/src/app.ts
@@ -1,7 +1,7 @@
 import { QueryRouterServer } from '@kevisual/router/src/route.ts'
 import { use } from '@kevisual/context'
 import { Query } from '@kevisual/query'
-import { Kevisual, ChatProviderMap } from '@kevisual/ai'
+import { Kevisual } from '@kevisual/ai'
 import mitt from 'mitt';
 const isBrowser = typeof window !== 'undefined'
 
@@ -17,6 +17,7 @@ type AppOptions = {
   token?: string
   initAI?: boolean
   storage?: Storage
+  ai?: Kevisual
 }
 export class App {
   #router: QueryRouterServer
@@ -32,6 +33,7 @@ export class App {
     const queryOptions = opts?.queryOptions || {}
     this.query = opts?.query || new Query({ url: queryOptions.url || 'https://kevisual.cn/api/router' })
     const initAI = opts?.initAI ?? true;
+    this.ai = opts?.ai || null as any;
     if (opts?.storage) {
       this.storage = opts.storage;
     } else {
@@ -57,13 +59,7 @@ export class App {
     try {
       const config = await this.getConfig('ai.json');
       if (config.token) {
-        if (config.provider) {
-          const provider = config.provider;
-          // @ts-ignore
-          this.ai = ChatProviderMap[provider] ? new ChatProviderMap[provider](config) : new Kevisual(config);
-        } else {
-          this.ai = new Kevisual(config);
-        }
+        this.ai = new Kevisual(config);
       }
     } catch (e) { }
     this.emitter.emit('ai-inited');