Skip to content

Commit 89f11aa

Browse files
author
yuan.wang
committed
modify bug in chat handle
1 parent 492571b commit 89f11aa

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

src/memos/api/handlers/chat_handler.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,9 @@ def handle_chat_complete(self, chat_req: APIChatCompleteRequest) -> dict[str, An
142142

143143
# Step 2: Build system prompt
144144
system_prompt = self._build_system_prompt(
145-
filtered_memories, search_response.data["pref_string"], chat_req.system_prompt
145+
filtered_memories,
146+
search_response.data.get("pref_string", ""),
147+
chat_req.system_prompt,
146148
)
147149

148150
# Prepare message history
@@ -257,7 +259,7 @@ def generate_chat_response() -> Generator[str, None, None]:
257259
# Step 2: Build system prompt with memories
258260
system_prompt = self._build_system_prompt(
259261
filtered_memories,
260-
search_response.data["pref_string"],
262+
search_response.data.get("pref_string", ""),
261263
chat_req.system_prompt,
262264
)
263265

@@ -436,7 +438,7 @@ def generate_chat_response() -> Generator[str, None, None]:
436438

437439
# Step 2: Build system prompt with memories
438440
system_prompt = self._build_enhance_system_prompt(
439-
filtered_memories, search_response.data["pref_string"]
441+
filtered_memories, search_response.data.get("pref_string", "")
440442
)
441443

442444
# Prepare messages

0 commit comments

Comments
 (0)