@miguel-carrera/aws-bedrock-mcp-server 1.0.0

package/README.md

@@ -0,0 +1,106 @@
+# aws-bedrock-mcp-server
+
+MCP server for AWS Bedrock Knowledge Base — exposes two tools for semantic search and RAG.
+
+| Tool | Description |
+|------|-------------|
+| `bedrock_retrieve` | Semantic / hybrid search — returns raw passages from the Knowledge Base |
+| `bedrock_retrieve_and_generate` | Full RAG pipeline — retrieves passages **and** generates a grounded answer |
+
+---
+
+## Usage with Claude Code
+
+```bash
+claude mcp add aws-bedrock -- npx -y @miguel-carrera/aws-bedrock-mcp-server
+```
+
+Or add to your `claude_desktop_config.json` / `.mcp.json`:
+
+```json
+{
+  "mcpServers": {
+    "aws-bedrock": {
+      "command": "npx",
+      "args": ["-y", "@miguel-carrera/aws-bedrock-mcp-server"],
+      "env": {
+        "BEDROCK_KNOWLEDGE_BASE_ID": "<your-kb-id>",
+        "AWS_REGION": "us-east-1",
+        "AWS_PROFILE": "default"
+      }
+    }
+  }
+}
+```
+
+---
+
+## Prerequisites
+
+- Node.js ≥ 18
+- AWS credentials configured in `~/.aws/credentials` or `~/.aws/config`
+
+Required IAM permissions:
+
+```json
+{
+  "Effect": "Allow",
+  "Action": [
+    "bedrock:Retrieve",
+    "bedrock:RetrieveAndGenerate"
+  ],
+  "Resource": "arn:aws:bedrock:*:*:knowledge-base/*"
+}
+```
+
+---
+
+## Configuration
+
+| Environment variable | Required | Default | Description |
+|---|---|---|---|
+| `BEDROCK_KNOWLEDGE_BASE_ID` | Recommended | — | Default Knowledge Base ID; can be overridden per tool call |
+| `AWS_REGION` | No | `us-east-1` | AWS region where your Knowledge Base lives |
+| `AWS_PROFILE` | No | `default` | AWS named profile from `~/.aws` |
+
+---
+
+## Tools
+
+### `bedrock_retrieve`
+
+Search the Knowledge Base and return raw passages.
+
+| Parameter | Type | Default | Description |
+|---|---|---|---|
+| `query` | string | — | Natural language search query (required) |
+| `knowledge_base_id` | string | env var | Bedrock KB ID |
+| `number_of_results` | number | 5 | Passages to return (1–100) |
+| `search_type` | `SEMANTIC` \| `HYBRID` | `SEMANTIC` | Search strategy |
+| `next_token` | string | — | Pagination cursor |
+| `response_format` | `markdown` \| `json` | `markdown` | Output format |
+
+### `bedrock_retrieve_and_generate`
+
+Retrieve passages and generate a grounded answer using a Bedrock model.
+
+| Parameter | Type | Default | Description |
+|---|---|---|---|
+| `query` | string | — | Question to answer (required) |
+| `knowledge_base_id` | string | env var | Bedrock KB ID |
+| `model_arn` | string | Claude 3 Haiku | ARN of the Bedrock model for generation |
+| `number_of_results` | number | 5 | Passages to retrieve (1–100) |
+| `search_type` | `SEMANTIC` \| `HYBRID` | `SEMANTIC` | Search strategy |
+| `session_id` | string | — | Session ID for multi-turn conversations |
+| `system_prompt` | string | — | Custom generation prompt (use `$search_results$` placeholder) |
+| `response_format` | `markdown` \| `json` | `markdown` | Output format |
+
+---
+
+## Development
+
+```bash
+npm install
+npm run dev      # watch mode
+npm run build    # compile to dist/
+```

package/dist/index.d.ts

@@ -0,0 +1,15 @@
+#!/usr/bin/env node
+/**
+ * MCP Server for AWS Bedrock Knowledge Base Retrieve API.
+ *
+ * Provides two tools:
+ *  - bedrock_retrieve              Semantic / hybrid search over a Knowledge Base
+ *  - bedrock_retrieve_and_generate RAG: retrieve + generate an answer via a Bedrock model
+ *
+ * Configuration (environment variables):
+ *  BEDROCK_KNOWLEDGE_BASE_ID  Default Knowledge Base ID (required unless passed per-call)
+ *  AWS_REGION                 AWS region (default: us-east-1)
+ *  AWS_PROFILE                AWS named profile (default: default)
+ */
+export {};
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AACA;;;;;;;;;;;GAWG"}

package/dist/index.js

@@ -0,0 +1,378 @@
+#!/usr/bin/env node
+/**
+ * MCP Server for AWS Bedrock Knowledge Base Retrieve API.
+ *
+ * Provides two tools:
+ *  - bedrock_retrieve              Semantic / hybrid search over a Knowledge Base
+ *  - bedrock_retrieve_and_generate RAG: retrieve + generate an answer via a Bedrock model
+ *
+ * Configuration (environment variables):
+ *  BEDROCK_KNOWLEDGE_BASE_ID  Default Knowledge Base ID (required unless passed per-call)
+ *  AWS_REGION                 AWS region (default: us-east-1)
+ *  AWS_PROFILE                AWS named profile (default: default)
+ */
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { z } from "zod";
+import { retrieve, retrieveAndGenerate, formatAwsError, } from "./services/bedrock.js";
+// ── Constants ─────────────────────────────────────────────────────────────────
+const CHARACTER_LIMIT = 25_000;
+const DEFAULT_KB_ID = process.env.BEDROCK_KNOWLEDGE_BASE_ID ?? "";
+// ── Helpers ───────────────────────────────────────────────────────────────────
+function formatLocation(location) {
+    if (!location)
+        return "unknown";
+    if (location.type === "S3")
+        return location.s3Location?.uri ?? "s3://unknown";
+    if (location.type === "WEB")
+        return location.webLocation?.url ?? "web://unknown";
+    if (location.type === "CONFLUENCE")
+        return location.confluenceLocation?.url ?? "confluence://unknown";
+    if (location.type === "SALESFORCE")
+        return location.salesforceLocation?.url ?? "salesforce://unknown";
+    if (location.type === "SHAREPOINT")
+        return location.sharePointLocation?.url ?? "sharepoint://unknown";
+    return location.type ?? "unknown";
+}
+function passagesToMarkdown(passages) {
+    return passages
+        .map((p, i) => {
+        const score = p.score !== undefined ? ` (score: ${p.score.toFixed(4)})` : "";
+        const src = formatLocation(p.location);
+        return `### Result ${i + 1}${score}\n**Source**: ${src}\n\n${p.content}`;
+    })
+        .join("\n\n---\n\n");
+}
+function citationsToMarkdown(citations) {
+    return citations
+        .map((c, i) => {
+        const refs = c.passages
+            .map((p) => `  - ${formatLocation(p.location)}`)
+            .join("\n");
+        return `**[${i + 1}]** ${c.generatedText ?? ""}\n*Sources*:\n${refs}`;
+    })
+        .join("\n\n");
+}
+function truncate(text) {
+    if (text.length <= CHARACTER_LIMIT)
+        return { text, truncated: false };
+    return {
+        text: text.slice(0, CHARACTER_LIMIT) + "\n\n[Response truncated. Use `numberOfResults` or a more specific query to reduce output size.]",
+        truncated: true,
+    };
+}
+// ── Server ────────────────────────────────────────────────────────────────────
+const server = new McpServer({
+    name: "bedrock-mcp-server",
+    version: "1.0.0",
+});
+// ── Tool: bedrock_retrieve ────────────────────────────────────────────────────
+const RetrieveInputSchema = z
+    .object({
+    query: z
+        .string()
+        .min(1, "Query must not be empty")
+        .max(1000, "Query must not exceed 1000 characters")
+        .describe("Natural language query to search the Knowledge Base"),
+    knowledge_base_id: z
+        .string()
+        .optional()
+        .describe("Bedrock Knowledge Base ID. Defaults to BEDROCK_KNOWLEDGE_BASE_ID env var if not provided."),
+    number_of_results: z
+        .number()
+        .int()
+        .min(1)
+        .max(100)
+        .default(5)
+        .describe("Number of passages to retrieve (1–100, default: 5)"),
+    search_type: z
+        .enum(["SEMANTIC", "HYBRID"])
+        .default("SEMANTIC")
+        .describe("Search strategy: 'SEMANTIC' for pure vector search, 'HYBRID' for vector + keyword (default: SEMANTIC)"),
+    next_token: z
+        .string()
+        .optional()
+        .describe("Pagination token from a previous response to fetch the next page of results"),
+    response_format: z
+        .enum(["markdown", "json"])
+        .default("markdown")
+        .describe("Output format: 'markdown' for human-readable, 'json' for structured data"),
+})
+    .strict();
+server.registerTool("bedrock_retrieve", {
+    title: "Retrieve from Bedrock Knowledge Base",
+    description: `Perform a semantic or hybrid search against an AWS Bedrock Knowledge Base and return the most relevant text passages.
+
+Use this tool when you need to find specific information stored in the Knowledge Base — for example, product documentation, FAQs, internal policies, or any indexed content.
+
+Args:
+  - query (string): Natural language question or search phrase (required)
+  - knowledge_base_id (string): Bedrock KB ID (e.g. "ABCDE12345"). Defaults to BEDROCK_KNOWLEDGE_BASE_ID env var.
+  - number_of_results (number): How many passages to return, 1–100 (default: 5)
+  - search_type ('SEMANTIC' | 'HYBRID'): Search strategy (default: 'SEMANTIC')
+  - next_token (string): Pagination cursor from a previous call
+  - response_format ('markdown' | 'json'): Output format (default: 'markdown')
+
+Returns for JSON format:
+  {
+    "passages": [
+      {
+        "content": string,        // Retrieved text chunk
+        "score": number,          // Relevance score (0–1)
+        "location": {             // Source location
+          "type": string,         // "S3" | "WEB" | "CONFLUENCE" | etc.
+          "uri": string           // Source URI
+        },
+        "metadata": object        // Optional key-value metadata
+      }
+    ],
+    "next_token": string | null   // Cursor for next page (null if no more results)
+  }
+
+Examples:
+  - "Find passages about return policy" → query="return policy", number_of_results=5
+  - "Get top 10 results about shipping" → query="shipping", number_of_results=10, search_type="HYBRID"
+  - "Next page of results" → query="shipping", next_token="<token from previous call>"
+
+Error handling:
+  - Returns "Error: Knowledge Base not found" if the KB ID is wrong
+  - Returns "Error: Access denied" if the AWS profile lacks bedrock:Retrieve permission
+  - Returns "Error: AWS request throttled" on rate limit`,
+    inputSchema: RetrieveInputSchema,
+    annotations: {
+        readOnlyHint: true,
+        destructiveHint: false,
+        idempotentHint: true,
+        openWorldHint: true,
+    },
+}, async (params) => {
+    const kbId = params.knowledge_base_id ?? DEFAULT_KB_ID;
+    if (!kbId) {
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: "Error: No Knowledge Base ID provided. Pass 'knowledge_base_id' or set the BEDROCK_KNOWLEDGE_BASE_ID environment variable.",
+                },
+            ],
+            isError: true,
+        };
+    }
+    try {
+        const result = await retrieve({
+            knowledgeBaseId: kbId,
+            query: params.query,
+            numberOfResults: params.number_of_results,
+            searchType: params.search_type,
+            nextToken: params.next_token,
+        });
+        if (result.passages.length === 0) {
+            return {
+                content: [{ type: "text", text: `No results found for query: "${params.query}"` }],
+            };
+        }
+        const structured = {
+            passages: result.passages.map((p) => ({
+                content: p.content,
+                score: p.score,
+                location: p.location
+                    ? { type: p.location.type, uri: formatLocation(p.location) }
+                    : null,
+                metadata: p.metadata ?? null,
+            })),
+            next_token: result.nextToken ?? null,
+        };
+        let text;
+        if (params.response_format === "json") {
+            text = JSON.stringify(structured, null, 2);
+        }
+        else {
+            text = [
+                `# Knowledge Base Results for: "${params.query}"`,
+                `Found ${result.passages.length} passages${result.nextToken ? " (more available)" : ""}.`,
+                "",
+                passagesToMarkdown(result.passages),
+                ...(result.nextToken
+                    ? ["", `---`, `*More results available. Pass \`next_token: "${result.nextToken}"\` to continue.*`]
+                    : []),
+            ].join("\n");
+        }
+        const { text: finalText } = truncate(text);
+        return {
+            content: [{ type: "text", text: finalText }],
+            structuredContent: structured,
+        };
+    }
+    catch (error) {
+        return {
+            content: [{ type: "text", text: formatAwsError(error) }],
+            isError: true,
+        };
+    }
+});
+// ── Tool: bedrock_retrieve_and_generate ──────────────────────────────────────
+const RetrieveAndGenerateInputSchema = z
+    .object({
+    query: z
+        .string()
+        .min(1, "Query must not be empty")
+        .max(1000, "Query must not exceed 1000 characters")
+        .describe("Natural language question to answer using the Knowledge Base"),
+    knowledge_base_id: z
+        .string()
+        .optional()
+        .describe("Bedrock Knowledge Base ID. Defaults to BEDROCK_KNOWLEDGE_BASE_ID env var if not provided."),
+    model_arn: z
+        .string()
+        .default("arn:aws:bedrock:us-east-1::foundation-model/anthropic.claude-3-haiku-20240307-v1:0")
+        .describe("ARN of the Bedrock foundation model used for generation (default: Claude 3 Haiku). Use a model available in your region."),
+    number_of_results: z
+        .number()
+        .int()
+        .min(1)
+        .max(100)
+        .default(5)
+        .describe("Number of passages to retrieve before generating (1–100, default: 5)"),
+    search_type: z
+        .enum(["SEMANTIC", "HYBRID"])
+        .default("SEMANTIC")
+        .describe("Search strategy: 'SEMANTIC' or 'HYBRID' (default: SEMANTIC)"),
+    session_id: z
+        .string()
+        .optional()
+        .describe("Session ID from a previous call to continue a multi-turn conversation in the same KB session"),
+    system_prompt: z
+        .string()
+        .optional()
+        .describe("Optional custom system prompt template for the generation step. Use $search_results$ as placeholder for the retrieved content."),
+    response_format: z
+        .enum(["markdown", "json"])
+        .default("markdown")
+        .describe("Output format: 'markdown' for human-readable, 'json' for structured data"),
+})
+    .strict();
+server.registerTool("bedrock_retrieve_and_generate", {
+    title: "Retrieve and Generate from Bedrock Knowledge Base",
+    description: `Perform a full RAG (Retrieval-Augmented Generation) pipeline: retrieve relevant passages from a Bedrock Knowledge Base and then generate a grounded answer using a Bedrock foundation model.
+
+Use this tool when you need a synthesized, natural language answer based on Knowledge Base content, with citations pointing back to the source passages.
+
+Args:
+  - query (string): Natural language question (required)
+  - knowledge_base_id (string): Bedrock KB ID. Defaults to BEDROCK_KNOWLEDGE_BASE_ID env var.
+  - model_arn (string): ARN of the Bedrock model to use for generation (default: Claude 3 Haiku)
+  - number_of_results (number): Passages to retrieve before generating, 1–100 (default: 5)
+  - search_type ('SEMANTIC' | 'HYBRID'): Search strategy (default: 'SEMANTIC')
+  - session_id (string): Session ID from a previous call for multi-turn conversation
+  - system_prompt (string): Custom generation prompt template (use $search_results$ placeholder)
+  - response_format ('markdown' | 'json'): Output format (default: 'markdown')
+
+Returns for JSON format:
+  {
+    "output": string,           // Generated answer grounded in KB content
+    "session_id": string,       // Session ID for follow-up questions
+    "citations": [
+      {
+        "generated_text": string,  // Span of generated text
+        "sources": [               // Passages that support this span
+          { "content": string, "location": { "type": string, "uri": string } }
+        ]
+      }
+    ]
+  }
+
+Examples:
+  - "What is the refund policy?" → query="refund policy"
+  - "Continue conversation" → query="Can I also return accessories?", session_id="<id from previous>"
+  - "Use Claude 3 Sonnet" → model_arn="arn:aws:bedrock:us-east-1::foundation-model/anthropic.claude-3-sonnet-20240229-v1:0"
+
+Error handling:
+  - Returns "Error: Knowledge Base not found" if the KB ID is wrong
+  - Returns "Error: Access denied" if missing bedrock:RetrieveAndGenerate permission
+  - Returns "Error: Invalid request" if the model ARN is incorrect`,
+    inputSchema: RetrieveAndGenerateInputSchema,
+    annotations: {
+        readOnlyHint: true,
+        destructiveHint: false,
+        idempotentHint: false,
+        openWorldHint: true,
+    },
+}, async (params) => {
+    const kbId = params.knowledge_base_id ?? DEFAULT_KB_ID;
+    if (!kbId) {
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: "Error: No Knowledge Base ID provided. Pass 'knowledge_base_id' or set the BEDROCK_KNOWLEDGE_BASE_ID environment variable.",
+                },
+            ],
+            isError: true,
+        };
+    }
+    try {
+        const result = await retrieveAndGenerate({
+            knowledgeBaseId: kbId,
+            modelArn: params.model_arn,
+            query: params.query,
+            numberOfResults: params.number_of_results,
+            searchType: params.search_type,
+            sessionId: params.session_id,
+            systemPrompt: params.system_prompt,
+        });
+        const structured = {
+            output: result.output,
+            session_id: result.sessionId ?? null,
+            citations: result.citations.map((c) => ({
+                generated_text: c.generatedText ?? null,
+                sources: c.passages.map((p) => ({
+                    content: p.content,
+                    location: p.location
+                        ? { type: p.location.type, uri: formatLocation(p.location) }
+                        : null,
+                    metadata: p.metadata ?? null,
+                })),
+            })),
+        };
+        let text;
+        if (params.response_format === "json") {
+            text = JSON.stringify(structured, null, 2);
+        }
+        else {
+            const citationBlock = result.citations.length > 0
+                ? `\n\n## Citations\n\n${citationsToMarkdown(result.citations)}`
+                : "";
+            const sessionNote = result.sessionId
+                ? `\n\n---\n*Session ID: \`${result.sessionId}\` — pass this as \`session_id\` to continue the conversation.*`
+                : "";
+            text = `# Answer\n\n${result.output}${citationBlock}${sessionNote}`;
+        }
+        const { text: finalText } = truncate(text);
+        return {
+            content: [{ type: "text", text: finalText }],
+            structuredContent: structured,
+        };
+    }
+    catch (error) {
+        return {
+            content: [{ type: "text", text: formatAwsError(error) }],
+            isError: true,
+        };
+    }
+});
+// ── Start ─────────────────────────────────────────────────────────────────────
+async function main() {
+    // Warn (but don't exit) if no default KB ID — users can still pass it per-call
+    if (!DEFAULT_KB_ID) {
+        console.error("WARNING: BEDROCK_KNOWLEDGE_BASE_ID is not set. " +
+            "You must pass 'knowledge_base_id' in every tool call, or set the env var.");
+    }
+    const transport = new StdioServerTransport();
+    await server.connect(transport);
+    console.error("Bedrock MCP Server running via stdio");
+}
+main().catch((error) => {
+    console.error("Fatal error:", error);
+    process.exit(1);
+});
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AACA;;;;;;;;;;;GAWG;AAEH,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACpE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EACL,QAAQ,EACR,mBAAmB,EACnB,cAAc,GAGf,MAAM,uBAAuB,CAAC;AAE/B,iFAAiF;AAEjF,MAAM,eAAe,GAAG,MAAM,CAAC;AAE/B,MAAM,aAAa,GAAG,OAAO,CAAC,GAAG,CAAC,yBAAyB,IAAI,EAAE,CAAC;AAElE,iFAAiF;AAEjF,SAAS,cAAc,CAAC,QAAsC;IAC5D,IAAI,CAAC,QAAQ;QAAE,OAAO,SAAS,CAAC;IAChC,IAAI,QAAQ,CAAC,IAAI,KAAK,IAAI;QAAE,OAAO,QAAQ,CAAC,UAAU,EAAE,GAAG,IAAI,cAAc,CAAC;IAC9E,IAAI,QAAQ,CAAC,IAAI,KAAK,KAAK;QAAE,OAAO,QAAQ,CAAC,WAAW,EAAE,GAAG,IAAI,eAAe,CAAC;IACjF,IAAI,QAAQ,CAAC,IAAI,KAAK,YAAY;QAAE,OAAO,QAAQ,CAAC,kBAAkB,EAAE,GAAG,IAAI,sBAAsB,CAAC;IACtG,IAAI,QAAQ,CAAC,IAAI,KAAK,YAAY;QAAE,OAAO,QAAQ,CAAC,kBAAkB,EAAE,GAAG,IAAI,sBAAsB,CAAC;IACtG,IAAI,QAAQ,CAAC,IAAI,KAAK,YAAY;QAAE,OAAO,QAAQ,CAAC,kBAAkB,EAAE,GAAG,IAAI,sBAAsB,CAAC;IACtG,OAAO,QAAQ,CAAC,IAAI,IAAI,SAAS,CAAC;AACpC,CAAC;AAED,SAAS,kBAAkB,CAAC,QAA4B;IACtD,OAAO,QAAQ;SACZ,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;QACZ,MAAM,KAAK,GAAG,CAAC,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;QAC7E,MAAM,GAAG,GAAG,cAAc,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;QACvC,OAAO,cAAc,CAAC,GAAG,CAAC,GAAG,KAAK,iBAAiB,GAAG,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC;IAC3E,CAAC,CAAC;SACD,IAAI,CAAC,aAAa,CAAC,CAAC;AACzB,CAAC;AAED,SAAS,mBAAmB,CAAC,SAAqB;IAChD,OAAO,SAAS;SACb,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;QACZ,MAAM,IAAI,GAAG,CAAC,CAAC,QAAQ;aACpB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,cAAc,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC;aAC/C,IAAI,CAAC,IAAI,CAAC,CAAC;QACd,OAAO,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,aAAa,IAAI,EAAE,iBAAiB,IAAI,EAAE,CAAC;IACxE,CAAC,CAAC;SACD,IAAI,CAAC,MAAM,CAAC,CAAC;AAClB,CAAC;AAED,SAAS,QAAQ,CAAC,IAAY;IAC5B,IAAI,IAAI,CAAC,MAAM,IAAI,eAAe;QAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;IACtE,OAAO;QACL,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,eAAe,CAAC,GAAG,iGAAiG;QACxI,SAAS,EAAE,IAAI;KAChB,CAAC;AACJ,CAAC;AAED,iFAAiF;AAEjF,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC;IAC3B,IAAI,EAAE,oBAAoB;IAC1B,OAAO,EAAE,OAAO;CACjB,CAAC,CAAC;AAEH,iFAAiF;AAEjF,MAAM,mBAAmB,GAAG,CAAC;KAC1B,MAAM,CAAC;IACN,KAAK,EAAE,CAAC;SACL,MAAM,EAAE;SACR,GAAG,CAAC,CAAC,EAAE,yBAAyB,CAAC;SACjC,GAAG,CAAC,IAAI,EAAE,uCAAuC,CAAC;SAClD,QAAQ,CAAC,qDAAqD,CAAC;IAClE,iBAAiB,EAAE,CAAC;SACjB,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CACP,2FAA2F,CAC5F;IACH,iBAAiB,EAAE,CAAC;SACjB,MAAM,EAAE;SACR,GAAG,EAAE;SACL,GAAG,CAAC,CAAC,CAAC;SACN,GAAG,CAAC,GAAG,CAAC;SACR,OAAO,CAAC,CAAC,CAAC;SACV,QAAQ,CAAC,oDAAoD,CAAC;IACjE,WAAW,EAAE,CAAC;SACX,IAAI,CAAC,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;SAC5B,OAAO,CAAC,UAAU,CAAC;SACnB,QAAQ,CACP,uGAAuG,CACxG;IACH,UAAU,EAAE,CAAC;SACV,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CAAC,6EAA6E,CAAC;IAC1F,eAAe,EAAE,CAAC;SACf,IAAI,CAAC,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;SAC1B,OAAO,CAAC,UAAU,CAAC;SACnB,QAAQ,CAAC,0EAA0E,CAAC;CACxF,CAAC;KACD,MAAM,EAAE,CAAC;AAIZ,MAAM,CAAC,YAAY,CACjB,kBAAkB,EAClB;IACE,KAAK,EAAE,sCAAsC;IAC7C,WAAW,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;yDAoCwC;IACrD,WAAW,EAAE,mBAAmB;IAChC,WAAW,EAAE;QACX,YAAY,EAAE,IAAI;QAClB,eAAe,EAAE,KAAK;QACtB,cAAc,EAAE,IAAI;QACpB,aAAa,EAAE,IAAI;KACpB;CACF,EACD,KAAK,EAAE,MAAqB,EAAE,EAAE;IAC9B,MAAM,IAAI,GAAG,MAAM,CAAC,iBAAiB,IAAI,aAAa,CAAC;IACvD,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,2HAA2H;iBAClI;aACF;YACD,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;IAED,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC;YAC5B,eAAe,EAAE,IAAI;YACrB,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,eAAe,EAAE,MAAM,CAAC,iBAAiB;YACzC,UAAU,EAAE,MAAM,CAAC,WAAW;YAC9B,SAAS,EAAE,MAAM,CAAC,UAAU;SAC7B,CAAC,CAAC;QAEH,IAAI,MAAM,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACjC,OAAO;gBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,gCAAgC,MAAM,CAAC,KAAK,GAAG,EAAE,CAAC;aACnF,CAAC;QACJ,CAAC;QAED,MAAM,UAAU,GAAG;YACjB,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBACpC,OAAO,EAAE,CAAC,CAAC,OAAO;gBAClB,KAAK,EAAE,CAAC,CAAC,KAAK;gBACd,QAAQ,EAAE,CAAC,CAAC,QAAQ;oBAClB,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,QAAQ,CAAC,IAAI,EAAE,GAAG,EAAE,cAAc,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE;oBAC5D,CAAC,CAAC,IAAI;gBACR,QAAQ,EAAE,CAAC,CAAC,QAAQ,IAAI,IAAI;aAC7B,CAAC,CAAC;YACH,UAAU,EAAE,MAAM,CAAC,SAAS,IAAI,IAAI;SACrC,CAAC;QAEF,IAAI,IAAY,CAAC;QACjB,IAAI,MAAM,CAAC,eAAe,KAAK,MAAM,EAAE,CAAC;YACtC,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QAC7C,CAAC;aAAM,CAAC;YACN,IAAI,GAAG;gBACL,kCAAkC,MAAM,CAAC,KAAK,GAAG;gBACjD,SAAS,MAAM,CAAC,QAAQ,CAAC,MAAM,YAAY,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,EAAE,GAAG;gBACzF,EAAE;gBACF,kBAAkB,CAAC,MAAM,CAAC,QAAQ,CAAC;gBACnC,GAAG,CAAC,MAAM,CAAC,SAAS;oBAClB,CAAC,CAAC,CAAC,EAAE,EAAE,KAAK,EAAE,gDAAgD,MAAM,CAAC,SAAS,mBAAmB,CAAC;oBAClG,CAAC,CAAC,EAAE,CAAC;aACR,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACf,CAAC;QAED,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;QAE3C,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;YAC5C,iBAAiB,EAAE,UAAU;SAC9B,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,cAAc,CAAC,KAAK,CAAC,EAAE,CAAC;YACxD,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;AACH,CAAC,CACF,CAAC;AAEF,gFAAgF;AAEhF,MAAM,8BAA8B,GAAG,CAAC;KACrC,MAAM,CAAC;IACN,KAAK,EAAE,CAAC;SACL,MAAM,EAAE;SACR,GAAG,CAAC,CAAC,EAAE,yBAAyB,CAAC;SACjC,GAAG,CAAC,IAAI,EAAE,uCAAuC,CAAC;SAClD,QAAQ,CAAC,8DAA8D,CAAC;IAC3E,iBAAiB,EAAE,CAAC;SACjB,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CACP,2FAA2F,CAC5F;IACH,SAAS,EAAE,CAAC;SACT,MAAM,EAAE;SACR,OAAO,CAAC,oFAAoF,CAAC;SAC7F,QAAQ,CACP,0HAA0H,CAC3H;IACH,iBAAiB,EAAE,CAAC;SACjB,MAAM,EAAE;SACR,GAAG,EAAE;SACL,GAAG,CAAC,CAAC,CAAC;SACN,GAAG,CAAC,GAAG,CAAC;SACR,OAAO,CAAC,CAAC,CAAC;SACV,QAAQ,CAAC,sEAAsE,CAAC;IACnF,WAAW,EAAE,CAAC;SACX,IAAI,CAAC,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;SAC5B,OAAO,CAAC,UAAU,CAAC;SACnB,QAAQ,CAAC,6DAA6D,CAAC;IAC1E,UAAU,EAAE,CAAC;SACV,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CACP,8FAA8F,CAC/F;IACH,aAAa,EAAE,CAAC;SACb,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CACP,gIAAgI,CACjI;IACH,eAAe,EAAE,CAAC;SACf,IAAI,CAAC,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;SAC1B,OAAO,CAAC,UAAU,CAAC;SACnB,QAAQ,CAAC,0EAA0E,CAAC;CACxF,CAAC;KACD,MAAM,EAAE,CAAC;AAIZ,MAAM,CAAC,YAAY,CACjB,+BAA+B,EAC/B;IACE,KAAK,EAAE,mDAAmD;IAC1D,WAAW,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;mEAoCkD;IAC/D,WAAW,EAAE,8BAA8B;IAC3C,WAAW,EAAE;QACX,YAAY,EAAE,IAAI;QAClB,eAAe,EAAE,KAAK;QACtB,cAAc,EAAE,KAAK;QACrB,aAAa,EAAE,IAAI;KACpB;CACF,EACD,KAAK,EAAE,MAAgC,EAAE,EAAE;IACzC,MAAM,IAAI,GAAG,MAAM,CAAC,iBAAiB,IAAI,aAAa,CAAC;IACvD,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,2HAA2H;iBAClI;aACF;YACD,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;IAED,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,mBAAmB,CAAC;YACvC,eAAe,EAAE,IAAI;YACrB,QAAQ,EAAE,MAAM,CAAC,SAAS;YAC1B,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,eAAe,EAAE,MAAM,CAAC,iBAAiB;YACzC,UAAU,EAAE,MAAM,CAAC,WAAW;YAC9B,SAAS,EAAE,MAAM,CAAC,UAAU;YAC5B,YAAY,EAAE,MAAM,CAAC,aAAa;SACnC,CAAC,CAAC;QAEH,MAAM,UAAU,GAAG;YACjB,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,UAAU,EAAE,MAAM,CAAC,SAAS,IAAI,IAAI;YACpC,SAAS,EAAE,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBACtC,cAAc,EAAE,CAAC,CAAC,aAAa,IAAI,IAAI;gBACvC,OAAO,EAAE,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;oBAC9B,OAAO,EAAE,CAAC,CAAC,OAAO;oBAClB,QAAQ,EAAE,CAAC,CAAC,QAAQ;wBAClB,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,QAAQ,CAAC,IAAI,EAAE,GAAG,EAAE,cAAc,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE;wBAC5D,CAAC,CAAC,IAAI;oBACR,QAAQ,EAAE,CAAC,CAAC,QAAQ,IAAI,IAAI;iBAC7B,CAAC,CAAC;aACJ,CAAC,CAAC;SACJ,CAAC;QAEF,IAAI,IAAY,CAAC;QACjB,IAAI,MAAM,CAAC,eAAe,KAAK,MAAM,EAAE,CAAC;YACtC,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QAC7C,CAAC;aAAM,CAAC;YACN,MAAM,aAAa,GACjB,MAAM,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC;gBACzB,CAAC,CAAC,uBAAuB,mBAAmB,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE;gBAChE,CAAC,CAAC,EAAE,CAAC;YACT,MAAM,WAAW,GAAG,MAAM,CAAC,SAAS;gBAClC,CAAC,CAAC,2BAA2B,MAAM,CAAC,SAAS,iEAAiE;gBAC9G,CAAC,CAAC,EAAE,CAAC;YACP,IAAI,GAAG,eAAe,MAAM,CAAC,MAAM,GAAG,aAAa,GAAG,WAAW,EAAE,CAAC;QACtE,CAAC;QAED,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;QAE3C,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;YAC5C,iBAAiB,EAAE,UAAU;SAC9B,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,cAAc,CAAC,KAAK,CAAC,EAAE,CAAC;YACxD,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;AACH,CAAC,CACF,CAAC;AAEF,iFAAiF;AAEjF,KAAK,UAAU,IAAI;IACjB,+EAA+E;IAC/E,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,OAAO,CAAC,KAAK,CACX,iDAAiD;YAC/C,2EAA2E,CAC9E,CAAC;IACJ,CAAC;IAED,MAAM,SAAS,GAAG,IAAI,oBAAoB,EAAE,CAAC;IAC7C,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAChC,OAAO,CAAC,KAAK,CAAC,sCAAsC,CAAC,CAAC;AACxD,CAAC;AAED,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;IACrB,OAAO,CAAC,KAAK,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC;IACrC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC"}

package/dist/services/bedrock.d.ts

@@ -0,0 +1,42 @@
+import { BedrockAgentRuntimeClient, type RetrievalResultLocation } from "@aws-sdk/client-bedrock-agent-runtime";
+export declare function getBedrockClient(): BedrockAgentRuntimeClient;
+export interface RetrievedPassage {
+    content: string;
+    score: number | undefined;
+    location: RetrievalResultLocation | undefined;
+    metadata: Record<string, unknown> | undefined;
+}
+export interface RetrieveResult {
+    passages: RetrievedPassage[];
+    nextToken: string | undefined;
+}
+export interface RetrieveAndGenerateResult {
+    output: string;
+    sessionId: string | undefined;
+    citations: Citation[];
+}
+export interface Citation {
+    generatedText: string | undefined;
+    passages: RetrievedPassage[];
+}
+export interface RetrieveParams {
+    knowledgeBaseId: string;
+    query: string;
+    numberOfResults?: number;
+    searchType?: "SEMANTIC" | "HYBRID";
+    nextToken?: string;
+    metadataFilter?: Record<string, unknown>;
+}
+export declare function retrieve(params: RetrieveParams): Promise<RetrieveResult>;
+export interface RetrieveAndGenerateParams {
+    knowledgeBaseId: string;
+    modelArn: string;
+    query: string;
+    numberOfResults?: number;
+    searchType?: "SEMANTIC" | "HYBRID";
+    sessionId?: string;
+    systemPrompt?: string;
+}
+export declare function retrieveAndGenerate(params: RetrieveAndGenerateParams): Promise<RetrieveAndGenerateResult>;
+export declare function formatAwsError(error: unknown): string;
+//# sourceMappingURL=bedrock.d.ts.map

package/dist/services/bedrock.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"bedrock.d.ts","sourceRoot":"","sources":["../../src/services/bedrock.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,yBAAyB,EAKzB,KAAK,uBAAuB,EAE7B,MAAM,uCAAuC,CAAC;AAY/C,wBAAgB,gBAAgB,IAAI,yBAAyB,CAQ5D;AAID,MAAM,WAAW,gBAAgB;IAC/B,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,GAAG,SAAS,CAAC;IAC1B,QAAQ,EAAE,uBAAuB,GAAG,SAAS,CAAC;IAC9C,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,CAAC;CAC/C;AAED,MAAM,WAAW,cAAc;IAC7B,QAAQ,EAAE,gBAAgB,EAAE,CAAC;IAC7B,SAAS,EAAE,MAAM,GAAG,SAAS,CAAC;CAC/B;AAED,MAAM,WAAW,yBAAyB;IACxC,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,GAAG,SAAS,CAAC;IAC9B,SAAS,EAAE,QAAQ,EAAE,CAAC;CACvB;AAED,MAAM,WAAW,QAAQ;IACvB,aAAa,EAAE,MAAM,GAAG,SAAS,CAAC;IAClC,QAAQ,EAAE,gBAAgB,EAAE,CAAC;CAC9B;AAID,MAAM,WAAW,cAAc;IAC7B,eAAe,EAAE,MAAM,CAAC;IACxB,KAAK,EAAE,MAAM,CAAC;IACd,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,UAAU,CAAC,EAAE,UAAU,GAAG,QAAQ,CAAC;IACnC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,cAAc,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAC1C;AAED,wBAAsB,QAAQ,CAAC,MAAM,EAAE,cAAc,GAAG,OAAO,CAAC,cAAc,CAAC,CAgC9E;AAID,MAAM,WAAW,yBAAyB;IACxC,eAAe,EAAE,MAAM,CAAC;IACxB,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,UAAU,CAAC,EAAE,UAAU,GAAG,QAAQ,CAAC;IACnC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,wBAAsB,mBAAmB,CACvC,MAAM,EAAE,yBAAyB,GAChC,OAAO,CAAC,yBAAyB,CAAC,CAgDpC;AAID,wBAAgB,cAAc,CAAC,KAAK,EAAE,OAAO,GAAG,MAAM,CAuBrD"}

package/dist/services/bedrock.js

@@ -0,0 +1,115 @@
+import { BedrockAgentRuntimeClient, RetrieveCommand, RetrieveAndGenerateCommand, } from "@aws-sdk/client-bedrock-agent-runtime";
+import { fromIni } from "@aws-sdk/credential-providers";
+// ── Configuration ────────────────────────────────────────────────────────────
+const AWS_REGION = process.env.AWS_REGION ?? "us-east-1";
+const AWS_PROFILE = process.env.AWS_PROFILE ?? "default";
+// ── Client (singleton) ───────────────────────────────────────────────────────
+let _client = null;
+export function getBedrockClient() {
+    if (!_client) {
+        _client = new BedrockAgentRuntimeClient({
+            region: AWS_REGION,
+            credentials: fromIni({ profile: AWS_PROFILE }),
+        });
+    }
+    return _client;
+}
+export async function retrieve(params) {
+    const client = getBedrockClient();
+    const input = {
+        knowledgeBaseId: params.knowledgeBaseId,
+        retrievalQuery: { text: params.query },
+        retrievalConfiguration: {
+            vectorSearchConfiguration: {
+                numberOfResults: params.numberOfResults ?? 5,
+                ...(params.searchType ? { overrideSearchType: params.searchType } : {}),
+                ...(params.metadataFilter
+                    ? { filter: params.metadataFilter }
+                    : {}),
+            },
+        },
+        ...(params.nextToken ? { nextToken: params.nextToken } : {}),
+    };
+    const command = new RetrieveCommand(input);
+    const response = await client.send(command);
+    const passages = (response.retrievalResults ?? []).map((r) => ({
+        content: r.content?.text ?? "",
+        score: r.score,
+        location: r.location,
+        metadata: r.metadata,
+    }));
+    return {
+        passages,
+        nextToken: response.nextToken,
+    };
+}
+export async function retrieveAndGenerate(params) {
+    const client = getBedrockClient();
+    const input = {
+        input: { text: params.query },
+        retrieveAndGenerateConfiguration: {
+            type: "KNOWLEDGE_BASE",
+            knowledgeBaseConfiguration: {
+                knowledgeBaseId: params.knowledgeBaseId,
+                modelArn: params.modelArn,
+                retrievalConfiguration: {
+                    vectorSearchConfiguration: {
+                        numberOfResults: params.numberOfResults ?? 5,
+                        ...(params.searchType ? { overrideSearchType: params.searchType } : {}),
+                    },
+                },
+                ...(params.systemPrompt
+                    ? {
+                        generationConfiguration: {
+                            promptTemplate: {
+                                textPromptTemplate: params.systemPrompt,
+                            },
+                        },
+                    }
+                    : {}),
+            },
+        },
+        ...(params.sessionId ? { sessionId: params.sessionId } : {}),
+    };
+    const command = new RetrieveAndGenerateCommand(input);
+    const response = await client.send(command);
+    const citations = (response.citations ?? []).map((c) => ({
+        generatedText: c.generatedResponsePart?.textResponsePart?.text,
+        passages: (c.retrievedReferences ?? []).map((ref) => ({
+            content: ref.content?.text ?? "",
+            score: undefined,
+            location: ref.location,
+            metadata: ref.metadata,
+        })),
+    }));
+    return {
+        output: response.output?.text ?? "",
+        sessionId: response.sessionId,
+        citations,
+    };
+}
+// ── Error helper ─────────────────────────────────────────────────────────────
+export function formatAwsError(error) {
+    if (error instanceof Error) {
+        const name = error.name;
+        const msg = error.message;
+        if (name === "ResourceNotFoundException") {
+            return `Error: Knowledge Base not found. Check that BEDROCK_KNOWLEDGE_BASE_ID is correct and the KB exists in region ${AWS_REGION}.`;
+        }
+        if (name === "AccessDeniedException") {
+            return `Error: Access denied. Ensure the AWS profile '${AWS_PROFILE}' has the 'bedrock:Retrieve' and 'bedrock:RetrieveAndGenerate' IAM permissions.`;
+        }
+        if (name === "ValidationException") {
+            return `Error: Invalid request — ${msg}. Check the parameters and try again.`;
+        }
+        if (name === "ThrottlingException") {
+            return "Error: AWS request throttled. Wait a moment and retry.";
+        }
+        if (name === "CredentialsProviderError") {
+            return `Error: Could not load AWS credentials for profile '${AWS_PROFILE}'. Run 'aws sso login --profile ${AWS_PROFILE}' or check your ~/.aws/credentials file.`;
+        }
+        return `Error [${name}]: ${msg}`;
+    }
+    return `Error: ${String(error)}`;
+}
+//# sourceMappingURL=bedrock.js.map

package/dist/services/bedrock.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bedrock.js","sourceRoot":"","sources":["../../src/services/bedrock.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,yBAAyB,EACzB,eAAe,EACf,0BAA0B,GAK3B,MAAM,uCAAuC,CAAC;AAC/C,OAAO,EAAE,OAAO,EAAE,MAAM,+BAA+B,CAAC;AAExD,gFAAgF;AAEhF,MAAM,UAAU,GAAG,OAAO,CAAC,GAAG,CAAC,UAAU,IAAI,WAAW,CAAC;AACzD,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC,WAAW,IAAI,SAAS,CAAC;AAEzD,gFAAgF;AAEhF,IAAI,OAAO,GAAqC,IAAI,CAAC;AAErD,MAAM,UAAU,gBAAgB;IAC9B,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,GAAG,IAAI,yBAAyB,CAAC;YACtC,MAAM,EAAE,UAAU;YAClB,WAAW,EAAE,OAAO,CAAC,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC;SAC/C,CAAC,CAAC;IACL,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAsCD,MAAM,CAAC,KAAK,UAAU,QAAQ,CAAC,MAAsB;IACnD,MAAM,MAAM,GAAG,gBAAgB,EAAE,CAAC;IAElC,MAAM,KAAK,GAAyB;QAClC,eAAe,EAAE,MAAM,CAAC,eAAe;QACvC,cAAc,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,KAAK,EAAE;QACtC,sBAAsB,EAAE;YACtB,yBAAyB,EAAE;gBACzB,eAAe,EAAE,MAAM,CAAC,eAAe,IAAI,CAAC;gBAC5C,GAAG,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,kBAAkB,EAAE,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACvE,GAAG,CAAC,MAAM,CAAC,cAAc;oBACvB,CAAC,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,cAA4C,EAAE;oBACjE,CAAC,CAAC,EAAE,CAAC;aACR;SACF;QACD,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAC7D,CAAC;IAEF,MAAM,OAAO,GAAG,IAAI,eAAe,CAAC,KAAK,CAAC,CAAC;IAC3C,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAE5C,MAAM,QAAQ,GAAuB,CAAC,QAAQ,CAAC,gBAAgB,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACjF,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,IAAI,IAAI,EAAE;QAC9B,KAAK,EAAE,CAAC,CAAC,KAAK;QACd,QAAQ,EAAE,CAAC,CAAC,QAAQ;QACpB,QAAQ,EAAE,CAAC,CAAC,QAA+C;KAC5D,CAAC,CAAC,CAAC;IAEJ,OAAO;QACL,QAAQ;QACR,SAAS,EAAE,QAAQ,CAAC,SAAS;KAC9B,CAAC;AACJ,CAAC;AAcD,MAAM,CAAC,KAAK,UAAU,mBAAmB,CACvC,MAAiC;IAEjC,MAAM,MAAM,GAAG,gBAAgB,EAAE,CAAC;IAElC,MAAM,KAAK,GAAoC;QAC7C,KAAK,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,KAAK,EAAE;QAC7B,gCAAgC,EAAE;YAChC,IAAI,EAAE,gBAAgB;YACtB,0BAA0B,EAAE;gBAC1B,eAAe,EAAE,MAAM,CAAC,eAAe;gBACvC,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,sBAAsB,EAAE;oBACtB,yBAAyB,EAAE;wBACzB,eAAe,EAAE,MAAM,CAAC,eAAe,IAAI,CAAC;wBAC5C,GAAG,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,kBAAkB,EAAE,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;qBACxE;iBACF;gBACD,GAAG,CAAC,MAAM,CAAC,YAAY;oBACrB,CAAC,CAAC;wBACE,uBAAuB,EAAE;4BACvB,cAAc,EAAE;gCACd,kBAAkB,EAAE,MAAM,CAAC,YAAY;6BACxC;yBACF;qBACF;oBACH,CAAC,CAAC,EAAE,CAAC;aACR;SACF;QACD,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAC7D,CAAC;IAEF,MAAM,OAAO,GAAG,IAAI,0BAA0B,CAAC,KAAK,CAAC,CAAC;IACtD,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAE5C,MAAM,SAAS,GAAe,CAAC,QAAQ,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACnE,aAAa,EAAE,CAAC,CAAC,qBAAqB,EAAE,gBAAgB,EAAE,IAAI;QAC9D,QAAQ,EAAE,CAAC,CAAC,CAAC,mBAAmB,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YACpD,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,IAAI,IAAI,EAAE;YAChC,KAAK,EAAE,SAAS;YAChB,QAAQ,EAAE,GAAG,CAAC,QAAQ;YACtB,QAAQ,EAAE,GAAG,CAAC,QAA+C;SAC9D,CAAC,CAAC;KACJ,CAAC,CAAC,CAAC;IAEJ,OAAO;QACL,MAAM,EAAE,QAAQ,CAAC,MAAM,EAAE,IAAI,IAAI,EAAE;QACnC,SAAS,EAAE,QAAQ,CAAC,SAAS;QAC7B,SAAS;KACV,CAAC;AACJ,CAAC;AAED,gFAAgF;AAEhF,MAAM,UAAU,cAAc,CAAC,KAAc;IAC3C,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;QAC3B,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;QACxB,MAAM,GAAG,GAAG,KAAK,CAAC,OAAO,CAAC;QAE1B,IAAI,IAAI,KAAK,2BAA2B,EAAE,CAAC;YACzC,OAAO,gHAAgH,UAAU,GAAG,CAAC;QACvI,CAAC;QACD,IAAI,IAAI,KAAK,uBAAuB,EAAE,CAAC;YACrC,OAAO,iDAAiD,WAAW,iFAAiF,CAAC;QACvJ,CAAC;QACD,IAAI,IAAI,KAAK,qBAAqB,EAAE,CAAC;YACnC,OAAO,4BAA4B,GAAG,uCAAuC,CAAC;QAChF,CAAC;QACD,IAAI,IAAI,KAAK,qBAAqB,EAAE,CAAC;YACnC,OAAO,wDAAwD,CAAC;QAClE,CAAC;QACD,IAAI,IAAI,KAAK,0BAA0B,EAAE,CAAC;YACxC,OAAO,sDAAsD,WAAW,mCAAmC,WAAW,0CAA0C,CAAC;QACnK,CAAC;QACD,OAAO,UAAU,IAAI,MAAM,GAAG,EAAE,CAAC;IACnC,CAAC;IACD,OAAO,UAAU,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;AACnC,CAAC"}

package/package.json

@@ -0,0 +1,49 @@
+{
+  "name": "@miguel-carrera/aws-bedrock-mcp-server",
+  "version": "1.0.0",
+  "description": "MCP server for AWS Bedrock Knowledge Base — semantic search and RAG via bedrock_retrieve and bedrock_retrieve_and_generate",
+  "type": "module",
+  "main": "dist/index.js",
+  "bin": {
+    "aws-bedrock-mcp-server": "./dist/index.js"
+  },
+  "files": [
+    "dist",
+    "README.md"
+  ],
+  "scripts": {
+    "start": "node dist/index.js",
+    "dev": "tsx watch src/index.ts",
+    "build": "tsc",
+    "clean": "rm -rf dist",
+    "prepublishOnly": "npm run build"
+  },
+  "engines": {
+    "node": ">=18"
+  },
+  "keywords": [
+    "mcp",
+    "aws",
+    "bedrock",
+    "knowledge-base",
+    "rag",
+    "claude"
+  ],
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/miguelcarrera/aws-bedrock-mcp-server.git"
+  },
+  "homepage": "https://github.com/miguelcarrera/aws-bedrock-mcp-server#readme",
+  "dependencies": {
+    "@aws-sdk/client-bedrock-agent-runtime": "^3.758.0",
+    "@aws-sdk/credential-providers": "^3.758.0",
+    "@modelcontextprotocol/sdk": "^1.6.1",
+    "zod": "^3.23.8"
+  },
+  "devDependencies": {
+    "@types/node": "^22.10.0",
+    "tsx": "^4.19.2",
+    "typescript": "^5.7.2"
+  }
+}