o jEiH@s6ddlZddlZddlmZeeZGdddZdS)N) getLoggerc@s:eZdZdZdddZddZddZd d Zd d Zd S)XaiCachez^ A simple persistent JSON-based cache for XAI API responses to avoid redundant calls. xai_cache.jsoncCs||_||_dS)N) cache_path _load_cachecache)selfrr 0/var/www/agentarbitrage/keepa_deals/xai_cache.py__init__ szXaiCache.__init__c Cstj|jstdiSz)t|jd }t|}Wdn1s%wYtdt |d|WSt tj fyOt d|jdiYSw)z!Loads the cache from a JSON file.z7XAI cache file not found. Starting with an empty cache.rNzSuccessfully loaded z items from XAI cache.zCould not read or parse 'z '. Starting with an empty cache.) ospathexistsrloggerinfoopenjsonloadlenIOErrorJSONDecodeErrorwarning)rfrr r r rs  zXaiCache._load_cachec Csz$t|jd}tj|j|ddWdWdS1swYWdStyC}ztd|jd|WYd}~dSd}~ww)z+Saves the in-memory cache to the JSON file.w)indentNzCould not save XAI cache to 'z': )rrrdumprrrerror)rrer r r _save_caches&&zXaiCache._save_cachecCs<|j|}|rtd|d|Std|d|S)zu Retrieves a value from the cache. Returns the cached value or None if the key is not found. zXAI Cache HIT for key: ''zXAI Cache MISS for key: ')rgetrdebug)rkeyresultr r r r"%s z XaiCache.getcCs(td|d||j|<|dS)zJ Adds a key-value pair to the cache and saves it to disk. zXAI Cache SET for key: 'r!N)rr#rr )rr$valuer r r set1s  z XaiCache.setN)r) __name__ __module__ __qualname____doc__r rr r"r'r r r r rs  r)rr loggingrr(rrr r r r s