Fix litellm issue (#1635)

2024-08-03 09:23:14 +05:30
parent 784b607613
commit 1c46fddce1
3 changed files with 1220 additions and 6 deletions
--- a/mem0/llms/litellm.py
+++ b/mem0/llms/litellm.py
@@ -1,10 +1,7 @@
 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
+import litellm

 from mem0.llms.base import LLMBase
 from mem0.configs.llms.base import BaseLlmConfig