klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix failing tests (#3162)

Browse Source
This commit is contained in:
Parshva Daftari
2025-07-25 00:58:45 +05:30
committed by GitHub
parent 37ee3c5eb2
commit 4433666117
11 changed files with 144 additions and 169 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
mem0/llms/azure_openai.py
View File

@@ -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,

4
mem0/llms/azure_openai_structured.py
View File

@@ -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,

2
mem0/llms/vllm.py
View File

@@ -4,8 +4,6 @@ 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
from mem0.memory.utils import extract_json