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

(fix) use OpenAI organization in ahealth_check

Browse source
This commit is contained in:
ishaan-jaff 2024-01-30 11:45:22 -08:00
parent 5c0f9f421c
commit e011c4a989
3 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/main.py
View file

@ -3227,6 +3227,7 @@ async def ahealth_check(
or custom_llm_provider == "text-completion-openai"
):
api_key = model_params.get("api_key") or get_secret("OPENAI_API_KEY")
organization = model_params.get("organization")
timeout = (
model_params.get("timeout")
@ -3244,6 +3245,7 @@ async def ahealth_check(
mode=mode,
prompt=prompt,
input=input,
organization=organization,
)
else:
if mode == "embedding":