phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

remove junk from completion

Browse source
This commit is contained in:
ishaan-jaff 2023-09-28 17:57:48 -07:00
parent 0f3be26778
commit 13ff65a8fe
1 changed files with 2 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/main.py
View file

@ -213,8 +213,6 @@ def completion(
litellm_logging_obj = kwargs.get('litellm_logging_obj', None) litellm_logging_obj = kwargs.get('litellm_logging_obj', None)
id = kwargs.get('id', None) id = kwargs.get('id', None)
metadata = kwargs.get('metadata', None) metadata = kwargs.get('metadata', None)
task = kwargs.get('task', "text-generation-inference")
return_full_text = kwargs.get('return_full_text', False)
request_timeout = kwargs.get('request_timeout', 0) request_timeout = kwargs.get('request_timeout', 0)
fallbacks = kwargs.get('fallbacks', []) fallbacks = kwargs.get('fallbacks', [])
######## end of unpacking kwargs ########### ######## end of unpacking kwargs ###########
@ -267,9 +265,9 @@ def completion(
model=model, model=model,
custom_llm_provider=custom_llm_provider, custom_llm_provider=custom_llm_provider,
top_k=kwargs.get('top_k', 40), top_k=kwargs.get('top_k', 40),
task=task, task=kwargs.get('task', "text-generation-inference"),
remove_input=kwargs.get('remove_input', True), remove_input=kwargs.get('remove_input', True),
return_full_text=return_full_text, return_full_text=kwargs.get('return_full_text', False),
) )
# For logging - save the values of the litellm-specific params passed in # For logging - save the values of the litellm-specific params passed in
litellm_params = get_litellm_params( litellm_params = get_litellm_params(