改进rerank效果

This commit is contained in:
2024-08-19 10:03:46 +08:00
parent 806b694b37
commit 22c51218b3
3 changed files with 51 additions and 23 deletions
+11 -10
View File
@@ -3,6 +3,16 @@
SQL_DATABASE_URL=mysql+pymysql://zjinfo1:Dy2Bcr53Hm5xRkba@110.42.234.166:3306/zjinfo1
#SQL_DATABASE_URL=mysql+pymysql://zjinfo2:GSKcziSdBixDXwcd@110.42.234.166:3306/zjinfo2
#--------------------------
# 是否启用检索重排功能
ENABLE_RERANK=true
# The number of similar embeddings to return when retrieving documents.
TOP_K=5
# Rerank model
RERANK_MODEL=bge-reranker-v2-m3
RERANK_BASE_URL=http://10.1.16.39:9995
RERANK_TOP_N=5
RERANK_THRESHOLD=0.3
#---------- Xinference ----------------
# The provider for the AI models to use.
MODEL_PROVIDER=xinference
@@ -19,9 +29,7 @@ EMBEDDING_MODEL=bge-m3
EMBEDDING_BASE_URL=http://10.1.16.39:9995
# Dimension of the embedding model to use.
EMBEDDING_DIM=1024
# Rerank model
RERANK_MODEL=bge-reranker-v2-m3
RERANK_BASE_URL=http://10.1.16.39:9995
##---------- OpenAI ----------------
## The provider for the AI models to use.
#MODEL_PROVIDER=openai
@@ -46,17 +54,10 @@ RERANK_BASE_URL=http://10.1.16.39:9995
## Name of the embedding model to use.
#EMBEDDING_MODEL=text-embedding-v2
#--------------------------
# 是否启用检索重排功能
ENABLE_RERANK=true
# The questions to help users get started (multi-line).
CONVERSATION_STARTERS=本工程指什么?\n总算表有哪些费用?\n项目划分哪些内容构成?\n其他费用表有哪些内容?
# The number of similar embeddings to return when retrieving documents.
TOP_K=5
# The time in milliseconds to wait for the stream to return a response.
STREAM_TIMEOUT=60000