Skip to content

Commit 59e701f

Browse files
committed
fix: excluding unnecessary metadata for reranker
1 parent 7b597cf commit 59e701f

File tree

2 files changed

+9
-7
lines changed

2 files changed

+9
-7
lines changed

src/core/services/llamaindex/retrieving.ts

+7-1
Original file line numberDiff line numberDiff line change
@@ -221,7 +221,13 @@ function transform (result: RetrievedChunk): TextNode {
221221
...result.metadata,
222222
...result.document_metadata
223223
},
224-
excludedLlmMetadataKeys: ['namespace_id'],
224+
excludedLlmMetadataKeys: [
225+
// Notice: Exclude several fields generated by the LLM to avoid passing too much text during rerank,
226+
// which may lead to exceeding the model's token limit.
227+
'sectionSummary',
228+
'questionsThisExcerptCanAnswer',
229+
'excerptKeywords'
230+
],
225231
relationships: {
226232
[NodeRelationship.NEXT]: result.relationships[NodeRelationship.NEXT] ? transformRef(result.relationships[NodeRelationship.NEXT]) : undefined,
227233
[NodeRelationship.PREVIOUS]: result.relationships[NodeRelationship.PREVIOUS] ? transformRef(result.relationships[NodeRelationship.PREVIOUS]) : undefined,

src/lib/llamaindex/postprocessors/postfilters/MetadataPostFilter.ts

+2-6
Original file line numberDiff line numberDiff line change
@@ -75,12 +75,8 @@ export class MetadataPostFilter implements BaseNodePostprocessor {
7575
console.log('Nodes after filter:', filteredNodes.length, 'nodes');
7676

7777
if (filteredNodes.length === 0) {
78-
const requiredFilters = filters.filter(f => !f.optional);
79-
console.info('No nodes left after filtering, fallback to filtering with required filters.', {
80-
requiredFilters
81-
});
82-
filteredNodes = await this.filterNodes(nodes, requiredFilters);
83-
console.log('Nodes after filtering with required filters:', filteredNodes.length, 'nodes');
78+
console.warn('No nodes left after filtering, fallback to using all nodes.');
79+
return nodes;
8480
}
8581

8682
return filteredNodes;

0 commit comments

Comments
 (0)