74 lines
15 KiB
Plaintext
74 lines
15 KiB
Plaintext
|
Ë
|
|||
|
|
Na×i+ ã óP — d Z ddlZddlZddlZddlZddlmZ ddlmZ ddl m
|
|||
|
|
Z
|
|||
|
|
ej ej dd¬« ej e« Zd Zd
|
|||
|
|
„ Zd efd „Zd
|
|||
|
|
„ Zded ee fd„Zd dedee fd„Zd„ Zd dedefd„Zd defd„Zd dedefd„Zd„ Z edk( r e « yy) uÿ
|
|||
|
|
Text2SQL 多智能体系统 - CLI演示入�
|
|||
|
|
|
|||
|
|
用法：
|
|||
|
|
python main.py "查询2024年1月销售�最高的�5个产�"
|
|||
|
|
python main.py --question "查询所有状�为Active的账户数�"
|
|||
|
|
python main.py --interactive # 交互�模�
|
|||
|
|
é N)ÚPath)ÚOptional)Ú load_dotenvz1%(asctime)s [%(levelname)s] %(name)s: %(message)sz%Y-%m-%d %H:%M:%S)ÚlevelÚformatÚdatefmtz"./data/models/Qwen3-Embedding-0.6Bc ój — t t t « j « j dz « y)uU åŠ è½½é¡¹ç›®æ ¹ç›®å½• .env（与 main.py å�Œç›®å½•ï¼‰ï¼Œä¾›å�Žç» os.getenv 使用。z.envN)r r Ú__file__ÚresolveÚparent© ó ú;C:\Users\24019\Desktop\backman\text2sql_agent_camel\main.pyÚ_load_project_envr s" € ä””X“×&Ñ&Ó(×/Ñ/°&Ñ8Õ9r Úreturnc óR — t j dt « j « S )NÚEMBEDDING_MODEL_PATH)ÚosÚgetenvÚ_DEFAULT_EMBEDDING_PATHÚstripr
|
|||
|
|
r r Ú_embedding_model_pathr % s € Ü
|
|||
|
|
�9‰9Ð+Ô-DÓ E× KÑ KÓ MÐMr c ó$ — t « t j dd« j « j « dv } | r€t t
|
|||
|
|
« « }|j « �s#t j d|› �« t j d« t j d|› d�« t j d« y t j d
|
|||
|
|
d « j « st j d « y t j d
|
|||
|
|
« xs t j dd « j « }|st j d« y t j dd « j « st j d« y t d« }|j « s.t j d|› �« t j d« y t j d« s+t j d« t j d« y y)u 环境检查ÚUSE_LOCAL_EMBEDDINGÚtrue)Ú1r ÚyesÚonu Embedding模型ä¸�å˜åœ¨: u 请先下载模型：zG modelscope download --model 'Qwen/Qwen3-Embedding-0.6B' --local_dir 'Ú'u† 或使用 DashScope API：在 .env 设置 USE_LOCAL_EMBEDDING=false 并é…�ç½® DASHSCOPE_API_KEY / DASHSCOPE_BASE_URL / DASHSCOPE_MODELFÚDASHSCOPE_API_KEYÚ u8 USE_LOCAL_EMBEDDING=false 但未设置 DASHSCOPE_API_KEYÚDASHSCOPE_BASE_URLÚDASHSCOPE_base_urlu8 未设置 DASHSCOPE_BASE_URL（或 DASHSCOPE_base_url）ÚDASHSCOPE_MODELu 未设置 DASHSCOPE_MODELú9./data/schemas/G3SB_MCDataDictionary_table_structure.jsonu Schema文件ä¸�å˜åœ¨: u7 请将G3SB Schema文件放置在 ./data/schemas/ 目录ÚDEEPSEEK_API_KEYu' 环境å�˜é‡� DEEPSEEK_API_KEY 未设置uB 请在 .env 文件ä¸é…�置，或 export DEEPSEEK_API_KEY=your_keyT) r r r r Úlowerr r ÚexistsÚloggerÚwarningÚinfo)Ú
|
|||
|
|
use_local_embÚ
|
|||
|
|
model_pathÚbaseÚ schema_paths r Úsetup_environmentr0 ) s¢ € äÔÜ—I‘IÐ 3°VÓ<×BÑBÓD×JÑJÓLð Qð €Mñ äÔ/Ó1Ó2ˆ
|
|||
|
|
Ø× Ñ Õ"Ü �N‰NÐ 7¸
|
|||
|
|
°|ÐDÔ EÜ �K‰KÐ/Ô 0Ü �K‰Kð (Ø(2 |°1ð6ô
|
|||
|
|
7ä �K‰Kð Sô
|
|||
|
|
Tàä�y‰yÐ,¨bÓ1×7Ñ7Ô9Ü �N‰NÐUÔ VØÜ— ‘ Ð.Ó/ÒV´2·9±9Ð=QÐSUÓ3V×]Ñ]Ó_ˆÙÜ �N‰NÐUÔ VØÜ�y‰yÐ*¨BÓ/×5Ñ5Ô7Ü �N‰NÐ6Ô 7Øô ÐRÓS€KØ × Ñ Ô Ü�‰Ð0°°
|
|||
|
|
Ð>Ô?Ü� ‰ ÐMÔNØô �9‰9Ð'Ô (Ü�‰Ð@ÔAÜ� ‰ ÐXÔYØà r Ústructure_pathc ó — t | « }d|j vry|j |j j dd« z }|j « r t |« S dS )uM è‹¥å˜åœ¨ä¸Ž table_structure å�Œå��çš„ table_meta 文件则返回其路径。Útable_structureNÚ
|
|||
|
|
table_meta)r Únamer ÚreplaceÚis_fileÚstr)r1 ÚpÚcands r Ú_default_g3sb_meta_pathr; V sP € ä ˆ^Ó €AØ §¡Ñ&ØØ �8‰8�a—f‘f—n‘nÐ%6¸ ÓEÑ E€DØ Ÿ ™ œŒ3ˆt‹9Ð0¨DÐ0r r/ Úschema_meta_pathc ó* — ddl m} |�|n
|
|||
|
|
t | « }t j d| › �« |rt j d|› �« |j | |¬« }|j
|
|||
|
|
« }t j d|d › d|d › d
|
|||
|
|
|d › d �« |S )
|
|||
|
|
uh åŠ è½½ Schemaï¼›G3SB structure JSON 会自动å°�试é…�对 table_meta（å�¯ç”¨ --schema-meta 指定）。r )Ú
|
|||
|
|
SchemaManageru åŠ è½½Schema: u 表注释(meta): )Úg3sb_meta_pathu ✅ SchemaåŠ è½½å®Œæˆ�: Údatabaseu , å…±Ú total_tablesu å¼ è¡¨, Ú
|
|||
|
|
total_columnsu 个å—段)Úschema.managerr> r; r) r+ Úload_from_jsonÚget_statistics)r/ r< r> ÚmetaÚ
|
|||
|
|
schema_mgrÚstatss r Ú load_schemarI _ sº € å,ð Ð 'ñ ä
|
|||
|
|
$ [Ó
|
|||
|
|
1ð ô
|
|||
|
|
‡K�K�.
|
|||
|
|
Ð.Ô/Ù Ü� ‰ Ð)¨$¨Ð0Ô1à ×-Ñ-Ø Dð .ó €Jð
|
|||
|
|
× %Ñ %Ó '€EÜ
|
|||
|
|
‡K�KØ
|
|||
|
|
" 5¨Ñ#4Ð"5ð 6Ø�NÑ#Ð
|
|||
|
|
$ H¨U°?Ñ-CÐ,DÀIð Oôð
|
|||
|
|
Ðr c ó¼ — ddl m} ddlm} |j xs t j d« xs dj « }|s t d« ‚t j d« xs dj « } ||||j xs d |j |j ¬
|
|||
|
|
« } || ||j |j |j |j ¬ « }|S ) u 创建编排器r )ÚText2SQLOrchestrator)ÚDeepSeekConfigr& r! uh 未é…�ç½® DeepSeek API Key：请在 .env ä¸è®¾ç½® DEEPSEEK_API_KEY，或使用命令行å�‚æ•° --api-keyÚDEEPSEEK_BASE_URLzhttps://api.deepseek.comú
|
|||
|
|
deepseek-chat)Úapi_keyÚbase_urlÚ
|
|||
|
|
model_nameÚ temperatureÚ
|
|||
|
|
max_tokens)Úschema_managerÚdeepseek_configÚembedding_model_pathÚvector_db_pathÚ max_retryÚuse_vector_search)Úagents.orchestratorrK Úllm.deepseek_clientrL rO r r r Ú
|
|||
|
|
ValueErrorÚmodelrR rS Úembedding_modelÚ vector_dbrX Úno_vector_search)rG ÚargsrK rL rO rP ÚconfigÚ orchestrators r Úcreate_orchestratorrd y sÌ € å8Ý2à�|‰|ÒBœrŸy™yÐ);Ó <ÒBÀ×IÑIÓK€GÙ Üð
|
|||
|
|
1ó
|
|||
|
|
ð
|
|||
|
|
ô — ‘ Ð-Ó.ÒLÐ2L×SÑSÓU€Há
|
|||
|
|
ØØØ—:‘:Ò0 Ø×$Ñ$Ø—?‘?ô €Fñ (Ø!Ø Ø !× 1Ñ 1Ø—~‘~Ø—.‘.Ø "× 3Ñ 3Ð3ô
|
|||
|
|
€Lð Ðr ÚquestionÚdialectc óx — ddl }t j d|› �« |j « }| j ||d¬« }|j « |z
|
|||
|
|
}t d« t d« t d« |j
|
|||
|
|
r3t d |d
|
|||
|
|
›d |j › d �« t |j « n8t d
|
|||
|
|
|j › d�« |j D ] }t d|› �« Œ |j r*t d« |j D ] }t d|› �« Œ t d|j › �« |S )u å�•æ¬¡æŸ¥è¯¢r Nu
|
|||
|
|
💬 问题: é )re rf Útop_k_candidatesú=
|
|||
|
|
============================================================u
|
|||
|
|
生æˆ�结果:ú<============================================================u ✅ SQL (耗时 z.2fu
|
|||
|
|
s, �试 u 次):
|
|||
|
|
u � 生�失败 (�试 u 次)z - u
|
|||
|
|
âš ï¸� è¦å‘Š:u
|
|||
|
|
使用表: ) Útimer) r+ ÚgenerateÚprintÚvalidÚattemptsÚsqlÚerrorsÚwarningsÚ tables_used) rc re rf rl ÚstartÚresultÚelapsedÚerrÚws r Ú single_queryrz š s € ãä
|
|||
|
|
‡K�K�- ˜zÐ*Ô+à �I‰I‹K€EØ
|
|||
|
|
×
|
|||
|
|
"Ñ
|
|||
|
|
"ØØØð #ó €Fð
|
|||
|
|
�i‰i‹k˜EÑ!€Gä ˆ/ÔÜ ˆ/ÔÜ ˆ(„Oà
|
|||
|
|
‡|‚|Ü
|
|||
|
|
Ð ¨
|
|||
|
|
¨Z¸¿¹Ð7HÈÐQÔRÜ
|
|||
|
|
ˆf�j‰jÕä
|
|||
|
|
Ð)¨&¯/©/Ð):¸%Ð@ÔAØ—=”=ˆCÜ �D˜˜�,Õ ð !𠇂Ü
|
|||
|
|
Ð!Ô"Ø— ” ˆAÜ �D˜˜�*Õ ð !ô
|