Fix not working with Gemini models (#2021)

2025-01-09 18:49:26 +07:00
parent c90f87e657
commit 4c31c65649
4 changed files with 63 additions and 17 deletions
--- a/mem0/llms/configs.py
+++ b/mem0/llms/configs.py
@@ -21,6 +21,7 @@ class LlmConfig(BaseModel):
            "azure_openai",
            "openai_structured",
            "azure_openai_structured",
+            "gemini",
        ):
            return v
        else: