File tree Expand file tree Collapse file tree 3 files changed +3
-2
lines changed
memories/textual/tree_text_memory/retrieve Expand file tree Collapse file tree 3 files changed +3
-2
lines changed Original file line number Diff line number Diff 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 ,
Original file line number Diff line number Diff 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
191191class APIADDRequest (BaseRequest ):
Original file line number Diff line number Diff 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 )
You can’t perform that action at this time.
0 commit comments