Remove session_id deprecation warning (#1994)
This commit is contained in:
@@ -1,6 +1,5 @@
|
||||
import logging
|
||||
import os
|
||||
import warnings
|
||||
from functools import wraps
|
||||
from typing import Any, Dict, List, Optional, Union
|
||||
|
||||
@@ -10,11 +9,6 @@ from mem0.memory.setup import get_user_id, setup_config
|
||||
from mem0.memory.telemetry import capture_client_event
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
warnings.filterwarnings(
|
||||
"always",
|
||||
category=DeprecationWarning,
|
||||
message="The 'session_id' parameter is deprecated. User 'run_id' instead.",
|
||||
)
|
||||
|
||||
# Setup user config
|
||||
setup_config()
|
||||
@@ -332,16 +326,6 @@ class MemoryClient:
|
||||
elif isinstance(messages, list):
|
||||
payload["messages"] = messages
|
||||
|
||||
# Handle session_id deprecation
|
||||
if "session_id" in kwargs:
|
||||
warnings.warn(
|
||||
"The 'session_id' parameter is deprecated and will be removed in version 0.1.20. "
|
||||
"Use 'run_id' instead.",
|
||||
DeprecationWarning,
|
||||
stacklevel=2,
|
||||
)
|
||||
kwargs["run_id"] = kwargs.pop("session_id")
|
||||
|
||||
payload.update({k: v for k, v in kwargs.items() if v is not None})
|
||||
return payload
|
||||
|
||||
@@ -355,16 +339,6 @@ class MemoryClient:
|
||||
A dictionary containing the prepared parameters.
|
||||
"""
|
||||
|
||||
# Handle session_id deprecation
|
||||
if "session_id" in kwargs:
|
||||
warnings.warn(
|
||||
"The 'session_id' parameter is deprecated and will be removed in version 0.1.20. "
|
||||
"Use 'run_id' instead.",
|
||||
DeprecationWarning,
|
||||
stacklevel=2,
|
||||
)
|
||||
kwargs["run_id"] = kwargs.pop("session_id")
|
||||
|
||||
return {k: v for k, v in kwargs.items() if v is not None}
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user