Option for LiteLLM to manage API keys (#604)

TheR1D · Aug 29, 2024 · aac2f54 · aac2f54
1 parent ab6b475
commit aac2f54
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -39,7 +39,7 @@ dependencies = [
 
 [project.optional-dependencies]
 litellm = [
-    "litellm == 1.24.5"
+    "litellm == 1.42.5"
 ]
 test = [
     "pytest >= 7.2.2, < 8.0.0",

diff --git a/sgpt/handlers/handler.py b/sgpt/handlers/handler.py
@@ -23,6 +23,7 @@
 
     completion = litellm.completion
     litellm.suppress_debug_info = True
+    additional_kwargs.pop("api_key")
 else:
     from openai import OpenAI