277 lines
27 KiB
Plaintext
277 lines
27 KiB
Plaintext
|
Ë
|
|||
|
|
y®Üižj ã óÊ — d Z ddlZddlZddlmZmZmZmZ ddlm Z m
|
|||
|
|
Z
|
|||
|
|
ddl m Z ddl
|
|||
|
|
mZ ddlmZmZ ddlmZ ej( e« Ze G d „ d
|
|||
|
|
« « Z G d „ d « Zy)
|
|||
|
|
u]
|
|||
|
|
Text2SQL 多智能体编排器
|
|||
|
|
�调 Schema Linker�SQL Generator�Validator 三个Agent
|
|||
|
|
é N)ÚDictÚListÚOptionalÚTuple)Ú dataclassÚfield)Ú
|
|||
|
|
SchemaManager)Ú
|
|||
|
|
SchemaIndexer)ÚDeepSeekClientÚDeepSeekConfig)ÚFewShotSelectorc óÔ — e Zd ZU dZeed< eed< ee¬« Z e
|
|||
|
|
e ed< ee¬« Z e
|
|||
|
|
e ed< ee¬« Z e
|
|||
|
|
e ed< dZ
|
|||
|
|
eed < d
|
|||
|
|
Zee ed < ee¬« Zeeef ed < y
|
|||
|
|
)
|
|||
|
|
ÚGenerationResultu SQL生æˆ�结果ÚsqlÚvalid)Údefault_factoryÚerrorsÚwarningsÚ tables_usedé ÚattemptsNÚ reasoningÚmetadata)Ú__name__Ú
|
|||
|
|
__module__Ú __qualname__Ú__doc__ÚstrÚ__annotations__Úboolr Úlistr r r r r Úintr r Údictr r Úany© ó úCC:\Users\24019\Desktop\backman-camel\backend\agents\orchestrator.pyr r sw … áØ ƒHØ ƒKÙ ¨dÔ3€FˆD�‰IÓ3Ù °Ô5€Hˆd�3‰iÓ5Ù "°4Ô 8€K��c‘Ó8Ø€HˆcÓØ #€Iˆx˜‰}Ó#Ù $°TÔ :€Hˆd�3˜�8‰nÔ:r& r c óà — e Zd ZdZ d*dedee dee dee deded e d
|
|||
|
|
e d ee d ed
|
|||
|
|
ede fd„Z
|
|||
|
|
de fd„Z d+dedede
|
|||
|
|
e fd„Z d,dede
|
|||
|
|
e dedee
|
|||
|
|
e ef fd„ZdZdede fd„Z d,dede
|
|||
|
|
e dede
|
|||
|
|
e fd„Zd e
|
|||
|
|
e de
|
|||
|
|
e fd „Z d-ded ed ed ee def
|
|||
|
|
d!„Z d.d"ed ed ededee e
|
|||
|
|
e e
|
|||
|
|
e ee ee f f
|
|||
|
|
d#„Z d/ded ed$ed%e def
|
|||
|
|
d&„Zd0d'e de fd(„Zdefd)„Zy)1ÚText2SQLOrchestratoruã
|
|||
|
|
Text2SQL 多智能体编排器
|
|||
|
|
|
|||
|
|
工作�程：
|
|||
|
|
1. ç²—ç›å€™é€‰è¡¨
|
|||
|
|
2. Schema Linker：LLM ç²¾ç›è¡¨
|
|||
|
|
3. 外键扩展 → 拼 Schema å�集
|
|||
|
|
4. SQL Generator：生� SQL
|
|||
|
|
5. Validator：验� SQL
|
|||
|
|
NÚschema_managerÚdeepseek_api_keyÚdeepseek_configÚembedding_model_pathÚvector_db_pathÚ max_retryÚuse_vector_searchÚfewshot_enabledÚfewshot_samples_pathÚ
|
|||
|
|
fewshot_top_kÚfewshot_min_ratingÚtranslate_english_to_zhc
|
|||
|
|
ó¨ — || _ || _ || _ | | _ |rt |« | _ nt t
|
|||
|
|
|¬« « | _ d| _ || _ || _ || _
|
|||
|
|
|
|
|||
|
|
| _ | | _ d| _
|
|||
|
|
| j rR | xs t j dd« }
|
|||
|
|
t! |
|
|||
|
|
| j ¬« | _
|
|||
|
|
t" j% d|
|
|||
|
|
› d| › �« t" j% d |› d |› �| j rd
|
|||
|
|
ndz | j r dz « ydz « y# t& $ r*}t" j) d|› d �« d
|
|||
|
|
| _
|
|||
|
|
Y d}~Œsd}~ww xY w)um
|
|||
|
|
�始化编排器
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
schema_manager: Schema管�器实例
|
|||
|
|
deepseek_api_key: DeepSeek API密钥（也�通过环境��DEEPSEEK_API_KEY）
|
|||
|
|
deepseek_config: DeepSeek�置对象（优先于api_key）
|
|||
|
|
embedding_model_path: Qwen3-Embedding模型路径
|
|||
|
|
vector_db_path: ��数�库路径
|
|||
|
|
max_retry: 最大�试次数（包�首次生�）
|
|||
|
|
use_vector_search: 是å�¦ä½¿ç”¨å�‘é‡�检索粗ç›
|
|||
|
|
translate_english_to_zh: æ— ä¸æ—¥éŸ©å—符的英文问å�¥æ˜¯å�¦å…ˆè¯‘为ä¸æ–‡å†�走检索与生æˆ�
|
|||
|
|
)Úapi_keyNÚFEWSHOT_DATA_PATHz$./data/experiences/all_samples.jsonl)r- u Few-shotå·²å�¯ç”¨: top_k=z
|
|||
|
|
, min_rating=u Few-shotåŠ è½½å¤±è´¥: u ï¼Œå°†ä½¿ç”¨æ ‡å‡†ç”Ÿæˆ�Fu4 [OK] Text2SQLOrchestratoråˆ�始化完æˆ�: max_retry=z, use_vector_search=z , fewshot=onÚ u , en→zh=onu
|
|||
|
|
, en→zh=off)r* r/ r0 r5 r Údeepseekr Ú
|
|||
|
|
_vector_indexÚ_vector_db_pathÚ_embedding_model_pathr1 r3 r4 Úfewshot_selectorÚosÚgetenvr
|
|||
|
|
ÚloggerÚinfoÚ ExceptionÚwarning)Úselfr* r+ r, r- r. r/ r0 r1 r2 r3 r4 r5 ÚpathÚes r' Ú__init__z Text2SQLOrchestrator.__init__, s� € ð: -ˆÔØ"ˆŒØ!2ˆÔ Ø'>ˆÔ$ñ Ü *¨?Ó ;ˆD�Mä *Ü Ð'7Ô8ó ˆDŒMð
|
|||
|
|
7;ˆÔØ -ˆÔ Ø%9ˆÔ"ð /ˆÔ Ø *ˆÔØ"4ˆÔ Ø $ˆÔ à × Ò ð
|
|||
|
|
-Ø+ò ¬r¯y©yØ'Ø:ó0�ô )8ØØ)-×)CÑ)Cô)�Ô%ô — ‘ Ø/°
|
|||
|
|
¨ð ?"Ø"4Ð!5ð7ôô � ‰ ðØ"˜ Ð#7Ð8IÐ7Jð
|
|||
|
|
Là"&×"6Ò"6‘ ¸Bñ
|
|||
|
|
@ð "&×!=Ò!=ˆ~ñ
|
|||
|
|
Tõ
|
|||
|
|
ð DSñ
|
|||
|
|
Tõ
|
|||
|
|
øô ò
|
|||
|
|
-Ü—‘Ð!7¸°sÐ:RÐ SÔTØ',�×$Ñ$ûð
|
|||
|
|
-ús ÂAD Ä EÄ' E Å EÚreturnc óš — | j €4ddlm} || j « }t || j
|
|||
|
|
¬« | _ | j S )u* 获å�–或创建å�‘é‡�ç´¢å¼•ï¼ˆæ‡’åŠ è½½ï¼‰r )Ú get_embedder)ÚembedderÚ persist_dir)r; Úutils.embeddingrK r= r
|
|||
|
|
r< )rE rK rL s r' Ú_get_vector_indexz&Text2SQLOrchestrator._get_vector_indexz sH € à × Ñ Ð %Ý 4á# D×$>Ñ$>Ó?ˆHÜ!.Ø!Ø × 0Ñ 0ô"ˆDÔ ð ×!Ñ!Ð!r& ÚquestionÚtop_kc ó¢ — | j s| j j « S | j « }|j « dk( r2t
|
|||
|
|
j
|
|||
|
|
d« |j | j d¬« |j ||d¬« }|D �cg c] }|d ‘Œ }}t
|
|||
|
|
j d|d d
|
|||
|
|
› d t |« › d �« |S c c}w )
|
|||
|
|
uÁ
|
|||
|
|
阶段1：粗ç›ï¼ˆå�‘é‡�检索）
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
question: 用户问题
|
|||
|
|
top_k: 返回�K个候选表
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
候选表�列表
|
|||
|
|
r u$ å�‘é‡�索引为空，æ£åœ¨æž„建...T©Ú
|
|||
|
|
force_rebuildgš™™™™™¹?)ÚqueryrQ Úscore_thresholdÚ
|
|||
|
|
table_nameu ç²—ç›å€™é€‰è¡¨ï¼šNé
|
|||
|
|
u ...（共u 个）) r0 r* Ú list_tablesrO ÚcountrA rB Ú build_indexÚsearchÚdebugÚlen)rE rP rQ ÚindexerÚresultsÚrÚcandidate_tabless r' Ú_coarse_filterz#Text2SQLOrchestrator._coarse_filter† s× € ð ×%Ò%à×&Ñ&×2Ñ2Ó4Ð 4à×(Ñ(Ó*ˆð �=‰=‹?˜aÒ Ü �K‰KÐ>Ô ?Ø × Ñ × 3Ñ 3À4Ð Ô Hð —.‘.ØØØ ð !ó
|
|||
|
|
ˆñ 6=Ó=±W°˜A˜l›O°WÐÐ=Ü� ‰ Ð)Ð*:¸3¸BÐ*?Ð)@À Ì#ÐN^ÓJ_ÐI`Ð`fÐgÔhØ Ð ùò >s  C rb Ú
|
|||
|
|
max_tablesc ó„ — g }|D ]G }| j j |« }|sŒ!|j xs d}|j d|› d|› �« ŒI dj |« }| j
|
|||
|
|
j
|
|||
|
|
||¬« } | j dg « }
|
|||
|
|
| j dd« } |
|
|||
|
|
d | }
|
|||
|
|
t j d
|
|||
|
|
|
|
|||
|
|
› �« |
|
|||
|
|
| fS ) u
|
|||
|
|
|
|||
|
|
阶段2：LLMç²¾ç›ï¼ˆSchema Linker Agent）
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
question: 用户问题
|
|||
|
|
candidate_tables: 候选表列表
|
|||
|
|
max_tables: 最多选择的表数
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
(相关表列表, 推��由)
|
|||
|
|
u æ— æ��è¿°ú- z: Ú
|
|||
|
|
)rP Ú
|
|||
|
|
table_listÚrelevant_tablesr r9 Nu LLMç²¾ç›é€‰ä¸è¡¨ï¼š)
|
|||
|
|
r* Ú get_tableÚcommentÚappendÚjoinr: Ú
|
|||
|
|
select_tablesÚgetrA rB ) rE rP rb rd Ú table_infosÚtbl_nameÚtablerk Útable_list_strÚresponseri r s r' Ú_llm_select_tablesz'Text2SQLOrchestrator._llm_select_tables« s× € ð$ ˆ Û(ˆHØ×'Ñ'×1Ñ1°(Ó;ˆEÚØ Ÿ-™-Ò6¨;�Ø×"Ñ" R¨ z°°G°9Ð#=Õ>ð )ð Ÿ™ ;Ó/ˆð —=‘=×.Ñ.Ø Ø%ð /ó
|
|||
|
|
ˆð
|
|||
|
|
#Ÿ,™,Ð'8¸"Ó=ˆØ —L‘L ¨bÓ1ˆ ð *¨+¨:Ð6ˆä� ‰ Ð+¨OÐ+<Ð=Ô>Ø Ð)Ð)r& )u 对手方u ç»�纪商u 券商u å°�手方c ól ‡— ‰syt ˆfd„| j D « « ryd‰j « v S )NFc 3 ó& •K — | ] }|‰v –— Œ
|
|||
|
|
y w)Nr% )Ú.0ÚkrP s €r' Ú <genexpr>zJText2SQLOrchestrator._question_implies_broker_dimension.<locals>.<genexpr>Ú s øè ø€ Ð?Ñ&> ˆq�HŒ}Ñ&>ùs ƒTÚbroker)r$ Ú_BROKER_KEYWORDS_CNÚlower)rE rP s `r' Ú"_question_implies_broker_dimensionz7Text2SQLOrchestrator._question_implies_broker_dimension× s2 ø€ ÙØÜ Ó? d×&>Ò&>Ó?Ô ?Øؘ8Ÿ>™>Ó+Ð+Ð+r& ri c ó — | j |« s|d| S ddg}|D �cg c] }| j j |« sŒ |‘Œ" }}|s|d| S t « }g }|D ]) }||vsŒ|j |« |j |« Œ+ |D ]U }t
|
|||
|
|
|« |k\ r nE||vsŒ| j j |« sŒ4|j |« |j |« ŒW t j d||« |d| S c c}w )uò
|
|||
|
|
问题涉å�Šå¯¹æ‰‹æ–¹/ç»�纪商时，优先纳入 TSBBrokerContract 与 MCBroker（若 Schema ä¸å˜åœ¨ï¼‰ï¼Œ
|
|||
|
|
é�¿å…�ä»…é€‰ä¸ VSBHK 报表视图å�´æ— BrokerID，模型å�ˆç…§æŠ„黄金范例列å��å¯¼è‡´æ ¡éªŒå¤±è´¥ã€‚
|
|||
|
|
NÚTSBBrokerContractÚMCBrokeruB 对手方/ç»�纪商问题：优先纳入 %s，调整å�Žé€‰è¡¨ï¼š%s) r~ r* rj Úsetrl Úaddr^ rA rB ) rE rP ri rd ÚpriorityÚtÚpresentÚseenÚmergeds r' Ú_prioritize_broker_tablesz.Text2SQLOrchestrator._prioritize_broker_tablesÞ s € ð ×6Ñ6°xÔ@Ø" ; JÐ/Ð /à'¨Ð4ˆÙ &ÓK™h˜¨$×*=Ñ*=×*GÑ*GÈÕ*J’1˜hˆÐKÙØ" ; JÐ/Ð /ä‹uˆØ ˆÛˆAؘŠ}Ø—
|
|||
|
|
‘
|
|||
|
|
˜aÔ Ø—‘˜• ð ó !ˆAÜ�6‹{˜jÒ(ÙؘŠ} ×!4Ñ!4×!>Ñ!>¸qÕ!AØ—
|
|||
|
|
‘
|
|||
|
|
˜aÔ Ø—‘˜• ð !ô � ‰ ÐXÐZaÐciÔjØ�k�zÐ"Ð"ùò# Ls Ÿ C=Á C=Ú table_namesc ó¶ — t |« }|D �]
|
|||
|
|
}| j j |« }|sŒ"|j D ]N }|j |vsŒ|j |j « t j d|j › �« ŒP | j j « D ]o }|j D ]^ }|j |k( sŒ|j |vsŒ"|j |j « t j d|j › �« Œ` Œq �Œ
|
|||
|
|
t |« }t |« t |« kD rt j d|› d|› �« |S )u®
|
|||
|
|
å¤–é”®æ‰©å±•ï¼šè‡ªåŠ¨æ·»åŠ å…³è�”表
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
table_names: 已选ä¸çš„表å��列表
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
扩展�的表�列表
|
|||
|
|
u å¤–é”®æ‰©å±•ï¼šæ·»åŠ å…³è�”表 u å¤–é”®æ‰©å±•ï¼šæ·»åŠ å¼•ç”¨è¡¨ u 外键扩展：u → )
|
|||
|
|
r‚ r* rj Ú foreign_keysÚ ref_tablerƒ rA r] Ú
|
|||
|
|
get_tablesÚnamer! r^ rB )rE rŠ Úresultrq rr ÚfkÚotherÚexpandeds r' Ú_expand_relationsz&Text2SQLOrchestrator._expand_relationsü s € ô �[Ó!ˆä#ˆHØ×'Ñ'×1Ñ1°(Ó;ˆEÙØð ×(Ô(�Ø—<‘< vÒ-Ø—J‘J˜rŸ|™|Ô,Ü—L‘LÐ#BÀ2Ç<Á<À.Ð!QÕRð )ð ×,Ñ,×7Ñ7Ö9�Ø ×,Ô,�BØ—|‘| xÓ/°E·J±JÀfÒ4LØ Ÿ
|
|||
|
|
™
|
|||
|
|
5§:¡:Ô.Ü Ÿ ™ Ð'FÀuÇzÁzÀlÐ%SÕTñ -ò :ð $ô$ ˜“<ˆÜ ˆx‹=œ3˜{Ó+Ò +Ü �K‰K˜/¨+¨ °e¸H¸:ÐFÔ Gàˆr& Ú
|
|||
|
|
schema_strÚdialectÚvalidation_feedbackc ó° — ddl m}m} ddlm} | j
|
|||
|
|
rÒ| j rÆ | j j || j | j ¬« }|r‘dj t |« D � �
|
|||
|
|
cg c]' \ } }
|
|||
|
|
d| dz › d|
|
|||
|
|
j › d |
|
|||
|
|
j › �‘Œ) c}
|
|||
|
|
} « } d
|
|||
|
|
| › d |› �}t j d t! |« › d
|
|||
|
|
|D �
|
|||
|
|
cg c] }
|
|||
|
|
|
|
|||
|
|
j" ‘Œ c}
|
|||
|
|
› �« |}
|
|||
|
|
|dk( rd}
|
|||
|
|
|j) |||
|
|||
|
|
¬« }|dk( r|dz
|
|||
|
|
}|r|d|› �z
|
|||
|
|
}d|dœd|dœg}| j* j- |dd¬« }|j. j1 « }d|v rH||j3 d« d z |j3 d |j3 d« d z « j1 « }nKd |v rG||j3 d « d z |j3 d |j3 d « d z « j1 « } |||« }t j d |dd › d!�« |S c c}
|
|||
|
|
} w c c}
|
|||
|
|
w # t$ $ r#} t j' d| › �« Y d} ~ �Œcd} ~ ww xY w)"uB
|
|||
|
|
SQL生�（SQL Generator Agent）
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
question: 用户问题
|
|||
|
|
schema_str: Schemaæ��è¿°å—符串
|
|||
|
|
dialect: SQL方言
|
|||
|
|
validation_feedback: é�žç©ºæ—¶é™„åŠ åˆ°ç”¨æˆ·æ��示（é‡�è¯•æ—¶ä¼ å…¥ä¸Šæ¬¡æ ¡éªŒé”™è¯¯ï¼‰
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
SQLè¯å�¥
|
|||
|
|
r )ÚSQL_GENERATOR_SYSTEMÚSQL_GENERATOR_USER)Únormalize_sql_for_dialect)rP rQ Ú
|
|||
|
|
min_ratingz
|
|||
|
|
|
|||
|
|
u 示例 r u
|
|||
|
|
：
|
|||
|
|
问题：u
|
|||
|
|
SQL：
|
|||
|
|
u/ å�‚考以下相似示例的SQLç¼–å†™é£Žæ ¼ï¼š
|
|||
|
|
|
|||
|
|
u
|
|||
|
|
|
|||
|
|
�当�Schema】
|
|||
|
|
u
|
|||
|
|
已注入 u 个few-shot示例: u Few-shot检索失败: NÚtsqlz Microsoft SQL Server (T-SQL))ÚschemarP r– u
|
|||
|
|
|
|||
|
|
ã€�硬性è¦�æ±‚ã€‘ç›®æ ‡åº“ä¸º SQL Server（T-SQL）：ç¦�æ¢ä½¿ç”¨ MySQL å��引å�· `ï¼›æ ‡è¯†ç¬¦å¦‚éœ€å¼•ç”¨è¯·ä½¿ç”¨æ–¹æ‹¬å�·ï¼Œä¾‹å¦‚ [TableName]ã€�[ColumnName]。å—符串连接使用 `+`（与系统æ��示ä¸çš„æ ‡å‡†ç‰ˆå¼�范例一致）。「今日ã€�「当天ã€�ç‰ä¸Žæ—¥æœŸåˆ—比较时，使用 `CAST(GETDATE() AS DATE)`，**ç¦�æ¢** `CURDATE()`ã€�`NOW()`ã€�`CURRENT_DATE`（MySQL）。æ�¡ä»¶è¯·ä½¿ç”¨ T-SQL 惯用写法（例如 IS NOT NULL）。排版ä»�é¡»é�µå®ˆï¼šå…³é”®å—大写ã€�SELECT æ¯�列一行缩进ã€�WHERE ç»è¡Œä»¥ AND 开头ã€�PascalCase 英文别å��。
|
|||
|
|
**ç¦�æ¢**在å�•å¼•å�·å—符串å—é�¢é‡�或 `N'…'` ä¸å‡ºçŽ°ä»»ä½•ä¸æ—¥éŸ©æ–‡å—；业务ä¸æ–‡é¡»æ˜ 射为 Schema 注释ä¸çš„代ç �或通过维表 JOIN，勿写 `= '过户费'` 这类比对。uµ
|
|||
|
|
|
|||
|
|
ã€�ä¸Šæ¬¡æ ¡éªŒæœªé€šè¿‡ã€‘è¯·æ ¹æ�®ä¸‹åˆ—é”™è¯¯ä¿®æ£ SQL，并输出完整å�¯æ‰§è¡ŒæŸ¥è¯¢ï¼›è¡¨å��ã€�列å��必须与「当å‰�Schemaã€�ä¸å®Œå…¨ä¸€è‡´ï¼Œä¸�è¦�è‡†é€ å—段å��。
|
|||
|
|
Úsystem)ÚroleÚcontentÚuserg g ð?)Ú temperatureÚtop_pz```sqlé z```é u 生æˆ�çš„SQL：éÈ ú...)Úconfig.promptsr™ rš Úutils.sql_parserr› r1 r> Úselectr3 r4 rm Ú enumerateÚ question_zhr rA r] r^ ÚqidrC rD Úformatr: Úchatr¡ ÚstripÚfind)rE rP r• r– r— r™ rš r› ÚexamplesÚiÚexÚexamples_promptrG Ú
|
|||
|
|
dialect_labelÚ user_contentÚmessagesrt r s r' Ú
|
|||
|
|
_generate_sqlz"Text2SQLOrchestrator._generate_sql s¦ € ÷& LÝ>ð × Ò D×$9Ò$9ð
|
|||
|
|
=Ø ×0Ñ0×7Ñ7Ø %Ø ×,Ñ,Ø #× 6Ñ 6ð 8ó �ñ
|
|||
|
|
Ø&,§k¡kä%.¨xÔ%8ô3á%8™E˜A˜rð " ! A¡# n°R·^±^Ð4DÀJÈrÏvÉvÈhÒWØ%8ò3ó '�Oð $UÐUdÐTeÐe}ð Ið ~Jð "K�JÜ—L‘L :¬c°(«m¨_Ð<PÑckÓQlÑckÐ]_ÐRT×RXÓRXÐckÑQlÐPmÐ!nÔoð ˆ
|
|||
|
|
Ø �fÒ Ø :ˆMà)×0Ñ0Ø Ø Ø!ð 1ó
|
|||
|
|
ˆ ð
|
|||
|
|
�fÒ Ø ðBñ
|
|||
|
|
|
|||
|
|
ˆLñ Ø ðià&Ð'ð)ñ
|
|||
|
|
ˆLð Ð*>Ñ ?ب Ñ 5ð
|
|||
|
|
ˆð —=‘=×%Ñ% h¸CÀsÐ%ÓKˆØ× Ñ ×$Ñ$Ó&ˆð �s‰?Ø�c—h‘h˜xÓ(¨1Ñ,¨S¯X©X°e¸S¿X¹XÀhÓ=OÐRSÑ=SÓ-TÐU×[Ñ[Ó]‰CØ
|
|||
|
|
�c‰\Ø�c—h‘h˜u“o¨Ñ)¨#¯(©(°5¸#¿(¹(À5»/ÈAÑ:MÓ*NÐO×UÑUÓWˆCá'¨¨WÓ5ˆä� ‰ � s¨4¨C y k°Ð5Ô6؈
|
|||
|
|
ùóq3ùò
|
|||
|
|
RmøÜ ò
|
|||
|
|
=Ü—‘Ð!7¸°sÐ ;×<Ò<ûð
|
|||
|
|
=ús6 ¨A
|
|||
|
|
H) Á5,H
|
|||
|
|
Â!0H) ÃH$Ã$ H) È H) È) IÈ2IÉIr c óD — g }g }d}d}ddl m} m}
|
|||
|
|
| ||¬« \ } } | s|j | « |
|
|||
|
|
|| j |¬« \ }
|
|||
|
|
}|
|
|||
|
|
s|j |« ddlm}m} ||« \ }}|s|j |« |dk( r ||« \ }}|s|j |« t |« dk( rGddl m
|
|||
|
|
} ||« \ }}|dk( r d }|r|d
|
|||
|
|
|› �z
|
|||
|
|
}|j |« n|€|j d « |d v }|sì | j j ||¬
|
|||
|
|
« }t |j dg « « }|
|
|||
|
|
rA|D �cg c]6 }t! |t" « r$|j% d« s|j% d« s|‘Œ8 }}|j dd« s|j |« |j |j dg « « |j dg « } | rt& j) d| › �« |dk( r6t |« dk( r(d} | j j/ |||¬« } | › d| › �}t |« dk( } | ||||fS c c}w # t* $ r"}t& j- d|› �« Y d}~Œ{d}~ww xY w# t* $ r'}t& j- d|› �« | › d�}Y d}~Œsd}~ww xY w) uT
|
|||
|
|
SQL验�（程�验� + 库探针 + 按探针分支的 LLM）
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
sql: SQLè¯å�¥
|
|||
|
|
schema_str: Schema�述
|
|||
|
|
dialect: 与生�一致的 SQL 方言（sqlglot �，默认 tsql）
|
|||
|
|
question: 用户自然è¯è¨€ï¼ˆæŽ¢é’ˆä¸º 0 时用于生æˆ�补充说明）
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
(是å�¦é€šè¿‡, 错误列表, è¦å‘Šåˆ—表, 库执行探针状æ€�, æ— æ•°æ�®æ—¶çš„用户说明)
|
|||
|
|
探针：``1`` 至少一行数�；``0`` 执行�功但行数为 0；``-1`` 执行失败；``None`` 未�置库或未跑探针
|
|||
|
|
Nr )Úvalidate_sql_syntaxÚvalidate_schema_consistency©r– )Úcheck_dangerous_operationsÚ#check_no_cjk_in_sql_string_literalsr� )Ú probe_sql_execution_status_exéÿÿÿÿun æ•°æ�®åº“执行验è¯�失败：SQL åœ¨ç›®æ ‡åº“æ‰§è¡ŒæŠ¥é”™ï¼ˆæŽ¢é’ˆçŠ¶æ€� -1），将æ�®æ¤é‡�新生æˆ� SQL。u æ•°æ�®åº“返回：u7 未é…�ç½® database_url，已跳过数æ�®åº“执行探针)r r r )r rž r zunknown_table:zunknown_column:r Tr Ú suggestionsu 优化建议：u/ LLM验è¯�失败（é™�级为仅程åº�验è¯�）: uÉ è¯¥ SQL 已在数æ�®åº“æˆ�功执行，但返回的数æ�®è¡Œæ•°ä¸º 0（未查到匹é…�记录）。请将下方 SQL ä¸Žè¯´æ˜Žä¸€å¹¶æ ¸å¯¹ï¼›è‹¥ä¸�符å�ˆé¢„期，请补充或调整æ�¡ä»¶å�Žå†�次æ��问。)rP r rž u
|
|||
|
|
|
|||
|
|
�分�与建议】
|
|||
|
|
u æ— æ•°æ�®è¯´æ˜Žç”Ÿæˆ�失败: u~
|
|||
|
|
|
|||
|
|
�分�与建议】
|
|||
|
|
未能自动生æˆ�详细分æž�。请补充时间范围ã€�ç›é€‰æ�¡ä»¶æˆ–业务对象å�Žé‡�æ–°æ��问。)rª r¼ r½ Úextendr* Úutils.validatorsr¿ rÀ r^ Údb.dbhub_toolsrÁ rl r: Ú validate_sqlr! ro Ú
|
|||
|
|
isinstancer Ú
|
|||
|
|
startswithrA r] rC rD Úempty_result_user_feedback) rE r r• r– rP r r Údb_execution_statusÚempty_feedbackr¼ r½ Ú syntax_okÚ
|
|||
|
|
syntax_errorsÚ schema_okÚ
|
|||
|
|
schema_errorsr¿ rÀ Ú danger_okÚ
|
|||
|
|
danger_errorsÚcjk_okÚ
|
|||
|
|
cjk_errorsrÁ Ú db_probe_errÚmsgÚskip_validator_llmÚ
|
|||
|
|
llm_resultÚ
|
|||
|
|
llm_errorsrG rà ÚprefixÚllm_fbÚis_valids r' Ú
|
|||
|
|
_validate_sqlz"Text2SQLOrchestrator._validate_sqly sú € ð( ˆØˆØ-1ÐØ(,ˆ÷ Vñ $7°sÀGÔ#LÑ ˆ �=ÙØ �M‰M˜-Ô (ñ $?Ø �×$Ñ$¨gô$
|
|||
|
|
Ñ ˆ �=ñ Ø �M‰M˜-Ô (÷ eÙ#=¸cÓ#BÑ ˆ �=ÙØ �M‰M˜-Ô (ð �fÒ Ù!DÀSÓ!IÑ ˆF�JÙØ—
|
|||
|
|
‘
|
|||
|
|
˜jÔ)ô ˆv‹;˜!Ò Ý Dá0MÈcÓ0RÑ -Ð Ø" bÒ(ð3ð ñ ØÐ 0° °Ð?Ñ?�CØ—
|
|||
|
|
‘
|
|||
|
|
˜cÕ"Ø$Ð,Ø—‘ØMôð 1°JÐ >Ðñ "ð
|
|||
|
|
VØ !Ÿ]™]× 7Ñ 7¸CÈ
|
|||
|
|
Ð 7Ó S�
|
|||
|
|
ä ! *§.¡.°¸2Ó">Ó ?�
|
|||
|
|
á ñ ",ó"á!+˜AÜ% a¬Ô-à ŸL™LÐ)9Ô :Ø Ÿ|™|Ð,=Ô >ò Ø!+ð ð "ð "—~‘~ g¨tÔ4Ø—M‘M *Ô-à—‘
|
|||
|
|
§¡¨z¸2Ó >Ô?Ø (Ÿn™n¨]¸BÓ ?� Ù Ü—L‘L ?°;°-Ð!@ÔAð !Ò #¬¨F« °qÒ(8ð{ð
|
|||
|
|
ð
|
|||
|
|
Ø Ÿ ™ ×AÑAØ %ØØ%ð Bó �ð
|
|||
|
|
%+ 8Ð+FÀvÀhÐ!O�ô �v“; !Ñ#ˆØ˜ Ð+>ÀÐNÐNùòU"øô$ ò
|
|||
|
|
VÜ—‘Ð!PÐQRÐPSÐ T×UÑUûð
|
|||
|
|
Vûô ò
|
|||
|
|
Ü—‘Ð!>¸q¸cÐ BÔCà �hð }ð }õ ûð
|
|||
|
|
úsC Ä>I Ä?;H<Å:A2I È%I/ È<I É I,É
|
|||
|
|
I'É'I,É/ J É8 JÊJ Útop_k_candidatesÚinclude_schema_in_resultc
|
|||
|
|
óà — ddl m} |xs dj « }i }|}| j r� ||« rˆ | j j |« j « } | rGt
|
|||
|
|
| « dk\ r9| }||d< | |d< t j d| dd t
|