format

Author: efrick2002

fastchat/serve/api_provider.py

@@ -433,7 +433,7 @@ def p2l_api_stream_iter(
     max_new_tokens,
     api_base=None,
     api_key=None,
-    extra_body=None
+    extra_body=None,
 ):
     import openai
 
@@ -462,7 +462,7 @@ def p2l_api_stream_iter(
         "temperature": None,
         "top_p": None,
         "max_new_tokens": max_new_tokens,
-        "extra_body": extra_body
+        "extra_body": extra_body,
     }
     logger.info(f"==== request ====\n{gen_params}")
 

fastchat/serve/gradio_web_server.py

@@ -527,7 +527,7 @@ def bot_response(
             conv.set_system_message("")
 
         extra_body = None
-        
+
         if use_recommended_config:
             recommended_config = model_api_dict.get("recommended_config", None)
             if recommended_config is not None:
@@ -546,7 +546,7 @@ def bot_response(
             top_p,
             max_new_tokens,
             state,
-            extra_body=extra_body
+            extra_body=extra_body,
         )
 
     html_code = ' <span class="cursor"></span> '