Merge pull request stanfordnlp#1542 from bluewin4/main

lm.py: disable telemetry in litellm
isaka · Sep 25, 2024 · 8a85337 · 8a85337
2 parents 2ce9a10 + a707b00
commit 8a85337
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/dspy/clients/lm.py b/dspy/clients/lm.py
@@ -7,7 +7,10 @@
     import warnings
     with warnings.catch_warnings():
         warnings.filterwarnings("ignore", category=UserWarning)
-        import litellm
+        if "LITELLM_LOCAL_MODEL_COST_MAP" not in os.environ:
+             os.environ["LITELLM_LOCAL_MODEL_COST_MAP"] = "True"
+        import litellm  
+        litellm.telemetry = False
 
     from litellm.caching import Cache
     disk_cache_dir = os.environ.get('DSPY_CACHEDIR') or os.path.join(Path.home(), '.dspy_cache')