phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 03:34:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(main.py): expose custom llm provider for text completions

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-02 07:55:54 -07:00
parent a652f05bb3
commit 943f9d9432
2 changed files with 3 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/main.py
View file

@ -496,7 +496,8 @@ def completion(
additional_args={"headers": headers},
)
elif (
model in litellm.open_ai_text_completion_models
custom_llm_provider == "text-completion-openai"
or model in litellm.open_ai_text_completion_models
or "ft:babbage-002" in model
or "ft:davinci-002" in model # support for finetuned completion models
# NOTE: Do NOT add custom_llm_provider == "openai".