@@ -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