o ̿Si5 @sddlZddlZddlZddlmZddlmZmZmZmZm Z ddl m Z e e Zeddp2dZdefdd ZeZeed d Zeed d ZeeddZeeddZeeddZdededeeeffddZdedeefddZdedeeeeffddZdeeefdeeeefdefd d!Z deeefdeeeefde eeeefeeffd"d#Z!defd$d%Z"deeeffd&d'Z#dS)(N)datetime)AnyDictListOptionalTuple)get_openai_clientEARLY_MOVERS_GPT_MODEoffreturncCs.dD]}t|dp d}|r|SqdS)N)EARLY_MOVERS_GPT_MODEL CHAT_MODELOPENAI_CHAT_MODEL OPENAI_MODEL)osgetenvstrip)kvr>/var/www/html/Trade-python/app/v1/services/early_movers_gpt.py _pick_model s rEARLY_MOVERS_GPT_TEMPERATUREz0.0EARLY_MOVERS_GPT_MAX_TOKENS1200EARLY_MOVERS_GPT_TIMEOUT45EARLY_MOVERS_GPT_MAX_CALLS10EARLY_MOVERS_GPT_LIMIT_PER_SIDE25model max_tokenscCs2|pd}|drdt|iSdt|iS)Nrzgpt-5max_completion_tokensr#)rlower startswithint)r"r#mrrr_max_tokens_kwarg"s   r)textcCsL|sdS|}|dr/|d}|dkr/||dd}|d}|dkr/|d|}|d}|d}|dkrk|dksE||krk|d}|dkrk||krk|||d}zt|WStyjYdSw|dkr|d}|dkr||kr|||d}zt|WStyYdSwzt|WStyYdSw) Nz``` {[]})rr&findrfindjsonloads Exception)r*snls2end start_obj start_arrchunkrrr _extract_json)sF             r>itemc Cst|tsdS|dp d}|dpd}|r#|dvr%dS|d}zt|}Wn ty;d}Ynw|durBdStdtd|}|d}d }t|t r[t |}n|durtt | }|d vrnd }n|d vrtd }|d }|durt |nd} | dvr|dkrd} n |dkrd} nd} g} |d} t| t r| D]} t| t r| r| | q|d} | durt | nd}||t|t || | |dddS)Nsymbolrbias)BULLISHBEARISH confidenceggY@bias_okT)false0noF)true1yes setup_qualityBADOKAYGOODKrP7rOrN risk_flagsnotesir@rArDrErLrSrT) isinstancedictgetrupperfloatr6maxminboolstrr%listappend)r?r@rAconfconf_f bias_ok_rawrEr7setup_quality_rawrLrSrfxrTnotes_srrr_validate_itemRsd             rhmarket candidatesc Cs ddgdddidddgdd d d d dd idgdddddidddiddd}ddgd|id}g}|p:gD])}t|trd|dpHd}|dpSd}|rd|rd|||dq;||tt|plg|ddddddddd|d}tj |d d!S)"zoSingle-call GPT prompt for multiple candidates. Candidates must already be compact and features-only. arrayobjectrUtypestringrBrC)rmenumnumberrd)rmminimummaximumbooleanrM)rmitems)rmrequired propertiesrur@rrA)r@rA)countkeyszGEvaluate each candidate for early breakout/breakdown watchlist quality.zReturn ONLY ONE valid JSON OBJECT with a single key 'items' (no markdown). items must be a JSON array matching the schema contract. You MUST return exactly one item per input candidate (expected.count). Do not omit any symbol. Do not add extra symbols.zN0-100, higher means higher probability of follow-through in next 1-5 sessions.zQtrue if the provided bias is supported by the provided features; otherwise false.zMBAD/OKAY/GOOD based on structure + context; do not use any outside knowledge.z Use short upper snake-case tags.zUDo not infer raw OHLC/candlestick details unless provided as explicit features/flags.)taskoutputrDrErLrSstrict)rirjexpected instructions output_schemaF) ensure_ascii) rVrWrXrrYr`r'lenr4dumps) rirjitems_contractcontractcandidate_keyscr7bpayloadrrr _build_promptsT      rc s g}tdkr ggfSts|dg|fSztWn ty<}z|dt|g|fWYd}~Sd}~wwdttttfdtt ttffdd}d ttttfdt ttttfttfffd d }||}t |}id }t |pgg} | rS|t d t tkrS| d } | sq|| } | rtfdd| Drq|d 7}|| \} } | r|| | D]$}|dpd|dpdf}|d r|d r||<qg}| D]-}t|tsq|dpd|dpdf}||vr|vr||q|rGt|dkr,|D] }| |gq!nt d t|d}| |d|| ||d| rS|t d t tksg}|D]}|}|rf||qW|rt|t|kr|dt|dt|d|d|||fS)zZEvaluate candidates via GPT in a single call. Returns: (validated_items, errors) r zGPT model not configuredzOpenAI unavailable: Nrowsr cSsdg}|pgD])}t|tsq|dpd}|dpd}|r/|r/|||fq|S)Nr@rrA)rVrWrXrrYr`)rksrr7rrrr_candidate_key_lists  z4gpt_evaluate_candidates.._candidate_key_listbatchc sg}t|d}dddd|dg}d}z@zjjjd|ttddid tt}WntyGjjjd|ttd tt}Ynw|j d j j pPd }Wn tyt}z| d t|g|fWYd}~Sd}~wwt|}t|tr|d }t|tr|}t|trdD]} || }t|tr|}nqt|tr|dr|dr|g}t|ts|dddd} | d| d| g|fSg} |D] } t| }|r| |q| s| d| |fS)N)rirjsystemzYou are a disciplined swing trader and risk manager. You only use the provided structured data. Output ONLY valid JSON. No markdown. No commentary.)rolecontentuserrrm json_object)r"messages temperaturetimeoutresponse_format)r"rrrrzOpenAI call failed: ru)resultsr{datar@rAir+ z"GPT response not a JSON items listz raw_head=zGPT returned no valid itemsr)rchat completionscreaterrr)rr6choicesmessagerrr`r^r>rVrWrXr_replacerh) r local_errorspromptrrresperawrrheadoutit)clientrir"rr _call_onces~                z+gpt_evaluate_candidates.._call_oncerr-c3s|]}|vVqdS)Nr).0r)results_by_keyrr Msz*gpt_evaluate_candidates..r@rrAz GPT returned z of z itemsz gpt_calls=)r r r`rr6r^rrrrsetr_r[r'rpopallextendrXrrYrVrWr)rirjerrorsrrr wanted_keys wanted_setcallsqueuer batch_keysruerrsrrmissingrr(midrrr)rrir"rrgpt_evaluate_candidatess  *> K   , , (   rcCstdttS)Nr)r[r'r rrrrgpt_limit_per_side{srcCstttttttdS)N)moder"rr#rlimit_per_side generated_at)r r rrrrrutcnowrrrrgpt_metasr)$r4loggingrrtypingrrrrrapp.v1.services.gpt_enginer getLogger__name__loggerrrr%r r^rr rZrr'rrrr r)r>rhrrrrrrrrs.    )*?BD'