Implemented Gemini (#1490) (#1965)

2024-10-21 16:23:26 +05:30
parent d4ffed9822
commit 078aa66b90
6 changed files with 309 additions and 1 deletions
--- a/mem0/utils/factory.py
+++ b/mem0/utils/factory.py
@@ -22,6 +22,7 @@ class LlmFactory:
        "openai_structured": "mem0.llms.openai_structured.OpenAIStructuredLLM",
        "anthropic": "mem0.llms.anthropic.AnthropicLLM",
        "azure_openai_structured": "mem0.llms.azure_openai_structured.AzureOpenAIStructuredLLM",
+        "gemini": "mem0.llms.gemini.GeminiLLM",
    }

    @classmethod