liuyimeta/training_data_chat
0
1a
2 .wdg � @ s� d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dlm Z d dl3m Z d dl m
Z
d dlmZ g Zd adad ad ad ad4d � ZdS ) � N)�LLMChain)�5OpenAIChat)�Prompt)�OpenAI)�get_openai_callback)�CallbackManager)� StreamingStdOutCallbackHandler� c C s� t d7 a t dkr�t�d�atdd�� }t�|�att_W d � n1 sJ0 Y tdd��}|�� }W d � n1 s|0 Y t |g d�d�a6t t7t d d8�d �a
d d
� }|| t�}t�d| � �� t�d|� �� |S )N� z after_training/training.indexzafter_training/faiss.pkl�rbztraining/master.txt�r)�history�context�question)�template�input_variablesr )� temperature)�prompt�llmc S sV t j| dd�}g }t|�D ]6\}}|�d|� d|j� �� tj| d�|�|d�}q|S )Nr9 )�kzContext z:10z11 12)r r r
)�store�similarity_search� enumerate�append� page_content�llmChain�predict�join)r Z history_p�docs�contexts�i�doc� ai_answer� r# �3/Users/kai/PycharmProjects/TTS/liuyimeta/process.py� onMessage" s z runPrompt.<locals>.onMessagezHuman: zBot: )r �faiss�13read_index�index�open�pickle�loadr �readr r r r r r
r )Z14user_input�f�promptTemplater% �answerr# r# r$ � runPrompt s 15 16$ &17r0 )r* r& �openai� langchainr �langchain.llms.openair Zlangchain.promptsr r �langchain.callbacksr �langchain.callbacks.baser Z$langchain.callbacks.streaming_stdoutr r
r( r r r r r0 r# r# r# r$ �<module> s