format spaces in build_server_utils.py

lang_agent/front_api/build_server_utils.py

@@ -18,9 +18,10 @@ def opt_to_config(save_path: str, *nargs):
         cwd=_PROJECT_ROOT,
     )
 
-def _build_and_load_pipeline_config(pipeline_id: str,
-                                    pipeline_config_dir: str,
-                                    cmd: List[str]):
+
+def _build_and_load_pipeline_config(
+    pipeline_id: str, pipeline_config_dir: str, cmd: List[str]
+):
     save_config_f = osp.join(pipeline_config_dir, f"{pipeline_id}.yml")
     opt_to_config(save_config_f, *cmd)
 
@@ -28,41 +29,43 @@ def _build_and_load_pipeline_config(pipeline_id: str,
     return load_tyro_conf(save_config_f)
 
 
-def update_pipeline_registry(pipeline_id:str, # the agent name -- xiaozhan, blueberry
-                             prompt_set:str,  # the version of the prompt for xiaozhan/blueberry
-                             graph_id: str,   # what type of graph is this pipeline.
+def update_pipeline_registry(
+    pipeline_id: str,
+    graph_id: str,
     config_file: str,
     llm_name: str,
     enabled: bool = True,
-                             registry_f:str="configs/pipeline_registry.json"):
+    registry_f: str = "configs/pipeline_registry.json",
+):
     if not osp.isabs(registry_f):
         registry_f = osp.join(_PROJECT_ROOT, registry_f)
     os.makedirs(osp.dirname(registry_f), exist_ok=True)
     if not osp.exists(registry_f):
         with open(registry_f, "w", encoding="utf-8") as f:
-            json.dump({"routes": {}, "api_keys": {}}, f, indent=4)
+            json.dump({"pipelines": {}, "api_keys": {}}, f, indent=4)
 
     with open(registry_f, "r") as f:
         registry = json.load(f)
 
-    routes: Dict[str, Dict[str, Any]] = registry.setdefault("routes", {})
-    route = routes.setdefault(pipeline_id, {})
-    route["enabled"] = bool(enabled)
-    route["config_file"] = config_file
-    route["prompt_pipeline_id"] = prompt_set
-    route["graph_id"] = graph_id
-    route["overrides"] = {"llm_name": llm_name}
+    pipelines: Dict[str, Dict[str, Any]] = registry.setdefault("pipelines", {})
+    pipeline = pipelines.setdefault(pipeline_id, {})
+    pipeline["enabled"] = bool(enabled)
+    pipeline["config_file"] = config_file
+    pipeline["graph_id"] = graph_id
+    pipeline["overrides"] = {"llm_name": llm_name}
 
     with open(registry_f, "w", encoding="utf-8") as f:
         json.dump(registry, f, indent=4)
 
 
-def build_route(pipeline_id:str,
+def build_route(
+    pipeline_id: str,
     prompt_set: str,
     tool_keys: List[str],
     api_key: str,
     llm_name: str = "qwen-plus",
-                pipeline_config_dir="configs/pipelines"):
+    pipeline_config_dir="configs/pipelines",
+):
     cmd_opt = [
         "route",  # ------------
         "--llm-name", llm_name,
@@ -84,12 +87,14 @@ def build_route(pipeline_id:str,
     return _build_and_load_pipeline_config(pipeline_id, pipeline_config_dir, cmd_opt)
 
 
-def build_react(pipeline_id:str,
+def build_react(
+    pipeline_id: str,
     prompt_set: str,
     tool_keys: List[str],
     api_key: str,
     llm_name: str = "qwen-plus",
-                pipeline_config_dir="configs/pipelines"):
+    pipeline_config_dir="configs/pipelines",
+):
     cmd_opt = [
         "react",  # ------------
         "--llm-name", llm_name,