80 lines
9.0 KiB
Plaintext
80 lines
9.0 KiB
Plaintext
|
Ë
|
|||
|
|
jD×iQ ã ó´ — d Z ddlmZmZ ddlmZmZmZmZ ddl m Z e G d„ d« « Z
|
|||
|
|
e G d„ d« « Z e G d „ d
|
|||
|
|
« « Z e G d „ d « « Z
|
|||
|
|
y
|
|||
|
|
)u
|
|||
|
|
Schema数�模型定义
|
|||
|
|
é )Ú dataclassÚfield)ÚListÚOptionalÚDictÚAny)Údatetimec ój — e Zd ZU dZeed< eed< dZee ed< dZe ed< dZ
|
|||
|
|
e ed < dZ e ed
|
|||
|
|
< d „ Z y) ÚColumnu 列å—段定义ÚnameÚ data_typeNÚcommentTÚnullableFÚis_primary_keyÚis_foreign_keyc ó¾ — | j rdnd}| j rdnd}| j rd| j › �nd}| j › d| j › |› |› |› �S )NÚNULLzNOT NULLz PKÚ ú -- Ú )r r r r r
|
|||
|
|
)ÚselfÚnull_strÚpk_strÚ comment_strs úDC:\Users\24019\Desktop\backman\text2sql_agent_camel\schema\models.pyÚ__str__zColumn.__str__ s\ € Ø !Ÿ]š]‘6°
|
|||
|
|
ˆØ ×-Ò-‘°2ˆØ/3¯|ª|˜˜TŸ\™\˜NÑ+Àˆ Ø—)‘)�˜A˜dŸn™nÐ -¨h¨Z¸°xÀ ¸}ÐMÐMó )
|
|||
|
|
Ú__name__Ú
|
|||
|
|
__module__Ú __qualname__Ú__doc__ÚstrÚ__annotations__r r r Úboolr r r © r r r r
|
|||
|
|
sA … áØ
|
|||
|
|
|
|||
|
|
ƒI؃NØ !€GˆX�c‰]Ó!Ø€HˆdÓØ €N�DÓ Ø €N�DÓ óNr r c ó< — e Zd ZU dZee ed< eed< ee ed< y)Ú
|
|||
|
|
ForeignKeyu 外键关系ÚcolumnsÚ ref_tableÚ ref_columnsN)r r r r! r r" r# r% r r r' r' s … áØ
|
|||
|
|
�#‰YÓ؃NØ�c‘Ôr r' c ó — e Zd ZU dZeed< dZee ed< ee ¬« Z
|
|||
|
|
e e ed< ee ¬« Z
|
|||
|
|
e e ed< ee ¬« Ze e ed< dZee ed < ed
|
|||
|
|
eee f fd „« Zed
|
|||
|
|
e e fd „« Zdd
|
|||
|
|
ed
|
|||
|
|
efd„Zd
|
|||
|
|
efd„Zy)ÚTableu æ•°æ�®è¡¨å®šä¹‰r Nr ©Údefault_factoryr( Ú primary_keysÚ foreign_keysÚ embeddingÚreturnc óV — | j D �ci c] }|j |“Œ c}S c c}w )u! å—段å��到Columnå¯¹è±¡çš„æ˜ å°„©r( r ©r Úcols r Ú column_dictzTable.column_dict. s( € ð *.¯ ª Ó6© #�—‘˜#‘
|
|||
|
|
¨ Ñ6Ð6ùÒ6ó �&c óT — | j D �cg c] }|j ‘Œ c}S c c}w )u 所有å—段å��列表r4 r5 s r Ú column_nameszTable.column_names3 s$ € ð %)§L¢LÓ1¡L˜S�—“ LÑ1Ð1ùÒ1s �%Ú max_columnsc ó^ — d| j › d�g}t | j « |kD r| j d| n | j }g }|D ]^ }d|j › d|j › �}|j r|dz
|
|||
|
|
}|j
|
|||
|
|
r|d|j
|
|||
|
|
› �z
|
|||
|
|
}|j
|
|||
|
|
|« Œ` |j
|
|||
|
|
dj |« « t | j « |kD r+|j
|
|||
|
|
d t | j « |z
|
|||
|
|
› d
|
|||
|
|
�« |j
|
|||
|
|
d « d j |« S )
|
|||
|
|
uÌ
|
|||
|
|
生æˆ�紧凑的Schemaå—符串（用于LLM输入）
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
max_columns: 最大å—段数，超过时截æ–
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
紧凑的Schemaæ��è¿°å—符串
|
|||
|
|
zTABLE z (Nú ú: ú [PK]r z,
|
|||
|
|
u
|
|||
|
|
|
|||
|
|
-- 还有 u 个å—段未显示Ú)r )r Úlenr( r
|
|||
|
|
r r ÚappendÚjoin)r r; ÚpartsÚdisplay_columnsÚcol_strsr6 Úcol_strs r Úto_compact_stringzTable.to_compact_string8 s € ð ˜$Ÿ)™)˜ BÐ'Ð(ˆô 9<¸D¿L¹LÓ8IÈKÒ8W˜$Ÿ,™, |¨ Ñ4Ð]a×]iÑ]iˆàˆÛ"ˆCؘ3Ÿ8™8˜* B s§}¡} oÐ6ˆGØ×!Ò!ؘ7Ñ"�Ø�{Š{ؘT #§+¡+ Ð/Ñ/�Ø �O‰O˜GÕ $ð
|
|||
|
|
#ð � ‰ �U—Z‘Z Ó)Ô*ä ˆt�|‰|Ó ˜{Ò *Ø �L‰L˜>¬#¨d¯l©lÓ*;¸kÑ*IÐ)JÐJ]Ð^Ô _à
|
|||
|
|
� ‰ �SÔØ�w‰w�u‹~Ð r c óv — | j | j | j D �cg c]< }|j |j |j |j |j
|
|||
|
|
dœ‘Œ> c}| j | j D �cg c]& }|j |j |j dœ‘Œ( c}dœS c c}w c c}w )u 转æ�¢ä¸ºå—å…¸æ ¼å¼�)r r
|
|||
|
|
r r r )r( r) r* )r r r( r/ r0 )
|
|||
|
|
r r r( r
|
|||
|
|
r r r/ r0 r) r* )r r6 Úfks r Úto_dictz
|
|||
|
|
Table.to_dictX s» € ð —I‘IØ—|‘|ð Ÿ<š<ó ñ (�Cð
|
|||
|
|
ŸH™HØ!$§ ¡ Ø "Ÿ{™{Ø #§ ¡ Ø&)×&8Ñ&8ó ð (ñ ð !× -Ñ -ð ×+Ò+ó
|
|||
|
|
ñ ,�Bð "Ÿz™zØ!#§ ¡ Ø#%§>¡>óð
|
|||
|
|
,ñ
|
|||
|
|
ñ
|
|||
|
|
ð
|
|||
|
|
ùò ùò s ¥AB1Â+B6 )é )r r r r! r" r# r r r Úlistr( r r r/ r0 r' r1 r Úpropertyr r7 r: ÚintrH rK r% r r r, r, # sÁ … áØ
|
|||
|
|
|
|||
|
|
ƒIØ !€GˆX�c‰]Ó!Ù !°$Ô 7€GˆT�&‰\Ó7Ù #°DÔ 9€L�$�s‘)Ó9Ù%*¸4Ô%@€L�$�zÑ"Ó@à #€Iˆx˜‰}Ó#à
|
|||
|
|
ð7˜T # v +Ñ .ò 7ó ð7ð ð2˜d 3™iò 2ó ð2ñ ¨Sð ¸#ó ð@
|
|||
|
|
˜ô
|
|||
|
|
r r, c ó — e Zd ZU dZeed< ee¬« Ze e
|
|||
|
|
ed< ee j ¬« Z
|
|||
|
|
e ed< edeee
|
|||
|
|
f fd„« Zdedee
|
|||
|
|
fd „Zdd
|
|||
|
|
edefd „Zdefd „Zy
|
|||
|
|
)ÚDatabaseSchemau' æ•°æ�®åº“Schema（多个表的集å�ˆï¼‰r r- ÚtablesÚ
|
|||
|
|
created_atr2 c óV — | j D �ci c] }|j |“Œ c}S c c}w )u 表å��å—å…¸)rR r ©r Útbls r Ú
|
|||
|
|
table_dictzDatabaseSchema.table_dictz s( € ð *.¯ªÓ5© #�—‘˜#‘
|
|||
|
|
¨Ñ5Ð5ùÒ5r8 Ú
|
|||
|
|
table_namec ó8 — | j j |« S )u 获å�–指定表)rW Úget)r rX s r Ú get_tablezDatabaseSchema.get_table s € à�‰×"Ñ" :Ó.Ð.r Úinclude_columnsc ó¶ — d| j › d�g}|j dt | j « › d�« |j d« | j D �]ï }|j d|j › �« |j r |j d|j › �« |j dt |j
|
|||
|
|
« › �« |râ|j
|
|||
|
|
rÖ|j d« |j
|