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

fix(text_completion): allow either model or engine to be set

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-17 18:25:21 -08:00
parent 8e430fcfbe
commit 3ea776bdc0
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/main.py
View file

@ -1845,7 +1845,7 @@ def embedding(
###### Text Completion ################
def text_completion(
prompt: Union[str, List[Union[str, List[Union[str, List[int]]]]]], # Required: The prompt(s) to generate completions for.
model: Optional[str], # Optional: either `model` or `engine` can be set
model: Optional[str]=None, # Optional: either `model` or `engine` can be set
best_of: Optional[int] = None, # Optional: Generates best_of completions server-side.
echo: Optional[bool] = None, # Optional: Echo back the prompt in addition to the completion.
frequency_penalty: Optional[float] = None, # Optional: Penalize new tokens based on their existing frequency.

2
litellm/tests/test_text_completion.py
View file

@ -61,7 +61,7 @@ def test_completion_openai_engine():
#print(response.choices[0].text)
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# test_completion_openai_engine()
test_completion_openai_engine()
def test_completion_chatgpt_prompt():