Skip to content

Commit a172146

Browse files
authored
feat: modify product config (#199)
1 parent cd99dc5 commit a172146

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

src/memos/api/config.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -100,9 +100,9 @@ def get_embedder_config() -> dict[str, Any]:
100100
"backend": "universal_api",
101101
"config": {
102102
"provider": os.getenv("MOS_EMBEDDER_PROVIDER", "openai"),
103-
"api_key": os.getenv("OPENAI_API_KEY", "sk-xxxx"),
103+
"api_key": os.getenv("MOS_EMBEDDER_API_KEY", "sk-xxxx"),
104104
"model_name_or_path": os.getenv("MOS_EMBEDDER_MODEL", "text-embedding-3-large"),
105-
"base_url": os.getenv("OPENAI_API_BASE", "http://openai.com"),
105+
"base_url": os.getenv("MOS_EMBEDDER_API_BASE", "http://openai.com"),
106106
},
107107
}
108108
else: # ollama
@@ -120,10 +120,9 @@ def get_embedder_config() -> dict[str, Any]:
120120
def get_internet_config() -> dict[str, Any]:
121121
"""Get embedder configuration."""
122122
return {
123-
"backend": "xinyu",
123+
"backend": "bocha",
124124
"config": {
125-
"api_key": os.getenv("XINYU_API_KEY"),
126-
"search_engine_id": os.getenv("XINYU_SEARCH_ENGINE_ID"),
125+
"api_key": os.getenv("BOCHA_API_KEY"),
127126
"max_results": 15,
128127
"num_per_request": 10,
129128
"reader": {

0 commit comments

Comments
 (0)