Class AmazonKnowledgeBaseRetriever

Class for interacting with Amazon Bedrock Knowledge Bases, a RAG workflow oriented service provided by AWS. Extends the BaseRetriever class.

Example

const retriever = new AmazonKnowledgeBaseRetriever({
  topK: 10,
  knowledgeBaseId: "YOUR_KNOWLEDGE_BASE_ID",
  region: "us-east-2",
  clientOptions: {
    credentials: {
      accessKeyId: "YOUR_ACCESS_KEY_ID",
      secretAccessKey: "YOUR_SECRET_ACCESS_KEY",
    },
  },
});

const docs = await retriever.getRelevantDocuments("How are clouds formed?");

Hierarchy (view full)

  • Toolkit
    • AmazonKnowledgeBaseRetriever

Constructors

constructor

Properties

bedrockAgentRuntimeClient knowledgeBaseId topK

Methods

cleanResult queryKnowledgeBase

Constructors

constructor

Properties

bedrockAgentRuntimeClient

bedrockAgentRuntimeClient: BedrockAgentRuntimeClient

knowledgeBaseId

knowledgeBaseId: string

topK

topK: number

Methods

cleanResult

queryKnowledgeBase

  • queryKnowledgeBase(query, topK, filter?): Promise<Document[]>

  • Parameters

    • query: string
    • topK: number
    • Optional filter: AndAllMember | EqualsMember | GreaterThanMember | GreaterThanOrEqualsMember | InMember | LessThanMember | LessThanOrEqualsMember | ListContainsMember | NotEqualsMember | NotInMember | OrAllMember | StartsWithMember | StringContainsMember | $UnknownMember

    Returns Promise<Document[]>

Settings

Member Visibility

Theme

On This Page

Generated using TypeDoc