UNPKG

@langchain/community

Version:
1 lines • 3.27 kB
{"version":3,"file":"ibm.d.cts","names":["DocumentInterface","BaseDocumentCompressor","WatsonXAI","TextRerankParams","WatsonxAuth","WatsonxRerankBasicOptions","WatsonxInputRerank","Omit","WatsonxRerank","Partial","Promise","Record","Array"],"sources":["../../src/document_compressors/ibm.d.ts"],"sourcesContent":["import { DocumentInterface } from \"@langchain/core/documents\";\nimport { BaseDocumentCompressor } from \"@langchain/core/retrievers/document_compressors\";\nimport { WatsonXAI } from \"@ibm-cloud/watsonx-ai\";\nimport { TextRerankParams } from \"@ibm-cloud/watsonx-ai/dist/watsonx-ai-ml/vml_v1.js\";\nimport { WatsonxAuth, WatsonxRerankBasicOptions } from \"../types/ibm.js\";\nexport interface WatsonxInputRerank extends Omit<TextRerankParams, \"modelId\" | \"inputs\" | \"query\">, WatsonxRerankBasicOptions {\n model: string;\n truncateInputTokens?: number;\n returnOptions?: {\n topN?: number;\n inputs?: boolean;\n };\n}\nexport declare class WatsonxRerank extends BaseDocumentCompressor implements WatsonxInputRerank {\n maxRetries: number;\n version: string;\n truncateInputTokens?: number | undefined;\n returnOptions?: {\n topN?: number;\n inputs?: boolean;\n query?: boolean;\n } | undefined;\n model: string;\n spaceId?: string | undefined;\n projectId?: string | undefined;\n maxConcurrency?: number | undefined;\n serviceUrl: string;\n service: WatsonXAI;\n constructor(fields: WatsonxInputRerank & WatsonxAuth);\n scopeId(): {\n projectId: string;\n modelId: string;\n spaceId?: undefined;\n } | {\n spaceId: string | undefined;\n modelId: string;\n projectId?: undefined;\n };\n invocationParams(options?: Partial<WatsonxInputRerank>): {\n truncate_input_tokens: number | undefined;\n return_options: {\n top_n: number | undefined;\n inputs: boolean | undefined;\n };\n };\n compressDocuments(documents: DocumentInterface[], query: string): Promise<DocumentInterface[]>;\n rerank(documents: Array<DocumentInterface | string | Record<\"pageContent\", string>>, query: string, options?: Partial<WatsonxInputRerank>): Promise<Array<{\n index: number;\n relevanceScore: number;\n input?: string;\n }>>;\n}\n//# sourceMappingURL=ibm.d.ts.map"],"mappings":";;;;;;;UAKiBM,kBAAAA,SAA2BC,KAAKJ,mDAAmDE;;EAAnFC,mBAAAA,CAAAA,EAAkB,MAAA;EAAcH,aAAAA,CAAAA,EAAAA;IAALI,IAAAA,CAAAA,EAAAA,MAAAA;IAAwDF,MAAAA,CAAAA,EAAAA,OAAAA;EAAyB,CAAA;AAQ7H;AAcaH,cAdQM,aAAAA,SAAsBP,sBAAAA,YAAkCK,kBAchEJ,CAAAA;EACWI,UAAAA,EAAAA,MAAAA;EAAqBF,OAAAA,EAAAA,MAAAA;EAUNE,mBAAAA,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;EAARG,aAAAA,CAAAA,EAAAA;IAOET,IAAAA,CAAAA,EAAAA,MAAAA;IAA6CA,MAAAA,CAAAA,EAAAA,OAAAA;IAARU,KAAAA,CAAAA,EAAAA,OAAAA;EAC1CV,CAAAA,GAAAA,SAAAA;EAA6BW,KAAAA,EAAAA,MAAAA;EAAnCC,OAAAA,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;EAAoGN,SAAAA,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;EAARG,cAAAA,CAAAA,EAAAA,MAAAA,GAAAA,SAAAA;EAAsCG,UAAAA,EAAAA,MAAAA;EAARF,OAAAA,EAnBnIR,SAmBmIQ;EAjCrGT,WAAAA,CAAAA,MAAAA,EAenBK,kBAfmBL,GAeEG,WAfFH;EAAkCK,OAAAA,CAAAA,CAAAA,EAAAA;IAAkB,SAAA,EAAA,MAAA;;;;;;;;6BAyBhEG,QAAQH;;;;;;;+BAONN,qCAAqCU,QAAQV;oBACxDY,MAAMZ,6BAA6BW,yDAAyDF,QAAQH,sBAAsBI,QAAQE"}