adapt to sglang v0.5.2rc1 on dcu
This commit is contained in:
47
docs/wrap_run_llm.py
Normal file
47
docs/wrap_run_llm.py
Normal file
@@ -0,0 +1,47 @@
|
||||
import os
|
||||
import re
|
||||
|
||||
|
||||
def insert_runllm_widget(html_content):
|
||||
# RunLLM Widget script to be inserted
|
||||
widget_script = """
|
||||
<!-- RunLLM Widget Script -->
|
||||
<script type="module" id="runllm-widget-script" src="https://widget.runllm.com" crossorigin="true" version="stable" runllm-keyboard-shortcut="Mod+j" runllm-name="SGLang Chatbot" runllm-position="BOTTOM_RIGHT" runllm-assistant-id="629" async></script>
|
||||
"""
|
||||
|
||||
# Find the closing body tag and insert the widget script before it
|
||||
return re.sub(r"</body>", f"{widget_script}\n</body>", html_content)
|
||||
|
||||
|
||||
def process_html_files(build_dir):
|
||||
for root, dirs, files in os.walk(build_dir):
|
||||
for file in files:
|
||||
if file.endswith(".html"):
|
||||
file_path = os.path.join(root, file)
|
||||
|
||||
# Read the HTML file
|
||||
with open(file_path, "r", encoding="utf-8") as f:
|
||||
content = f.read()
|
||||
|
||||
# Insert the RunLLM widget
|
||||
modified_content = insert_runllm_widget(content)
|
||||
|
||||
# Write back the modified content
|
||||
with open(file_path, "w", encoding="utf-8") as f:
|
||||
f.write(modified_content)
|
||||
|
||||
|
||||
def main():
|
||||
# Get the build directory path
|
||||
build_dir = os.path.join(
|
||||
os.path.dirname(os.path.abspath(__file__)), "_build", "html"
|
||||
)
|
||||
# Process all HTML files
|
||||
if os.path.exists(build_dir):
|
||||
process_html_files(build_dir)
|
||||
else:
|
||||
print(f"Build directory not found: {build_dir}")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user