AWS Bedrock KB Retrieval

Amazon Bedrock Knowledge Base Retrieval MCP Server

เซิร์ฟเวอร์ MCP สำหรับการเข้าถึง Amazon Bedrock Knowledge Bases

คุณสมบัติ

ค้นพบฐานความรู้และแหล่งข้อมูลของพวกมัน

• ค้นหาและสำรวจฐานความรู้ที่มีอยู่ทั้งหมด
• ค้นหาฐานความรู้ตามชื่อหรือแท็ก
• แสดงรายการแหล่งข้อมูลที่เชื่อมโยงกับแต่ละฐานความรู้

สอบถามฐานความรู้ด้วยภาษาธรรมชาติ

• ดึงข้อมูลโดยใช้คำถามเชิงสนทนา
• รับข้อความที่เกี่ยวข้องจากฐานความรู้ของคุณ
• เข้าถึงข้อมูลการอ้างอิงสำหรับผลลัพธ์ทั้งหมด

กรองผลลัพธ์ตามแหล่งข้อมูล

• เน้นคำถามของคุณไปที่แหล่งข้อมูลเฉพาะ
• รวมหรือยกเว้นแหล่งข้อมูลเฉพาะ
• จัดลำดับความสำคัญผลลัพธ์จากแหล่งข้อมูลเฉพาะ

จัดอันดับผลลัพธ์ใหม่

• ปรับปรุงความเกี่ยวข้องของผลลัพธ์การดึงข้อมูล
• ใช้ความสามารถการจัดอันดับใหม่ของ Amazon Bedrock
• เรียงลำดับผลลัพธ์ตามความเกี่ยวข้องกับคำถามของคุณ

ข้อกำหนดเบื้องต้น

ข้อกำหนดในการติดตั้ง

1. ติดตั้ง uv จาก Astral หรือ GitHub README
2. ติดตั้ง Python โดยใช้ uv python install 3.10

ข้อกำหนดของ AWS

1. การกำหนดค่า AWS CLI: คุณต้องกำหนดค่า AWS CLI ด้วยข้อมูลประจำตัวและ AWS_PROFILE ที่มีสิทธิ์เข้าถึง Amazon Bedrock และ Knowledge Bases
2. Amazon Bedrock Knowledge Base: คุณต้องมี Amazon Bedrock Knowledge Base อย่างน้อยหนึ่งรายการที่มีคีย์แท็ก mcp-multirag-kb พร้อมค่า true
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 knowledge bases
• สิทธิ์สำหรับการจัดอันดับใหม่ใน 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 หลังจาก docker build -t awslabs/bedrock-kb-retrieval-mcp-server . สำเร็จ:

# 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 และมีให้บริการเฉพาะในบางภูมิภาคเท่านั้น