fix(openai.py): support passing prompt as list instead of concat string

This commit is contained in:
Krrish Dholakia 2024-04-03 15:23:01 -07:00
parent 6048582f95
commit 475144e5b7
2 changed files with 4 additions and 4 deletions

View file

@ -1033,7 +1033,7 @@ class OpenAITextCompletion(BaseLLM):
):
prompt = messages[0]["content"]
else:
prompt = " ".join([message["content"] for message in messages]) # type: ignore
prompt = [message["content"] for message in messages] # type: ignore
# don't send max retries to the api, if set