diff --git a/docusaurus.config.js b/docusaurus.config.js
index 924fc46..42f1fbb 100644
--- a/docusaurus.config.js
+++ b/docusaurus.config.js
@@ -115,11 +115,21 @@ markdown: {
         ],
       },
       // algolia search plugin
-     algolia: {
-      appId: 'QMWL076P1O',
-      apiKey: '76c365bb35304a755c612e0d58b1df5e',
-      indexName: 'tapdata-io',
-      contextualSearch: true,
+      algolia: {
+        appId: 'QMWL076P1O',
+        apiKey: '76c365bb35304a755c612e0d58b1df5e',
+        indexName: 'tapdata-io',
+        contextualSearch: true,
+
+        // Ask AI only works with DocSearch v4
+        ...(require('@docsearch/react').version.startsWith('4.')
+          ? {
+              askAi: {
+                assistantId: 'BSuLtTkwafCP',
+                indexName: 'tapdata-io',
+              },
+            }
+          : {}),
       },
       footer: {
         style: 'dark',