o i+(@sddlZddlZddlZddlZddlZddlZddlmZm Z ddl m Z ddl m Z e Z e Z eeZedddZddZdd d Zd d ZddZdS)N)datetime timedelta)XaiTokenManager)XaiCacheicCs0tj|ddd}tj|dtdd|d<|S)zTConverts a DataFrame's timestamp column from Keepa Time Minutes to datetime objects. timestampcoerce)errorsm)unitoriginzdatetime64[ns])pd to_numeric to_datetime KEEPA_EPOCHastype)dfnumeric_timestampsr:/var/www/agentarbitrage/keepa_deals/xai_sales_inference.py_convert_ktm_to_datetimesrdc sB|d|dg}t|trt|dkrdS|d}|d}|d}|r(|s*dSfdd }||d }||d }||d } |dusG|durIdStt|d } ||d| kd} ||d| kd} | | g} | dur}| | d| kd}| |t | }| rt dd| Drt |g}n%t |  jdd}|jd|krt t |g|g jdd}td|i}z&tj||ddd}tj||ddd}| durtj|| ddd}nd|d <Wnty}ztdd|WYd}~dSd}~ww|jd d gd}|jrdSg}|d|dd}d}|D]s\}}|dd}t|d sAt|d nd}t|d sUd|d dd nd}t|d sl|d dkrlt|d nd}|||f}||kr||d!|d!|d!||}|d"7}|d#kr|d$nq(d%|S)&z Extracts relevant history (Rank, Used Price, Used Offer Count) for the last n days and formats it as a compact time-series text for the LLM. Ensures initial state (at start of window) is captured. asincsv N c s|rt|dkr dSz%t|}t|ddkr|dd}tj|ddd|gd}t|WStyM}zt dd|WYd}~dSd}~ww)Nrrr)columnsz#Error converting history to DF for : ) lennparrayr DataFramereshaper Exceptionloggerwarning)histcol_namearrrerrrto_df1s   z%format_history_for_xai..to_dfRankPriceOffersdaysrcss|]}|jVqdS)N)empty).0tsrrr Wsz)format_history_for_xai..T)droprbackward)on directionzN/AzError during merge_asof for r )subsetz!Time | Rank | Used Price | Offersz---|---|---|---z%Y-%m-%d %H:%M-$rz.2fz | rz... (truncated) ... )get isinstancelistr!rnowrappendr TimestampallSeriesconcat sort_valuesdrop_duplicates reset_indexilocr$ merge_asofr&r'errordropnar4iterrowsstrftimeisnaintjoin)productr3csv_data rank_historyused_price_historyused_offer_count_historyr.df_rankdf_price df_offers start_datets_rankts_pricetimestamps_list ts_offers start_date_ts timestamps df_mergedr,lines last_row_valscount_rowts_strrankpriceoffers current_valsrr-rformat_history_for_xais        "   (.      rpc Cstd}|stddS|dd}|dg}|r"|ddnd}ts3td |d dSd |d |d |d}d|dgdddd}d|dd}zutjddd} | j d||d} | | dddd} z0d | vr| d d! d"d } n d"| vr| d"d! } t | } | WWdWSt jytd#| YWdWdSw1swYWdSty} ztd$| WYd} ~ dSd} ~ ww)%z@ Queries XAI to identify sales in the provided history. XAI_TOKENzNo XAI_TOKEN found.NtitleUnknown categoryTreernamez=XAI daily limit reached. Cannot perform sales inference for 'z'.z You are an expert Amazon Arbitrage Analyst. Your goal is to identify "Inferred Sales" from the following historical data for the product: "z" (a). Logic for a sale: - Typically, a drop in "Offers" (Used Offer Count) suggests a unit was sold. - However, offers can also be removed by sellers. - A corresponding drop (improvement) in "Rank" (Sales Rank gets smaller) shortly after an offer drop is a strong confirmation of a sale. - Sometimes Rank drops without an offer drop (if stock depth > 1). - Sometimes Offers drop without a rank drop (delisted). - Sometimes Rank drops and recovers between data points (sparse data). Task: Analyze the provided time-series data. Identify the most likely "Sold Price" for any VALID sales you detect in this period. Ignore price drops that look like repricing wars unless accompanied by a sale signal. Return a JSON object with: 1. "sales_found": count of confirmed sales. 2. "events": list of objects, each with {"date": "YYYY-MM-DD", "price": float (number only)}. 3. "estimated_market_price": your best estimate of the current true market value for a Used copy based on these sales. 4. "confidence": "High", "Medium", or "Low". 5. "reasoning": brief explanation. Data (Last ~100 days): z user)rolecontentzgrok-4-fast-reasoningg?i)messagesmodel temperature max_tokenszBearer zapplication/json) Authorizationz Content-TypegN@)timeoutz$https://api.x.ai/v1/chat/completions)headersjsonchoicesrmessagerxz```jsonrz```z&Failed to parse XAI response as JSON: z"XAI Error during sales inference: )osgetenvr'r(rAxai_token_managerrequest_permissionhttpxClientpostraise_for_statusrsplitstriploadsJSONDecodeErrorrOr&) history_textrVapi_keyrrcat_treecategorypromptpayloadrclientresponserxresultr,rrrquery_xai_sales_inferences^      rc Cs8|di}|ddgdd}|r|dkrdSt|dd}|s#dSt||}|r|d d vr|d g}g}|D]J}z+|d }|d } |rg| rgt|dtdd} tt| d} || | dWq;t y} zt d|d| WYd} ~ q;d} ~ ww|rt dt |d|d|SdS)zg Wrapper to call XAI inference if eligible. Returns a list of sale events (dicts) or None. statscurrentNrirr2 confidence)HighMediumeventsdatermz%Y-%m-%dr)hours)event_timestampinferred_sale_price_centszError parsing XAI event r z XAI Rescued z sales for ASIN r)rArprrstrptimerrTfloatrEr&r'r(infor!) rVr current_rankrrrconfirmed_saleseventdate_strrmdt price_centsr,rrrinfer_sales_with_xais>       " r)r)loggingpandasr numpyr"rrrrrrr xai_cacher getLogger__name__r'rrrprrrrrrs"     t M