phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-15 14:08:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

error handling for scan result failure

Browse source
This commit is contained in:
Swapna Lekkala 2025-08-14 13:56:01 -07:00
parent 900424ef61
commit 4e720dcbef
2 changed files with 17 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

16
llama_stack/providers/inline/safety/code_scanner/code_scanner.py
View file

@ -109,8 +109,20 @@ class MetaReferenceCodeScannerSafetyImpl(Safety):
for text_input in inputs:
log.info(f"Running CodeScannerShield moderation on input: {text_input[:100]}...")
scan_result = await CodeShield.scan_code(text_input)
moderation_result = self.get_moderation_object_results(scan_result)
try:
scan_result = await CodeShield.scan_code(text_input)
moderation_result = self.get_moderation_object_results(scan_result)
except Exception as e:
log.error(f"CodeShield.scan_code failed: {e}")
# create safe fallback response on scanner failure to avoid blocking legitimate requests
moderation_result = ModerationObjectResults(
flagged=False,
categories={},
category_scores={},
category_applied_input_types={},
user_message=None,
metadata={"scanner_error": str(e)},
)
results.append(moderation_result)
return ModerationObject(id=str(uuid.uuid4()), model=model, results=results)