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

docs(main.py): adding docstring for text_completion

Browse source
This commit is contained in:
Krrish Dholakia 2023-10-16 07:31:39 -07:00
parent 09b51fcdb4
commit 1a09bbd4a9
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/main.py
View file

@ -1551,6 +1551,9 @@ def embedding(
###### Text Completion ################ ###### Text Completion ################
def text_completion(*args, **kwargs): def text_completion(*args, **kwargs):
"""
This maps to the Openai.Completion.create format, which has a different I/O (accepts prompt, returning ["choices"]["text"].
"""
if "prompt" in kwargs: if "prompt" in kwargs:
messages = [{"role": "system", "content": kwargs["prompt"]}] messages = [{"role": "system", "content": kwargs["prompt"]}]
kwargs["messages"] = messages kwargs["messages"] = messages