AWS Bedrock KB Retrieval

Amazon Bedrock Knowledge Base Retrieval MCP 서버

Amazon Bedrock Knowledge Base에 접근하기 위한 MCP 서버

기능

지식 베이스 및 데이터 소스 검색

• 사용 가능한 모든 지식 베이스 찾기 및 탐색
• 이름 또는 태그로 지식 베이스 검색
• 각 지식 베이스와 연결된 데이터 소스 나열

자연어로 지식 베이스 쿼리

• 대화형 쿼리를 사용하여 정보 검색
• 지식 베이스에서 관련 구절 가져오기
• 모든 결과에 대한 인용 정보 접근

데이터 소스별 결과 필터링

• 특정 데이터 소스에 쿼리 집중
• 특정 데이터 소스 포함 또는 제외
• 특정 데이터 소스의 결과 우선 처리

결과 재순위화

• 검색 결과의 관련성 향상
• Amazon Bedrock 재순위화 기능 사용
• 쿼리 관련성에 따라 결과 정렬

사전 요구 사항

설치 요구 사항

1. Astral 또는 GitHub README에서 uv 설치
2. uv python install 3.10를 사용하여 Python 설치

AWS 요구 사항

1. AWS CLI 구성: Amazon Bedrock 및 Knowledge Base에 접근할 수 있는 자격 증명과 AWS_PROFILE로 AWS CLI가 구성되어 있어야 합니다.
2. Amazon Bedrock Knowledge Base: 태그 키 mcp-multirag-kb의 값이 true인 Amazon Bedrock Knowledge Base가 하나 이상 있어야 합니다.
3. IAM 권한: IAM 역할/사용자에게 다음을 수행할 적절한 권한이 있어야 합니다:
   • 지식 베이스 나열 및 설명
   • 데이터 소스 접근
   • 지식 베이스 쿼리

재순위화 요구 사항

재순위화 기능을 사용하려면 Bedrock Knowledge Base에 추가 권한이 필요합니다:

1. IAM 역할에 bedrock:Rerank 및 bedrock:InvokeModel 작업에 대한 권한이 있어야 합니다.
2. Amazon Bedrock Knowledge Bases 서비스 역할에도 이러한 권한이 있어야 합니다.
3. 재순위화는 특정 리전에서만 사용할 수 있습니다. 지원되는 리전의 최신 목록은 공식 문서를 참조하십시오.
4. 지정된 리전에서 사용 가능한 재순위화 모델에 대한 모델 접근을 활성화하십시오.

재순위화 제어

재순위화는 BEDROCK_KB_RERANKING_ENABLED 환경 변수를 사용하여 전역적으로 활성화 또는 비활성화할 수 있습니다:

• false로 설정 (기본값): 명시적으로 활성화되지 않은 모든 쿼리에 대해 재순위화 비활성화
• true로 설정: 명시적으로 비활성화되지 않은 모든 쿼리에 대해 재순위화 활성화

환경 변수는 다양한 형식을 허용합니다:

• 활성화: 'true', '1', 'yes', 또는 'on' (대소문자 구분 없음)
• 비활성화: 다른 값 또는 설정되지 않음 (기본 동작)

이 설정은 전역 기본값을 제공하며, 개별 API 호출 시 reranking 매개변수를 명시적으로 설정하여 재정의할 수 있습니다.

지식 베이스 설정에 대한 자세한 지침은 다음을 참조하십시오:

• 지식 베이스 생성
• Amazon Bedrock 지식 베이스 권한 관리
• Amazon Bedrock 재순위화 권한

설치

Kiro	Cursor	VS Code

MCP 클라이언트 구성에서 MCP 서버를 구성합니다 (예: Kiro의 경우 ~/.kiro/settings/mcp.json 편집):

{
  "mcpServers": {
    "awslabs.bedrock-kb-retrieval-mcp-server": {
      "command": "uvx",
      "args": ["awslabs.bedrock-kb-retrieval-mcp-server@latest"],
      "env": {
        "AWS_PROFILE": "your-profile-name",
        "AWS_REGION": "us-east-1",
        "FASTMCP_LOG_LEVEL": "ERROR",
        "KB_INCLUSION_TAG_KEY": "optional-tag-key-to-filter-kbs",
        "BEDROCK_KB_RERANKING_ENABLED": "false"
      },
      "disabled": false,
      "autoApprove": []
    }
  }
}

Windows 설치

Windows 사용자의 경우 MCP 서버 구성 형식이 약간 다릅니다:

{
  "mcpServers": {
    "awslabs.bedrock-kb-retrieval-mcp-server": {
      "disabled": false,
      "timeout": 60,
      "type": "stdio",
      "command": "uv",
      "args": [
        "tool",
        "run",
        "--from",
        "awslabs.bedrock-kb-retrieval-mcp-server@latest",
        "awslabs.bedrock-kb-retrieval-mcp-server.exe"
      ],
      "env": {
        "FASTMCP_LOG_LEVEL": "ERROR",
        "AWS_PROFILE": "your-aws-profile",
        "AWS_REGION": "us-east-1"
      }
    }
  }
}

또는 docker build -t awslabs/bedrock-kb-retrieval-mcp-server . 성공 후 docker:

# fictitious `.env` file with AWS temporary credentials
AWS_ACCESS_KEY_ID=ASIAIOSFODNN7EXAMPLE
AWS_SECRET_ACCESS_KEY=wJalrXUtnFEMI/K7MDENG/bPxRfiCYEXAMPLEKEY
AWS_SESSION_TOKEN=AQoEXAMPLEH4aoAH0gNCAPy...truncated...zrkuWJOgQs8IZZaIv2BXIa2R4Olgk

  {
    "mcpServers": {
      "awslabs.bedrock-kb-retrieval-mcp-server": {
        "command": "docker",
        "args": [
          "run",
          "--rm",
          "--interactive",
          "--env",
          "FASTMCP_LOG_LEVEL=ERROR",
          "--env",
          "KB_INCLUSION_TAG_KEY=optional-tag-key-to-filter-kbs",
          "--env",
          "BEDROCK_KB_RERANKING_ENABLED=false",
          "--env",
          "AWS_REGION=us-east-1",
          "--env-file",
          "/full/path/to/file/above/.env",
          "awslabs/bedrock-kb-retrieval-mcp-server:latest"
        ],
        "env": {},
        "disabled": false,
        "autoApprove": []
      }
    }
  }

참고: 자격 증명은 호스트에서 계속 새로 고쳐야 합니다.

제한 사항

• IMAGE 콘텐츠 유형의 결과는 KB 쿼리 응답에 포함되지 않습니다.
• reranking 매개변수는 추가 권한, Amazon Bedrock 모델 접근이 필요하며 특정 리전에서만 사용할 수 있습니다.