o TLNiY@s:ddlmZmZmZmZddlmZmZddlZddlZddl Z ddl Z ddl m Z ddl mZddlmZddlmZzddlmZWn eyOdZYnweeZeejed d Zed d Zeed dZddZ de!fddZ"hdZ#dddddddddZ$de!de!fddZ%dee!effdd Z&dee!fd!d"Z'deee!effd#d$Z(d%e!deefd&d'Z)dSd)ed*e*dee!effd+d,Z+dTd)ed/ed%e!d0ee!d1ed2edee!effd3d4Z,d)ed%e!dee!effd5d6Z-d7e!deee!effd8d9Z.d:e!deee!effd;d<Z/d=ee!efdee!effd>d?Z0dUd@ee!efdAee!dee!effdBdCZ1d)edDee!efd%e!dEee!efdee!eff dFdGZ2dVd)edJe!dKedLee!dAee!dee!eff dMdNZ3d)edDee!efdEee!efdee!effdOdPZ4dEee!efdee!effdQdRZ5dS)W)ListDictAnyOptional)datetime timedeltaN)ObjectId) HTTPException)database) ZerodhaClient)OpenAIOPENAI_API_KEY OPENAI_MODELz gpt-4o-miniLLM_TIMEOUT_SEC20cCs&tdurtdtstdttdS)Nz= 1.0.0.z/OPENAI_API_KEY environment variable is not set.)api_key)r RuntimeErrorr rr6/var/www/html/Trade-python/app/v1/services/tradeGPT.py_get_openai_clients  r method_namec Os zRt||rt|||i|WSt|dd}|r)t||r)t|||i|WS|dd}t||r?t|||i|WS|rQt||rQt|||i|WSWn ty_}zd}~wwg}z ddt|D}Wn tyxg}Ynw|rd|ddnd }td |d |) z Defensive call helper: tries method on wrapper, then on zerodha.kite. Raises AttributeError with diagnostics if not found. kiteN_rcSsg|] }|ds|qS)r) startswith).0arrr >z!_call_zerodha..z, (z zZerodha client missing method 'z'. Sample attrs: )hasattrgetattrreplace ExceptiondirjoinAttributeError) zerodharargskwargsralte sample_attrssamplerrr _call_zerodha%s0     r.> dayweekmonthminute3minute5minute10minute15minute30minute60minuter2r4r6r7r8)1minute1min5min15min30min60min1hourhourintervalreturncCs~|stdt|}t||}|tvr|Std|}|r8t | ddvr8t | ddkr6dS|Std|)NzEmpty intervalz ^(\d+)minute$)rC <r2zUnsupported interval: ) ValueErrorstrstriplower_INTERVAL_ALIASESget_VALID_KITE_INTERVALSrematchintgroup)rAivmrrr_normalize_interval_for_kiteHs  rWc Cszddlm}||}|\}}dd}||}||}Wnty0tdg}g}Ynwdt||tt ||d}|dj d did |id d t |}d |d <|d ||S)z Try to populate db['movers'] using project scrapers if available. Minimal: if scrapers unavailable, returns empty lists. r)TopMoversFetchercSsb|sgSg}|D]&}t|tr$dD]}||r"|||nqqt|tr.||q|S)N)symbol nse_symbol tradingsymboltickername) isinstancedictrOappendrK)arroutitemkrrr_norm_s    z%refresh_movers_service.._normz fetch candles & quote -> save. Minimal mapping: does not perform fuzzy name mapping here. T)rNz(Instrument token not found for symbol %sz&Instrument token not found for symbol r)rYrp created_at snapshotsr) rrr#rterrorr rrrvrz inserted_id)r{r'rYrrprresrrrcreate_snapshot_services     r snapshot_idc CsPz|ddt|i}|WSty'}z td|WYd}~dSd}~ww)Nrrzget_snapshot_service error: %s)rrr#rtr)r{rrr+rrrget_snapshot_services rtextc Cs|sdSzt|WStyYnwd}d}t|D]?\}}|dkr/|dur*|}|d7}q|dkr[|d8}|dkr[|dur[|||d}zt|WStyZd}YqwqdS)zD Extract first JSON object from text. Returns dict or None. Nr{rC})jsonloadsr# enumerate)rstartdepthrch candidaterrr_extract_json_from_text$s.     rrespcCst|ts dddgdSt|dd}|dvr!dddgdS|d d}t|tr7|}|d vr6d}nd}||d <||d<d |vrJd g|d <|S) zn Minimal validation: ensure decision exists and is one of BUY/SELL/HOLD. Keep other fields as-is. HOLDLOWzInvalid LLM response formatdecision confidence rationalerr)BUYSELLrzInvalid decision value from LLMr)HIGHMEDIUMrrzNo rationale provided by model)r^r_rKrOr)rrconfrrr_validate_llm_response=s     rrpquestionc CsZtstddddgdSd}||pdd}zqt}|jjjtd |d d tj |t d d gd dt d}d}z |j dj d}Wn'tygzt|j dj ddpUt |}Wn tydt |}YnwYnwt|}|std|ptddddddgdWSt|}|WSty} ztd| dddt | gdWYd} ~ Sd} ~ ww)z Send snapshot JSON to ChatGPT, expect a single JSON object back (model must return JSON). Returns validated dict. No local algorithms applied. u2OPENAI_API_KEY not set — returning fallback HOLDrrzLLM not configuredruYou are an intraday trading assistant. You will be given a JSON payload containing a 'snapshot' key with raw OHLCV arrays and a 'quote' key. Analyze only the provided data and return EXACTLY one JSON object with keys: decision (BUY|SELL|HOLD), confidence (HIGH|MEDIUM|LOW), optional entry:{low,high}, optional stop_loss, optional targets[], rationale[] . Return only JSON — no extra commentary. If uncertain, return decision=HOLD and explain why in rationale.r)rprsystem)rolecontentuser)defaulti )modelmessages temperature max_tokenstimeoutrrz2LLM did not return parseable JSON. Raw excerpt: %sNrzLLM did not return JSONzLLM call failed: %sz LLM error: )r rtrrchat completionscreaterrdumpsrKrchoicesmessager#r!rrr) rpr system_prompt user_payloadclientrrparsed validatedr+rrrcall_chatgpt_analyze_serviceUsN       $rrpayloadc Cs8|d}|dp d}d}|rt||}|sBzt|||}Wn ty)tyA}z td||tdddd}~ww|d pHi} t| d pS| d } | set d |d ddgd} nt | |} t | t rsd| vrzd ddgd} |d|t |dt |d|| td} |d| | S)z Create (or load) snapshot, send to ChatGPT, persist chat record, return LLM response. No local heuristics or fallback algorithms applied. rrrNz#snapshot creation failed for %s: %srzSnapshot creation failedrrprrz*Snapshot for %s has no data; skipping LLM.rrzSnapshot emptyrrzInvalid LLM responserYr)rYruser_idr llm_responserchats)rOrLrrr r#rtrboolrrr^r_rKrrvrz) r{r'rrYrrr snapshot_docr+rphas_datarchat_docrrrchat_symbol_services@         rri2moverlimitrc Csg}z|dddipi}|dkr|dn|dp#|dp#g}dd|p*gD}|s:td |d giWS|d |D]} zt||| } | d pOi} t| |d } WnWty} z%td| tt | drl| j n| dddt| gd} d } WYd } ~ n+d } ~ wt y}zt d| |dddt|gd} d } WYd }~nd }~ww| |pd || rt| dnd | t d}z|d|}t|j}Wnt yt d| d }Ynw| r| d idinipi}|dd }|| | d| d|| | rt| dnd d}||q@d |iWSt y?}zt d|gt|d WYd }~Sd }~ww)!z For each mover symbol, create snapshot, call ChatGPT, persist the LLM response. Returns list of LLM-decisions for frontend. rlrgrfrirjrkcSsg|] }t|tr|qSr)r^rK)rrrrrrrz,get_live_signals_service..zNo movers found for mover=%sresultsNrp)rz)create_snapshot_service failed for %s: %srrrzSnapshot failed: rzError processing %s: %szError: r)rYr mover_typerrrstreamz#Failed to persist stream doc for %sr last_pricerr)idrYrrltpllmrz#get_live_signals_service failed: %s)r r)rrOrtrurrr rrKr rr#rrrvrzrr`)r{r'r r rrr  movers_doc candidates raw_symbolrrp llm_decisionher+ stream_docinserted stream_idrrrowrrrget_live_signals_servicesn*   "      rc Cs|ds tdddt|d}|dd|ipi}|dd s)td d d|d }t|d dp6d}|d}|dd} ttdd} |rQt|nd|} |ra| | kratdddt||} | sntddd|dd|d| d} | dkr|rt|| d<zt |d| }Wnt y}zt d|tdt|dd}~ww||| |td }|d!|||d"S)#Nuser_confirmationizuser_confirmation requiredrr user_settingsr allow_shortsFiz User not permitted to short-sellrYquantityrprice order_typeMARKETMAX_NOTIONAL_PER_ORDER200000rzOrder exceeds notional limitrzInstrument token not foundrrMIS)r[rtransaction_typerproductr!LIMIT place_orderzOrder placement failed: %sr)rrY order_payloadzerodha_responserorders)orderr+)rOr rKrrSfloatosgetenvrrr.r#rtrrrvrz)r{r'rrrrrYqtyr r! MAX_NOTIONALnotionalrr*placedr+ order_docrrrplace_short_order_servicesH              r6cCsXt}|d|td|d}|dvr(d|td}|d|dd iS) Nzerodha_postbacks)r received_atstatus)REJECTED CANCELLEDCOMPLETETRIGGER_PENDINGorder_postback)rgrralertsrT)r get_mongo_dbrzrrvrO)rr{r9alertrrrkite_postback_services rB)F)NrCr)N)rirNN)6typingrrrrrrloggingr/rrQbsonrfastapir app.dbr app.v1.services.zerodha.clientr openair r# getLogger__name__rtsetLevelINFOr0r rrSrrrKr.rPrNrWrrrrrrrrrrrrrrr6rBrrrrsP          - 4E ",624+2@&'