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

4
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,7 +1214,8 @@ def completion(
) )
response = model_response response = model_response
elif ("clarifai" in model elif (
"clarifai" in model
or custom_llm_provider == "clarifai" or custom_llm_provider == "clarifai"
or model in litellm.clarifai_models or model in litellm.clarifai_models
): ):