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

fix text-003 bug

Browse source
This commit is contained in:
ishaan-jaff 2023-08-17 10:48:22 -07:00
parent 5ef5af8ddc
commit 35fd82fd3f
2 changed files with 9 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/main.py
View file

@ -158,7 +158,7 @@ def completion(
model=model,
prompt = prompt
)
completion_response = response["choices"]["text"]
completion_response = response["choices"][0]["text"]
## LOGGING
logging(model=model, input=prompt, custom_llm_provider=custom_llm_provider, additional_args={"max_tokens": max_tokens, "original_response": completion_response}, logger_fn=logger_fn)
## RESPONSE OBJECT