Don't return find_datasets usage via api

whiletrue-industries · Aug 5, 2023 · efca998 · efca998
1 parent ad05acb
commit efca998
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ckangpt/frontend/api/routers/find_datasets.py b/ckangpt/frontend/api/routers/find_datasets.py
@@ -15,6 +15,7 @@ async def find_datasets_(
         document_ids: typing.List[str] = fastapi.Query(None),
         num_results: int = find_datasets.config.DEFAULT_NUM_RESULTS,
 ):
-    return find_datasets.main(
+    _, answer = find_datasets.main(
         user_prompt, db_queries.split('\n'), document_ids=document_ids, num_results=num_results
     )
+    return answer