o gJ@sddlmZmZddlmZmZddlmZddlm Z m Z ddl m Z ddl mZddlmZddlmZdd lmZdd lmZGd d d eZGd ddZdS)) ChatOpenAIOpenAIEmbeddings)ChatPromptTemplateMessagesPlaceholder)ChromaDB)create_history_aware_retrievercreate_retrieval_chain)create_stuff_documents_chain)BaseChatMessageHistory)RunnableWithMessageHistoryDocument) BaseRetriever)JsonOutputParsercsFeZdZ  d fdd ZddZdefdd Zdefd d ZZS) ChromaCustomRetrieverNcs\tt|d|t|d|t|d|t|d||t|dtdS)Nchroma_collectionembedding_functiontop_kfilter openai_ef)super__init__object __setattr__ build_filterr)selfrrselected_filesr __class__1/var/www/html/XCapMarket/services/docchat/chat.pyrs zChromaCustomRetriever.__init__cCs>g}|r |dd|ii|siSt|dkr|dSd|iS)N file_namez$inrz$and)appendlen)rfilesfiltersr r r!rs z"ChromaCustomRetriever.build_filterquerycCsH|j|}|jj|g|j|jd}ddt|dd|ddDS)N)query_embeddings n_resultswherecSsg|] \}}t||dqS)) page_contentmetadatar ).0docmetar r r! -s zAChromaCustomRetriever._get_relevant_documents.. documentsr metadatas)r embed_queryrr(rrzip)rr( embeddingresultsr r r!_get_relevant_documents(s  z-ChromaCustomRetriever._get_relevant_documentscs ||SN)r8)rr(r r r!_aget_relevant_documents2s z.ChromaCustomRetriever._aget_relevant_documents)Nr) __name__ __module__ __qualname__rrstrr8r: __classcell__r r rr!rs  rc@sPeZdZddZddZddZddZd ed efd d Z d ed efddZ dS)ConversationalRAGcCsp||_t|ddd|_t|_||_d||_|jjj|j|jj d|_ | |_ | |_||_||_dS)Ng333333?T)model temperature streamingchat_)r)rrllmr chroma_db project_idcollection_name chroma_clientget_collectionr collectionrhistory_aware_retrievercreate_qa_chainquestion_answer_chaincreate_conversational_rag_chainconversational_rag_chainhistory_provider)rrrGrQai_modelr r r!r7s      zConversationalRAG.__init__cCsBd}td|ftddg}t|j|jj|jdd}t|j ||S)NaGiven a chat history and the latest user question which might reference context in the chat history, formulate a standalone question which can be understood without the chat history. Do NOT answer the question, just reformulate it if needed and otherwise return it as is.system chat_historyhumanz{input}r)rrrr) r from_messagesrrrKrFrrrrE)rcontextualize_q_system_promptcontextualize_q_prompt retrieverr r r!rGs"z0ConversationalRAG.create_history_aware_retrievercCs(d}td|ftddg}t|j|S)Nuo You are an AI-powered Investment Banking Analyst. Your role is to support deal teams with research, analysis, and presentation material preparation across M&A, capital markets, and strategic advisory mandates. Maintain a highly professional, data-driven, and concise tone. Use financial terminology appropriately and ensure all insights are substantiated by logic, precedent, or relevant data. ## Core Functions: - Conduct industry and company research using credible sources and databases - Build and analyze financial models including DCF, comparable company, and precedent transaction analyses - Create polished, professional PowerPoint slides for pitch books, CIMs, management presentations, and market updates - Synthesize complex financial and strategic information into concise insights for internal and client-facing use - Monitor M&A, equity, and debt capital markets to identify relevant transactions and trends - Assist with due diligence by summarizing key findings from data rooms and public filings - Maintain accuracy, clarity, and consistency across deliverables ## Communication Style: - Use a professional and analytical tone - Be concise and structured in all written responses - Support assertions with data or citations when possible - Prioritize clarity and eliminate fluff - Never speculate—only provide informed, defensible perspectives ## Formatting Guidelines: - Use bullet points and short paragraphs for readability - Clearly label financial metrics, dates, and sources - Avoid jargon unless appropriate for an investment banking audience - Keep bullet points clean and without terminal punctuation {context} rSrTrU)rrWrr rE)r system_prompt qa_promptr r r!rM`s! z!ConversationalRAG.create_qa_chaincCstt|j|j|jddddS)NinputrTanswer)input_messages_keyhistory_messages_keyoutput_messages_key)r rrLrNget_session_history)rr r r!rOsz1ConversationalRAG.create_conversational_rag_chain session_idreturncCs|jSr9)rQ)rrcr r r!rbsz%ConversationalRAG.get_session_historycombined_answercCs.ddg}t|}||jB}|d|i}|S)z Generate a summary of the combined answer using the language model. :param combined_answer: The full answer to be summarized. :return: A summarized version of the answer. )rSzp Summarize the following answer in a concise manner in less than 5 words: )rVz2Here is the answer to summarize: {combined_answer}re)rrWrEinvoke)rre llm_message chat_promptchainresponser r r!get_summary_of_combined_answers z0ConversationalRAG.get_summary_of_combined_answerN) r;r<r=rrrMrOr>r rbrkr r r r!r@6s+ r@N)langchain_openairrlangchain.prompts.chatrrutils.chroma_dbrlangchain.chainsrr(langchain.chains.combine_documents.stuffr langchain_core.chat_historyr langchain_core.runnables.historyr langchain_core.documentsr langchain_core.retrieversrlangchain_core.output_parsersrrr@r r r r!s       (