76 lines
6.1 KiB
Plaintext
76 lines
6.1 KiB
Plaintext
|
Ë
|
|||
|
|
ðD×i ã óŒ — d Z ddlZddlmZmZmZmZ ddlZddlm Z ddl
|
|||
|
|
m Z ddl m
|
|||
|
|
Z
|
|||
|
|
ej e« Z G d„ d« Zy) u^
|
|||
|
|
Schema Linker Agent - 表ç›é€‰ä¸“家
|
|||
|
|
从大é‡�æ•°æ�®è¡¨ä¸è¯†åˆ«ä¸Žç”¨æˆ·é—®é¢˜ç›¸å…³çš„表
|
|||
|
|
é N)ÚDictÚListÚTupleÚOptional)Ú ChatAgent)Ú ChatModel)ÚSCHEMA_LINKER_SYSTEMc
|
|||
|
|
ó¦ — e Zd ZdZ ddedee fd„Z ddedee dee eef d e
|
|||
|
|
d
|
|||
|
|
e ee ef f
|
|||
|
|
d „Z d ed
|
|||
|
|
e fd
|
|||
|
|
„Z
|
|||
|
|
ddee de
|
|||
|
|
d
|
|||
|
|
ee fd„Zy)ÚSchemaLinkerAgentu¡
|
|||
|
|
Schema Linker Agent
|
|||
|
|
|
|||
|
|
�责：
|
|||
|
|
- 分æž�用户问题ä¸çš„实体和æ„�图
|
|||
|
|
- 从候选表ä¸ç›é€‰çœŸæ£ç›¸å…³çš„表
|
|||
|
|
- �供选择�由
|
|||
|
|
NÚmodelÚsystem_messagec ó„ — |xs t | _ t | j |¬« | _ t j d« y)u©
|
|||
|
|
�始化Agent
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
model: CAMEL AI模型实例
|
|||
|
|
system_message: 系统�示�（默认使用SCHEMA_LINKER_SYSTEM）
|
|||
|
|
)r
|
|||
|
|
r u$ ✅ SchemaLinkerAgent�始化完�N)r r
|
|||
|
|
r ÚagentÚloggerÚinfo)Úselfr r
|
|||
|
|
s úagents/schema_linker.pyÚ__init__zSchemaLinkerAgent.__init__ s9 € ð -Ò DÔ0DˆÔÜ Ø ×.Ñ.Øô
|
|||
|
|
ˆŒ
|
|||
|
|
ô � ‰ Ð:Õ;ó ÚquestionÚcandidate_tablesÚtable_metadataÚ
|
|||
|
|
max_tablesÚreturnc óB — |r6dj |D �cg c] }d|› d|j |d« › �‘Œ c}« }n"dj |D �cg c] }d|› �‘Œ c}« }ddlm} |j ||¬« } | j
|
|||
|
|
j
|
|||
|
|
|« } | j j j « }
|
|||
|
|
| j |
|
|||
|
|
« } | j dg « } | j d d
|
|||
|
|
« }
|
|||
|
|
| d | } t j d t | « › d
|
|||
|
|
| › �« | |
|
|||
|
|
fS c c}w c c}w # t j $ r/}t j! d|› d
|
|||
|
|
d d › �« |d | dfcY d }~S d }~wt" $ r5}t j! d|› �« |d | dt% |« › �fcY d }~S d }~ww xY w)u9
|
|||
|
|
选择相关表
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
question: 用户问题
|
|||
|
|
candidate_tables: 候选表列表（粗ç›ç»“果）
|
|||
|
|
table_metadata: 表元数� {表�: 注释}
|
|||
|
|
max_tables: 最多返回表数�
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
(相关表列表, 推��由)
|
|||
|
|
Ú
|
|||
|
|
z- z: u æ— æ��è¿°r )ÚSCHEMA_LINKER_USER)r Ú
|
|||
|
|
table_listÚrelevant_tablesÚ reasoningÚ Nu SchemaLinkeré€‰ä¸ u å¼ è¡¨: u JSON解æž�失败: u , 原始内容: éÈ u% JSON解æž�失败，使用粗ç›ç»“æžœu Agent调用失败: u
|
|||
|
|
Agent错误: )ÚjoinÚgetÚconfig.promptsr Úformatr ÚstepÚmsgÚcontentÚstripÚ_parse_json_responser r ÚlenÚjsonÚJSONDecodeErrorÚerrorÚ ExceptionÚstr)r r r r r Útblr r ÚpromptÚresponser) Úresultr r Úes r Ú
|
|||
|
|
select_tablesz SchemaLinkerAgent.select_tables. sÆ € ñ( Ø Ÿ™á+ó$á+�Cð �S�E˜˜N× .Ñ .¨s°KÓ @ÐAÒBØ+ñ$ó ‰Jð
|
|||
|
|
Ÿ™Ñ:JÓ#KÑ:J°3 b¨¨¢JÐ:JÑ#KÓLˆJå5Ø#×*Ñ*Ø Ø!ð +ó
|
|||
|
|
ˆð
|
|||
|
|
KØ—z‘z—‘ vÓ.ˆHØ —l‘l×*Ñ*×0Ñ0Ó2ˆGð ×.Ñ.¨wÓ7ˆFà $Ÿj™jÐ):¸BÓ ?ˆOØ Ÿ
|
|||
|
|
™
|
|||
|
|
;°Ó3ˆIð .¨k¨zÐ :ˆOä �K‰KØ%¤c¨/Ó&:Ð%;¸9À_ÐDUÐVô
|
|||
|
|
ð # IÐ-Ð -ùò;$ùò
|
|||
|
|
$Løô4 ×#Ñ#ò ZÜ �L‰LÐ-¨a¨SÐ0@ÀÈÈ#À ÀÐPÔ Qà# K ZÐ0Ð2YÐYÕ YûÜò KÜ �L‰LÐ.¨q¨cÐ2Ô 3Ø# K ZÐ0°MÄ#ÀaÃ&ÀÐ2JÐJÕ Jûð Kús< ’ DÁ DÁ5B D Ä F Ä3$E ÅF Å F Å)*FÆF ÆF r) c ó8 — ddl }d|v r?|j d« dz }|j d|« }|dk7 rV||| j « }nBd|v r>|j d« dz }|j d|« }|dk7 r||| j « } |j |« S )uZ
|
|||
|
|
解�Agent的JSON�应
|
|||
|
|
|
|||
|
|
处ç�†å�¯èƒ½çš„markdown代ç �å�—包裹
|
|||
|
|
r Nz```jsoné z```éÿÿÿÿé )r- Úfindr* Úloads)r r) r- ÚstartÚends r r+ z&SchemaLinkerAgent._parse_json_responsej s¨ € ó ð ˜Ñ Ø—L‘L Ó+¨aÑ/ˆEØ—,‘,˜u eÓ,ˆCØ�bŠyØ! %¨Ð,×2Ñ2Ó4‘Ø
|
|||
|
|
�gÑ
|
|||
|
|
Ø—L‘L Ó'¨!Ñ+ˆEØ—,‘,˜u eÓ,ˆCØ�bŠyØ! %¨Ð,×2Ñ2Ó4�àˆt�z‰z˜'Ó"Ð"r Úselected_tablesÚdepthc óš — t |« }|D ]± }|j |« }|sŒ|j D ], }|j |vsŒ|j |j « Œ. |j « D ]M }|j D ]< }|j |k( sŒ|j |vsŒ"|j |j « Œ> ŒO Œ³ t |« S )u
|
|||
|
|
通过外键关系扩展表（备选方案，也å�¯åœ¨Orchestratorä¸å®Œæˆ�）
|
|||
|
|
|
|||
|
|
Args:
|