phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

Merge pull request #3439 from phact/patch-3

Browse source 
add_function_to_prompt bug fix
This commit is contained in:
Ishaan Jaff 2024-05-07 19:31:19 -07:00 • committed by GitHub
commit 4e7b5aa9d7
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 12 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/utils.py
View file

@ -4836,7 +4836,7 @@ def get_optional_params(
**kwargs,
):
# retrieve all parameters passed to the function
passed_params = locals()
passed_params = locals().copy()
special_params = passed_params.pop("kwargs")
for k, v in special_params.items():
if k.startswith("aws_") and (
@ -4933,6 +4933,7 @@ def get_optional_params(
and custom_llm_provider != "mistral"
and custom_llm_provider != "anthropic"
and custom_llm_provider != "cohere_chat"
and custom_llm_provider != "cohere"
and custom_llm_provider != "bedrock"
and custom_llm_provider != "ollama_chat"
):
@ -4957,7 +4958,7 @@ def get_optional_params(
litellm.add_function_to_prompt
): # if user opts to add it to prompt instead
optional_params["functions_unsupported_model"] = non_default_params.pop(
"tools", non_default_params.pop("functions")
"tools", non_default_params.pop("functions", None)
)
else:
raise UnsupportedParamsError(