Skip to content

Commit 4f02be4

Browse files
authored
Feat: change deafult memos cube (#765)
feat: update config
1 parent 4c6a114 commit 4f02be4

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/memos/api/config.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -378,7 +378,7 @@ def get_reranker_config() -> dict[str, Any]:
378378
return {
379379
"backend": embedder_backend,
380380
"config": {
381-
"url": os.getenv("MOS_RERANKER_URL"),
381+
"url": os.getenv("MOS_RERANKER_URL", "localhost:8000/v1/rerank"),
382382
"model": os.getenv("MOS_RERANKER_MODEL", "bge-reranker-v2-m3"),
383383
"timeout": 10,
384384
"headers_extra": json.loads(os.getenv("MOS_RERANKER_HEADERS_EXTRA", "{}")),
@@ -404,7 +404,7 @@ def get_feedback_reranker_config() -> dict[str, Any]:
404404
return {
405405
"backend": embedder_backend,
406406
"config": {
407-
"url": os.getenv("MOS_RERANKER_URL"),
407+
"url": os.getenv("MOS_RERANKER_URL", "localhost:8000/v1/rerank"),
408408
"model": os.getenv("MOS_FEEDBACK_RERANKER_MODEL", "bge-reranker-v2-m3"),
409409
"timeout": 10,
410410
"headers_extra": json.loads(os.getenv("MOS_RERANKER_HEADERS_EXTRA", "{}")),
@@ -671,7 +671,7 @@ def is_scheduler_enabled() -> bool:
671671
@staticmethod
672672
def is_default_cube_config_enabled() -> bool:
673673
"""Check if default cube config is enabled via environment variable."""
674-
return os.getenv("MOS_ENABLE_DEFAULT_CUBE_CONFIG", "false").lower() == "true"
674+
return os.getenv("MOS_ENABLE_DEFAULT_CUBE_CONFIG", "true").lower() == "true"
675675

676676
@staticmethod
677677
def is_dingding_bot_enabled() -> bool:

0 commit comments

Comments
 (0)