Skip to content
Snippets Groups Projects
Commit 68fc6e8b authored by Marcus Schiesser's avatar Marcus Schiesser
Browse files

fix: don't need similarityTopK parameter for LlamaCloud

parent ea0331ef
No related branches found
No related tags found
No related merge requests found
......@@ -13,7 +13,7 @@ export class LlamaCloudIndex {
this.params = params;
}
asRetriever(params: RetrieveParams): BaseRetriever {
asRetriever(params: RetrieveParams = {}): BaseRetriever {
return new LlamaCloudRetriever({ ...this.params, ...params });
}
......
......@@ -38,8 +38,9 @@ export class LlamaCloudRetriever implements BaseRetriever {
constructor(params: CloudConstructorParams & RetrieveParams) {
this.clientParams = { apiKey: params.apiKey, baseUrl: params.baseUrl };
params.denseSimilarityTopK =
params.similarityTopK ?? params.denseSimilarityTopK;
if (params.similarityTopK) {
params.denseSimilarityTopK = params.similarityTopK;
}
this.retrieveParams = params;
this.pipelineName = params.name;
if (params.projectName) {
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment