Fixed errors and added more support

This commit is contained in:
jvinolus
2025-02-04 13:04:36 -08:00
parent 47b8412695
commit 7b8e5d4e7c
3 changed files with 20 additions and 16 deletions

View File

@@ -1330,16 +1330,16 @@ RAG_EMBEDDING_BATCH_SIZE = PersistentConfig(
),
)
RAG_EMBEDDING_PASSAGE_PREFIX = PersistentConfig(
"RAG_EMBEDDING_PASSAGE_PREFIX",
"rag.embedding_passage_prefix",
os.environ.get("RAG_EMBEDDING_PASSAGE_PREFIX", False),
RAG_EMBEDDING_QUERY_PREFIX = (
os.environ.get("RAG_EMBEDDING_QUERY_PREFIX", None)
)
RAG_EMBEDDING_QUERY_PREFIX = PersistentConfig(
"RAG_EMBEDDING_QUERY_PREFIX",
"rag.embedding_query_prefix",
os.environ.get("RAG_EMBEDDING_QUERY_PREFIX", False),
RAG_EMBEDDING_PASSAGE_PREFIX = (
os.environ.get("RAG_EMBEDDING_PASSAGE_PREFIX", None)
)
RAG_EMBEDDING_PREFIX_FIELD_NAME = (
os.environ.get("RAG_EMBEDDING_PREFIX_FIELD_NAME", "input_type")
)
RAG_RERANKING_MODEL = PersistentConfig(