Browse Source

roll back rerank topn setting (#11297)

tags/0.13.0
Jyong 11 months ago
parent
commit
c58d2fce89
No account linked to committer's email address

+ 0
- 1
api/.env.example View File



CREATE_TIDB_SERVICE_JOB_ENABLED=false CREATE_TIDB_SERVICE_JOB_ENABLED=false


RETRIEVAL_TOP_N=0

+ 0
- 2
api/configs/feature/__init__.py View File

default=30, default=30,
) )


RETRIEVAL_TOP_N: int = Field(description="number of retrieval top_n", default=0)



class WorkspaceConfig(BaseSettings): class WorkspaceConfig(BaseSettings):
""" """

+ 3
- 4
api/core/rag/datasource/retrieval_service.py View File



from flask import Flask, current_app from flask import Flask, current_app


from configs import DifyConfig
from core.rag.data_post_processor.data_post_processor import DataPostProcessor from core.rag.data_post_processor.data_post_processor import DataPostProcessor
from core.rag.datasource.keyword.keyword_factory import Keyword from core.rag.datasource.keyword.keyword_factory import Keyword
from core.rag.datasource.vdb.vector_factory import Vector from core.rag.datasource.vdb.vector_factory import Vector
query=query, query=query,
documents=all_documents, documents=all_documents,
score_threshold=score_threshold, score_threshold=score_threshold,
top_n=DifyConfig.RETRIEVAL_TOP_N or top_k,
top_n=top_k,
) )


return all_documents return all_documents
query=query, query=query,
documents=documents, documents=documents,
score_threshold=score_threshold, score_threshold=score_threshold,
top_n=DifyConfig.RETRIEVAL_TOP_N or len(documents),
top_n=len(documents),
) )
) )
else: else:
query=query, query=query,
documents=documents, documents=documents,
score_threshold=score_threshold, score_threshold=score_threshold,
top_n=DifyConfig.RETRIEVAL_TOP_N or len(documents),
top_n=len(documents),
) )
) )
else: else:

Loading…
Cancel
Save