Hierarchy

  • BaseDocumentCompressor
    • WatsonxRerank

Implements

Constructors

Properties

maxConcurrency?: number
maxRetries: number = 0
model: string
projectId?: string
returnOptions?: {
    inputs?: boolean;
    query?: boolean;
    topN?: number;
}
service: WatsonxAiMlVml_v1
serviceUrl: string
spaceId?: string
truncateInputTokens?: number
version: string = "2024-05-31"

Methods

  • Abstract method that must be implemented by any class that extends BaseDocumentCompressor. This method takes an array of Document objects and a query string as parameters and returns a Promise that resolves with an array of compressed Document objects.

    Parameters

    • documents: DocumentInterface<Record<string, any>>[]

      An array of Document objects to be compressed.

    • query: string

      A query string.

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

    A Promise that resolves with an array of compressed Document objects.

  • Parameters

    Returns {
        return_options: {
            inputs: undefined | boolean;
            top_n: undefined | number;
        };
        truncate_input_tokens: undefined | number;
    }

    • return_options: {
          inputs: undefined | boolean;
          top_n: undefined | number;
      }
      • inputs: undefined | boolean
      • top_n: undefined | number
    • truncate_input_tokens: undefined | number
  • Parameters

    • documents: (string | DocumentInterface<Record<string, any>> | Record<"pageContent", string>)[]
    • query: string
    • Optionaloptions: Partial<WatsonxInputRerank>

    Returns Promise<{
        index: number;
        input?: string;
        relevanceScore: number;
    }[]>

  • Returns {
        modelId: string;
        projectId: string;
        spaceId?: undefined;
    } | {
        modelId: string;
        projectId?: undefined;
        spaceId: undefined | string;
    }

  • Parameters

    • x: any

    Returns x is BaseDocumentCompressor