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.py
+++ b/mem0/llms/azure_openai.py
@@ -82,11 +82,11 @@ class AzureOpenAILLM(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

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