Fix failing tests (#3162)

2025-07-25 00:58:45 +05:30
parent 37ee3c5eb2
commit 4433666117
11 changed files with 144 additions and 169 deletions
--- a/mem0/llms/azure_openai_structured.py
+++ b/mem0/llms/azure_openai_structured.py
@@ -49,11 +49,11 @@ class AzureOpenAIStructuredLLM(LLMBase):
            str: The generated response.
        """

-        user_prompt = messages[-1]['content']
+        user_prompt = messages[-1]["content"]

        user_prompt = user_prompt.replace("assistant", "ai")

-        messages[-1]['content'] = user_prompt
+        messages[-1]["content"] = user_prompt

        params = {
            "model": self.config.model,