Poetry and LLM fixes (#1508)

This commit is contained in:
Dev Khant
2024-07-20 10:49:15 +05:30
committed by GitHub
parent 51fd7db205
commit 0c9c5fe9c2
9 changed files with 26 additions and 1556 deletions

View File

@@ -1,7 +1,10 @@
import json
from typing import Dict, List, Optional
import litellm
try:
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.configs.llms.base import BaseLlmConfig