From 3267e6c654cf0030cf8e53010231434d8a61a3a2 Mon Sep 17 00:00:00 2001 From: Tadashi Date: Thu, 5 Sep 2024 23:08:28 +0700 Subject: [PATCH] fix: disable default install for google-genai package --- flowsettings.py | 16 ++++++++-------- libs/kotaemon/pyproject.toml | 1 - 2 files changed, 8 insertions(+), 9 deletions(-) diff --git a/flowsettings.py b/flowsettings.py index 281fd63..e4b95b9 100644 --- a/flowsettings.py +++ b/flowsettings.py @@ -184,14 +184,14 @@ KH_LLMS["claude"] = { }, "default": False, } -KH_LLMS["gemini"] = { - "spec": { - "__type__": "kotaemon.llms.chats.LCGeminiChat", - "model_name": "gemini-1.5-pro", - "api_key": "your-key", - }, - "default": False, -} +# KH_LLMS["gemini"] = { +# "spec": { +# "__type__": "kotaemon.llms.chats.LCGeminiChat", +# "model_name": "gemini-1.5-pro", +# "api_key": "your-key", +# }, +# "default": False, +# } KH_LLMS["groq"] = { "spec": { "__type__": "kotaemon.llms.ChatOpenAI", diff --git a/libs/kotaemon/pyproject.toml b/libs/kotaemon/pyproject.toml index afd3854..37ce1dd 100644 --- a/libs/kotaemon/pyproject.toml +++ b/libs/kotaemon/pyproject.toml @@ -31,7 +31,6 @@ dependencies = [ "langchain-community>=0.0.34,<0.1.0", "langchain-openai>=0.1.4,<0.2.0", "langchain-anthropic", - "langchain-google-genai", "llama-hub>=0.0.79,<0.1.0", "llama-index>=0.10.40,<0.11.0", "llama-index-vector-stores-chroma>=0.1.9",