Skip to content

Commit 991c6bf

Browse files
committed
update config
1 parent f622114 commit 991c6bf

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/memos/api/config.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -100,8 +100,10 @@ def get_reranker_config() -> dict[str, Any]:
100100
"backend": "http_bge",
101101
"config": {
102102
"url": os.getenv("MOS_RERANKER_URL"),
103-
"model": "bge-reranker-v2-m3",
103+
"model": os.getenv("MOS_RERANKER_MODEL", "bge-reranker-v2-m3"),
104104
"timeout": 10,
105+
"headers_extra": os.getenv("MOS_RERANKER_HEADERS_EXTRA"),
106+
"rerank_source": os.getenv("MOS_RERANK_SOURCE"),
105107
},
106108
}
107109
else:

0 commit comments

Comments
 (0)