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

mem0/memory/utils.py

@@ -1,5 +1,5 @@
-import re
 import hashlib
+import re
 
 from mem0.configs.prompts import FACT_RETRIEVAL_PROMPT
 

mem0/vector_stores/azure_ai_search.py

@@ -5,8 +5,8 @@ from typing import List, Optional
 
 from pydantic import BaseModel
 
-from mem0.vector_stores.base import VectorStoreBase
 from mem0.memory.utils import extract_json
+from mem0.vector_stores.base import VectorStoreBase
 
 try:
     from azure.core.credentials import AzureKeyCredential

mem0/vector_stores/baidu.py

@@ -8,12 +8,31 @@ from mem0.vector_stores.base import VectorStoreBase
 
 try:
     import pymochow
-    from pymochow.configuration import Configuration
     from pymochow.auth.bce_credentials import BceCredentials
-    from pymochow.model.enum import FieldType, MetricType, IndexType, TableState, ServerErrCode
-    from pymochow.model.schema import Field, Schema, VectorIndex, FilteringIndex, HNSWParams, AutoBuildRowCountIncrement
-    from pymochow.model.table import Partition, Row, VectorSearchConfig, VectorTopkSearchRequest, FloatVector
+    from pymochow.configuration import Configuration
     from pymochow.exception import ServerError
+    from pymochow.model.enum import (
+        FieldType,
+        IndexType,
+        MetricType,
+        ServerErrCode,
+        TableState,
+    )
+    from pymochow.model.schema import (
+        AutoBuildRowCountIncrement,
+        Field,
+        FilteringIndex,
+        HNSWParams,
+        Schema,
+        VectorIndex,
+    )
+    from pymochow.model.table import (
+        FloatVector,
+        Partition,
+        Row,
+        VectorSearchConfig,
+        VectorTopkSearchRequest,
+    )
 except ImportError:
     raise ImportError("The 'pymochow' library is required. Please install it using 'pip install pymochow'.")
 

mem0/vector_stores/mongodb.py

@@ -1,12 +1,12 @@
 import logging
-from typing import List, Optional, Dict, Any
+from typing import Any, Dict, List, Optional
 
 from pydantic import BaseModel
 
 try:
     from pymongo import MongoClient
-    from pymongo.operations import SearchIndexModel
     from pymongo.errors import PyMongoError
+    from pymongo.operations import SearchIndexModel
 except ImportError:
     raise ImportError("The 'pymongo' library is required. Please install it using 'pip install pymongo'.")
 

mem0/vector_stores/redis.py

@@ -11,8 +11,8 @@ from redisvl.index import SearchIndex
 from redisvl.query import VectorQuery
 from redisvl.query.filter import Tag
 
-from mem0.vector_stores.base import VectorStoreBase
 from mem0.memory.utils import extract_json
+from mem0.vector_stores.base import VectorStoreBase
 
 logger = logging.getLogger(__name__)