Use gapminder fork of litellm with safety_settings support for Gemini…

… provider
Gapminder · Feb 1, 2024 · 9deecd8 · 9deecd8
1 parent 33675e3
commit 9deecd8
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 245 deletions.
diff --git a/automation-api/poetry.lock b/automation-api/poetry.lock
diff --git a/automation-api/pyproject.toml b/automation-api/pyproject.toml
@@ -41,7 +41,7 @@ types-requests = "^2.28.11.6"
 bracelogger = "^1.1.0"
 
 # for access to AI models
-litellm = "^1.19.4"
+litellm = {git = "https://github.com/Gapminder/litellm#gemini-provider-safety-settings"}
 langchain = "^0.0.275"
 
 # for hugging face

diff --git a/automation-api/yival_experiments/custom-litellm-llms-gemini.py b/automation-api/yival_experiments/custom-litellm-llms-gemini.py