[Feature] Add support for Google Gemini (#1009)

2023-12-15 06:10:55 +05:30
parent c0ee680546
commit 151746beec
9 changed files with 211 additions and 14 deletions
--- a/embedchain/factory.py
+++ b/embedchain/factory.py
@@ -18,6 +18,7 @@ class LlmFactory:
        "llama2": "embedchain.llm.llama2.Llama2Llm",
        "openai": "embedchain.llm.openai.OpenAILlm",
        "vertexai": "embedchain.llm.vertex_ai.VertexAILlm",
+        "google": "embedchain.llm.google.GoogleLlm",
    }
    provider_to_config_class = {
        "embedchain": "embedchain.config.llm.base.BaseLlmConfig",