Skip to content

Commit 09f219f

Browse files
authored
Feat: add func Base class and add strategy logs, low search mode (#552)
* feat: update memos headers * feat: headers add * feat: update search agent * feat: upadte mem story * feat: update mem scehduler * feat: update deepsearch mem code * feat: update deepsearch agent * feat: update test code * fix: remove dup config * feat: dock search pipeline * fix: code test * feat: add test scripts * feat: add test * feat: update need_raw process * fix: add initter * fix: change agent search func name * feat: update logs and defined
1 parent 8374f96 commit 09f219f

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

src/memos/multi_mem_cube/single_cube.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -35,14 +35,17 @@
3535

3636
if TYPE_CHECKING:
3737
from memos.api.product_models import APIADDRequest, APISearchRequest
38+
from memos.mem_cube.navie import NaiveMemCube
39+
from memos.mem_reader.simple_struct import SimpleStructMemReader
40+
from memos.mem_scheduler.optimized_scheduler import OptimizedScheduler
3841

3942

4043
@dataclass
4144
class SingleCubeView(MemCubeView):
4245
cube_id: str
43-
naive_mem_cube: Any
44-
mem_reader: Any
45-
mem_scheduler: Any
46+
naive_mem_cube: NaiveMemCube
47+
mem_reader: SimpleStructMemReader
48+
mem_scheduler: OptimizedScheduler
4649
logger: Any
4750
searcher: Any
4851
deepsearch_agent: Any | None = None
@@ -155,7 +158,7 @@ def _search_text(
155158
Args:
156159
search_req: Search request
157160
user_context: User context
158-
search_mode: Search mode (FAST, FINE, or MIXTURE)
161+
search_mode: Search mode (fast, fine, or mixture)
159162
160163
Returns:
161164
List of formatted memory items
@@ -227,6 +230,7 @@ def _fine_search(
227230
Returns:
228231
List of enhanced search results
229232
"""
233+
logger.info(f"Fine strategy: {FINE_STRATEGY}")
230234
if FINE_STRATEGY == FineStrategy.DEEP_SEARCH:
231235
return self._deep_search(search_req=search_req, user_context=user_context)
232236
elif FINE_STRATEGY == FineStrategy.AGENTIC_SEARCH:

0 commit comments

Comments
 (0)