farcasclaudiu/Flowise
1
0
Fork 0
mirror of https://github.com/farcasclaudiu/Flowise.git synced 2026-06-29 11:01:18 +03:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1656 from FlowiseAI/bugfix/Rerank-Top-K

Browse Source 
Bugfix/Cohere top K
This commit is contained in:
Henry Heng
2024-02-01 16:47:17 +00:00
committed by GitHub
parent 7e475f1288 5c6b5b233c
commit 48677c33cc
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/components/nodes/retrievers/CohereRerankRetriever/CohereRerank.ts
+1 -1
View File
@@ -47,7 +47,7 @@ export class CohereRerank extends BaseDocumentCompressor {
doc.metadata.relevance_score = result.relevance_score doc.metadata.relevance_score = result.relevance_score
finalResults.push(doc) finalResults.push(doc)
}) })
return finalResults return finalResults.splice(0, this.k)
} catch (error) { } catch (error) {
return documents return documents
} }