tPjMUddlZddlmZddlmZmZmZmZddlm Z m Z m Z ddl m Z ddlmZddlmZmZmZddlmZdd lmZmZdd lmZdd lmZdd lmZdd lm Z m!Z!m"Z"m#Z#m$Z$m%Z%erddl&m'Z'e Z(ee)d<eZ*ee)d<GddeZ+dddee+fdZ,de-de-de-fdZ.dZ/e(jaddge ege%e efdefd Z1e(jeddge ege#e efd!e$defd"Z3d#Z4ee)d$<de5fd%Z6de5fd&Z7de5fd'Z8e(jsd(dge ege#e efd!e"defd)Z:e(jsd*dge ege!e efd!e defd+Z;e(jsd,dge ege!e efd!e defd-Zy)0N)Mapping) TYPE_CHECKINGFinalProtocolcast) APIRouterDepends HTTPException)verbose_proxy_logger)SensitiveDataMasker)CommonProxyErrorsLitellmUserRolesUserAPIKeyAuth)user_api_key_auth)decrypt_value_helperencrypt_value_helper)_user_has_admin_view)ConfigRepository) TableActions)CloudZeroExportRequestCloudZeroExportResponseCloudZeroInitRequestCloudZeroInitResponseCloudZeroSettingsUpdateCloudZeroSettingsView) PrismaClientrouter_sensitive_maskerc<eZdZdZedeeeefzdzfdZy)_CloudZeroConfigRowzSThe ``LiteLLM_Config`` row holding ``cloudzero_settings``, as this module reads it.returnNcyN)selfs u/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/proxy/spend_tracking/cloudzero_endpoints.py param_valuez_CloudZeroConfigRow.param_value-s=@)__name__ __module__ __qualname____doc__propertystrrr'r$r(r&r r *s+] @S738#44t;@@r(r prisma_clientrr!cZt|j}ttt|Sr#)rtablerrr )r/repository_tables r& _config_tabler31s'.}=CC  013C DDr(api_key connection_idtimezonecLKddlm}|'tddtjj it |}|||d}t|jjdd id tj|d d tj|id  d{y7w)z Store CloudZero settings in the database with encrypted API key. Args: api_key: CloudZero API key to encrypt and store connection_id: CloudZero connection ID timezone: Timezone for date handling rr/Nerror status_codedetailr4r5r6 param_namecloudzero_settings)r?r'r')createupdate)wheredata) litellm.proxy.proxy_serverr/r r db_not_connected_errorvaluerrr1upsertjsondumps)r4r5r6r/encrypted_api_keyr@s r&_set_cloudzero_settingsrL6s9.EEKKL  4G<%&! = ) / / 6 6123#zz*<=%djj1C&DE   7   sBB$B"B$c4Kddlm}|'tddtjj it |jddi d{}| |jiSt|jtr |j}nOt|jtr tj|j}nt|j}|jd }|r$t|d d }|tddd i||d <|S7íw)z Retrieve CloudZero settings from the database with decrypted API key. Returns: dict: CloudZero settings with decrypted API key, or empty dict if not configured rr8Nr9r:r;r?r@rCr4cloudzero_api_key)keyexception_typezGFailed to decrypt CloudZero API key. Check your salt key configuration.)rEr/r r rFrGr3 find_firstr' isinstancedictr.rIloadsgetr)r/cloudzero_configsettingsrKdecrypted_api_keys r&_get_cloudzero_settingsrZ\s#9.EEKKL  %2-$@$K$KS_auRv$K$ww#3#?#?#G "..5#// $00# 6::.::;(445 (||I601BH[lst  $!jk 0 O/xsADDCDz/cloudzero/settings CloudZero)tags dependenciesresponse_modeluser_api_key_dictcKt|s'tddtjji t d{}|st ddddStj|}t |jd|jd|jdd S7f#t$r}|d}~wt$r-}tjd |td dd |id}~wwxYww) a View current CloudZero settings. Returns the current CloudZero configuration with the API key masked for security. Only the first 4 and last 4 characters of the API key are shown. Returns null/empty values when settings are not configured (consistent with other settings endpoints). Only admin users (Proxy Admin or Admin Viewer) can view CloudZero settings. r:r;N)api_key_maskedr5r6statusr4r5r6 configuredz'Error retrieving CloudZero settings: %sr9z'Failed to retrieve CloudZero settings: ) rr r not_allowed_accessrGrZrr mask_dictrV Exceptionr r:)r_rXmasked_settingses r&get_cloudzero_settingsrjs& 1 2.AAGGH   7 99(#"  "3!z'CloudZero settings updated successfullysuccessmessagerczTCloudZero settings not found. Please initialize settings first using /cloudzero/initz%Error updating CloudZero settings: %sr9z%Failed to update CloudZero settings: ) user_roler PROXY_ADMINr r rerGanyr4r5r6rZrLr inforr<rgr:)rkr_current_settingsupdated_api_keyupdated_connection_idupdated_timezoneris r&update_cloudzero_settingsrys0""&6&B&BB.AAGGH  !6!68H8HI JMN  % (?(A"A56F6F6R'"2"2XhisXt&#/%   !!"KL$-V_hii%#B    ==C s    ""#JANDQCHI   s[BF  D(D$A,D(=D&>%D(#F $D(&D(( F 1 E F (FF  F F%_cloudzero_background_job_initializedcK ddlm}|yt|jddid{}|duxr|jduS7#t $r }t jd|Yd}~yd}~wwxYww) a Check if CloudZero is setup in the database. CloudZero is considered setup in the database if: - CloudZero settings exist in the database - The settings have a non-None value Returns: bool: True if CloudZero is active, False otherwise rr8NFr?r@rNz#Error checking CloudZero status: %s)rEr/r3rRr'rgr r:)r/rWris r&is_cloudzero_setup_in_dbr|s<  )6m(D(O(O!56)P) #   t+X0@0L0LTX0XX #  ""#H!LsHA4AA4 AAAA4A A1A,'A4,A11A4c&ddl}d|jvS)a  Check if CloudZero is setup in config.yaml or environment variables. CloudZero is considered setup in config if: - "cloudzero" is in the callbacks list in config.yaml, OR Returns: bool: True if CloudZero is configured, False otherwise rN cloudzero)litellm callbacks)rs r&is_cloudzero_setup_in_configr/s '++ ++r(cK trytd{ryy7#t$r }tjd|Yd}~yd}~wwxYww)aj Check if CloudZero is setup in either config.yaml/env vars OR database. CloudZero is considered setup if: - CloudZero is configured in config.yaml callbacks, OR - CloudZero environment variables are set, OR - CloudZero settings exist in the database Returns: bool: True if CloudZero is configured anywhere, False otherwise TNFz"Error checking CloudZero setup: %s)rr|rgr r:)ris r&is_cloudzero_setupr=sR  ' )*+ + + , ""#GKs?A %A %#%A% AA A AAz/cloudzero/initcK|jtjk7r'tddtj j i t|j|j|jd{tjdtddS7&#t$r-}tjd |td dd |id}~wwxYww) a Initialize CloudZero settings and store in the database. This endpoint stores the CloudZero API key, connection ID, and timezone configuration in the proxy database for use by the CloudZero logger. Parameters: - api_key: CloudZero API key for authentication - connection_id: CloudZero connection ID for data submission - timezone: Timezone for date handling (default: UTC) Only admin users can configure CloudZero settings. rar:r;r>Nz+CloudZero settings initialized successfullyrmrnz)Error initializing CloudZero settings: %sr9z)Failed to initialize CloudZero settings: )rqrrrr r rerGrLr4r5r6r rtrrgr:)rkr_ris r&init_cloudzero_settingsrYs0""&6&B&BB.AAGGH   %OO!//%%   !!"OP$-Zclmm   ""#NPQRHLM   s<AC/B 7B8%B CB C)(CCCz/cloudzero/dry-runcK|jtjk7r'tddtj j i ddlm}|}|j|jd{}tjdtd d ||r|jd  Sd S7@#t$r-}tj d |tddd|id}~wwxYww)aO Perform a dry run export using the CloudZero logger. This endpoint uses the CloudZero logger to perform a dry run export, which returns the data that would be exported without actually sending it to CloudZero. Parameters: - limit: Optional limit on number of records to process (default: 10000) Returns: - usage_data: Sample of the raw usage data (first 50 records) - cbf_data: CloudZero CBF formatted data ready for export - summary: Statistics including total cost, tokens, and record counts Only admin users can perform CloudZero exports. rar:r;rCloudZeroLogger)limitNz/CloudZero dry run export completed successfullyz0CloudZero dry run export completed successfully.rmsummaryrorc dry_run_datarz-Error performing CloudZero dry run export: %sr9z,Failed to perform CloudZero dry run export: )rqrrrr r rerG(litellm.integrations.cloudzero.cloudzerordry_run_export_usage_datarr rtrrVrgr:)rkr_rloggerdry_run_resultris r&cloudzero_dry_run_exportrs6""&6&B&BB.AAGGH   L()&,&F&FW]]&F&[ [!!"ST&F'5CN&&y1   JN   !\  ""#RTUVKA3OP   sHAC0,B74B558B7-C0.B74C05B77 C-(C((C--C0z/cloudzero/exportcK|jtjk7r'tddtj j i td{}ddlm }||jd|jd|jd  }|j|j|j|j|j d{t!j"d t%d d ddS77*#t&$r-}t!j(d|tddd|id}~wwxYww)a| Perform an actual export using the CloudZero logger. This endpoint uses the CloudZero logger to export usage data to CloudZero AnyCost API. Parameters: - limit: Optional limit on number of records to export - operation: CloudZero operation type ("replace_hourly" or "sum", default: "replace_hourly") Only admin users can perform CloudZero exports. rar:r;Nrrr4r5r6r>)r operationstart_time_utc end_time_utcz'CloudZero export completed successfullyrmrz%Error performing CloudZero export: %sr9z$Failed to perform CloudZero export: )rqrrrr r rerGrZrrrVexport_usage_datarrrrr rtrrgr:)rkr_rXrrris r&cloudzero_exportrs@,""&6&B&BB.AAGGH  " 7 99 M(LL+",,7\\*-  &&--''"11 -- '    !!"KL&=   ):  ""#JANCA3GH   sOAD> DDBDD'DD>DD D;(D66D;;D>z/cloudzero/deletecK|jtjk7r'tddtj j i ddlm}|'tddtjj it|jdd i d{}|td dd it|jjdd i d{tj d t#d dS7f7(#t$r}|d}~wt$$r-}tj&d|tddd|id}~wwxYww)a[ Delete CloudZero settings from the database. This endpoint removes the CloudZero configuration (API key, connection ID, timezone) from the proxy database. Only the CloudZero settings entry will be deleted; other configuration values in the database will remain unchanged. Only admin users can delete CloudZero settings. rar:r;rr8Nr9r?r@rNrpzCloudZero settings not foundz'CloudZero settings deleted successfullyrmrnz%Error deleting CloudZero settings: %sz%Failed to delete CloudZero settings: )rqrrrr r rerGrEr/rFr3rRrr1deleter rtrrgr:)r_r/rWris r&delete_cloudzero_settingsrsf&""&6&B&BB.AAGGH  # <  !2!I!I!O!OP  )6m(D(O(O!56)P) #   #!?@ }-33::,PdAe:fff!!"KL$-V_hii!#  g   ""#JANDQCHI   s[AE ADC>?DD%D=E >DD E D  E(EEE )?rIcollections.abcrtypingrrrrfastapirr r litellm._loggingr 0litellm.litellm_core_utils.sensitive_data_maskerr litellm.proxy._typesr rr$litellm.proxy.auth.user_api_key_authr0litellm.proxy.common_utils.encrypt_decrypt_utilsrr/litellm.proxy.management_endpoints.common_utilsr&litellm.repositories.config_repositoryr%litellm.repositories.prisma_protocolsr'litellm.types.proxy.cloudzero_endpointsrrrrrrrErr__annotations__rr r3r.rLrZrVrjputryrzboolr|rrpostrrrrrr$r(r&rs #651PTTBQC>7 /050A(AEELAT4UE #3#s#c#L&R +,-(  )00A(B2 %2  2 j +,-(  )00A(BD $D %D  D P05%u4< ,d ,$8 +,-(  )00A(B) !) %)  ) X +,-*  )00A(B1 #1 %1  1 h +,-*  )00A(B8 #8 %8  8 v +,-( )00A(B6 %6  6 r(