에이전트 위즈: AI 인텔리전스 워크플로 및 보안 위험 분석

일반 소개

Agent-Wiz는 개발자, 연구원 및 보안 팀을 위해 설계된 오픈 소스 Python 명령줄 도구입니다. LangGraph, CrewAI, AutoGen 등과 같은 주요 AI 인텔리전스 프레임워크에서 복잡한 워크플로를 추출하고 직관적인 시각적 다이어그램을 생성하며 MAESTRO 위협 모델링 프레임워크로 잠재적인 보안 위험을 자동으로 분석합니다.Agent-Wiz는 사용자에게 인텔리전스, 도구 및 데이터 흐름 간의 상호작용 로직에 대한 명확한 이해를 제공하며 코드 디버깅, 최적화 설계 또는 시스템 보안 평가에 적합합니다.

기능 목록

• AI 인텔리전스의 워크플로우를 추출하여 자세한 노드 및 연결 데이터를 생성합니다.
• 인텔리전스, 도구 및 데이터 흐름 간의 동적 관계를 보여주는 대화형 시각화 차트를 생성합니다.
• MAESTRO 프레임워크에 기반한 위협 모델링 및 보안 위험 보고서 출력.
• AutoGen, CrewAI, LangGraph, LlamaIndex 등 다양한 AI 프레임워크를 지원합니다.
• 간단한 명령줄 인터페이스를 제공하고, JSON 형식 출력을 지원하며, 개발자를 쉽게 통합할 수 있습니다.
• 사용자 정의 구문 분석기가 새로운 프레임워크에 대한 지원을 확장할 수 있도록 허용합니다.

도움말 사용

Agent-Wiz는 설치가 간편하고 강력한 Python 기반의 경량 명령줄 도구입니다. 다음은 사용자가 빠르게 시작하고 기능을 최대한 활용할 수 있도록 도와주는 자세한 설치 및 사용 가이드입니다.

설치 프로세스

Agent-Wiz는 Python 3.8 이상이 필요하며 일부 기능에 대해 OpenAI API를 사용합니다. 다음은 설치 단계입니다:

1. Agent-Wiz 설치
   터미널을 열고 다음 명령을 실행하여 pip를 통해 직접 설치합니다:

   pip install repello-agent-wiz
   

   그러면 에이전트 마법사와 클릭, 그래프 시각화 등과 같은 종속 라이브러리가 자동으로 설치됩니다.

2. OpenAI API 키 설정
   일부 기능(예: 위협 모델링)은 OpenAI API 지원이 필요합니다. 다음 명령을 실행하여 환경 변수를 설정하세요:
   • macOS/Linux:

     export OPENAI_API_KEY=sk-...
     

   • Windows:

     set OPENAI_API_KEY=sk-...
     

   설정을 유지하려면 위의 명령어를.bashrc및.zshrc또는 환경 변수 구성 파일에서 설정할 수 있습니다.

3. 설치 확인
   설치가 완료되면 다음 명령을 실행하여 성공 여부를 확인합니다:

   agent-wiz --help
   

   명령 도움말 메시지가 표시되면 설치가 올바른 것입니다.

사용법

Agent-Wiz는 세 가지 핵심 명령을 제공합니다:extract(추출 워크플로),visualize(시각화 차트 생성) 및analyze(위협 모델링 수행). 다음은 이를 수행하기 위한 자세한 지침입니다.

1. 추출 워크플로

이 함수는 코드를 정적으로 구문 분석하여 지능형 객체의 워크플로우를 추출합니다. CrewAI 기반 프로젝트가 있다고 가정하고 다음 명령을 실행합니다:

agent-wiz extract --framework crewai --directory ./my_project --output workflow.json

• --framework지원 대상 프레임워크 지정autogen및crewai및langgraph및llamaindex및n8n및swarm등
• --directory프로젝트 폴더 경로를 지정하면 에이전트 마법사가 코드 구조를 분석합니다.
• --output출력 파일 경로를 지정하여 워크플로 데이터를 JSON 형식으로 생성합니다.
  출력 JSON 파일에는 다음과 같은 구조의 노드(인텔리전스, 도구 등)와 에지(상호 작용)가 포함되어 있습니다:

{
"nodes": [{"id": "agent1", "type": "agent"}, ...],
"edges": [{"from": "agent1", "to": "tool1"}, ...],
"metadata": {"framework": "crewai"}
}

2. 시각화 차트 생성

시각화 기능은 워크플로를 인텔리전스 간의 연결을 보여주는 대화형 다이어그램으로 변환합니다. 다음 명령을 실행합니다:

agent-wiz visualize --input workflow.json --open

• --input: 추출할 JSON 파일을 지정합니다.
• --open선택 사항, 기본 브라우저에서 생성된 HTML 차트를 자동으로 엽니다.
  차트는 D3.js를 기반으로 하며 사용자가 확대/축소하고 끌어서 세부 정보를 볼 수 있는 인텔리전스, 도구 및 데이터 흐름 간의 동적 관계를 보여줍니다.

3. 위협 모델링 수행

위협 모델링 기능은 MAESTRO 프레임워크를 기반으로 하며 워크플로우의 보안 위험을 분석합니다. 다음 명령을 실행하세요:

agent-wiz analyze --input workflow.json

• --input: 추출한 JSON 파일을 사용합니다.
  실행 후 도구는 마크다운 형식의 보고서를 생성합니다(예crewai_report.md), 콘텐츠 포함:
• 시스템 목표 및 자산 목록.
• 공격 포털 및 잠재적 위협.
• 위험 평가 및 안전 권장 사항
  MAESTRO 프레임워크는 구체적으로 다음과 같은 측면을 분석합니다:
• 미션 목표시스템 기능 및 보안 요구 사항을 명확히 합니다.
• 부채인텔리전스, 도구 및 데이터 스트림을 나열합니다.
• 입구: 가능한 공격 지점을 식별합니다.
• 제어기존 방어 메커니즘에 대한 평가.
• 위협취약점 및 공격 시나리오를 나열합니다.
• 노출영향 및 가능성을 계산합니다.
• 정기 서비스(기차, 버스 등)작동 안전 문제를 확인합니다.

4. 도움말 정보 보기

명령에 대한 자세한 내용을 보려면 실행하세요:

agent-wiz --help

여기에는 사용 가능한 모든 명령과 해당 매개변수가 나열되어 사용자가 기능을 탐색하는 데 도움이 됩니다.

지원되는 프레임워크

Agent-Wiz는 다음과 같은 주요 AI 인텔리전스 프레임워크를 지원하며, 각각 정확한 추출을 보장하는 전용 AST 파서를 갖추고 있습니다:

• 자동 생성핵심 기능 및 상담원 채팅 모듈을 지원합니다.
• CrewAI: 팀워크 인텔리전스 구문 분석.
• LangGraph복잡한 지시형 그래프 워크플로 처리하기.
• LlamaIndex데이터 인덱싱 및 쿼리 인텔리전스 지원.
• n8n: 자동화된 워크플로 구문 분석.
• OpenAI 에이전트OpenAI의 지능형 신체 API와 호환됩니다.
• Pydantic-AIPydantic 기반 인텔리전스 지원.
• 스웜다중 지능 협업 처리.

고급 사용

• 배치 분석반복할 스크립트 작성extract노래로 응답visualize를 사용하여 여러 프로젝트를 처리합니다.
• 사용자 지정 출력차트 스타일을 조정하고 프로젝트 문서에서 D3.js 구성을 참조하세요.
• 확장 프레임워크개발자는 SDK를 기반으로 새로운 구문 분석기를 작성하여 다른 프레임워크에 대한 지원을 확장할 수 있습니다.
• 보안 보고서 최적화수동 검토와 함께 위협 모델링 보고서를 개선합니다.

주의

• 프로젝트 코드에 구문 분석을 쉽게 할 수 있도록 명확한 구조와 주석이 포함되어 있는지 확인하세요.
• OpenAI API 키가 유효해야 하며 그렇지 않으면 위협 모델링 기능을 사용할 수 없습니다.
• 종속성 문제가 발생하면 다음 명령을 실행하여 pip를 업그레이드하세요:

  pip install --upgrade pip
  

• 시각화 차트는 최상의 경험을 위해 Chrome과 같은 최신 브라우저에서 볼 것을 권장합니다.

이러한 단계를 통해 사용자는 Agent-Wiz를 쉽게 설치하고, 워크플로우를 추출하고, 차트를 생성하고, 보안 위험을 분석할 수 있습니다.

애플리케이션 시나리오

1. 개발자가 지능형 신체 시스템을 최적화합니다.
   LangGraph 인텔리전스를 개발할 때 개발자는 복잡한 상호작용 로직을 이해하는 데 어려움을 겪을 수 있습니다. Agent-Wiz는 워크플로우를 추출하고 다이어그램을 생성하여 문제를 빠르게 파악하고 시스템 효율성을 개선할 수 있도록 도와줍니다.
2. 보안 팀이 잠재적 위협을 식별합니다.
   보안 전문가는 취약점에 대한 AutoGen 인텔리전스를 평가해야 합니다. Agent-Wiz의 MAESTRO 위협 모델링 기능은 공격 진입 지점 및 개선 권장 사항을 나열하는 자세한 보고서를 생성합니다.
3. AI 프레임워크 시연을 위한 기술 교육
   교육 강사는 Agent-Wiz를 사용하여 CrewAI 워크플로우의 대화형 다이어그램을 생성하여 학생들에게 다중 지능 협업의 원리를 시각적으로 보여줌으로써 학습 문턱을 낮춥니다.
4. 연구원들이 새로운 프레임워크 분석
   연구자들이 스웜과 같은 새로운 프레임워크를 탐색할 때 에이전트 위즈를 사용하여 워크플로를 추출하고, 다양한 프레임워크의 장단점을 비교하고, 학술 연구를 지원할 수 있습니다.

QA

1. 에이전트 위즈에 OpenAI API가 필요한가요?
   위협 모델링 기능에는 OpenAI API 키가 필요하지만, 추출 및 시각화 기능에는 API가 필요하지 않습니다.
2. 어떤 위협 모델링 프레임워크가 지원되나요?
   현재 MAESTRO가 지원되고 있으며, STRIDE 및 PASTA와 같은 프레임워크는 개발 중입니다.
3. 대규모 프로젝트를 처리할 수 있나요?
   예, 하지만 프로젝트 코드가 잘 구조화되어 있고 더 효율적으로 구문 분석되는 것이 좋습니다.
4. 출력 차트를 다른 형식으로 내보낼 수 있나요?
   현재 HTML과 JSON이 지원되며 향후 PNG 또는 SVG가 지원될 수 있습니다.