j/#VUdZddlmZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z ddlmZej eZd Zed hZd d d d Zded< ddZddZddZddZd ddd ddZd d ddZy)aB author: Yagnik Poshiya organization: Webbrains Technologies Private Limited Optional LangChain SQL path (kept separate from custom 3-LLM and LlamaIndex pipelines). Uses LangChain's SQL Database Chain to convert visitor questions into SELECT queries. Dialect-aware: supports MySQL now; architecture ready for PostgreSQL. ) 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 K/var/www/html/agent-engine/apps/agents/rag-agent/src/agent/sql_langchain.py_build_connection_urir#sS! $ $W.A'.J KFXSD)*!Jx,@+A4&$qQYPZ [[c|dvryy)z,Human-readable dialect name for LLM prompts.)r r PostgreSQLMySQLrs r_dialect_labelr"*s,, rct|jxsdjj}t|jxsdj}|dk(rFt|j xsdj}|s t d ddlm}|||dS|d k(rFt|jxsdj}|s t d  dd l m }|||d S|dk(rFt|jxsdj}|s t d ddlm}|||dS|dk(rGt|j xsdj}|s t d dd l m }|||ddSt d#t$r}t d|d}~wwxYw#t$r}t d |d}~wwxYw#t$r}t d|d}~wwxYw#t$r}t d |d}~wwxYw)NgooglezAGOOGLE_LLM_API_KEY is required for LANGCHAIN_SQL_PROVIDER=google.r)ChatGoogleGenerativeAIz+Missing dependency: langchain-google-genai.g)modelgoogle_api_key temperatureopenaizAOPENAI_LLM_API_KEY is required for LANGCHAIN_SQL_PROVIDER=openai.) ChatOpenAIz%Missing dependency: langchain-openai.)r'api_keyr) anthropiczGANTHROPIC_LLM_API_KEY is required for LANGCHAIN_SQL_PROVIDER=anthropic.) ChatAnthropicz(Missing dependency: langchain-anthropic.)r'anthropic_api_keyr) openrouterzIOPENROUTER_LLM_API_KEY is required for LANGCHAIN_SQL_PROVIDER=openrouter.zhttps://openrouter.ai/api/v1)r'r,base_urlr)zUUnsupported LANGCHAIN_SQL_PROVIDER. Supported: google, openai, anthropic, openrouter.)strlangchain_sql_providerstriplowerlangchain_sql_modelgoogle_llm_api_key ValueErrorlangchain_google_genair& ImportErroropenai_llm_api_keylangchain_openair+anthropic_llm_api_keylangchain_anthropicr.openrouter_llm_api_key)settingsproviderr'r,r&er+r.s r_build_langchain_llmrC1s(8228b9??AGGIH ,,2 3 9 9 ;E8h117R8>>@`a a S E&E'WZ[[8h117R8>>@`a a M 3wCHH;h44:;AACfg g P 95GQTUU<h55;<BBDhi i M 33    _ O SJKQR R S MDE1 L M PGHa O P MDE1 L Ms`F"F2-G8G, F/ F**F/2 G ; GG  G) G$$G), H5 HHc^|j}tjd|tjtjz}|r|j dj}tjd|}|r.|j djj dSy)Nz```(?:sql)?\s*\n?(.*?)```z(?is)\bSELECT\b.+r;r$)r4researchDOTALL IGNORECASEgrouprstrip)textfencems r_extract_sql_from_textrPgs :: *F F6%= AB "3 +E"5')<]OL&&*iiu &>%?@&&2^4++9*:;++0'1El^  U #5')<]OL&&*iiu &>%?@!!&';L> K  \\:v. /F fl  ! ! #C !# &&=  D  sC&& D/ C;;Dr!c|tvr.td|ddjttdddlm}d}d} d} d} t } t|jd xsd} t|jd xsd}t|jd xsd}t|jd xsd }t|| ||||}ttD]H}t||||||| | }|stjd||n|} tj!d|| | |||| ||| |t|jdxsdt|jdxsdd \}}|r t#j$||ddt}nd}t'|||| |\}}tj!d|||jd|jd |jdd!k(rn-| } |jd"xs|jd xs|} K|| | fS#t$r(}tjd|||d|d}Yd}~d}~wwxYw)#zn LangChain SQL pipeline for live databases (dialect-aware). Returns (db_answer, sql_used, usage). z5LangChain SQL pipeline does not yet support dialect 'z'. Supported: rY.r)run_select_mysqlr$Nusernamerrri )rarbrcrdrrQrRzFDB pipeline source=langchain dialect=%s generated empty SQL attempt=%dzCDB pipeline source=langchain dialect=%s generated_sql=%s attempt=%dssl_moderequired ssl_ca_pem) rrrrrsqlallowed_tablesrqrsmax_rows)columnsrowsF) ensure_asciidefaultz[Query returned 0 rows]z@DB pipeline source=langchain dialect=%s exec_error=%s attempt=%dz [SQL error: ])rarbru query_resultzPDB pipeline source=langchain dialect=%s verifier attempt=%d verdict=%s reason=%sverdictreasonpassfeedback)_SUPPORTED_DIALECTSr8r_r^agent.mysql_runnerrorr2rintrrangeMAX_LANGCHAIN_ATTEMPTSrlloggerwarninginfojsondumpsr)r@rarbdb_labellive_ctxrcrro db_answersql_used prev_errorprev_sqlusagerrrrrdattempt sql_used_newcolsryrBr~_s rrun_langchain_live_sqlrs))CG9M))F+>$?@A D  4IH!JH JE x||J'-2 .D8<< +1r2H x||F#)r *D x||F#+t ,D*7D(D$PXYN/0:S. %')!%   NNcelnu v  Y[bdlnuv ,)!!$X\\*5CDx||L9?R@ JD$ JJ $d3!& 6 ' %'"    ^ Wgkk)4gkk(6K ;;y !V + [[,R H0ER u:Sx h %%- , NN]_fhikr s&qc+I ,s4A1H77 I(I##I()rr2rr2rr2rr2rrrr2returnr2)rr2rr2)r@rrr)rMr2rr2)r@rrar2rbr2rcset[str]rdr2rr2rQ str | NonerRrrr2)r@rrar2rbr2rr2rzdict[str, Any]rcrrr2rztuple[str, str, LLMUsage])__doc__ __future__rrloggingrGtypingr urllib.parseragent.sql_verifierrconfig.settingsrllm.baser getLogger__name__rr frozensetrr __annotations__rr"rCrPrlrr rrrs#  #0$   8 $ *'%'^\ \\'*\25\=@\LO\\3l$#!%-'-'-' -'  -'  -'-'-'-' -'p^&^&^& ^&  ^&  ^& ^&^&^&r