AI_NovelGenerator/__pycache__/novel_generator.cpython-310.pyc

o
ŠCšgp;ã@sÖ
dd
lZdd
l
Z
ddlmZmZmZ
zddlmZ
Wney)


ddlmZ
Yn
wddl	m
Z

ddlmZm
Z
mZ
ddl	mZ
ddlmZ
ddlmZ
dd	lmZmZmZmZ
dd
lmZmZmZmZmZmZm Z m!Z!m"Z"
e
j#e
j$dd�
d
Z%de&de&dee&defdd„Z'de&de&deefdd„Z(de&de&de&dd
fdd„Z)d/de&de&de&de*de&f
dd„Z+Gdd„deƒZ,de&de&d e&d!e&d"e&d#e*d$e*d%e&dd
fd&d'„Z-d(e&d)e&de&de&d*e&d+e*d%e&d$e*d,e&de&fd-d.„Z.d
S)0éN)ÚDictÚListÚOptional)
Ú	TypedDict)
Ú
ChatOpenAI)Ú
StateGraphÚSTARTÚEND)
ÚOpenAIEmbeddings)
ÚChroma©
ÚDocument)Ú	read_fileÚappend_text_to_fileÚclear_file_contentÚsave_string_to_txt)	Ú
set_promptÚcharacter_promptÚdark_lines_promptÚfinalize_setting_promptÚnovel_directory_promptÚsummary_promptÚupdate_character_state_promptÚchapter_outline_promptÚchapter_write_promptz'%(asctime)s [%(levelname)s] %(message)s)ÚlevelÚformatÚvectorstoreÚapi_keyÚbase_urlÚtextsÚreturncCs6t||
d
�}dd„|Dƒ
}t
j||td�}| ¡
|S)uŸ
    �始化并返回一个Chroma��库，将传入的文本进行嵌入并�存到本地目录。
    如果�存在该目录，会自动创建。
    ©Zopenai_api_keyZopenai_api_basec
Ssg|]}
t|
d�
‘qS)
©
Zpage_contentr)Ú.0Ú
t©r&ú;c:\Users\xieli\Desktop\AI_NovelGenerator\novel_generator.pyÚ
<listcomp>*sz%init_vector_store.<locals>.<listcomp>)Z	embeddingÚpersist_directory)r
rZfrom_documentsÚVECTOR_STORE_DIRÚpersist)rrr Ú
embeddingsZ	documentsrr&r&r'Úinit_vector_store!s

þ



ý
r-cCs(tj
 t¡
sd
St||
d�}tt|d�S)u;读å�–已存在的å�‘é‡�库。若ä¸�存在则返回 None。Nr")r)Zembedding_function)ÚosÚpathÚexistsr*r
r)rrr,r&r&r'Úload_vector_store4s



þr1Únew_chaptercCsLt||
ƒ}|st
 d
¡

t||
|g
ƒ
dSt|d�
}| |g
¡

| ¡
dS)uf将最新章节文本æ�’入到å�‘é‡�库里，用于å�Žç»­æ£€ç´¢å�‚考。若库ä¸�存在则åˆ�始化。z6Vector store does not exist. Initializing a new one...Nr#)r1ÚloggingÚinfor-r
Z
add_documentsr+)rrr2ÚstoreZnew_docr&r&r'Úupdate_vector_store?s








r6éÚqueryÚ
kcCsBt||
ƒ}|st
 d
¡

dS|j||d�}d dd„|Dƒ
¡
}|S)uˆ
    从��库中检索与 query 最相关的 k �文本，拼接�返回。
    若��库�存在则返回空字符串。
    z0Vector store not found. Returning empty context.Ú©
r9Ú

c
Ssg|]}
|
j‘qSr&r#)r$Ú
dr&r&r'r(Vsz:get_relevant_context_from_vector_store.<locals>.<listcomp>)r1r3ÚwarningZsimilarity_searchÚjoin)rrr8r9r5ZdocsZcombinedr&r&r'Ú&get_relevant_context_from_vector_storeLs







r@c@sVeZ
dZUeed
<eed<eed<eed<eed<eed<eed<eed<eed	<d
S)ÚOverallStateÚtopicÚgenreÚnumber_of_chaptersÚword_numberÚnovel_setting_baseÚcharacter_settingÚ
dark_linesÚfinal_novel_settingÚnovel_directoryN)Ú__name__Ú
__module__Ú__qualname__ÚstrÚ__annotations__Úintr&r&r&r'rA\s









rAÚ	llm_modelrBrCrDrEÚfilepathcsþ
tj
|d
d�
t|||
d�‰dtdtttff‡f
dd„}dtdtttff‡f
dd	„}	dtdtttff‡f
d
d„}
dtdtttff‡f
dd
„}dtdtttff‡f
dd„}ttƒ
}
|
 d|¡
|
 d|	¡
|
 d|
¡
|
 d|¡
|
 d|¡
|
 t	d¡
|
 dd¡
|
 dd¡
|
 dd¡
|
 dd¡
|
 dt
¡
|
 ¡}||||dœ}| |¡
}|s¸t
 d¡

dS| dd¡}| dd¡}|rÈ|sÏt
 d¡

dStj |d¡}tj |d¡}dtdtfdd „}||ƒ
}||ƒ
}t||ƒ
t||ƒ
t
 d!¡

dS)"uØ

    使用多步�程，生� Novel_setting.txt 与 Novel_directory.txt 并�存到 filepath。

    :param api_key:       OpenAI API key
    :param base_url:      OpenAI API base url
    :param llm_model:     所使用的 LLM 模型�称
    :param topic:         �说主题
    :param genre:         �说类型
    :param number_of_chapters: 章节数
    :param word_number:   �章目标字数
    :param filepath:      存放生�文件的目录路径
    T©
Úexist_ok)ÚmodelrrÚstater!c
sPtj
|d
|d|d|dd�}
ˆ |
¡
}|s!t d¡

ddi
Sd|j ¡i
S)	NrBrCrDrE©rBrCrDrEz#generate_base_setting: No response.rFr:)rrÚinvoker3r>ÚcontentÚstrip©rVÚpromptÚresponse©
rUr&r'Úgenerate_base_setting‡s




ü





z=Novel_novel_directory_generate.<locals>.generate_base_settingc
ó>tj
|d
d�
}
ˆ |
¡
}|st d¡

ddi
Sd|j ¡i
S)NrF)
Ú
novel_settingz(generate_character_setting: No response.rGr:)rrrXr3r>rYrZr[r^r&r'Úgenerate_character_setting”ó

ÿ





zBNovel_novel_directory_generate.<locals>.generate_character_settingc
r`)NrG)
Zcharacter_infoz!generate_dark_lines: No response.rHr:)rrrXr3r>rYrZr[r^r&r'Úgenerate_dark_linesžrcz;Novel_novel_directory_generate.<locals>.generate_dark_linesc
sJtj
|d
|d|dd�}
ˆ |
¡
}|st d¡

ddi
Sd|j ¡i
S)NrFrGrH)rFrGrHz$finalize_novel_setting: No response.rIr:)rrrXr3r>rYrZr[r^r&r'Úfinalize_novel_setting¨s



ý





z>Novel_novel_directory_generate.<locals>.finalize_novel_settingc
sDtj
|d
|dd�}
ˆ |
¡
}|st d¡

ddi
Sd|j ¡i
S)NrIrD)rIrDz&generate_novel_directory: No response.rJr:)rrrXr3r>rYrZr[r^r&r'Úgenerate_novel_directory´s


þ





z@Novel_novel_directory_generate.<locals>.generate_novel_directoryr_rbrdrerfrWuGNovel_novel_directory_generate: invoke() 结果为空，生æˆ�失败。NrIr:rJu@生æˆ�失败：缺少 final_novel_setting 或 novel_directory。zNovel_setting.txtzNovel_directory.txtÚtxtc

Ss| d
d¡ dd¡S)Nú
#r:Ú
*)
Úreplace)
rgr&r&r'Ú
clean_textés
z2Novel_novel_directory_generate.<locals>.clean_textz4Novel settings and directory generated successfully.)r.ÚmakedirsrrArrNrZadd_nodeZadd_edgerr	ÚcompilerXr3r>Úgetr/r?rr4)rrrQrBrCrDrErRr_rbrdrerfZgraphÚappZinput_paramsÚresultrIZfinal_novel_directoryZfilename_setZfilename_novel_directoryrkZfinal_novel_setting_cleanedZfinal_novel_directory_cleanedr&r^r'ÚNovel_novel_directory_generatehs\


ý















ü












rqÚnovel_settingsÚnovel_novel_directoryÚ
model_nameÚnovel_numberÚlastchapterc	sì
tj
|d
d�
t|||dd�‰tj |d¡}	tj
|	d
d�
tj |	d|›d�¡}
tj |d¡}tj |d	¡}tj |d
¡}
t|ƒ
}t|
ƒ
}dtdtd
tf‡f
dd„}| ¡r\|||ƒ}n|}dtdtd
tf‡f
dd„}| ¡ru|||ƒ}n|}t||ddd�}dtdtdtdt	dtd
tf‡f
dd„}||||||ƒ}dtdtdtdtdt	dtd
tf‡f
dd „}|||||||ƒ}|rët
||
ƒ
t|ƒ

t
||ƒ
t|ƒ

t
||ƒ
t|
ƒ

t
||
ƒ
t|||ƒ
t
 d!|›d"�¡

|St
 d!|›d#�¡

|S)$uŽ
    多步�程:
      1) 更新/创建全局摘�
      2) 更新/生�角色状�文档
      3) ��检索获�往期上下文
      4) 大纲 -> 正文
      5) 写入 chapter_{novel_number}.txt, 更新 last_chapter.txt
      6) 更新��库

    :param novel_settings:        最终的作�设定（字符串）