Fix proxy for Mem0 (#2616)
This commit is contained in:
@@ -25,7 +25,7 @@ except ImportError:
|
||||
|
||||
from mem0 import Memory, MemoryClient
|
||||
from mem0.configs.prompts import MEMORY_ANSWER_PROMPT
|
||||
from mem0.memory.telemetry import capture_client_event
|
||||
from mem0.memory.telemetry import capture_client_event, capture_event
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
@@ -143,7 +143,10 @@ class Completions:
|
||||
api_key=api_key,
|
||||
model_list=model_list,
|
||||
)
|
||||
capture_client_event("mem0.chat.create", self.mem0_client)
|
||||
if isinstance(self.mem0_client, Memory):
|
||||
capture_event("mem0.chat.create", self.mem0_client)
|
||||
else:
|
||||
capture_client_event("mem0.chat.create", self.mem0_client)
|
||||
return response
|
||||
|
||||
def _prepare_messages(self, messages: List[dict]) -> List[dict]:
|
||||
|
||||
Reference in New Issue
Block a user