Skip to content

Commit ef67c6f

Browse files
Wang-Daojiyuan.wangfridayL
authored
Feat/fix palyground bug (#691)
* fix playground bug, internet search judge * fix playground internet bug * modify delete mem * modify tool resp bug in multi cube * fix bug in playground chat handle and search inter * modify prompt * fix bug in playground * fix bug playfround * fix bug * fix code * fix model bug in playground * modify plan b * llm param modify * add logger in playground * modify code * fix bug * modify code * modify code * fix bug * fix search bug in plarground * fixx bug * move schadualr to back * modify pref location * modify fast net search * add tags and new package * modify prompt fix bug * remove nltk due to image promblem * prompt modify * modify bug remove redundant field * modify bug * fix playground bug * fix bug * bust internet topk --------- Co-authored-by: yuan.wang <[email protected]> Co-authored-by: chunyu li <[email protected]>
1 parent 48bbe92 commit ef67c6f

File tree

1 file changed

+2
-2
lines changed
  • src/memos/memories/textual/tree_text_memory/retrieve

1 file changed

+2
-2
lines changed

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -531,14 +531,14 @@ def _retrieve_from_internet(
531531
return []
532532
logger.info(f"[PATH-C] '{query}' Retrieving from internet...")
533533
items = self.internet_retriever.retrieve_from_internet(
534-
query=query, top_k=top_k, parsed_goal=parsed_goal, info=info, mode=mode
534+
query=query, top_k=2 * top_k, parsed_goal=parsed_goal, info=info, mode=mode
535535
)
536536
logger.info(f"[PATH-C] '{query}' Retrieved from internet {len(items)} items: {items}")
537537
return self.reranker.rerank(
538538
query=query,
539539
query_embedding=query_embedding[0],
540540
graph_results=items,
541-
top_k=min(top_k, 5),
541+
top_k=top_k,
542542
parsed_goal=parsed_goal,
543543
)
544544

0 commit comments

Comments
 (0)