126 lines
10 KiB
Plaintext
126 lines
10 KiB
Plaintext
|
Ë
|
|||
|
|
E×i®# ã ó¨ — d Z ddlZddlZddlmZmZmZmZ ddlm Z ddl
|
|||
|
|
m Z ddl m
|
|||
|
|
Z
|
|||
|
|
ddlmZ ddlmZmZ ej& e« Z G d „ d
|
|||
|
|
« Zy) uB
|
|||
|
|
Validator Agent - SQLå®¡æ ¸å‘˜
|
|||
|
|
验è¯�SQLçš„æ£ç¡®æ€§å’Œå®‰å…¨æ€§
|
|||
|
|
é N)ÚDictÚListÚTupleÚOptional)Ú ChatAgent)Ú ChatModel)ÚVALIDATOR_SYSTEM)Úfull_validation_pipeline)Úvalidate_sql_syntaxÚvalidate_schema_consistencyc óÎ — e Zd ZdZ ddedee fd„Z ddededed ed
|
|||
|
|
e f
|
|||
|
|
d „Z
|
|||
|
|
deded ed
|
|||
|
|
e fd „Z deded
|
|||
|
|
e fd
|
|||
|
|
„Z ded
|
|||
|
|
e fd„Z
|
|||
|
|
ded
|
|||
|
|
eeee f fd„Zdedee ded
|
|||
|
|
ee fd„Zy)ÚValidatorAgentu»
|
|||
|
|
Validator Agent
|
|||
|
|
|
|||
|
|
�责：
|
|||
|
|
- è¯æ³•æ£ç¡®æ€§éªŒè¯�
|
|||
|
|
- Schema一致性检查
|
|||
|
|
- 安全性检查（ç¦�æ¢DML/DDL）
|
|||
|
|
- 性能问题识别
|
|||
|
|
- æ��供修æ£å»ºè®®
|
|||
|
|
NÚmodelÚsystem_messagec ó’ — || _ |xs t | _ t | j |¬« | _ t
|
|||
|
|
j
|
|||
|
|
d« y)uÄ
|
|||
|
|
�始化Agent
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
model: CAMEL AI模型实例
|
|||
|
|
system_message: 系统�示�
|
|||
|
|
schema_manager: Schema管�器（程�验�用）
|
|||
|
|
)r r u! ✅ ValidatorAgentåˆ�始化完æˆ�N)Úschema_managerr r r ÚagentÚloggerÚinfo)Úselfr r r s úagents/validator.pyÚ__init__zValidatorAgent.__init__ sA € ð -ˆÔØ ,Ò @Ô0@ˆÔÜ Ø ×.Ñ.Øô
|
|||
|
|
ˆŒ
|
|||
|
|
ô � ‰ Ð7Õ8ó ÚsqlÚ
|
|||
|
|
schema_strÚdialectÚcheck_dangerousÚreturnc ó — dg g g dœ}| j |||« }|d j |j dg « « |d j |j dg « « |d j |j dg « « |j dd« rd|d< |S | j ||« }|d j |j dg « « |d j |j dg « « |d j |j dg « « t |d « d k( |d< |S # t $ rB}t
|
|||
|
|
j
|
|||
|
|
d |› �« |d j d
|
|||
|
|
t |« › �« Y d }~Œ\d }~ww xY w)
|
|||
|
|
u¯
|
|||
|
|
完整验��程（程� + LLM��验�）
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
sql: SQLè¯å�¥
|
|||
|
|
schema_str: Schema�述
|
|||
|
|
dialect: SQL方言
|
|||
|
|
check_dangerous: 是�检查�险�作
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
验è¯�结果å—å…¸ {
|
|||
|
|
"valid": bool,
|
|||
|
|
"errors": [...],
|
|||
|
|
"warnings": [...],
|
|||
|
|
"suggestions": [...]
|
|||
|
|
}
|
|||
|
|
T)ÚvalidÚerrorsÚwarningsÚ suggestionsr! r" r# ÚfatalFr u, LLM验è¯�失败，使用程åº�验è¯�结果: õ LLM验è¯�异常: Nr )
|
|||
|
|
Ú_program_validationÚextendÚgetÚ_llm_validationÚ Exceptionr ÚwarningÚappendÚstrÚlen) r r r r r ÚresultÚprogram_resultÚ
|
|||
|
|
llm_resultÚes r ÚvalidatezValidatorAgent.validate6 s� € ð2 ØØØ ñ
|
|||
|
|
ˆð ×1Ñ1Ø �˜/ó
|
|||
|
|
ˆð ˆxÑ× Ñ × 2Ñ 2°8¸RÓ @ÔA؈zÑ×!Ñ! .×"4Ñ"4°ZÀÓ"DÔE؈}Ñ ×$Ñ$ ^×%7Ñ%7¸
|
|||
|
|
ÀrÓ%JÔKð × Ñ ˜g uÔ -Ø #ˆF�7‰O؈Mð DØ ×-Ñ-¨c°:Ó>ˆJØ �8Ñ × #Ñ # J§N¡N°8¸RÓ$@Ô AØ �:Ñ × %Ñ % j§n¡n°ZÀÓ&DÔ EØ �=Ñ !× (Ñ (¨¯ © ¸
|
|||
|
|
ÀrÓ)JÔ Kô
|
|||
|
|
˜f XÑ .Ó/°1Ñ4ˆˆw‰Øˆ
|
|||
|
|
øô ò DÜ �N‰NÐ IÈ!ÈÐMÔ NØ �:Ñ × %Ñ %Ð(9¼#¸a»&¸Ð&B× CÑ Cûð Dús Â!A>D5 Ä5 F Ä>8E;Å;F c ó` — g }g }g }d}t ||« \ }} |s|j | « d}||||dœS | j r/t || j |« \ }
|
|||
|
|
} |
|
|||
|
|
s|j | « d}|r8ddlm} | |« \ }
|
|||
|
|
}|
|
|||
|
|
s%|j
|
|||
|
|
ddj |« › �« d}||||dœS )u�
|
|||
|
|
程�验�（规则引擎）
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
{"errors": [], "warnings": [], "suggestions": [], "fatal": bool}
|
|||
|
|
FT)r! r" r# r$ r ©Úcheck_dangerous_operationsu 包å�«å�±é™©æ“�作: ú, )r r' r r Úutils.validatorsr6 r, Újoin)r r r r r! r" r# r$ Ú syntax_okÚ
|
|||
|
|
syntax_errorsÚ schema_okÚ
|
|||
|
|
schema_errorsr6 ÚsafeÚdangerss r r& z"ValidatorAgent._program_validationp sä € ð ˆØˆØˆ ؈ô $7°s¸GÓ#DÑ ˆ �=ÙØ �M‰M˜-Ô (؈Eà ¨hØ *°Uñð
|
|||
|
|
ð × Ò Ü'BØ�T×(Ñ(¨'ó(Ñ $ˆI�}ñ Ø—
|
|||
|
|
‘
|
|||
|
|
˜mÔ,à �ñ Ý CÙ 6°sÓ ;‰MˆD�'ÙØ—
|
|||
|
|
‘
|
|||
|
|
Ð 4°T·Y±Y¸wÓ5GÐ4HÐ IÔJØ �ð Ø Ø&Øñ
|
|||
|
|
ð
|
|||
|
|
r c óÄ — ddl m} |j ||¬« } | j j |« }|j
|
|||
|
|
j j « }| j |« }|j dg « |j dg « |j dg « dœS # t $ r5}t j d|› �« d t |« › �gg g dœcY d
|
|||
|
|
}~S d
|
|||
|
|
}~ww xY w) u›
|
|||
|
|
LLMè¯ä¹‰éªŒè¯�
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
sql: SQLè¯å�¥
|
|||
|
|
schema_str: Schema�述
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
验�结果
|
|||
|
|
r )ÚVALIDATOR_USER)r Úschemar! r" r# ©r! r" r# r% u LLM验è¯�失败: N)Úconfig.promptsrA Úformatr ÚstepÚmsgÚcontentÚstripÚ_parse_json_responser( r* r Úerrorr- ) r r r rA ÚpromptÚresponserH r/ r2 s r r) z ValidatorAgent._llm_validation¤ sÚ € õ 2à ×&Ñ&¨3°zÐ&ÓBˆð Ø—z‘z—‘ vÓ.ˆHØ —l‘l×*Ñ*×0Ñ0Ó2ˆGð ×.Ñ.¨wÓ7ˆFð !Ÿ*™* X¨rÓ2Ø "ŸJ™J z°2Ó 6Ø %Ÿz™z¨-¸Ó <ñð
|
|||
|
|
øô ò Ü �L‰LÐ,¨Q¨CÐ0Ô 1à .¬s°1«v¨hÐ7Ð8Ø Ø !ñõ
|
|||
|
|
ûð ús ›BB! Â! C Â**CÃC ÃC rH c ó< — d|v r>|j d« dz }|j d|« }|dk7 r||| j « }t j |« S # t j $ r/}t
|
|||
|
|
j
|
|||
|
|
d|› d|dd › �« d gg g d
|
|||
|
|
œcY d}~S d}~ww xY w) u 解æž�JSONå“�应z```jsoné z```éÿÿÿÿu JSON解æž�失败: z
|
|||
|
|
, content=NéÈ u 验è¯�结果解æž�失败rC )ÚfindrI ÚjsonÚloadsÚJSONDecodeErrorr r+ )r rH ÚstartÚendr2 s r rJ z#ValidatorAgent._parse_json_responseÉ s§ € ð _à˜GÑ#Ø Ÿ ™ YÓ/°!Ñ3�Ø —l‘l 5¨%Ó0�ؘ"’9Ø % e¨CÐ 0× 6Ñ 6Ó 8�Gä—:‘:˜gÓ&Ð &øÜ×#Ñ#ò _Ü �N‰NÐ /°¨s°*¸WÀTÀc¸]¸OÐLÔ MØ 9Ð :ÈÐ[]Ñ^Õ ^ûð _ús ‚AA ÁBÁ,$BÂBÂBc óÌ — g }t |« \ }}|s|j |« d|fS ddlm} ||« \ }}|s'|j ddj |« › �« d|fS dg fS )uw
|
|||
|
|
快速检查（仅è¯æ³•å’Œå�±é™©æ“�作）
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
(是�通过, 错误列表)
|
|||
|
|
Fr r5 u å�±é™©æ“�作: r7 T)r r' r8 r6 r, r9 )r r r! r: r; r6 r> r? s r Ú quick_checkzValidatorAgent.quick_checkØ sz € ð ˆô $7°sÓ#;Ñ ˆ �=ÙØ �M‰M˜-Ô (ؘ&�=Ð õ @Ù2°3Ó7‰
|
|||
|
|
ˆˆgÙØ �M‰M˜N¨4¯9©9°WÓ+=Ð*>Ð?Ô @ؘ&�=Ð à�Rˆxˆr r! c ó° — g }|D ]s }|j « }d|v sd|v r|j d« d|v sd|v r|j d« d|v sd|v r|j d « d
|
|||
|
|
|v sŒc|j d « Œu t |« t |« k rŽ d |› d
|
|||
|
|
|› d|dd › d�}| j j |« }|j
|
|||
|
|
j j « j d« } |j | D �
|
|||
|
|
cg c] }
|
|||
|
|
|
|
|||
|
|
j « sŒ|
|
|||
|
|
‘Œ c}
|
|||
|
|
« |dd S c c}
|
|||
|
|
w # t $ r"} t j d| › �« Y d} ~ Œ0d} ~ ww xY w)uÑ
|
|||
|
|
æ ¹æ�®é”™è¯¯å»ºè®®ä¿®å¤�方案
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
sql: 原始SQL
|
|||
|
|
errors: 错误列表
|
|||
|
|
schema_str: Schema�述
|