Skip to content

Commit 2b797f6

Browse files
committed
feat: modify search rephrase prompt
1 parent 48da7ca commit 2b797f6

File tree

4 files changed

+9
-4
lines changed

4 files changed

+9
-4
lines changed

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

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -218,7 +218,9 @@ def _process_result(
218218
memory_items = []
219219
for read_item_i in read_items[0]:
220220
read_item_i.memory = (
221-
f"Title: {title}\nNewsTime: {publish_time}\nSummary: {summary}\n"
221+
f"[Outer internet view] Title: {title}\nNewsTime:"
222+
f" {publish_time}\nSummary:"
223+
f" {summary}\n"
222224
f"Content: {read_item_i.memory}"
223225
)
224226
read_item_i.metadata.source = "web"

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -179,7 +179,7 @@ def _vector_recall(
179179
query_embedding: list[list[float]],
180180
memory_scope: str,
181181
top_k: int = 20,
182-
max_num: int = 5,
182+
max_num: int = 3,
183183
cube_name: str | None = None,
184184
) -> list[TextualMemoryItem]:
185185
"""

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

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -108,9 +108,10 @@ def _parse_task(self, query, info, mode, top_k=5):
108108
context = list({node["memory"] for node in related_nodes})
109109

110110
# optional: supplement context with internet knowledge
111-
if self.internet_retriever:
111+
"""if self.internet_retriever:
112112
extra = self.internet_retriever.retrieve_from_internet(query=query, top_k=3)
113113
context.extend(item.memory.partition("\nContent: ")[-1] for item in extra)
114+
"""
114115

115116
# parse goal using LLM
116117
parsed_goal = self.task_goal_parser.parse(

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

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,9 @@
1919
Context (if any):
2020
\"\"\"$context\"\"\"
2121
22-
Return strictly in this JSON format:
22+
Return strictly in this JSON format, note that the
23+
keys/tags/rephrased_instruction/memories should use the same language as the
24+
input query:
2325
{
2426
"keys": [...],
2527
"tags": [...],

0 commit comments

Comments
 (0)