j9"BUdZddlmZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z ddlmZej eZd Zed hZd d d d Zded< ddZddZddZd d ddZd d ddZy)z author: Yagnik Poshiya organization: Webbrains Technologies Private Limited Optional LlamaIndex SQL path (kept separate from custom 3-LLM SQL orchestration). ) annotationsN)Any) quote_plus)verify_sql_result)Settings)LLMUsagemysqlz mysql+pymysqlzpostgresql+psycopg2)r postgresqlpostgreszdict[str, str]_SQLALCHEMY_DRIVERSc tj|td}|dt|dt|d|d|d| S)Nr z://:@/)r getr)dialectuserpasswordhostportdatabasedrivers L/var/www/html/agent-engine/apps/agents/rag-agent/src/agent/sql_llamaindex.py_build_connection_urir!sS! $ $W.A'.J KFXSD)*!Jx,@+A4&$qQYPZ [[c|dvryy)N)r r PostgreSQLMySQLrs r_dialect_labelr"(s,, rct|dd}t|trZdD]U}|j|}t|ts%|j s6|j j dcSt t|d|xsdj }tjd|}|r.|jdj j dSy)Nmetadata) sql_querysqlquery;responsez(?is)\bselect\b.+r) getattr isinstancedictrstrstriprstripresearchgroup)respr$keyvaltextms r%_extract_sql_from_llamaindex_responser9.stZ.H(D!0 /C,,s#C#s# yy{))#.. / wtZ.4" 5 ; ; =D &-Awwqz!((-- rr!c t|jxsdjj}t |} ddlm}ddlm} m } ddl m }  ddlm} | d | _|d k(rRt|j$xsdj}|s td  dd lm}||j*|d}n|dk(rQt|j,xsdj}|s td ddlm}||j*|d}n|dk(rQt|j2xsdj}|s td ddlm}||j*|d}nc|dk(rSt|j8xsdj}|s td ddlm}||j*|ddd}n td||}| |t?| }| |t?||d!"}d#|d$d%jAt?|d&|d'|d( }|jC|}tE|S#t$r} td| d} ~ wwxYw#t"$r d| _Y wxYw#t$r} td | d} ~ wwxYw#t$r} td| d} ~ wwxYw#t$r} td| d} ~ wwxYw#t$r} td| d} ~ wwxYw))Nr*r) create_engine) SQLDatabaser)NLSQLTableQueryEnginezULlamaIndex dependencies are not installed. Install llama-index + sqlalchemy packages.) MockEmbedding) embed_dimopenaizBOPENAI_LLM_API_KEY is required for LLAMAINDEX_SQL_PROVIDER=openai.)OpenAIz,Missing dependency: llama-index-llms-openai.g)modelapi_key temperaturegooglezBGOOGLE_LLM_API_KEY is required for LLAMAINDEX_SQL_PROVIDER=google.) GoogleGenAIz2Missing dependency: llama-index-llms-google-genai. anthropiczHANTHROPIC_LLM_API_KEY is required for LLAMAINDEX_SQL_PROVIDER=anthropic.) Anthropicz/Missing dependency: llama-index-llms-anthropic. openrouterzJOPENROUTER_LLM_API_KEY is required for LLAMAINDEX_SQL_PROVIDER=openrouter.) OpenAILikez1Missing dependency: llama-index-llms-openai-like.zhttps://openrouter.ai/api/v1T)rCrDapi_base is_chat_modelrEzVUnsupported LLAMAINDEX_SQL_PROVIDER. Supported: google, openai, anthropic, openrouter.)engineinclude_tablesF) sql_databasetablesllmsynthesize_responsezGenerate exactly one z5 SELECT query. Use ONLY tables from this allow-list: , z Schema details: z Visitor question:  )#r.llamaindex_sql_providerr/lowerr" sqlalchemyr;llama_index.corer<rllama_index.core.query_enginer= Exception ValueErrorllama_index.core.embeddingsr> embed_model ImportErroropenai_llm_api_keyllama_index.llms.openairBllamaindex_sql_modelgoogle_llm_api_keyllama_index.llms.google_genairGanthropic_llm_api_keyllama_index.llms.anthropicrIopenrouter_llm_api_keyllama_index.llms.openai_likerKsortedjoinr'r9)settings user_message schema_detailallowconnection_urirproviderlabelr;r< LlamaSettingsr=er>rDrBrRrGrIrKrNsql_db query_enginepromptr4s r_llamaindex_generate_sqlrw>s+8339r:@@BHHJH 7 #E,KG)=$1A$> ! 8h117R8>>@ab b T 6888'WZ[ X h117R8>>@ab b Z A = =w\_` [ h44:;AACgh h W <h;;WZ]^ \ !h55;<BBDij j Y ?//3   d  > *F ve} EF(e} ! L'ug.''+yy'?&@A F   f %D 0 66_  9   )$( !) TKLRS S T ZQRXY Y Z WNOUV V W YPQWX X YsII!%I8<JJ2(K I II!I54I58 J J  J J/ J**J/2 K ; KK  K) K$$K)cl|tvr.td|ddjttdddlm}d}d} d} t } t|jd xsd} t|jd xsd} t|jd xsd}t|jd xsd }t|| | |||}ttD]}t||||||} | stjd||ntj!d|| | |||| | || |t|jdxsdt|jdxsdd \}}|r t#j$||ddt}nd}t'|||| |\}}|jddk(rn6|jd xs|jd!xs|} | s|d"| }|| | fS#t$r(}tjd|||d|d}Yd}~d}~wwxYw)#z LlamaIndex SQL pipeline for live databases (dialect-aware). Returns (db_answer, sql_used, usage). Usage is currently zero-filled since provider usage extraction differs by backend. z6LlamaIndex SQL pipeline does not yet support dialect 'z'. Supported: rT.r)run_select_mysqlr*Nusernamerrri )rlrmrnrorzGDB pipeline source=llamaindex dialect=%s generated empty SQL attempt=%dzDDB pipeline source=llamaindex dialect=%s generated_sql=%s attempt=%dssl_moderequired ssl_ca_pem) rrrrrr&allowed_tablesr|r~max_rows)columnsrowsF) ensure_asciidefaultz[Query returned 0 rows]zADB pipeline source=llamaindex dialect=%s exec_error=%s attempt=%dz [SQL error: ])rlrmr& query_resultverdictpassfeedbackreasonz Previous issue: )_SUPPORTED_DIALECTSr\rjriagent.mysql_runnerrzrr.rintrrangeMAX_LLAMAINDEX_ATTEMPTSrwloggerwarninginfojsondumpsr)rkrlrmdb_labellive_ctxrnrrz db_answersql_used prev_errorusagerrrrroattemptcolsrrsr_s rrun_llamaindex_live_sqlrsh))DWIN))F+>$?@A D  4IH!J JE x||J'-2 .D8<< +1r2H x||F#)r *D x||F#+t ,D*7D(D$PXYN013M+ %')   NNdfmov w  Z\cemovw ,)!!$X\\*5CDx||L9?R@ JD$ JJ $d3!& 6 ' %'"   ;;y !V + [[,R H0ER *^+? |LLg3Mj h %%% , NN^`gijls t&qc+I ,s-A1H H3 H..H3)rr.rr.rr.rr.rrrr.returnr.)rr.rr.)r4rrr.)rkrrlr.rmr.rnset[str]ror.rr.rr.)rkrrlr.rmr.rr.rzdict[str, Any]rnrrr.rztuple[str, str, LLMUsage])__doc__ __future__rrloggingr1typingr urllib.parseragent.sql_verifierrconfig.settingsrllm.baser getLogger__name__rr frozensetrr __annotations__rr"r9rwrr rrrsm#  #0$   8 $ *'%'^\ \\'*\25\=@\LO\\ ._7_7_7 _7  _7  _7_7 _7TW&W&W& W&  W&  W& W&W&W&r