Skip to content

Commit 39cab8c

Browse files
authored
Feat: update code (#533)
* feat: add full text memory * fix: remove search * feat: update pulgin
1 parent 57e42f1 commit 39cab8c

File tree

3 files changed

+3
-2
lines changed

3 files changed

+3
-2
lines changed

src/memos/api/handlers/search_handler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -191,7 +191,7 @@ def _fast_search(
191191
"""
192192
target_session_id = search_req.session_id or "default_session"
193193
search_filter = {"session_id": search_req.session_id} if search_req.session_id else None
194-
plugin = bool(search_req.info is not None and search_req.info.get("origin_model"))
194+
plugin = bool(search_req.source is not None and search_req.source == "plugin")
195195
search_results = self.naive_mem_cube.text_mem.search(
196196
query=search_req.query,
197197
user_name=user_context.mem_cube_id,

src/memos/api/product_models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -185,7 +185,7 @@ class APISearchRequest(BaseRequest):
185185
)
186186
include_preference: bool = Field(True, description="Whether to handle preference memory")
187187
pref_top_k: int = Field(6, description="Number of preference results to return")
188-
info: dict | None = Field(None, description="Info for search")
188+
source: str | None = Field(None, description="Source of the search")
189189

190190

191191
class APIADDRequest(BaseRequest):

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

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -275,6 +275,7 @@ def _retrieve_simple(
275275
user_name=user_name,
276276
use_fast_graph=self.use_fast_graph,
277277
)
278+
logger.info(f"[SIMPLESEARCH] Items count: {len(items)}")
278279
documents = [getattr(item, "memory", "") for item in items]
279280
documents_embeddings = self.embedder.embed(documents)
280281
similarity_matrix = cosine_similarity_matrix(documents_embeddings)

0 commit comments

Comments
 (0)