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

refactor(main.py): trigger new build

Browse source
This commit is contained in:
Krrish Dholakia 2024-05-11 22:53:09 -07:00
parent 99e8f0715e
commit 61143c8b45
1 changed files with 11 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

12
litellm/main.py
View file

@ -14,6 +14,7 @@ import dotenv, traceback, random, asyncio, time, contextvars
from copy import deepcopy
import httpx
import litellm
from ._logging import verbose_logger
from litellm import ( # type: ignore
client,
@ -1213,11 +1214,12 @@ def completion(
)
response = model_response
elif ("clarifai" in model
or custom_llm_provider == "clarifai"
or model in litellm.clarifai_models
):
clarifai_key = None
elif (
"clarifai" in model
or custom_llm_provider == "clarifai"
or model in litellm.clarifai_models
):
clarifai_key = None
clarifai_key = (
api_key
or litellm.clarifai_key