phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: refactor add models tab on UI to enable setting credentials

Browse source
This commit is contained in:
Krrish Dholakia 2025-03-12 20:32:01 -07:00
parent 52926408cd
commit d604f52884
6 changed files with 192 additions and 40 deletions
Download patch file Download diff file Expand all files Collapse all files

17
litellm/proxy/credential_endpoints/endpoints.py
View file

@ -6,6 +6,7 @@ from fastapi import APIRouter, Depends, HTTPException, Request, Response
import litellm
from litellm._logging import verbose_proxy_logger
from litellm.litellm_core_utils.credential_accessor import CredentialAccessor
from litellm.proxy._types import CommonProxyErrors, UserAPIKeyAuth
from litellm.proxy.auth.user_api_key_auth import user_api_key_auth
from litellm.proxy.common_utils.encrypt_decrypt_utils import encrypt_value_helper
@ -51,8 +52,10 @@ async def create_credential(
detail={"error": CommonProxyErrors.db_not_connected_error.value},
)
credential = CredentialHelperUtils.encrypt_credential_values(credential)
credentials_dict = credential.model_dump()
encrypted_credential = CredentialHelperUtils.encrypt_credential_values(
credential
)
credentials_dict = encrypted_credential.model_dump()
credentials_dict_jsonified = jsonify_object(credentials_dict)
await prisma_client.db.litellm_credentialstable.create(
data={
@ -62,6 +65,9 @@ async def create_credential(
}
)
## ADD TO LITELLM ##
CredentialAccessor.upsert_credentials([credential])
return {"success": True, "message": "Credential created successfully"}
except Exception as e:
verbose_proxy_logger.exception(e)
@ -146,6 +152,13 @@ async def delete_credential(
await prisma_client.db.litellm_credentialstable.delete(
where={"credential_name": credential_name}
)
## DELETE FROM LITELLM ##
litellm.credential_list = [
cred
for cred in litellm.credential_list
if cred.credential_name != credential_name
]
return {"success": True, "message": "Credential deleted successfully"}
except Exception as e:
return handle_exception_on_proxy(e)