86 lines
8.9 KiB
Plaintext
86 lines
8.9 KiB
Plaintext
|
o
|
|||
|
|
jšgØ/ ã @ sv d dl Z d dlmZ d dlmZ d dlmZmZmZ d dl m
|
|||
|
|
Z
|
|||
|
|
d dl m Z m
|
|||
|
|
Z
|
|||
|
|
mZmZ d dlmZmZmZmZmZmZmZmZmZ d dlmZ d d l m Z d d
|
|||
|
|
l m Z d Z G d d
|
|||
|
|
„ d
|
|||
|
|
eƒZ!de"de"de"de"de"de#de#de"fdd„Z$de"de%e" de fdd„Z&de"de fd d „Z'de"d e"fd d „Z(d-de"d"e"d#e#de"fd$d%„Z)d&e"d'e"de"de"d(e"d)e#de"de#d*e"de"fd+d,„Z*dS ).é N)Ú TypedDict)Ú
|
|||
|
|
ChatOpenAI)Ú
|
|||
|
|
StateGraphÚSTARTÚEND)ÚDict)Ú 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_prompt)ÚOpenAIEmbeddings)ÚChroma©ÚDocumentÚ vectorstorec @ sV e Zd ZU eed< 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__Úint© r* r* ú;c:\Users\xieli\Desktop\AI_NovelGenerator\novel_generator.pyr s
|
|||
|
|
r Úapi_keyÚbase_urlÚ llm_modelr r r r Úfilepathc s¬ t || |d�‰ dtf‡ fdd„ }dtf‡ fdd„ } dtf‡ fdd„ }
|
|||
|
|
dtf‡ fd d
|
|||
|
|
„ } dtf‡ fd d „ } ttƒ}
|
|||
|
|
|
|
|||
|
|
d
|
|||
|
|
|¡ |
|
|||
|
|
d| ¡ |
|
|||
|
|
d|
|
|||
|
|
¡ |
|
|||
|
|
d| ¡ |
|
|||
|
|
d| ¡ |
|
|||
|
|
td
|
|||
|
|
¡ |
|
|||
|
|
d
|
|||
|
|
d¡ |
|
|||
|
|
dd¡ |
|
|||
|
|
dd¡ |
|
|||
|
|
dd¡ |
|
|||
|
|
dt¡ |
|
|||
|
|
¡ }||||dœ}| |¡}|s’t dƒ dS |
|
|||
|
|
dd¡}|
|
|||
|
|
dd¡}|r¢|s¨t dƒ dS t j
|
|||
|
|
|d¡}t j
|
|||
|
|
|d¡}| dd¡ d d¡}| dd¡ d d¡}t||ƒ t||ƒ dS ) uO
|
|||
|
|
使用多æ¥æµ�程，生æˆ� Novel_setting.txt 与 Novel_directory.txt
|
|||
|
|
)Úmodelr, r- Ústatec sF t j| d | d | d | d d�}ˆ |¡}|s ddiS d|j ¡ iS )Nr r r r ©r r r r r Ú )r ÚformatÚinvokeÚcontentÚstrip©r1 ÚpromptÚresponse©r0 r* r+ Úgenerate_base_setting8 s ü
|
|||
|
|
z=Novel_novel_directory_generate.<locals>.generate_base_settingc ó4 t j| d d�}ˆ |¡}|sddiS d|j ¡ iS )Nr )Ú
|
|||
|
|
novel_settingr r3 )r
|
|||
|
|
r4 r5 r6 r7 r8 r; r* r+ Úgenerate_character_settingD ó
|
|||
|
|
|
|||
|
|
zBNovel_novel_directory_generate.<locals>.generate_character_settingc r= )Nr )Zcharacter_infor! r3 )r r4 r5 r6 r7 r8 r; r* r+ Úgenerate_dark_linesK r@ z;Novel_novel_directory_generate.<locals>.generate_dark_linesc s@ t j| d | d | d d�}ˆ |¡}|sddiS d|j ¡ iS )Nr r r! )r r r! r" r3 )r r4 r5 r6 r7 r8 r; r* r+ Úfinalize_novel_settingR s ý
|
|||
|
|
z>Novel_novel_directory_generate.<locals>.finalize_novel_settingc s: t j| d | d d�}ˆ |¡}|sddiS d|j ¡ iS )Nr" r )r" r r# r3 )r r4 r5 r6 r7 r8 r; r* r+ Úgenerate_novel_directory] s þ
|
|||
|
|
z@Novel_novel_directory_generate.<locals>.generate_novel_directoryr< r? rA rB rC r2 u. âš ï¸� invoke() 结果为空，生æˆ�失败。Nr" r3 r# uG âš ï¸� 生æˆ�失败：缺少 final_novel_setting 或 novel_directory。zNovel_setting.txtzNovel_directory.txtú#Ú*)r r r Zadd_nodeZadd_edger r Úcompiler5 ÚprintÚgetÚosÚpathÚjoinÚreplacer )r, r- r. r r r r r/ r< r? rA rB rC ZgraphÚappZ input_paramsÚresultr" Zfinal_novel_directoryZ filename_setZfilename_novel_directoryZfinal_novel_setting_cleanedZ final_novel_directory_cleanedr* r; r+ Ú Novel_novel_directory_generate% sV
|
|||
|
|
ý
|
|||
|
|
ü
|
|||
|
|
|
|||
|
|
rO ÚtextsÚreturnc C s4 t | d�}dd„ |D ƒ}tj||td�}| ¡ |S )u©
|
|||
|
|
åˆ�始化并返回一个Chromaå�‘é‡�åº“ï¼Œå°†ä¼ å…¥çš„æ–‡æœ¬è¿›è¡ŒåµŒå…¥ã€‚
|
|||
|
|
若需��对 texts �分�或分�处�；这里�演示简�用法。
|
|||
|
|
©Zopenai_api_keyc S s g | ]}t |d �‘qS )©Z page_contentr )Ú.0Útr* r* r+ Ú
|
|||
|
|
<listcomp>œ s z%init_vector_store.<locals>.<listcomp>)Z embeddingÚpersist_directory)r r Zfrom_documentsÚVECTOR_STORE_DIRÚpersist)r, rP Ú
|
|||
|
|
embeddingsZ documentsr r* r* r+ Úinit_vector_store– s
|
|||
|
|
|
|||
|
|
r[ c C s& t j t¡sdS t| d�}tt|d�S )uY
|
|||
|
|
读å�–å·²å˜åœ¨çš„å�‘é‡�库。若ä¸�å˜åœ¨åˆ™è¿”回None或新建一个空的。
|
|||
|
|
NrR )rW Zembedding_function)rI rJ ÚexistsrX r r )r, rZ r* r* r+ Úload_vector_store¡ s
|
|||
|
|
r] Ú new_chapterc C sH t | ƒ}|st| |gƒ}dS t| d�}t|d�}| |g¡ | ¡ dS )u™
|
|||
|
|
å°†æœ€æ–°ç« èŠ‚æ–‡æœ¬æ�’入到å�‘é‡�库里，用于å�Žç»æ£€ç´¢å�‚考。
|
|||
|
|
å�¯æ ¹æ�®å®žé™…需求å�šåˆ†å�—处ç�†ã€‚æ¤å¤„仅作简å�•ç¤ºèŒƒã€‚
|
|||
|
|
NrR rS )r] r[ r r Z
|
|||
|
|
add_documentsrY )r, r^ ÚstorerZ Znew_docr* r* r+ Úupdate_vector_store« s
|
|||
|
|
|
|||
|
|
r` é ÚqueryÚkc C s6 t | ƒ}|sdS |j||d�}d dd„ |D ƒ¡}|S )u¦
|
|||
|
|
从å�‘é‡�库ä¸æ£€ç´¢ä¸Ž query 最相关的 k æ�¡æ–‡æœ¬ï¼Œæ‹¼æŽ¥å�Žè¿”回。
|
|||
|
|
用于在生æˆ�大纲或写æ£æ–‡æ—¶ï¼Œä¸ºå¤§æ¨¡åž‹æ��供更多上下文。
|
|||
|
|
r3 ©rc Ú
|
|||
|
|
c S s g | ]}|j ‘qS r* rS )rT Údr* r* r+ rV Æ s z:get_relevant_context_from_vector_store.<locals>.<listcomp>)r] Zsimilarity_searchrK )r, rb rc r_ ZdocsZcombinedr* r* r+ Ú&get_relevant_context_from_vector_store¼ s rg Únovel_settingsÚnovel_novel_directoryÚ
|
|||
|
|
model_nameÚ novel_numberÚ lastchapterc sŽ t |||dd�‰ tj |d¡} tj |d¡}
|
|||
|
|
tj |d¡} tj |d¡} t| ƒ}
|
|||
|
|
t|
|
|||
|
|
ƒ}dtdtd tf‡ fd
|
|||
|
|
d „ }| ¡ rC|||ƒ}n|}dtd td tf‡ fd
|
|||
|
|
d„ }| ¡ r\|||
|
|||
|
|
ƒ}n|
|
|||
|
|
}t|ddd�}dtdtdtdtdtd tf ‡ fdd„ }|| ||||ƒ}dtdtdtdtdtdtd tf‡ fdd „ }|| |||||ƒ}|rÅt || ƒ t
|
|||
|
|
| ƒ t || ƒ t
|
|||
|
|
| ƒ t || ƒ t
|
|||
|
|
|
|
|||
|
|
ƒ t ||
|
|||
|
|
ƒ t ||ƒ |S ) u*
|
|||
|
|
多æ¥æµ�程:
|