Fix: Changed keyword from assisstant to secretary (#2937)

mem0/llms/aws_bedrock.py

@@ -11,7 +11,6 @@ except ImportError:
 from mem0.configs.llms.base import BaseLlmConfig
 from mem0.llms.base import LLMBase
 
-
 PROVIDERS = ["ai21", "amazon", "anthropic", "cohere", "meta", "mistral", "stability", "writer"]
 
 

mem0/llms/azure_openai.py

@@ -82,6 +82,12 @@ class AzureOpenAILLM(LLMBase):
             str: The generated response.
         """
 
+        user_prompt = messages[-1]['content']
+
+        user_prompt = user_prompt.replace("assistant", "ai")
+
+        messages[-1]['content'] = user_prompt
+
         common_params = {
             "model": self.config.model,
             "messages": messages,

mem0/llms/azure_openai_structured.py

@@ -48,6 +48,13 @@ class AzureOpenAIStructuredLLM(LLMBase):
         Returns:
             str: The generated response.
         """
+
+        user_prompt = messages[-1]['content']
+
+        user_prompt = user_prompt.replace("assistant", "ai")
+
+        messages[-1]['content'] = user_prompt
+
         params = {
             "model": self.config.model,
             "messages": messages,

mem0/llms/sarvam.py

@@ -1,6 +1,8 @@
 import os
-import requests
 from typing import Dict, List, Optional
+
+import requests
+
 from mem0.configs.llms.base import BaseLlmConfig
 from mem0.llms.base import LLMBase
 

mem0/llms/vllm.py

@@ -1,8 +1,8 @@
 import json
 import os
 from typing import Dict, List, Optional
-from openai import OpenAI
 
+from openai import OpenAI
 
 from mem0.configs.llms.base import BaseLlmConfig
 from mem0.llms.base import LLMBase