ai-g3sb-backman2.0/scripts/__pycache__/build_fewshot_chroma_index.cpython-312.pyc

Ë
?êÞiVã
ó’
—dZd
dl
mZ
d
dlZd
dlZd
dlZd
dlZd
dlZd
dlm	Z	
e	e
«
j«jdZ
e
dzZee«
ej v
r"ej j#d
ee«
«
ej$ej&dd¬	«
ej(d
«
Zd
d„Zedk(r
ee««
‚
y)uÒ

将 Few-shot �验数�集（JSONL）写入 Chroma �久化��库。

用法（仓库根目录）：
  python scripts/build_fewshot_chroma_index.py
  python scripts/build_fewshot_chroma_index.py --samples data/experiences/all_samples.jsonl --force

构建完��在 .env 中设置：
  FEWSHOT_USE_CHROMA=true
  FEWSHOT_CHROMA_PATH=./data/embeddings/chroma_fewshot

Embedding 与 Schema ��一致，由 OPENAI_* / MODELSCOPE_* / DASHSCOPE_* 等决定。
é)
ÚannotationsN)
ÚPathé
Úbackendz1%(asctime)s [%(levelname)s] %(name)s: %(message)sz%Y-%m-%d %H:%M:%S)ÚlevelÚformatÚdatefmtÚbuild_fewshot_chromac	
ó¶—t
jd
¬«
}|jdtjdd«d¬«
|jdtjd	d
«d¬«
|jdd
d¬«
|j«}
t
|
j«
}|j«s	t|z}|j«stjd|«
yddl
m}
|tdz«

ddlm}
ddlm}
ddlm}
g}|j+dd¬«5}t-|d«D]L\}	}
|
j/«}
|
s
Œ	|j1|j3t5j6|
«
«
«

ŒN	ddd«
|stjd«

y|«}t
|
j<«
}
|
j«s	t|
z}
||t?|
«
d¬«}|jA||
jB¬«}tjEd ||
«
y#t4j8$r"}tj;d|	|«
Yd}~�
Œd}~w
wxYw
#1s
w
Y

Œ×xYw
)!Nu&Few-shot JSONL → Chroma ��索引)
Údescriptionz	--samplesÚFEWSHOT_DATA_PATHz"data/experiences/all_samples.jsonlu0JSONL 路径（相对仓库根或ç»�对路径）)ÚdefaultÚhelpz
--persist-dirÚFEWSHOT_CHROMA_PATHzdata/embeddings/chroma_fewshotuChroma æŒ�久化目录z--forceÚ
store_trueu!清空已有集å�ˆå¹¶å…¨é‡�é‡�建)Úactionru样本文件ä¸�存在: %srr)
Úload_dotenvz.env)
Úget_embedder)
ÚFewShotChromaStore)
ÚExperienceSampleÚ
rzutf-8)
Úencodingu 跳过第 %s è¡Œ JSON 错误: %su未解æž�到任何样本T)Úpersist_dirÚpersist_to_disk)
Ú
force_rebuildu完æˆ�: 写入 %s æ�¡ → %s)#ÚargparseÚArgumentParserÚadd_argumentÚosÚgetenvÚ
parse_argsrÚsamplesÚis_absoluteÚ
_REPO_ROOTÚis_fileÚloggerÚerrorÚdotenvrÚutils.embeddingrÚutils.fewshot_chroma_storerÚutils.fewshot_selectorrÚopenÚ	enumerateÚstripÚappendÚ	from_dictÚjsonÚloadsÚJSONDecodeErrorÚwarningrÚstrÚbuild_from_samplesÚforceÚinfo)ÚparserÚargsÚsamples_pathrrrrÚrowsÚ
fÚline_noÚlineÚ
eÚembedderÚpersistÚstoreÚ
ns                ú%scripts/build_fewshot_chroma_index.pyÚmainrF&s€Ü
×