AWS Knowledge Base

🚀 AWS知識庫檢索MCP服務器

本項目是一個用於從AWS知識庫中檢索信息的MCP服務器實現，藉助Bedrock代理運行時，能高效地從AWS知識庫中獲取所需信息。

🚀 快速開始

本服務器藉助Bedrock代理運行時，可實現從AWS知識庫中檢索信息。在使用前，請確保完成相關配置，具體配置步驟見下文。

✨ 主要特性

• RAG（基於檢索的增強生成）：根據查詢和知識庫ID從AWS知識庫中檢索上下文。
• 支持多結果檢索：可以選擇檢索自定義數量的結果。

📦 安裝指南

本地開發

安裝依賴：

npm install

啟動開發服務器：

npm start

💻 使用示例

基礎用法

從本地存儲的JSON文件加載知識庫

將知識庫內容存儲為knowledge.json，然後使用以下命令進行檢索：

node index.js --file knowledge.json --query "你的查詢問題"

使用Docker運行服務器

構建鏡像：

docker build -t mcp-aws-kb .

運行容器並掛載知識庫文件夾：

docker run -it --mount type=bind,source=/path/to/knowledge,target=/app/knowledge mcp-aws-kb

📚 詳細文檔

工具

• retrieve_from_aws_kb
  • 執行使用AWS知識庫的檢索操作。
  • 輸入：
    • query（字符串）：檢索查詢。
    • knowledgeBaseId（字符串）：AWS知識庫ID。
    • n（數字，可選）：要檢索的結果數量，默認為10。
  • 輸出：
    • 返回與查詢相關的上下文段落列表，每個結果包含相關性評分。

配置

AWS憑證配置

• 使用IAM訪問密鑰： 在環境變量中設置以下參數：

AWS_ACCESS_KEY_ID=your_access_key_id
AWS_SECRET_ACCESS_KEY=your_secret_access_key

• 使用AWS SSO： 在環境變量中設置以下參數：

AWS_PROFILE=your_sso_profile_name

知識庫配置

在環境變量中設置知識庫ID：

AWS_KB_ID=your_knowledge_base_id

📄 許可證

本MCP服務器根據MIT許可證發佈。這意味著您可以自由使用、修改和分發軟件，前提是您遵守MIT許可證的條款和條件。有關詳細信息，請參閱項目存儲庫中的LICENSE文件。

⚠️ 重要提示

此README假設您的服務器包名為@modelcontextprotocol/server-aws-kb-retrieval。如果您的設置不同，請調整包名和安裝詳細信息。同時，請確保您的服務器腳本正確構建，並且所有依賴項在package.json中正確管理。