Amazon Bedrock Knowledge Base MCP Server

Instructions

Implementation Reference

• src/bedrock_kb_mcp_server/main.py:439-466 (handler)

  The 'get_knowledge_base' MCP tool handler function in the main module. It validates the input and calls the Bedrock client.

  @mcp.tool()  # MCPツールとして公開
  @handle_errors  # エラーハンドリングデコレータを適用
  def get_knowledge_base(knowledge_base_id: str) -> KnowledgeBaseDetailDict:
      """
      特定のAmazon Bedrock Knowledge Baseの詳細情報を取得します。
      
      Args:
          knowledge_base_id: Knowledge BaseのID
  
      Returns:
          KnowledgeBaseDetailDict: Knowledge Baseの詳細情報
              - id: Knowledge BaseのID
              - name: Knowledge Baseの名前
              - status: Knowledge Baseのステータス
              - description: Knowledge Baseの説明（オプション）
              - arn: Knowledge BaseのARN（オプション）
              - その他の設定情報
      
      Raises:
          ValueError: knowledge_base_idが空の場合
      """
      # 入力値のバリデーション（共通関数を使用）
      knowledge_base_id = validate_required_string(knowledge_base_id, "knowledge_base_id")
      
      # BedrockクライアントからKnowledge Baseの詳細を取得
      kb = bedrock_client.get_knowledge_base(knowledge_base_id)
      return kb

• src/bedrock_kb_mcp_server/bedrock_client.py:236-275 (handler)

  The underlying implementation of 'get_knowledge_base' within the BedrockKBClient class, which interacts directly with the AWS SDK (boto3).

  def get_knowledge_base(self, knowledge_base_id: str) -> Dict[str, Any]:
      """
      特定のKnowledge Baseの詳細情報を取得します。
      
      Knowledge Baseの設定、ステータス、ストレージ設定などの
      詳細な情報を取得します。
  
      Args:
          knowledge_base_id: Knowledge BaseのID
  
      Returns:
          Dict[str, Any]: Knowledge Baseの詳細情報
              - id: Knowledge BaseのID
              - name: Knowledge Baseの名前
              - status: Knowledge Baseのステータス
              - description: Knowledge Baseの説明
              - roleArn: 使用されているIAMロールのARN
              - storageConfiguration: ストレージ設定
              - knowledgeBaseArn: Knowledge BaseのARN
              - createdAt: 作成日時
              - updatedAt: 最終更新日時
      
      Raises:
          ClientError: AWS API呼び出しが失敗した場合
      """
      try:
          # AWS Bedrock APIを呼び出してKnowledge Baseの詳細を取得
          response = self.bedrock_agent.get_knowledge_base(
              knowledgeBaseId=knowledge_base_id
          )
          
          # 取得成功をログに記録
          logger.info(f"Retrieved knowledge base: {knowledge_base_id}")
          
          # Knowledge Baseの詳細情報を返す
          return response["knowledgeBase"]
      except ClientError as e:
          logger.error(f"Error getting knowledge base {knowledge_base_id}: {e}")
          raise