From f4fe2575cc3e6b1eb6866e2f8388eaa02c891a2e Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Wed, 27 Dec 2023 15:19:59 +0530 Subject: [PATCH] (fix) use client for text_completion() --- litellm/main.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/litellm/main.py b/litellm/main.py index 2f4894410c..9b6e57d7c8 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -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]]]]]