diff --git a/docs/_templates/base.html b/docs/_templates/base.html
index 0cc647529d8..423a0e28c26 100644
--- a/docs/_templates/base.html
+++ b/docs/_templates/base.html
@@ -77,11 +77,11 @@
     <link rel="preconnect" href="https://WLG0MZB58Q-dsn.algolia.net" crossorigin />
 
     <!-- RunLLM Configuration -->
-    <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/@runllm/search-widget@0.0.1/dist/main.css" />
-    <script defer id="runllm-widget-script" type="module" src='https://cdn.jsdelivr.net/npm/@runllm/search-widget@0.0.1/dist/run-llm-search-widget.es.js'
-      runllm-server-address="https://api.runllm.com" runllm-assistant-id="51" runllm-position="BOTTOM_RIGHT"
+    <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/@runllm/search-widget@0.0.2/dist/main.css" />
+    <script defer id="runllm-widget-script" type="module" src='https://cdn.jsdelivr.net/npm/@runllm/search-widget@0.0.2/dist/run-llm-search-widget.es.js'
+      runllm-server-address="https://api.runllm.com" runllm-assistant-id="60" runllm-position="BOTTOM_RIGHT"
       runllm-keyboard-shortcut="Mod+k"
-      version="0.0.1"
+      version="0.0.2"
       runllm-slack-community-url="https://flyte-org.slack.com/join/shared_invite/zt-2eq2fgs5f-UGUWnMYVB9agervilmlyaw#/shared-invite/email"
       runllm-name="RunLLM" />
     {%- endblock extra_styles -%}