summaryrefslogtreecommitdiffstats
path: root/config.example.yaml
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-06-21 06:00:26 +0800
committerGitHub <noreply@github.com>2024-06-21 06:00:26 +0800
commitabc588daac6053ec2edbdcde3f5a2dc5eb7d50b8 (patch)
tree9f1cd8a40bd959420dfdf4ac3ae8b52f823aa1e9 /config.example.yaml
parent2eab71a641827e503b14952373aec82661192ba2 (diff)
downloadaichat-abc588daac6053ec2edbdcde3f5a2dc5eb7d50b8.tar.gz
feat: support rerank (#620)
Diffstat (limited to 'config.example.yaml')
-rw-r--r--config.example.yaml15
1 files changed, 11 insertions, 4 deletions
diff --git a/config.example.yaml b/config.example.yaml
index ead96f4..15877b3 100644
--- a/config.example.yaml
+++ b/config.example.yaml
@@ -35,16 +35,20 @@ bots:
# Specifies the embedding model to use
rag_embedding_model: null
+# Specifies the rerank model to use
+rag_rerank_model: null
# Specifies the chunk size
rag_chunk_size: null
# Specifies the chunk overlap
rag_chunk_overlap: null
# Specifies the number of documents to retrieve
rag_top_k: 4
-# Specifies the minimum relevance score for vector search
-rag_min_score_vector: 0
-# Specifies the minimum relevance score for full-text search
-rag_min_score_text: 0
+# Specifies the minimum relevance score for vector searching
+rag_min_score_vector_search: 0
+# Specifies the minimum relevance score for full-text searching
+rag_min_score_fulltext_search: 0
+# Specifies the minimum relevance score for reranking
+rag_min_score_rerank: 0
# Defines the query structure using variables like __CONTEXT__ and __INPUT__ to tailor searches to specific needs
rag_template: |
@@ -88,6 +92,9 @@ clients:
# max_input_tokens: 2048
# default_chunk_size: 2000
# max_concurrent_chunks: 100
+ # - name: xxxx
+ # mode: rerank # Rerank model
+ # max_input_tokens: 2048
# patches:
# <regex>: # The regex to match model names, e.g. '.*' 'gpt-4o' 'gpt-4o|gpt-4-.*'
# chat_completions_body: # The JSON to be merged with the chat completions request body.