Format, async test_chat_api.

server/main.py

@@ -9,30 +9,34 @@ from . import workspace
 here = pathlib.Path(__file__).parent
 lynxkite_modules = {}
 for _, name, _ in pkgutil.iter_modules([str(here)]):
-    if name.endswith('_ops') and not name.startswith('test_'):
-        print(f'Importing {name}')
-        name = f'server.{name}'
+    if name.endswith("_ops") and not name.startswith("test_"):
+        print(f"Importing {name}")
+        name = f"server.{name}"
         lynxkite_modules[name] = importlib.import_module(name)
 
 app = fastapi.FastAPI()
 
 
+
 @app.get("/api/catalog")
 def get_catalog():
     return {
         k: {op.name: op.model_dump() for op in v.values()}
-        for k, v in ops.CATALOGS.items()}
+        for k, v in ops.CATALOGS.items()
+    }
 
 
 class SaveRequest(workspace.BaseConfig):
     path: str
     ws: workspace.Workspace
 
+
 def save(req: SaveRequest):
     path = DATA_PATH / req.path
     assert path.is_relative_to(DATA_PATH)
     workspace.save(req.ws, path)
 
+
 @app.post("/api/save")
 def save_and_execute(req: SaveRequest):
     save(req)
@@ -40,6 +44,7 @@ def save_and_execute(req: SaveRequest):
     save(req)
     return req.ws
 
+
 @app.get("/api/load")
 def load(path: str):
     path = DATA_PATH / path
@@ -48,31 +53,41 @@ def load(path: str):
         return workspace.Workspace()
     return workspace.load(path)
 
-DATA_PATH = pathlib.Path.cwd() / 'data'
+
+DATA_PATH = pathlib.Path.cwd() / "data"
+
 
 @dataclasses.dataclass(order=True)
 class DirectoryEntry:
     name: str
     type: str
 
+
 @app.get("/api/dir/list")
 def list_dir(path: str):
     path = DATA_PATH / path
     assert path.is_relative_to(DATA_PATH)
-    return sorted([
-        DirectoryEntry(p.relative_to(DATA_PATH), 'directory' if p.is_dir() else 'workspace')
-        for p in path.iterdir()])
+    return sorted(
+        [
+            DirectoryEntry(
+                p.relative_to(DATA_PATH), "directory" if p.is_dir() else "workspace"
+            )
+            for p in path.iterdir()
+        ]
+    )
+
 
 @app.post("/api/dir/mkdir")
 def make_dir(req: dict):
-    path = DATA_PATH / req['path']
+    path = DATA_PATH / req["path"]
     assert path.is_relative_to(DATA_PATH)
     assert not path.exists()
     path.mkdir()
     return list_dir(path.parent)
 
+
 @app.post("/api/service")
 async def service(req: dict):
-    '''Executors can provide extra HTTP APIs through the /api/service endpoint.'''
-    module = lynxkite_modules[req['module']]
+    """Executors can provide extra HTTP APIs through the /api/service endpoint."""
+    module = lynxkite_modules[req["module"]]
     return await module.api_service(req)