AI Infra Guard

AI-Infra-Guard is a security scanning platform designed to detect vulnerabilities across large language model deployments, AI agent skills, and the underlying infrastructure. It functions as a security toolset for auditing source code, evaluating model robustness, and identifying insecure network configurations.

The project provides a red teaming framework that uses curated attack datasets to test for jailbreak vulnerabilities and prompt injections. It also includes an infrastructure auditor that employs network fingerprinting and asset discovery to match running components against known common vulnerabilities and exposures databases.

The system covers a broad range of security assessment capabilities, including agent workflow auditing, remote source code scanning, and automated security pipelines. These processes are accessible via programmatic interfaces for triggering audits and system integrity checks.

Features

LLM Security - Provides a comprehensive platform for identifying and mitigating security vulnerabilities specific to LLM deployments.

Adversarial Robustness Testing - Evaluates model stability and security using adversarial attack methods and curated datasets.

Model Red-Teaming - Evaluates model robustness using curated attack datasets to detect potential jailbreak vulnerabilities.

Infrastructure Scanning - Scans the infrastructure supporting AI systems to detect security weaknesses and facilitate risk self-examination.

AI Infrastructure Auditors - Identifies CVEs and insecure configurations within the network components supporting AI agents.

Infrastructure Security Audits - Performs automated security assessments of network assets and cloud infrastructure supporting AI environments.

Agent Security Auditing - Scans AI agent configurations, source code, and remote endpoints for vulnerabilities and injection risks.

CVE Mapping - Links detected infrastructure software versions to known CVE identifiers using network fingerprinting.

Infrastructure Fingerprinters - Detects running infrastructure components and configurations through network probing to map the attack surface.

File and URL Scans - Analyzes remote URLs and source files to detect security risks within AI agent skills and servers.

Adversarial Red Teaming Toolkits - Provides a framework for testing model robustness against safety guardrails using adversarial red teaming techniques.

Vulnerability Matching - Compares detected network fingerprints and software versions against known CVE databases to identify insecure components.

AI Agent Scanners - Provides a dedicated tool for auditing the source code and remote endpoints of AI agent skills.

AI Model Vulnerabilities - Provides a red teaming framework to evaluate and identify jailbreak vulnerabilities and security risks in large language models.

Security Testing Pipelines - Implements automated workflows that orchestrate a sequence of infrastructure audits and model evaluations for continuous risk assessment.

Vulnerability Management - Monitors and identifies known security flaws in infrastructure components by matching them against CVE databases.

AI Application Security - Security assessment tool for AI infrastructure and LLM deployments.

TencentAI-Infra-Guard

Features

Star history