FewShotRAG ignores configured retrieval threshold

[abhishekreddyallu]

FewShotRAG does not forward threshold to retriever postprocessing

Location: ontoaligner/aligner/fewshot/fewshot.py:80

Current code:

ir_output_cleaned = process.retriever_postprocessor(predicts=ir_output)

This ignores the configured retrieval threshold from:

self.kwargs["retriever_config"]["threshold"]

The base RAG implementation forwards the threshold, but the FewShotRAG override does not.

Root cause:

FewShotRAG.generate() calls retriever_postprocessor() with only predicts, so the postprocessor uses its default threshold instead of the user-configured threshold.

Fix:

        ir_output = self.ir_generate(input_data=input_data)
        if 'threshold' in self.kwargs['retriever_config']:
            threshold = self.kwargs['retriever_config']['threshold']
        else:
            threshold = 0.0
        ir_output_cleaned = process.retriever_postprocessor(predicts=ir_output, threshold=threshold)