diff --git a/cmd/testdata/stores.yaml b/cmd/testdata/stores.yaml
index e81815804..cfa520514 100644
--- a/cmd/testdata/stores.yaml
+++ b/cmd/testdata/stores.yaml
@@ -6,7 +6,30 @@ stores:
       url: xxx
       readonly: false
       disabled: false
+    - name: ai
+      kind:
+        name: atest-ext-ai
+        enabled: true
+      url: ""
+      readonly: false
+      disabled: false
+      properties:
+        - key: "provider"
+          description: "AI provider (local, openai, claude)"
+          defaultValue: "local"
+        - key: "model"
+          description: "AI model name"
+          defaultValue: "codellama"
+        - key: "endpoint"
+          description: "AI service endpoint"
+          defaultValue: "http://localhost:11434"
 plugins:
     - name: atest-store-git
       url: unix:///tmp/atest-store-git.sock
       enabled: true
+    - name: atest-ext-ai
+      url: unix:///tmp/atest-ext-ai.sock
+      enabled: true
+      description: "AI Extension Plugin for intelligent SQL generation and execution"
+      version: "latest"
+      registry: "ghcr.io/linuxsuren/atest-ext-ai"
diff --git a/console/atest-desktop/forge.config.js b/console/atest-desktop/forge.config.js
index cd3c74359..e000ef4f7 100644
--- a/console/atest-desktop/forge.config.js
+++ b/console/atest-desktop/forge.config.js
@@ -71,6 +71,13 @@ module.exports = {
 
     </Product>`
           );
+
+          // Ensure INSTALLDIR is properly defined in the existing ProgramFilesFolder
+          msiCreator.wixTemplate = msiCreator.wixTemplate.replace(
+            '<Directory Id="ProgramFilesFolder">',
+            `<Directory Id="ProgramFilesFolder">
+        <Directory Id="INSTALLDIR" Name="API Testing" />`
+          );
         }
       }
     }
diff --git a/console/atest-ui/src/views/store.ts b/console/atest-ui/src/views/store.ts
index b86ee8b42..0f8ef5bdb 100644
--- a/console/atest-ui/src/views/store.ts
+++ b/console/atest-ui/src/views/store.ts
@@ -38,6 +38,7 @@ const ExtensionKindRedis = "atest-store-redis"
 const ExtensionKindMongoDB = "atest-store-mongodb"
 const ExtensionKindElasticsearch = "atest-store-elasticsearch"
 const ExtensionKindOpengeMini = "atest-store-opengemini"
+const ExtensionKindAI = "atest-ext-ai"
 
 export const ExtensionKind = {
     ExtensionKindGit,
@@ -49,7 +50,8 @@ export const ExtensionKind = {
     ExtensionKindRedis,
     ExtensionKindMongoDB,
     ExtensionKindElasticsearch,
-    ExtensionKindOpengeMini
+    ExtensionKindOpengeMini,
+    ExtensionKindAI
 }
 
 const MySQL = "mysql";
diff --git a/sample/ai-extension.yaml b/sample/ai-extension.yaml
new file mode 100644
index 000000000..e429114bd
--- /dev/null
+++ b/sample/ai-extension.yaml
@@ -0,0 +1,26 @@
+stores:
+  - name: ai
+    kind:
+      name: atest-ext-ai
+      enabled: true
+    url: ""
+    readonly: false
+    disabled: false
+    properties:
+      - key: "provider"
+        description: "AI provider (local, openai, claude)"
+        defaultValue: "local"
+      - key: "model"
+        description: "AI model name"
+        defaultValue: "codellama"
+      - key: "endpoint"
+        description: "AI service endpoint"
+        defaultValue: "http://localhost:11434"
+
+plugins:
+  - name: atest-ext-ai
+    url: unix:///tmp/atest-ext-ai.sock
+    enabled: true
+    description: "AI Extension Plugin for intelligent SQL generation and execution"
+    version: "latest"
+    registry: "ghcr.io/linuxsuren/atest-ext-ai"
\ No newline at end of file