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
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

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