Class for interacting with Amazon Kendra, an intelligent search service provided by AWS. Extends the BaseRetriever class.

Hierarchy

Constructors

Properties

indexId: string
kendraClient: KendraClient
topK: number
attributeFilter?: AttributeFilter
callbacks?: Callbacks
metadata?: Record<string, unknown>
tags?: string[]
verbose?: boolean
lc_runnable: boolean = true

Methods

  • Cleans the result text by replacing sequences of whitespace with a single space and removing ellipses.

    Parameters

    • resText: string

      The result text to clean.

    Returns string

    The cleaned result text.

  • Combines title and excerpt into a single string.

    Parameters

    • Optional title: string

      The title of the document.

    • Optional excerpt: string

      An excerpt from the document.

    Returns string

    A single string combining the title and excerpt.

  • Converts a QueryResultItem object into a Document object.

    Parameters

    • item: QueryResultItem

      The QueryResultItem object to convert.

    Returns Document<{
        document_attributes: {
            [key: string]: unknown;
        };
        excerpt: string;
        source: undefined | string;
        title: string;
    }>

    A Document object.

  • Converts a RetrieveResultItem object into a Document object.

    Parameters

    • item: RetrieveResultItem

      The RetrieveResultItem object to convert.

    Returns Document<{
        document_attributes: {
            [key: string]: unknown;
        };
        excerpt: string;
        source: undefined | string;
        title: string;
    }>

    A Document object.

  • Extracts the attribute value from a DocumentAttributeValue object.

    Parameters

    • value: DocumentAttributeValue

      The DocumentAttributeValue object to extract the value from.

    Returns string | number | string[] | Date

    The extracted attribute value.

  • Extracts the attribute key-value pairs from an array of DocumentAttribute objects.

    Parameters

    • Optional documentAttributes: DocumentAttribute[]

      The array of DocumentAttribute objects to extract the key-value pairs from.

    Returns {
        [key: string]: unknown;
    }

    An object containing the extracted attribute key-value pairs.

    • [key: string]: unknown
  • Extracts the top-k documents from a QueryCommandOutput object.

    Parameters

    • response: QueryCommandOutput

      The QueryCommandOutput object to extract the documents from.

    • pageSize: number

      The number of documents to extract.

    Returns Document<{
        document_attributes: {
            [key: string]: unknown;
        };
        excerpt: string;
        source: undefined | string;
        title: string;
    }>[]

    An array of Document objects.

  • Extracts the excerpt text from a QueryResultItem object.

    Parameters

    • item: QueryResultItem

      The QueryResultItem object to extract the excerpt text from.

    Returns string

    The extracted excerpt text.

  • Main method used to retrieve relevant documents. It takes a query string and an optional configuration object, and returns a promise that resolves to an array of Document objects. This method handles the retrieval process, including starting and ending callbacks, and error handling.

    Parameters

    • query: string

      The query string to retrieve relevant documents for.

    • Optional config: Callbacks | BaseCallbackConfig

      Optional configuration object for the retrieval process.

    Returns Promise<Document<Record<string, any>>[]>

    A promise that resolves to an array of Document objects.

  • Extracts the top-k documents from a RetrieveCommandOutput object.

    Parameters

    • response: RetrieveCommandOutput

      The RetrieveCommandOutput object to extract the documents from.

    • pageSize: number

      The number of documents to extract.

    Returns Document<Record<string, any>>[]

    An array of Document objects.

  • Create a new runnable sequence that runs each individual runnable in series, piping the output of one runnable into another runnable or runnable-like.

    Type Parameters

    • NewRunOutput

    Parameters

    • coerceable: RunnableLike<Document<Record<string, any>>[], NewRunOutput>

      A runnable, function, or object whose values are functions or runnables.

    Returns RunnableSequence<string, Exclude<NewRunOutput, Error>>

    A new runnable sequence.

  • Sends a retrieve or query request to Kendra and returns the top-k documents.

    Parameters

    • query: string

      The query to send to Kendra.

    • topK: number

      The number of top documents to return.

    • Optional attributeFilter: AttributeFilter

      Optional filter to apply when retrieving documents.

    Returns Promise<Document<Record<string, any>>[] | Document<{
        document_attributes: {
            [key: string]: unknown;
        };
        excerpt: string;
        source: undefined | string;
        title: string;
    }>[]>

    A Promise that resolves to an array of Document objects.

  • Stream all output from a runnable, as reported to the callback system. This includes all inner runs of LLMs, Retrievers, Tools, etc. Output is streamed as Log objects, which include a list of jsonpatch ops that describe how the state of the run has changed in each step, and the final state of the run. The jsonpatch ops can be applied in order to construct state.

    Parameters

    • input: string
    • Optional options: Partial<BaseCallbackConfig>
    • Optional streamOptions: Omit<LogStreamCallbackHandlerInput, "autoClose">

    Returns AsyncGenerator<RunLogPatch, any, unknown>

  • Default implementation of transform, which buffers input and then calls stream. Subclasses should override this method if they can start producing output while input is still being generated.

    Parameters

    Returns AsyncGenerator<Document<Record<string, any>>[], any, unknown>

  • Helper method to transform an Iterator of Input values into an Iterator of Output values, with callbacks. Use this to implement stream() or transform() in Runnable subclasses.

    Type Parameters

    • I extends string

    • O extends Document<Record<string, any>>[]

    Parameters

    • inputGenerator: AsyncGenerator<I, any, unknown>
    • transformer: ((generator, runManager?, options?) => AsyncGenerator<O, any, unknown>)
        • (generator, runManager?, options?): AsyncGenerator<O, any, unknown>
        • Parameters

          Returns AsyncGenerator<O, any, unknown>

    • Optional options: BaseCallbackConfig & {
          runType?: string;
      }

    Returns AsyncGenerator<O, any, unknown>

Generated using TypeDoc