110 lines
6.9 KiB
Plaintext
110 lines
6.9 KiB
Plaintext
|
Ë
|
|||
|
|
øD×iß ã óŒ — d Z ddlZddlZddlZddlmZmZ ddlmZ ddl m
|
|||
|
|
Z
|
|||
|
|
ddl m Z ej e« Z G d„ d« Zy) uO
|
|||
|
|
SQL Generator Agent - SQL生�专家
|
|||
|
|
æ ¹æ�®Schema和问题生æˆ�高质é‡�SQL
|
|||
|
|
é N)ÚDictÚOptional)Ú ChatAgent)Ú ChatModel)ÚSQL_GENERATOR_SYSTEMc óº — e Zd ZdZ ddedee defd„Z ddeded eee dee d
|
|||
|
|
ef
|
|||
|
|
d „Z
|
|||
|
|
d ed
|
|||
|
|
efd
|
|||
|
|
„Z ded ee d
|
|||
|
|
efd„Z ddededee d
|
|||
|
|
e
|
|||
|
|
eef fd„Zy)ÚSQLGeneratorAgentuÎ
|
|||
|
|
SQL Generator Agent
|
|||
|
|
|
|||
|
|
�责：
|
|||
|
|
- �解用户问题和Schema结构
|
|||
|
|
- 生æˆ�准确的SQLè¯å�¥
|
|||
|
|
- 处ç�†å¤�æ�‚çš„JOINã€�è�šå�ˆã€�å�查询
|
|||
|
|
- �循金�/�券业务特殊规则
|
|||
|
|
NÚmodelÚsystem_messageÚdialectc óš — || _ |xs t | _ t | j |¬« | _ t
|
|||
|
|
j
|
|||
|
|
d|› d�« y)u¢
|
|||
|
|
�始化Agent
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
model: CAMEL AI模型实例
|
|||
|
|
system_message: 系统�示�
|
|||
|
|
dialect: SQL方言
|
|||
|
|
)r r
|
|||
|
|
u. ✅ SQLGeneratorAgentåˆ�始化完æˆ� (dialect=Ú)N)r r r r ÚagentÚloggerÚinfo)Úselfr
|
|||
|
|
r r s úagents/sql_generator.pyÚ__init__zSQLGeneratorAgent.__init__ sI € 𠈌 Ø ,Ò DÔ0DˆÔÜ Ø ×.Ñ.Øô
|
|||
|
|
ˆŒ
|
|||
|
|
ô � ‰ ÐDÀWÀIÈQÐOÕPó ÚquestionÚ
|
|||
|
|
schema_strÚexamplesÚreturnc ó� — ddl m} |xs | j }|j |||¬« }|r| j ||« } | j
|
|||
|
|
j
|
|||
|
|
|« }| j |j j « }t j d|dd › d�« |S # t $ r } t j d| › �« ‚ d} ~ ww xY w) u
|
|||
|
|
生�SQL
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
question: 用户问题
|
|||
|
|
schema_str: Schemaæ��è¿°å—符串
|
|||
|
|
examples: Few-shot示例列表
|
|||
|
|
dialect: 覆盖默认dialect
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
SQLè¯å�¥å—符串
|
|||
|
|
r ©ÚSQL_GENERATOR_USER©Úschemar r u 生æˆ�çš„SQL: NéÈ z...u SQL生æˆ�失败: )Úconfig.promptsr r ÚformatÚ_inject_examplesr ÚstepÚ _extract_sqlÚmsgÚcontentr ÚdebugÚ ExceptionÚerror)
|
|||
|
|
r r r r r r ÚpromptÚresponseÚsqlÚes
|
|||
|
|
r ÚgeneratezSQLGeneratorAgent.generate3 s € õ& 6àÒ)˜TŸ\™\ˆØ#×*Ñ*Ø Ø Øð +ó
|
|||
|
|
ˆñ Ø×*Ñ*¨6°8Ó<ˆFð Ø—z‘z—‘ vÓ.ˆHØ×#Ñ# H§L¡L×$8Ñ$8Ó9ˆCä �L‰L˜>¨#¨d¨s¨)¨°CÐ8Ô 9؈Jøäò Ü �L‰LÐ,¨Q¨CÐ0Ô 1Ø ûð ús Á A B  CÂ'C à Cr& c ó¢ — |j « }d|v r>|j d« dz }|j d|« }|dk7 r||| j « S d|v r>|j d« dz }|j d|« }|dk7 r||| j « S t j |« }d|v r|d S |S # t j $ r Y |S w xY w)u
|
|||
|
|
从Agentå“�应ä¸æ��å�–SQLè¯å�¥
|
|||
|
|
|
|||
|
|
处�：
|
|||
|
|
- Markdown代ç �å�— (```sql ... ```)
|
|||
|
|
- 纯SQL文本
|
|||
|
|
- JSONæ ¼å¼� {"sql": "..."}
|
|||
|
|
z```sqlé ú```éÿÿÿÿé r, )ÚstripÚfindÚjsonÚloadsÚJSONDecodeError)r r& ÚstartÚendÚdatas r r$ z SQLGeneratorAgent._extract_sql^ sô € ð —-‘-“/ˆð �wÑ Ø—L‘L Ó*¨QÑ.ˆEØ—,‘,˜u eÓ,ˆCØ�bŠyØ ˜u SÐ)×/Ñ/Ó1Ð1ð �GÑ Ø—L‘L Ó'¨!Ñ+ˆEØ—,‘,˜u eÓ,ˆCØ�bŠyØ ˜u SÐ)×/Ñ/Ó1Ð1ð Ü—:‘:˜gÓ&ˆDؘ‰}ؘE‘{Ð"ð ð ˆøô ×#Ñ#ò Ø ð ˆð ús  B7 Â7CÃ
|
|||
|
|
Cr* c
|
|||
|
|
ó˜ — g }|dd D ]( }|j d|d › d|d dd › d|d › �« Œ* d
|
|||
|
|
j |« }|› d |› d �S )
|
|||
|
|
uû
|
|||
|
|
注入few-shot示例到�示�
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
prompt: 原始�示�
|
|||
|
|
examples: 示例列表，�项为 {"question": "...", "schema": "...", "sql": "..."}
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
增强�的�示�
|
|||
|
|
Nr3 u 示例：
|
|||
|
|
问题：r z
|
|||
|
|
Schema: r r z ...
|
|||
|
|
SQL: r, z
|
|||
|
|
|
|||
|
|
u
|
|||
|
|
|
|||
|
|
�考示例：
|
|||
|
|
u
|
|||
|
|
|
|||
|
|
请生æˆ�SQL：)ÚappendÚjoin)r r* r Ú
|
|||
|
|
examples_textÚexÚexamples_blocks r r" z"SQLGeneratorAgent._inject_examples‚ s„ € ð ˆ
|
|||
|
|
ؘ2˜A“,ˆBØ × Ñ Ø& r¨*¡~Ð&6ð 7Ø ˜h™<¨¨Ð-Ð.ð /ؘ5™ �{ð$õ
|
|||
|
|
ð ð Ÿ™ ]Ó3ˆð �Ð.¨~Ð.>Ð>QÐRÐRr c óŒ — ddl m} |xs | j }dt j |||¬« › d�} | j
|
|||
|
|
j
|
|||
|
|
|« }|j j j « } d|v r9|j d« dz }|j d|« } ||| j « }t j |« }
|
|||
|
|
|
|
|||
|
|
j d d
|
|||
|
|
« |
|
|||
|
|
j d d
|
|||
|
|
« fS # t j t f$ r | j! |« d
|
|||
|
|
fcY S w xY w# t" $ r } t$ j' d | › �« ‚ d
|
|||
|
|
} ~ ww xY w)uÒ
|
|||
|
|
生�SQL并返回解释
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
question: 用户问题
|
|||
|
|
schema_str: Schema�述
|
|||
|
|
dialect: SQL方言
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
(SQLè¯å�¥, 解释)
|
|||
|
|
r r Ú
|
|||
|
|
r uu
|
|||
|
|
|
|||
|
|
请å�Œæ—¶è¾“出SQL和简è¦�解释（JSONæ ¼å¼�）：
|
|||
|
|
{
|
|||
|
|
"sql": "SELECT ...",
|