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 from copy import deepcopy
import httpx import httpx
import litellm import litellm
from ._logging import verbose_logger from ._logging import verbose_logger
from litellm import ( # type: ignore from litellm import ( # type: ignore
client, client,
@ -1213,11 +1214,12 @@ def completion(
) )
response = model_response response = model_response
elif ("clarifai" in model elif (
or custom_llm_provider == "clarifai" "clarifai" in model
or model in litellm.clarifai_models or custom_llm_provider == "clarifai"
): or model in litellm.clarifai_models
clarifai_key = None ):
clarifai_key = None
clarifai_key = ( clarifai_key = (
api_key api_key
or litellm.clarifai_key or litellm.clarifai_key