fix: properly parsing comma-separated list on REST API

varfish-org · Sep 13, 2023 · 5dc5cf7 · 5dc5cf7
1 parent 278086a
commit 5dc5cf7
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/cada_prio/rest_server.py b/cada_prio/rest_server.py
@@ -59,12 +59,14 @@ class PredictionResult(BaseModel):
 # Register endpoint for the prediction
 @app.get("/predict")
 async def handle_predict(
-    hpo_terms: typing.Annotated[typing.List[str], Query()],
-    genes: typing.Annotated[typing.Optional[typing.List[str]], Query()] = [],
+    hpo_terms: typing.Annotated[str, Query()],
+    genes: typing.Annotated[typing.Optional[str], Query()] = [],
 ):
+    hpo_terms_list = hpo_terms.split(",")
+    genes_list = genes.split(",")
     _, sorted_scores = predict.run_prediction(
-        hpo_terms,
-        genes,
+        hpo_terms_list,
+        genes_list,
         GLOBAL_STATIC["all_to_hgnc"],
         GLOBAL_STATIC["graph"],
         GLOBAL_STATIC["model"],