Skip to content

Commit c3c8403

Browse files
Wang-Daojiyuan.wangCaralHsi
authored
Feat/merge api refactor to dev (#569)
* new type * llm reconstruct and add search api modify * llm construction * add delete and get, modify chat * modify code * modify code * modify code * coding chat * fix bug in get and delete * add internet reference in playground chat stream * remove moscube * modify code * fix pre_commit * fix make test * finish info transfer * add info and custom tags * modify model product fileds * fix get api bug * fix bug * fix bug in pref add info * modify code * fix bug in get and delete * modify delete code * new package * fix bug * delete mem, add writeble ids * change internet search to False * modify --------- Co-authored-by: yuan.wang <[email protected]> Co-authored-by: CaralHsi <[email protected]>
1 parent f714027 commit c3c8403

File tree

2 files changed

+8
-3
lines changed

2 files changed

+8
-3
lines changed

src/memos/api/handlers/memory_handler.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -187,6 +187,9 @@ def handle_get_memories(
187187

188188

189189
def handle_delete_memories(delete_mem_req: DeleteMemoryRequest, naive_mem_cube: NaiveMemCube):
190+
logger.info(
191+
f"[Delete memory request] writable_cube_ids: {delete_mem_req.writable_cube_ids}, memory_ids: {delete_mem_req.memory_ids}"
192+
)
190193
# Validate that only one of memory_ids, file_ids, or filter is provided
191194
provided_params = [
192195
delete_mem_req.memory_ids is not None,
@@ -201,7 +204,8 @@ def handle_delete_memories(delete_mem_req: DeleteMemoryRequest, naive_mem_cube:
201204

202205
try:
203206
if delete_mem_req.memory_ids is not None:
204-
naive_mem_cube.text_mem.delete(delete_mem_req.memory_ids)
207+
for cube_id in delete_mem_req.writable_cube_ids:
208+
naive_mem_cube.text_mem.delete(delete_mem_req.memory_ids, user_name=cube_id)
205209
if naive_mem_cube.pref_mem is not None:
206210
naive_mem_cube.pref_mem.delete(delete_mem_req.memory_ids)
207211
elif delete_mem_req.file_ids is not None:

src/memos/api/product_models.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,7 @@ class ChatRequest(BaseRequest):
111111
)
112112

113113
# ==== Extended capabilities ====
114-
internet_search: bool = Field(True, description="Whether to use internet search")
114+
internet_search: bool = Field(False, description="Whether to use internet search")
115115
threshold: float = Field(0.5, description="Threshold for filtering references")
116116

117117
# ==== Backward compatibility ====
@@ -699,7 +699,7 @@ class APIChatCompleteRequest(BaseRequest):
699699
)
700700

701701
# ==== Extended capabilities ====
702-
internet_search: bool = Field(True, description="Whether to use internet search")
702+
internet_search: bool = Field(False, description="Whether to use internet search")
703703
threshold: float = Field(0.5, description="Threshold for filtering references")
704704

705705
# ==== Backward compatibility ====
@@ -728,6 +728,7 @@ class GetMemoryRequest(BaseRequest):
728728
class DeleteMemoryRequest(BaseRequest):
729729
"""Request model for deleting memories."""
730730

731+
writable_cube_ids: list[str] = Field(..., description="Writable cube IDs")
731732
memory_ids: list[str] | None = Field(None, description="Memory IDs")
732733
file_ids: list[str] | None = Field(None, description="File IDs")
733734
filter: dict[str, Any] | None = Field(None, description="Filter for the memory")

0 commit comments

Comments
 (0)