Refactor artifact recommendation error handling (#263)
taylor_socfortress committed
Jul 2, 2024 at 11:57 UTC
82b650535629754d5eafa5bf2c2319a1333ff207
1 file changed
+7
backend/app/connectors/velociraptor/services/artifacts.py
+7
@@ -301,4 +301,11 @@ async def post_to_copilot_ai_module(data: ArtifactReccomendationRequest) -> Arti
301
json=data.dict(),
302
timeout=120,
303
)
304
+ data_json = data.json()
305
+ if data_json.success is False:
306
+ logger.error(f"Failed to get artifact recommendation: {data_json.message}")
307
+ raise HTTPException(
308
+ status_code=500,
309
+ detail=f"Failed to get artifact recommendation: {data_json.message}",
310
+ )
311
return ArtifactReccomendationResponse(**data.json())