uPj UdZddlmZmZddlmZmZmZmZm Z ddl m Z ddl m Z ddlmZmZddlmZerddlmZeej,Ze ed <Gd d eZGd d eZGddZy)a Credentials repository for database operations on LiteLLM_CredentialsTable. This is the only place that talks to ``litellm_credentialstable``. Encryption of credential values is the caller's responsibility (see ``CredentialHelperUtils``), so reads return the stored values verbatim. )MappingSequence) TYPE_CHECKINGAnyFinalProtocol TypeAlias)CredentialItem)"wrap_table_actions_for_config_sync)DbRecordrecord_to_dict) TableActions)models_CredentialsTableceZdZeddZy)_PrismaCredentialsDbcyNselfs p/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/repositories/credentials_repository.pylitellm_credentialstablez-_PrismaCredentialsDb.litellm_credentialstables?BNreturnr)__name__ __module__ __qualname__propertyrrrrrrs BBrrc"eZdZedefdZy)_PrismaClientViewrcyrrrs rdbz_PrismaClientView.dbs*-rN)rrrr rr$rrrr"r"s -(--rr"ceZdZdZdefdZedefdZeddZ e de dzde dzfd Z ded fd Zd eeefdd fd Zdede dzfdZded eeefddfdZdeddfdZy)CredentialsRepositoryzIRepository for credentials database operations, keyed by credential name. prisma_clientc||_yr)_prisma_client)rr's r__init__zCredentialsRepository.__init__$s +rrcL|j td|j}|S)NzFNo DB Connected. See - https://docs.litellm.ai/docs/proxy/virtual_keys)r) RuntimeError)rclients rr'z#CredentialsRepository.prisma_client's*    &gh h+/+>+> rcXt|jjjdS)Nr)actions table_name)r r'r$rrs rtablezCredentialsRepository.table.s'1&&))BB1  rrecordNc|yt|}tj|d|jdxsi|jdxsidS)Ncredential_namecredential_valuescredential_info)r4r5r6)r r model_validateget)r2datas r _to_modelzCredentialsRepository._to_model5sY >$V,,,#'(9#:%)XX.A%B%Hb#'88,=#>#D"   rz&prisma_models.LiteLLM_CredentialsTablecRK|jjd{S7wr)r1 find_manyrs rfind_allzCredentialsRepository.find_allBsZZ))++++s '%'r9cVK|jj|d{S7w)N)r9)r1create)rr9s rr?zCredentialsRepository.createEs$ZZ&&D&1111s )')r4c|K|jjd|id{}|j|S7wNr4)where)r1 find_uniquer:)rr4r2s r find_by_namez"CredentialsRepository.find_by_nameHs;"jj44^ 8^ SCSrYsc.AA5\I>.#/ 0V0V#WyWC8C .. 1S1Sr