Amazon Bedrock Knowledge Base MCP Server

Instructions

Implementation Reference

• src/bedrock_kb_mcp_server/main.py:727-759 (handler)

  The implementation of the `list_data_sources` MCP tool, which validates the knowledge_base_id and delegates the data retrieval to the `bedrock_client.list_data_sources` method.

  @mcp.tool()  # MCPツールとして公開
  @handle_errors  # エラーハンドリングデコレータを適用
  def list_data_sources(knowledge_base_id: str) -> DataSourceListResponseDict:
      """
      指定されたKnowledge Baseのデータソース一覧を取得します。
      
      Knowledge Baseに紐づけられているすべてのデータソースを取得します。
  
      Args:
          knowledge_base_id: Knowledge BaseのID
  
      Returns:
          DataSourceListResponseDict: データソース一覧
              - count: データソースの数
              - data_sources: データソースの詳細情報のリスト
                  各要素には以下の情報が含まれます:
                  - id: データソースのID
                  - name: データソースの名前
                  - status: データソースのステータス
                  - dataSourceConfiguration: データソースの設定情報
      
      Raises:
          ValueError: knowledge_base_idが空の場合
      """
      # 入力値のバリデーション（共通関数を使用）
      knowledge_base_id = validate_required_string(knowledge_base_id, "knowledge_base_id")
      
      # Bedrockクライアントからデータソース一覧を取得
      data_sources = bedrock_client.list_data_sources(knowledge_base_id)
      return {
          "count": len(data_sources),
          "data_sources": data_sources,
      }