o
gh
@ s d dl mZmZ d dlmZ d dlmZmZ d dlm Z d dl
m Z d dl m
Z
d dlmZ d dlmZ d d lmZ d d
lmZ d d lZdddZdd Zd dlmZmZ dd Zd S ) ) create_history_aware_retrievercreate_retrieval_chain) create_stuff_documents_chain)ChatPromptTemplateMessagesPlaceholder)
ChatOpenAI) ConversationSummaryBufferMemory)MongoDBAtlasVectorSearch)OpenAIEmbeddings)get_character_by_id) count_tokens)metadataN
similarityc C s" t | t dd}|j|d|idS )NZvector_search_index)
collection embedding
index_namek) search_type
search_kwargs)r r
as_retriever)r r r vector_store r g/var/www/html/aiverses.in/BackEnd/Ai_Agents/services/process_requests/process_request_knowledge_bank.pycreate_retriever s r c C sn t ddd}t|ddd}|t| d< ttddg}t|||}td tddg}t||d
}t||S ) Nz
gpt-3.5-turbog?)
model_name temperature chat_historyi )llmZ
memory_keyZmax_token_limitmemory)userz{input})systemz Use the context below:
{context})r prompt) r r r
r
from_messagesr r r r )sid retrieverr r Z
contextualizeZhistory_aware_retrieverZ
answer_promptZ
combine_chainr r r create_rag_chain s* r& ) HumanMessage AIMessagec C sL |d }|d }t |}|dd}|dd}|dd}t|d d d t| d d u rB| d
| } t| }
t||
t| d < t| d } t| d } g }
| ry| jry| jjD ] }|jd krj|
t
|j d
qY|jdkrx|
t |j d
qYd| d| d| d| |
d}|
|}t||d }|d dd|d|dS )N character_idmessagecharacter_nameUnknowngender backstory )r rag_chainr0 ZKnowledge_Bankr human)contentaizYou are z, a z. Your backstory- z.
User: )inputr answer
z
section_id)responser7 token_usage)r getr
setdefaultr r& Z chat_memorymessagestypeappendr' r2 r( invoker replace)clientdatar$ r) question characternamer- r. r r% r0 r r msgZ llm_inputr8 r9 r r r process_request4 s<
rG )r r ) langchain.chainsr r Z"langchain.chains.combine_documentsr langchain.promptsr r langchain_openair Zlangchain.memoryr Zlangchain_mongodbr r
app.database.fetch_datar -Ai_Agents.services.llm_functions.count_tokensr app.corer
osr r& langchain_core.messagesr' r( rG r r r r s