update file format

Author: Changxf5

router_inference/router/__init__.py

@@ -8,4 +8,5 @@
 from router_inference.router.vllm_sr import VLLMSR
 from router_inference.router.api_router import APIRouter
 from router_inference.router.auto_router import auto_router
+
 __all__ = ["BaseRouter", "ExampleRouter", "VLLMSR", "APIRouter", "auto_router"]

router_inference/router/auto_router.py

@@ -38,37 +38,28 @@ def __init__(self, router_name: str):
 
     def get_routing_result(self, query):
         """Test the routing API with the specified request"""
-        url = 'http://10.109.4.26:8501/api/v1/routing'
-        
-        headers = {
-            'Content-Type': 'application/json'
-        }
+        url = "http://10.109.4.26:8501/api/v1/routing"
+
+        headers = {"Content-Type": "application/json"}
 
         data = {
             "default": 287,
             "models": self.id_list,
-            "messages": [
-                {
-                "role": "user",
-                "content": query
-                }
-            ],
+            "messages": [{"role": "user", "content": query}],
             "strategy": 1,
             "qualityPercentage": 60,
             "option": {
                 "multiRoundJudgeMode": 0,
                 "enableRagWebJudge": False,
                 "enableDomainJudge": False,
-            }
+            },
+        }
 
-            }
-        
         # Send POST request
         response = httpx.post(url, headers=headers, json=data)
-        
 
-        model_name = self.id_to_modelname[response.json()['data']['id']]
-        
+        model_name = self.id_to_modelname[response.json()["data"]["id"]]
+
         return model_name
 
     def _get_prediction(self, query: str) -> str:
@@ -85,9 +76,9 @@ def _get_prediction(self, query: str) -> str:
         Returns:
             Name of the selected model
         """
-        
+
         # Simple example: cycle through models
         model_name = self.get_routing_result(query)
-        print('----------------model_name:',model_name)
+        print("----------------model_name:", model_name)
         self.counter += 1
         return model_name