ai-g3sb-backman2.0/__pycache__/main.cpython-312.pyc

Ë
	ÊØi80ãóN
—dZd
dl
Z
d
dlZd
dlZd
dlZd
dlmZ
d
dlmZ
d
dl	m
Z

ejejdd¬«
eje«
Zd	Zd
„Zdefd„Zd
edefd„Zd„Zdedeefd„Zddedeefd„Zd„Zddedefd„Zddefd„Zd„Zedk(re«
yy)uC
Text2SQL 多智能体系统 - CLI 入�

用法：python main.py
é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
tt«
j«jd
z«

y)uU加载项目根目录 .env（与 main.py å�Œç›®å½•ï¼‰ï¼Œä¾›å�Žç»­ os.getenv 使用。z.envN)rrÚ__file__ÚresolveÚparent©óúAC:\Users\24019\Desktop\backman-camel\text2sql_agent_camel\main.pyÚ_load_project_envrs"€ä””X“×&Ñ&Ó(×/Ñ/°&Ñ8Õ9rÚreturncóR—t
jd
t«j«S)NÚEMBEDDING_MODEL_PATH)ÚosÚgetenvÚ_DEFAULT_EMBEDDING_PATHÚstripr
rrÚ_embedding_model_pathr!s€Ü
�9‰9Ð+Ô-DÓE×KÑKÓMÐMrÚnamec
óT—|x
s
d
j
«j«}
|
dvr
y|
S)uTCLI / é…�置中的方言别å��统一为 sqlglot 方言å��（SQL Server -> tsql）。Ú	sqlserver)rÚmssqlÚtsql)Úlowerr)rÚ
ns  rÚresolve_sql_dialectr %s/€à	
Ò	�×#Ñ#Ó%×+Ñ+Ó-€AØÐ"Ñ"ØØ€Hrcó†—t
«
tjd
d«j«j	«dv}|r€tt
««
}
|
j«�
sÔtjd|
›�«

tjd«

tjd|
›d�«

tjd«

y	tjd
d«j«}tjdd«j«}|x
r%
|jd
«
x
s
|jd«
}|r�
n|r:tjdd«j«sÝtjd«

y	tjdd«j«stjd«

y	tjd«
x
s
tjdd«j«}|stjd«

y	tjdd«j«stjd«

y	td«
}|j«s.tjd|›�«

tjd«

y	tjd«
s+tjd«

tjd«

y	y)u环境检查ÚUSE_LOCAL_EMBEDDINGÚtrue)Ú
1r#ÚyesÚonuEmbedding模型ä¸�存在: u请先下载模型：zG  modelscope download --model 'Qwen/Qwen3-Embedding-0.6B' --local_dir 'Ú
'uÁ或使用远程 Embedding API：USE_LOCAL_EMBEDDING=false，并é…�ç½® MODELSCOPE_API_KEYã€�或 OPENAI_API_KEY+OPENAI_EMBEDDING_MODEL（å�Šå�¯é€‰ OPENAI_BASE_URL）ã€�或 DASHSCOPE_*（百炼）FÚMODELSCOPE_API_KEYÚÚOPENAI_API_KEYzhttp://zhttps://ÚOPENAI_EMBEDDING_MODELu=USE_LOCAL_EMBEDDING=false 但未设置 OPENAI_EMBEDDING_MODELÚDASHSCOPE_API_KEYuwUSE_LOCAL_EMBEDDING=false 但未设置 MODELSCOPE_API_KEYã€�OPENAI_API_KEY+OPENAI_EMBEDDING_MODEL 或 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.jsonuSchema文件ä¸�存在: u7请将G3SB Schema文件放置在 ./data/schemas/ 目录ÚDEEPSEEK_API_KEYu'环境å�˜é‡� DEEPSEEK_API_KEY 未设置uB请在 .env 文件中é…�置，或 export DEEPSEEK_API_KEY=your_keyT)rrrrrrrÚexistsÚloggerÚwarningÚinfoÚ
startswith)Ú
use_local_embÚ
model_pathÚms_keyÚoa_keyÚ	oa_key_okÚbaseÚschema_paths       rÚsetup_environmentr>-sB€äÔÜ—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ðX
ô
Y
ðä—‘Ð/°Ó4×:Ñ:Ó<ˆÜ—‘Ð+¨RÓ0×6Ñ6Ó8ˆØò
Ø×Ñ˜iÓ(ÒI¨F×,=Ñ,=¸jÓ,Ið 
ˆ	ñÙÙ
Ü—9‘9Ð5°rÓ:×@Ñ@ÔBÜ—‘Ð^Ô_Øä—9‘9Ð0°"Ó5×;Ñ;Ô=Ü—‘ð
R
ôðä—	‘	Ð.Ó/ÒV´2·9±9Ð=QÐSUÓ3V߉e‹gð
ñÜ—‘ÐYÔZØÜ—9‘9Ð.°Ó3×9Ñ9Ô;Ü—‘Ð:Ô;ØôÐRÓS€KØ×ÑÔÜ�‰Ð0°°
Ð>Ô?Ü�‰ÐMÔNØô�9‰9Ð'Ô(Ü�‰Ð@ÔAÜ�‰ÐXÔYØàrÚstructure_pathc
ó—t
|«
}
d
|
jv
r
y|
j|
jjd
d«z}|j	«rt|«
SdS)uM若存在与 table_structure å�Œå��çš„ table_meta 文件则返回其路径。Útable_structureNÚ
table_meta)rrrÚreplaceÚis_fileÚstr)r?Ú
pÚcands   rÚ_default_g3sb_meta_pathrHlsP€äˆ^Ó€AØ 
§¡Ñ&ØØ�8‰8�a—f‘f—n‘nÐ%6¸ÓEÑE€DØŸ™œŒ3ˆt‹9Ð0¨DÐ0rr=Úschema_meta_pathc	ó*
—d
dlm
}
|
�|
n
t|«
}tj	d|›�«

|rtj	d|›�«

|j||¬«}|j
«}tj	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[OK] Schema加载完æˆ�: Údatabaseu, å…±Útotal_tablesu张表, Ú
total_columnsu	个字段)Úschema.managerrKrHr3r5Úload_from_jsonÚget_statistics)r=rIrKÚmetaÚ
schema_mgrÚstatss      rÚload_schemarVusº€å,ðÐ'ñ	ä
$ [Ó
1ð	ô
‡K�K�.  
Ð.Ô/ÙÜ�‰Ð)¨$¨Ð0Ô1à×-Ñ-Ø Dð.ó€Jð
×%Ñ%Ó'€EÜ
‡K�KØ
# E¨*Ñ$5Ð#6ð
7Ø�NÑ#Ð
$ H¨U°?Ñ-CÐ,DÀIð
	O
ôð
Ðrcó—d
dlm
}
d
dlm}
|
jx
s
tjd«
x
s
dj«}|std«
‚
tjd«
x
s
dj«}||||
jx
s
d	|
j|
j¬
«}||||
j|
j|
j|
j|
j |
j"|
j$¬«	}|S)u创建编排器r)
ÚText2SQLOrchestrator)
ÚDeepSeekConfigr1r)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Úfewshot_enabledÚ
fewshot_top_kÚfewshot_min_rating)Úagents.orchestratorrXÚllm.deepseek_clientrYr\rrrÚ
ValueErrorÚmodelr_r`Úembedding_modelÚ	vector_dbreÚno_vector_searchÚ
no_fewshotrhri)rTÚargsrXrYr\r]ÚconfigÚorchestrators        rÚcreate_orchestratorru�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à ŸO™OÐ+Ø×(Ñ(Ø×2Ñ2ô€LðÐrÚquestionÚdialectc
óØ—d
dl}d
dl
m}
d
dlm}m}
tjd|
›�«

||
«
}|j|jk(r�|jx
s
d}tjd«

td«

td	«

td
«

t|«

td«

|ddgggd
d
|jji
¬«S|j
«}	|j|
|d¬«}
|j
«|	z
}td«

td«

td
«

|
jr3td|d›d|
j›d�«

t|
j «

n8td|
j›d�«

|
j"D]}td|›�«

Œ|
j$r*td«

|
j$D]}
td|
›�«

Œtd|
j&›�«

|
S)u�次查询rN)
ÚGenerationResult)ÚDialogIntentÚclassify_dialogu[Q] 问题: r)u/[Q] æ„�图: conversation（跳过 SQL 生æˆ�）ú=
============================================================u2对� / �查询输入（未触� SQL 生�）z<============================================================u
使用表: []FÚ
dialog_intent)ÚsqlÚvalidÚerrorsÚwarningsÚtables_usedÚattemptsÚmetadataé)rvrwÚtop_k_candidatesu
生�结果:u[OK] SQL (耗时 z.2fu
s, �试 u 次):
u[FAIL] 生�失败 (�试 u 次)z  - u
[WARN] 警告:u
使用表: )ÚtimerjryÚutils.dialog_classifierrzr{r3r5ÚintentÚCONVERSATIONÚreply_suggestionÚprintÚvalueÚgeneraterrƒr~r€r�r‚)rtrvrwr‡ryrzr{Ú
classifiedÚreplyÚstartÚresultÚelapsedÚerrÚ
ws              rÚsingle_queryr–´s·
€ãå4ßEä
‡K�K�,˜x˜jÐ)Ô*á  Ó*€JØ×јL×5Ñ5Ò5Ø×+Ñ+Ò1¨rˆÜ�‰ÐEÔFÜ
ˆoÔÜ
ÐBÔCÜ
ˆhŒÜ
ˆeŒÜ
�Ô!ÙØØØØØØØ% |×'@Ñ'@×'FÑ'FÐGô
ð	
ð