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

9
litellm/tests/test_get_optional_params_functions_not_supported.py Normal file
View file

@ -0,0 +1,9 @@
import litellm
from litellm import get_optional_params
litellm.add_function_to_prompt = True
optional_params = get_optional_params(
tools= [{'type': 'function', 'function': {'description': 'Get the current weather in a given location', 'name': 'get_current_weather', 'parameters': {'type': 'object', 'properties': {'location': {'type': 'string', 'description': 'The city and state, e.g. San Francisco, CA'}, 'unit': {'type': 'string', 'enum': ['celsius', 'fahrenheit']}}, 'required': ['location']}}}],
tool_choice= 'auto',
)
assert optional_params is not None

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(