Merge pull request #1513 from costly-ai/main

Allow overriding headers for anthropic
BerriAI · Jan 19, 2024 · 6134b65 · 6134b65
2 parents 6bc7cc4 + 3b719b2
commit 6134b65
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/litellm/llms/anthropic.py b/litellm/llms/anthropic.py
@@ -105,8 +105,9 @@ def completion(
     optional_params=None,
     litellm_params=None,
     logger_fn=None,
+    headers={},
 ):
-    headers = validate_environment(api_key)
+    headers = { **validate_environment(api_key), **headers }
     if model in custom_prompt_dict:
         # check if the model has a registered custom prompt
         model_prompt_details = custom_prompt_dict[model]

diff --git a/litellm/main.py b/litellm/main.py
@@ -972,6 +972,7 @@ def completion(
                 encoding=encoding,  # for calculating input/output tokens
                 api_key=api_key,
                 logging_obj=logging,
+                headers=headers,
             )
             if "stream" in optional_params and optional_params["stream"] == True:
                 # don't try to access stream object,