Poetry and LLM fixes (#1508)

2024-07-20 10:49:15 +05:30
parent 51fd7db205
commit 0c9c5fe9c2
9 changed files with 26 additions and 1556 deletions
--- a/mem0/llms/configs.py
+++ b/mem0/llms/configs.py
@@ -14,7 +14,7 @@ class LlmConfig(BaseModel):
    @field_validator("config")
    def validate_config(cls, v, values):
        provider = values.data.get("provider")
-        if provider in ["openai", "ollama", "groq"]:
+        if provider in ("openai", "ollama", "groq", "together", "aws_bedrock", "litellm"):
            return v
        else:
            raise ValueError(f"Unsupported LLM provider: {provider}")