feat add safe_memory_mode

This commit is contained in:
Ishaan Jaff 2024-07-12 18:18:39 -07:00
parent fd743aaefd
commit 08efef5316
2 changed files with 3 additions and 0 deletions

View file

@ -113,6 +113,7 @@ ssl_verify: bool = True
ssl_certificate: Optional[str] = None
disable_streaming_logging: bool = False
in_memory_llm_clients_cache: dict = {}
safe_memory_mode: bool = False
### DEFAULT AZURE API VERSION ###
AZURE_DEFAULT_API_VERSION = "2024-02-01" # this is updated to the latest
### GUARDRAILS ###

View file

@ -83,6 +83,8 @@ def safe_deep_copy(data):
Use this function to safely deep copy the LiteLLM Request
"""
if litellm.safe_memory_mode is True:
return data
# Step 1: Remove the litellm_parent_otel_span
if isinstance(data, dict):