format

lm-sys · May 20, 2024 · 1180bd1 · 1180bd1
1 parent 78ba314
commit 1180bd1
Show file tree

Hide file tree

Showing 6 changed files with 33 additions and 27 deletions.
diff --git a/fastchat/serve/gradio_block_arena_anony.py b/fastchat/serve/gradio_block_arena_anony.py
@@ -538,10 +538,10 @@ def build_side_by_side_ui_anony(models):
                         height=550,
                         show_copy_button=True,
                         latex_delimiters=[
-                            {"left": '$$', "right": '$$', "display": True},
-                            {"left": '$', "right": '$', "display": False},
-                            {"left": '\\(', "right": '\\)', "display": False},
-                            {"left": '\\[', "right": '\\]', "display": True}
+                            {"left": "$$", "right": "$$", "display": True},
+                            {"left": "$", "right": "$", "display": False},
+                            {"left": "\\(", "right": "\\)", "display": False},
+                            {"left": "\\[", "right": "\\]", "display": True},
                         ],
                     )
 

diff --git a/fastchat/serve/gradio_block_arena_named.py b/fastchat/serve/gradio_block_arena_named.py
@@ -344,10 +344,10 @@ def build_side_by_side_ui_named(models):
                         height=550,
                         show_copy_button=True,
                         latex_delimiters=[
-                            {"left": '$$', "right": '$$', "display": True},
-                            {"left": '$', "right": '$', "display": False},
-                            {"left": '\\(', "right": '\\)', "display": False},
-                            {"left": '\\[', "right": '\\]', "display": True}
+                            {"left": "$$", "right": "$$", "display": True},
+                            {"left": "$", "right": "$", "display": False},
+                            {"left": "\\(", "right": "\\)", "display": False},
+                            {"left": "\\[", "right": "\\]", "display": True},
                         ],
                     )
 

diff --git a/fastchat/serve/gradio_block_arena_vision.py b/fastchat/serve/gradio_block_arena_vision.py
@@ -227,12 +227,14 @@ def build_single_vision_language_model_ui(
             )
         with gr.Column(scale=8):
             chatbot = gr.Chatbot(
-                elem_id="chatbot", label="Scroll down and start chatting", height=550,
+                elem_id="chatbot",
+                label="Scroll down and start chatting",
+                height=550,
                 latex_delimiters=[
-                    {"left": '$$', "right": '$$', "display": True},
-                    {"left": '$', "right": '$', "display": False},
-                    {"left": '\\(', "right": '\\)', "display": False},
-                    {"left": '\\[', "right": '\\]', "display": True}
+                    {"left": "$$", "right": "$$", "display": True},
+                    {"left": "$", "right": "$", "display": False},
+                    {"left": "\\(", "right": "\\)", "display": False},
+                    {"left": "\\[", "right": "\\]", "display": True},
                 ],
             )
 

diff --git a/fastchat/serve/gradio_block_arena_vision_anony.py b/fastchat/serve/gradio_block_arena_vision_anony.py
@@ -398,10 +398,10 @@ def build_side_by_side_vision_ui_anony(models, random_questions=None):
                                 height=550,
                                 show_copy_button=True,
                                 latex_delimiters=[
-                                    {"left": '$$', "right": '$$', "display": True},
-                                    {"left": '$', "right": '$', "display": False},
-                                    {"left": '\\(', "right": '\\)', "display": False},
-                                    {"left": '\\[', "right": '\\]', "display": True}
+                                    {"left": "$$", "right": "$$", "display": True},
+                                    {"left": "$", "right": "$", "display": False},
+                                    {"left": "\\(", "right": "\\)", "display": False},
+                                    {"left": "\\[", "right": "\\]", "display": True},
                                 ],
                             )
 

diff --git a/fastchat/serve/gradio_block_arena_vision_named.py b/fastchat/serve/gradio_block_arena_vision_named.py
@@ -283,10 +283,10 @@ def build_side_by_side_vision_ui_named(models, random_questions=None):
                                 height=550,
                                 show_copy_button=True,
                                 latex_delimiters=[
-                                    {"left": '$$', "right": '$$', "display": True},
-                                    {"left": '$', "right": '$', "display": False},
-                                    {"left": '\\(', "right": '\\)', "display": False},
-                                    {"left": '\\[', "right": '\\]', "display": True}
+                                    {"left": "$$", "right": "$$", "display": True},
+                                    {"left": "$", "right": "$", "display": False},
+                                    {"left": "\\(", "right": "\\)", "display": False},
+                                    {"left": "\\[", "right": "\\]", "display": True},
                                 ],
                             )
 

diff --git a/fastchat/serve/gradio_web_server.py b/fastchat/serve/gradio_web_server.py
@@ -514,7 +514,9 @@ def wrap_output(output, strip=True, add_html_code=True, escape_backslash=False):
         data = {"text": ""}
         for i, data in enumerate(stream_iter):
             if data["error_code"] == 0:
-                output = wrap_output(data["text"], strip=True, add_html_code=True, escape_backslash=True)
+                output = wrap_output(
+                    data["text"], strip=True, add_html_code=True, escape_backslash=True
+                )
                 # conv.update_last_message(output + "▌")
                 yield (state, state.to_gradio_chatbot()) + (disable_btn,) * 5
             else:
@@ -528,7 +530,9 @@ def wrap_output(output, strip=True, add_html_code=True, escape_backslash=False):
                     enable_btn,
                 )
                 return
-        output = wrap_output(data["text"], strip=True, add_html_code=False, escape_backslash=True)
+        output = wrap_output(
+            data["text"], strip=True, add_html_code=False, escape_backslash=True
+        )
         conv.update_last_message(output)
         yield (state, state.to_gradio_chatbot()) + (enable_btn,) * 5
     except requests.exceptions.RequestException as e:
@@ -800,10 +804,10 @@ def build_single_model_ui(models, add_promotion_links=False):
             height=550,
             show_copy_button=True,
             latex_delimiters=[
-                {"left": '$$', "right": '$$', "display": True},
-                {"left": '$', "right": '$', "display": False},
-                {"left": '\\(', "right": '\\)', "display": False},
-                {"left": '\\[', "right": '\\]', "display": True}
+                {"left": "$$", "right": "$$", "display": True},
+                {"left": "$", "right": "$", "display": False},
+                {"left": "\\(", "right": "\\)", "display": False},
+                {"left": "\\[", "right": "\\]", "display": True},
             ],
         )
     with gr.Row():