Skip to content

Commit 7541827

Browse files
fridayLCaralHsi
andauthored
Feat:reorg playground code (#497)
* feat: re org code * feat: code reorg and merge API and playground * feat: update memcube info * feat: remove act mem and params mem * feat: upadte init * code suffix * feat: update internet search mode --------- Co-authored-by: CaralHsi <[email protected]>
1 parent 465fba3 commit 7541827

File tree

3 files changed

+5
-5
lines changed

3 files changed

+5
-5
lines changed

src/memos/api/handlers/chat_handler.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,7 @@ def handle_chat_complete(self, chat_req: APIChatCompleteRequest) -> dict[str, An
116116
query=chat_req.query,
117117
top_k=chat_req.top_k or 10,
118118
session_id=chat_req.session_id,
119-
mode=SearchMode.FINE,
119+
mode=SearchMode.FAST,
120120
internet_search=chat_req.internet_search,
121121
moscube=chat_req.moscube,
122122
chat_history=chat_req.history,
@@ -213,8 +213,8 @@ def generate_chat_response() -> Generator[str, None, None]:
213213
query=chat_req.query,
214214
top_k=20,
215215
session_id=chat_req.session_id,
216-
mode=SearchMode.FINE,
217-
internet_search=chat_req.internet_search,
216+
mode=SearchMode.FINE if chat_req.internet_search else SearchMode.FAST,
217+
internet_search=chat_req.internet_search, # TODO this param is not worked at fine mode
218218
moscube=chat_req.moscube,
219219
chat_history=chat_req.history,
220220
)

src/memos/api/handlers/search_handler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -245,7 +245,7 @@ def _fine_search(
245245
query=search_req.query,
246246
user_name=user_context.mem_cube_id,
247247
top_k=search_req.top_k,
248-
mode=SearchMode.FAST,
248+
mode=SearchMode.FINE,
249249
manual_close_internet=not search_req.internet_search,
250250
moscube=search_req.moscube,
251251
search_filter=search_filter,

src/memos/memories/textual/tree_text_memory/retrieve/searcher.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -182,7 +182,7 @@ def _parse_task(
182182
query_embedding = None
183183

184184
# fine mode will trigger initial embedding search
185-
if mode == "fine":
185+
if mode == "fine_old":
186186
logger.info("[SEARCH] Fine mode: embedding search")
187187
query_embedding = self.embedder.embed([query])[0]
188188

0 commit comments

Comments
 (0)