klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix litellm issue (#1635)

Browse Source
This commit is contained in:
Dev Khant
2024-08-03 09:23:14 +05:30
committed by GitHub
parent 784b607613
commit 1c46fddce1
3 changed files with 1220 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
mem0/llms/litellm.py
View File

@@ -1,10 +1,7 @@
import json import json
from typing import Dict, List, Optional from typing import Dict, List, Optional
try: import litellm
import litellm
except ImportError:
raise ImportError("litellm requires extra dependencies. Install with `pip install litellm`") from None
from mem0.llms.base import LLMBase from mem0.llms.base import LLMBase
from mem0.configs.llms.base import BaseLlmConfig from mem0.configs.llms.base import BaseLlmConfig

1218
poetry.lock generated
View File

File diff suppressed because it is too large Load Diff

3
pyproject.toml
View File

@@ -1,6 +1,6 @@
[tool.poetry] [tool.poetry]
name = "mem0ai" name = "mem0ai"
version = "0.0.11" version = "0.0.12"
description = "Long-term memory for AI Agents" description = "Long-term memory for AI Agents"
authors = ["Mem0 <founders@mem0.ai>"] authors = ["Mem0 <founders@mem0.ai>"]
exclude = [ exclude = [
@@ -22,6 +22,7 @@ openai = "^1.33.0"
posthog = "^3.5.0" posthog = "^3.5.0"
pytz = "^2024.1" pytz = "^2024.1"
sqlalchemy = "^2.0.31" sqlalchemy = "^2.0.31"
litellm = "^1.42.7"
[tool.poetry.group.test.dependencies] [tool.poetry.group.test.dependencies]
pytest = "^8.2.2" pytest = "^8.2.2"