From 94cc3a96c23aa18efeaa3f086ad36f396f82635f Mon Sep 17 00:00:00 2001 From: saidmukhamad Date: Fri, 27 Sep 2024 07:15:42 +0300 Subject: [PATCH] fix: add langchain google dependency (#329) * add-gemini-deps * uncomment gemeni flow settings --- flowsettings.py | 16 ++++++++-------- libs/kotaemon/pyproject.toml | 1 + 2 files changed, 9 insertions(+), 8 deletions(-) diff --git a/flowsettings.py b/flowsettings.py index 52553c3..bb2cb7a 100644 --- a/flowsettings.py +++ b/flowsettings.py @@ -191,14 +191,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 fb3fbcf..59a31d0 100644 --- a/libs/kotaemon/pyproject.toml +++ b/libs/kotaemon/pyproject.toml @@ -33,6 +33,7 @@ dependencies = [ "langchain>=0.1.16,<0.2.16", "langchain-community>=0.0.34,<=0.2.11", "langchain-openai>=0.1.4,<0.2.0", + "langchain-google-genai>=1.0.3,<2.0.0", "langchain-anthropic", "langchain-cohere>=0.2.4,<0.3.0", "llama-hub>=0.0.79,<0.1.0",