o ̿Si7@sDdZddlZddlZddlZddlmZmZddlmZddl m Z z ddl m Z dZ Wn ey9dZ dZ YnweeZed d Zed d Zed dZeeddZeeddZeeddZdedefddZdededeeeffddZdeeeffddZ de fdd Z!d!edeeeffd"d#Z"dS)$z GPT-specific configuration and helper functions for the ChatGPT-based trading engine. This module is responsible **only** for talking to OpenAI models and building prompts; it has no Zerodha or DB logic. N)DictAny) HTTPException)prepare_market_data_prompt)OpenAITFOPENAI_API_KEY OPENAI_MODELzgpt-4oOPENAI_MODEL_FALLBACKz gpt-4o-miniOPENAI_TEMPERATUREz0.1 MAX_TOKENS1000REQUEST_TIMEOUT30excreturncCsHt|pd}|s dSd|vo#d|vp#d|vp#d|vp#d|vp#d|vS) NrFmodelz not foundzdoes not existz no such modelmodel_not_foundzunsupported model)strlower)rmsgr8/var/www/html/Trade-python/app/v1/services/gpt_engine.py_looks_like_model_error#srr max_tokenscCs2|pd}|drdt|iSdt|iS)zReturn the correct completion-length kwarg for the chosen model. Some newer models (e.g., GPT-5.*) do not accept `max_tokens` and instead require `max_completion_tokens`. rzgpt-5max_completion_tokensr)stripr startswithint)rrmrrr_max_tokens_kwarg3s   r c CstsdddStsdddSz"ttd}|jjjdtdddgd ttd d d tdWStyL}zdd t |dWYd}~Sd}~ww)zValidate OpenAI configuration and basic connectivity. This is used by the `/health` endpoint to report whether GPT analysis is available. FzOpenAI package not installed) configuredmessagezOPENAI_API_KEY not setapi_keyuserTestrolecontent)rmessages TzOpenAI configured with model zOpenAI test failed: Nr) OPENAI_AVAILABLErrchat completionscreater r Exceptionr)clienterrrvalidate_openai_setup@s&      r3cCs*tstdddtstdddttdS)zCReturn an authenticated OpenAI client instance or raise HTTP error.izOpenAI package not available) status_codedetailzOpenAI API key not configuredr#)r,rrrrrrrget_openai_client\s    r6promptc Csz,tdtdtt}dddd|dg}ttt}z|jjj dZt|td|dt i}Wn]t y}zQt |}d |vr\d |vr\t d t|jjj t|tttt d }n*trttkrt|rt}t d t||jjj dZ||tdt|tdt i}nWYd}~nd}~wwzt|dd}|rtdt|ddt|ddt|ddWn t yYnw|jdjj} td| dddz!| d} | dd} td| d| | dkr| | kr| | | } td| t| } tdtdt| d | vrd!| d <d"| vr0d#| vr0| d#| d"<d#| vrAd"| vrA| d"| d#<d"| vrJd$| d"<d#| vrW| d"d$| d#<d%| vrad&g| d%<zt | d pkd'd!krsd$| d"<d$| d#<| d(}z |durt|nd)}Wn t yd)}Ynwttd*t dt|| d(<| d+}t!|t"rd,|vr|d,n|d-}d.|vr|d.n|d/}z|durt|nd}|durt|nd}Wnt yd0\}}Ynw|dur|dks |dur|dkrd| d+<n |d1vrd| d+<d2D](}z| |}|dur+Wqt|dkr6d| |<Wqt yCYqw| d3}t!|trjz d4d5|D}Wn t ydg}Ynw|| d3<n |d6vrsg| d3<Wn t y~Ynw| d7}z@|durt|}n4d(| vr| d(durt| d(}n t | d#| d"d$}|d8krd9}n |d:krd;}nd<}Wn t yd=}Ynw|durt d)td>|}|| d7<| WWSt#d?| t$d@tj%y-}z.t#dA|t#dB| ddCd!d$dDdEt |gdF| dddGWYd}~WSd}~wwt y}zt|dHdpFtt|dIddHd}t |}d}z=t|dJdpatt|dIddKdLdM}t!|t"r|dNpoi}t!|t"r|dOp|dP}|dQp|}Wn t yYnw|dRkr|dSksdS|vrt#dTd!d$dUdVg|dRdSdWWYd}~St&dXd!d$dY|g|||dWWYd}~Sd}~ww)[z@Call ChatGPT for structured JSON analysis with detailed logging.u🤖 CHATGPT API CALL - Model: z, Temp: systema?You are an expert institutional intraday trader and risk manager. You infer the best intraday decision purely from provided data. Priority: capital protection, trade only when edge is clear, precision over frequency. If no clear edge exists, you MUST choose HOLD. Output ONLY valid JSON (no code blocks, no extra text).r'r%)rr* temperaturetimeoutrrzTRetrying OpenAI call with max_completion_tokens due to parameter mismatch (model=%s))rr*r9rr:z9OpenAI model failed (%s). Retrying with fallback model=%sNusageu1📦 GPT usage | prompt=%s completion=%s total=%s prompt_tokenscompletion_tokens total_tokensru-📥 CHATGPT RAW RESPONSE (first 500 chars): iz...{}u🔍 JSON EXTRACTION - Start: z, End: u📋 EXTRACTED JSON STRING: u✅ JSON PARSED SUCCESSFULLYu🎯 PARSED ANALYSIS KEYS: decisionHOLD conviction confidenceLOW rationalezNo analysis providedrscore' entry_zonelowrhighupper)NN)rrI) entry_price stop_lossrisk_reward_ratiotargetscSs(g|]}|durt|dkrt|qS)Nr)float).0xrrr s(z)call_chatgpt_analysis..)rrINdecision_probabilityHIGHgT@MEDIUMgN@gD@gI@gY@u(❌ NO JSON FOUND in response. Content: zNo JSON found in responseu❌ JSON DECODE ERROR: u🔍 FAILED JSON STRING: iz.Analysis failed - invalid JSON response formatzError: zJSON parse error)rBrErGerror raw_responser4responsebodyjsoncSsdS)Nrrrrr<sz'call_chatgpt_analysis..rZcodetyper"iinsufficient_quotaz*OpenAI quota exceeded (insufficient_quota)z6GPT analysis temporarily unavailable (quota exceeded).z?Using non-GPT fallback is recommended until billing is updated.)rBrErGrZ error_code error_typezChatGPT analysis failedzAnalysis failed: r)'loggerinfor r r6r r r-r.r/rr0rwarningrr rgetattrchoicesr"r)rfindrfindr^loadslistkeysgetrNrSminmax isinstancedictrZ ValueErrorJSONDecodeError exception) r7r1r* token_kwargsr\r2rfbr;r) start_idxend_idxjson_stranalysisrHscore_fezrLrMlow_fhigh_fkvtcleanedprobconfr4r"rdr]errrrrcall_chatgpt_analysisgs      "                  (               "(     r)#__doc__osr^loggingtypingrrfastapirapp.v1.services.gpt_promptsropenairr, ImportError getLogger__name__regetenvrr r rSr rr rr0boolrrr r3r6rrrrrs4