feat: automates package installation (#1780)

mem0/llms/litellm.py

@@ -1,12 +1,21 @@
+import subprocess
+import sys
 import json
 from typing import Dict, List, Optional
 
 try:
     import litellm
 except ImportError:
-    raise ImportError(
-        "litellm requires extra dependencies. Install with `pip install litellm`"
-    ) from None
+    user_input = input("The 'litellm' library is required. Install it now? [y/N]: ")
+    if user_input.lower() == 'y':
+        try:
+            subprocess.check_call([sys.executable, "-m", "pip", "install", "litellm"])
+            import litellm
+        except subprocess.CalledProcessError:
+            print("Failed to install 'litellm'. Please install it manually using 'pip install litellm'.")
+            sys.exit(1)
+    else:
+        raise ImportError("The required 'litellm' library is not installed.")
 
 from mem0.llms.base import LLMBase
 from mem0.configs.llms.base import BaseLlmConfig