mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-11 22:51:28 +00:00
3130c4f8f9
* refactor: move guardrails to pip * refactor: move guardrails to pip * testing fix: move guardrails to pip * git commit setup_litellm_enterprise_pip
26 lines
710 B
Python
26 lines
710 B
Python
"""
|
|
This plugin searches for Hugging Face Access and Organization API Tokens.
|
|
"""
|
|
|
|
import re
|
|
|
|
from detect_secrets.plugins.base import RegexBasedDetector
|
|
|
|
|
|
class HuggingFaceDetector(RegexBasedDetector):
|
|
"""Scans for Hugging Face Tokens."""
|
|
|
|
@property
|
|
def secret_type(self) -> str:
|
|
return "Hugging Face Token"
|
|
|
|
@property
|
|
def denylist(self) -> list[re.Pattern]:
|
|
return [
|
|
# Hugging Face Access token
|
|
re.compile(r"""(?:^|[\\'"` >=:])(hf_[a-zA-Z]{34})(?:$|[\\'"` <])"""),
|
|
# Hugging Face Organization API token
|
|
re.compile(
|
|
r"""(?:^|[\\'"` >=:\(,)])(api_org_[a-zA-Z]{34})(?:$|[\\'"` <\),])"""
|
|
),
|
|
]
|