feat: Adds user safeguard on compute

quack-ai · Nov 7, 2023 · 1f8a2a1 · 1f8a2a1
1 parent 6b49009
commit 1f8a2a1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/app/api/api_v1/endpoints/compute.py b/src/app/api/api_v1/endpoints/compute.py
@@ -31,7 +31,7 @@ async def check_code_against_repo_guidelines(
     # Fetch guidelines
     guideline_list = [elt for elt in await guidelines.fetch_all(("repo_id", repo_id))]
     # Run analysis
-    return openai_client.analyze_multi(payload.code, guideline_list, mode=ExecutionMode.MULTI)
+    return openai_client.analyze_multi(payload.code, guideline_list, mode=ExecutionMode.MULTI, user_id=str(user.id))
 
 
 @router.post("/check/{guideline_id}", status_code=status.HTTP_200_OK)
@@ -47,4 +47,4 @@ async def check_code_against_guideline(
         user.id, event="compute-check", properties={"repo_id": guideline.repo_id, "guideline_id": guideline_id}
     )
     # Run analysis
-    return openai_client.analyze_mono(payload.code, guideline)
+    return openai_client.analyze_mono(payload.code, guideline, user_id=str(user.id))