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

fix(main.py): automatically infer mode for text completion models

Browse source
This commit is contained in:
Krrish Dholakia 2024-04-12 14:16:21 -07:00
parent b0770cf8e2
commit 74aa230eac
3 changed files with 11 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/main.py
View file

@ -3784,6 +3784,9 @@ async def ahealth_check(
api_base = model_params.get("api_base") or get_secret("OPENAI_API_BASE")
if custom_llm_provider == "text-completion-openai":
mode = "completion"
response = await openai_chat_completions.ahealth_check(
model=model,
messages=model_params.get(