i ddlZddlmZddlmZmZmZddlmZddl m Z ddl m Z ddl mZmZmZmZe eZed ed eeeeffd Zy) N)Path)ListDictAny)tool) db_session) get_logger)MAX_SQL_TOOL_ROWSmssql_wrap_top_subquerynormalize_llm_sqlvalidate_llm_sqlqueryreturnctjd|xsddd t|}t|}|rtj d|d|igSt |t }tj|}t|t kDr |dt }tjdt||S#t$r}|xsd}t|dkDr|ddd zn|}tj|j}|r|d nd} | r.t| jj d | j"nd } tj%d | tt&j |ddigcYd}~Sd}~wwxYw)z Executes a read-only SQL query against the MS SQL Server database. Only a single SELECT (or WITH ... SELECT) is allowed. Returns: List of dictionaries (rows), at most MAX_SQL_TOOL_ROWS from the server. z"llm query (truncated for logs): %sNi,zSQL tool rejected query: %serrorzwas successful True (%s rows)iu…:unknownzHSQL tool execution failed (raised at %s; handler %s; query for logs): %szQuery execution failed.)loggerinfor rrr r rexecutelen Exception traceback extract_tb __traceback__rfilenamenamelineno exception__file__) rqnerrwrappedrowse_q _q_preview_tb_loc _raised_wheres B/var/www/html/userprofiledev.eatanceapp.com/deep_analysis/tools.pyexecute_sql_queryr/s^ KK4u{DS6IJ(6 u %r"  LL6 <cN# #)".?@!!'* t9( (**+D 3SY? 6 [b+.r7S=b#h&b ""1??3s2w4;?tDMM"''($++ 7Y   V  N     "345536s%2B3A B33 E2r:s_""%)) H /6S/6T$sCx.%9/6/6r9