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

(fix) use client for text_completion()

Browse source
This commit is contained in:
ishaan-jaff 2023-12-27 15:19:59 +05:30
parent e70f588b87
commit f4fe2575cc
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/main.py
View file

@ -2362,6 +2362,7 @@ def embedding(
###### Text Completion ################
@client
async def atext_completion(*args, **kwargs):
"""
Implemented to handle async streaming for the text completion endpoint
@ -2426,6 +2427,7 @@ async def atext_completion(*args, **kwargs):
)
@client
def text_completion(
prompt: Union[
str, List[Union[str, List[Union[str, List[int]]]]]