o jEiS @sBddlZddlZddlmZddlmZeeZGdddZdS)N)date) getLoggerc@sBeZdZdZdddZddZdd Zd d Zd d ZddZ dS)XaiTokenManagerzb Manages the daily call quota for the XAI API to prevent rate-limiting and control costs. settings.jsonxai_token_state.jsoncCs(||_|||_||_|dS)N) state_path_load_daily_limit daily_limit _load_statestate_check_and_reset_daily_count)self settings_pathrr8/var/www/agentarbitrage/keepa_deals/xai_token_manager.py__init__ s   zXaiTokenManager.__init__c Csz,t|d }t|}Wdn1swY|dd}td|d|WSttjfyBtd|dYdSw) zALoads the max_xai_calls_per_day from the main settings.json file.rNmax_xai_calls_per_dayiz+XAI Token Manager: Daily call limit set to .zCould not load 'z%'. Using default daily limit of 1000.) openjsonloadgetloggerinfoFileNotFoundErrorJSONDecodeErrorwarning)r rfsettingslimitrrrrs   z!XaiTokenManager._load_daily_limitc Cszz&t|jd }t|}Wdn1swYtd||WSttjfy<tddddYSw)z"Loads the usage state from a file.rNzLoaded XAI token state: zDXAI token state file not found or invalid. Initializing fresh state.z 1970-01-01r)last_reset_date calls_today) rrrrrrrrr)r rr rrrr s  zXaiTokenManager._load_statec Csz$t|jd}tj|j|ddWdWdS1swYWdStyC}ztd|jd|WYd}~dSd}~ww)z(Saves the current usage state to a file.w)indentNz#Could not save XAI token state to 'z': )rrrdumpr IOErrorrerror)r rerrr _save_state)s&&zXaiTokenManager._save_statecCsXtt}|jd|kr*td|jdd||jd<d|jd<|dSdS)z4Resets the daily call count if the date has changed.r!z6New day detected. Resetting XAI daily call count from r"z to 0.rN)strrtodayr rrrr*)r today_strrrrr 1s    z,XaiTokenManager._check_and_reset_daily_countcCsl||jd|jkr*|jdd7<|td|jdd|jdStd|jddS) z Checks if an XAI API call can be made. If yes, increments the count. Returns True if the call is permitted, False otherwise. r"z!XAI call permitted. Usage today: /TzXAI daily call limit reached (z8). Call denied. No further XAI calls will be made today.F)r r r r*rrr)r rrrrequest_permission:s z"XaiTokenManager.request_permissionN)rr) __name__ __module__ __qualname____doc__rrr r*r r0rrrrrs    r) rosdatetimerloggingrr1rrrrrrs