format

Author: efrick2002

fastchat/serve/api_provider.py

@@ -433,7 +433,7 @@ def p2l_api_stream_iter(
     api_key=None,
 ):
     import openai
-    
+
     client = openai.OpenAI(
         base_url=api_base,
         api_key=api_key or "-",
@@ -479,13 +479,12 @@ def p2l_api_stream_iter(
             }
 
             if chunk_idx == 0:
-
                 if hasattr(chunk.choices[0].delta, "model"):
                     data["ans_model"] = chunk.choices[0].delta.model
-                
+
                 if hasattr(chunk, "router_outputs"):
                     data["router_outputs"] = chunk.router_outputs
-            
+
             yield data
 
 

fastchat/serve/gradio_web_server.py

@@ -131,14 +131,11 @@ def __init__(self, model_name, is_vision=False):
         self.init_system_prompt(self.conv, is_vision)
 
     def update_ans_models(self, ans: str) -> None:
-
         self.ans_models.append(ans)
 
     def update_router_outputs(self, outputs: Dict[str, float]) -> None:
-
         self.router_outputs.append(outputs)
 
-
     def init_system_prompt(self, conv, is_vision):
         system_prompt = conv.get_system_message(is_vision)
         if len(system_prompt) == 0:
@@ -171,7 +168,7 @@ def dict(self):
                     "ans_models": self.ans_models,
                 }
             )
-        
+
         if self.router_outputs:
             base.update(
                 {
@@ -557,23 +554,18 @@ def bot_response(
     try:
         data = {"text": ""}
         for i, data in enumerate(stream_iter):
-            
             # Change for P2L:
             if i == 0:
-
                 if "ans_model" in data:
-
                     ans_model = data.get("ans_model")
-                
+
                     state.update_ans_models(ans_model)
-                
-                if "router_outputs" in data:
 
+                if "router_outputs" in data:
                     router_outputs = data.get("router_outputs")
 
                     state.update_router_outputs(router_outputs)
 
-
             if data["error_code"] == 0:
                 output = data["text"].strip()
                 conv.update_last_message(output + "▌")