Cherrypick: Add the RunLLM widget to the website (pytorch#159592)

Add the RunLLM widget to the website (pytorch#152055)

Fixes #ISSUE_NUMBER

Pull Request resolved: pytorch#152055
Approved by: https://github.com/albanD

(cherry picked from commit e4e2701)

Author: svekars

.ci/docker/requirements-docs.txt

@@ -1,7 +1,7 @@
 sphinx==5.3.0
 #Description: This is used to generate PyTorch docs
 #Pinned versions: 5.3.0
--e git+https://github.com/pytorch/pytorch_sphinx_theme.git@pytorch_sphinx_theme2#egg=pytorch_sphinx_theme2
+-e git+https://github.com/pytorch/pytorch_sphinx_theme.git@722b7e6f9ca512fcc526ad07d62b3d28c50bb6cd#egg=pytorch_sphinx_theme2
 
 # TODO: sphinxcontrib.katex 0.9.0 adds a local KaTeX server to speed up pre-rendering
 # but it doesn't seem to work and hangs around idly. The initial thought that it is probably
@@ -49,8 +49,8 @@ IPython==8.12.0
 #Pinned versions: 8.12.0
 
 myst-nb==0.17.2
-#Description: This is used to generate PyTorch functorch docs
-#Pinned versions: 0.13.2
+#Description: This is used to generate PyTorch functorch and torch.compile docs.
+#Pinned versions: 0.17.2
 
 # The following are required to build torch.distributed.elastic.rendezvous.etcd* docs
 python-etcd==0.4.5

docs/source/_static/js/runllm-widget.js

@@ -0,0 +1,17 @@
+document.addEventListener("DOMContentLoaded", function () {
+  var script = document.createElement("script");
+  script.type = "module";
+  script.id = "runllm-widget-script"
+
+  script.src = "https://widget.runllm.com";
+
+  script.setAttribute("version", "stable");
+  script.setAttribute("crossorigin", "true");
+  script.setAttribute("runllm-keyboard-shortcut", "Mod+j");
+  script.setAttribute("runllm-name", "PyTorch");
+  script.setAttribute("runllm-position", "BOTTOM_RIGHT");
+  script.setAttribute("runllm-assistant-id", "834");
+
+  script.async = true;
+  document.head.appendChild(script);
+});

docs/source/conf.py

@@ -3564,6 +3564,8 @@ def linkcode_resolve(domain, info):
     "https://cdn.jsdelivr.net/npm/[email protected]/dist/katex.min.css",
 ]
 
+html_js_files = ["js/runllm-widget.js"]
+
 from sphinx.ext.coverage import CoverageBuilder