Option for LiteLLM to manage API keys (#604)

abhishekbhakat · web-flow · commit aac2f5461b0d · 2024-08-29T23:27:44.000+02:00
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
 

Original file line number	Diff line number	Diff line change
`@@ -39,7 +39,7 @@ dependencies = [`
`39`	`39`
`40`	`40`	`[project.optional-dependencies]`
`41`	`41`	`litellm = [`
`42`		`- "litellm == 1.24.5"`
	`42`	`+ "litellm == 1.42.5"`
`43`	`43`	`]`
`44`	`44`	`test = [`
`45`	`45`	`"pytest >= 7.2.2, < 8.0.0",`