Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 5 additions & 0 deletions src/memos/api/handlers/chat_handler.py
Original file line number Diff line number Diff line change
Expand Up @@ -110,6 +110,7 @@ def handle_chat_complete(self, chat_req: APIChatCompleteRequest) -> dict[str, An
Raises:
HTTPException: If chat fails
"""
self.logger.info(f"[ChatHandler] Chat Req is: {chat_req}")
try:
# Resolve readable cube IDs (for search)
readable_cube_ids = chat_req.readable_cube_ids or [chat_req.user_id]
Expand Down Expand Up @@ -241,6 +242,7 @@ def handle_chat_stream(self, chat_req: ChatRequest) -> StreamingResponse:
Raises:
HTTPException: If stream initialization fails
"""
self.logger.info(f"[ChatHandler] Chat Req is: {chat_req}")
try:

def generate_chat_response() -> Generator[str, None, None]:
Expand Down Expand Up @@ -422,6 +424,7 @@ def handle_chat_stream_playground(self, chat_req: ChatPlaygroundRequest) -> Stre
Raises:
HTTPException: If stream initialization fails
"""
self.logger.info(f"[ChatHandler] Chat Req is: {chat_req}")
try:

def generate_chat_response() -> Generator[str, None, None]:
Expand Down Expand Up @@ -585,6 +588,8 @@ def generate_chat_response() -> Generator[str, None, None]:
# get internet reference
internet_reference = self._get_internet_reference(
search_response.data.get("text_mem")[0]["memories"]
if search_response.data.get("text_mem")
else []
)
yield f"data: {json.dumps({'type': 'reference', 'data': reference}, ensure_ascii=False)}\n\n"

Expand Down
7 changes: 0 additions & 7 deletions src/memos/mem_reader/multi_modal_struct.py
Original file line number Diff line number Diff line change
Expand Up @@ -464,13 +464,6 @@ def _get_maybe_merged_memory(
status="activated",
threshold=merge_threshold,
user_name=user_name,
filter={
"or": [
{"memory_type": "LongTermMemory"},
{"memory_type": "UserMemory"},
{"memory_type": "WorkingMemory"},
]
},
)

if not search_results:
Expand Down
2 changes: 2 additions & 0 deletions src/memos/mem_reader/read_multi_modal/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@
TextualMemoryItem,
TreeNodeTextualMemoryMetadata,
)
from memos.utils import timed

from .utils import detect_lang, get_text_splitter

Expand Down Expand Up @@ -245,6 +246,7 @@ def parse(
else:
raise ValueError(f"Unknown mode: {mode}. Must be 'fast' or 'fine'")

@timed
def _split_text(self, text: str, is_markdown: bool = False) -> list[str]:
"""
Split text into chunks using text splitter from utils.
Expand Down