Skip to content

Commit

Permalink
add support for gemini
Browse files Browse the repository at this point in the history
Signed-off-by: Myrausman <maira.usman5703@gmail.com>
  • Loading branch information
Myrausman committed Aug 18, 2024
1 parent 16712c9 commit 9a13fd5
Show file tree
Hide file tree
Showing 3 changed files with 16 additions and 12 deletions.
20 changes: 10 additions & 10 deletions kai/config.toml
Original file line number Diff line number Diff line change
Expand Up @@ -135,18 +135,18 @@ password = "dog8code"
# ```


# [models]
# provider = "ChatIBMGenAI"
[models]
provider = "ChatIBMGenAI"

# [models.args]
# model_id = "mistralai/mixtral-8x7b-instruct-v01"
[models.args]
model_id = "mistralai/mixtral-8x7b-instruct-v01"

# [embeddings]
# todo = true
[embeddings]
todo = true

# **Google Gemini Pro**
[models]
provider = "ChatGoogleGenerativeAI"
# [models]
# provider = "ChatGoogleGenerativeAI"

[models.args]
model = "gemini-pro"
# [models.args]
# model = "gemini-pro"
1 change: 1 addition & 0 deletions requirements.in
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@ ibm-generative-ai==2.2.0
Jinja2==3.1.4
langchain==0.2.11
langchain-community==0.2.10
langchain-google-genai==1.0.9
langchain-openai==0.1.17
langchain-experimental==0.0.63
gunicorn==22.0.0
Expand Down
7 changes: 5 additions & 2 deletions requirements.txt
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
#
# This file is autogenerated by pip-compile with Python 3.12
# This file is autogenerated by pip-compile with Python 3.11
# by the following command:
#
# pip-compile --allow-unsafe
Expand Down Expand Up @@ -39,7 +39,8 @@ astunparse==1.6.3
async-lru==2.0.4
# via jupyterlab
async-timeout==4.0.3
# via -r requirements.in
# via
# -r requirements.in
attrs==23.2.0
# via
# aiohttp
Expand Down Expand Up @@ -107,6 +108,8 @@ gitdb==4.0.11
# via gitpython
gitpython==3.1.43
# via -r requirements.in
greenlet==3.0.3
# via sqlalchemy
gunicorn==22.0.0
# via -r requirements.in
h11==0.14.0
Expand Down

0 comments on commit 9a13fd5

Please sign in to comment.