fix: apply remove_code_blocks() to LangChain path in async _create_procedural_memory (#5711)

This commit is contained in:
Hrushikesh Yadav
2026-06-25 16:34:28 +05:30
committed by GitHub
parent af70668308
commit b91c745fbc
2 changed files with 34 additions and 1 deletions
+1 -1
View File
@@ -3464,7 +3464,7 @@ class AsyncMemory(MemoryBase):
if llm is not None:
parsed_messages = convert_to_messages(parsed_messages)
response = await asyncio.to_thread(llm.invoke, input=parsed_messages)
procedural_memory = response.content
procedural_memory = remove_code_blocks(response.content)
else:
procedural_memory = await asyncio.to_thread(self.llm.generate_response, messages=parsed_messages)
procedural_memory = remove_code_blocks(procedural_memory)