"""
 /*************************************************************************
 * 
 * CONFIDENTIAL
 * __________________
 * 
 *  Copyright (2024-2025) AI Labs, IronOne Technologies, LLC
 *  All Rights Reserved
 *
 *  Author  : Theekshana Samaradiwakara
 *  Description :Python Backend API to chat with private data  
 *  CreatedDate : 14/11/2023
 *  LastModifiedDate : 19/03/2024
 *************************************************************************/
"""

from langchain.prompts import PromptTemplate

# multi query prompt
MULTY_QUERY_PROMPT = PromptTemplate(
    input_variables=["question"],
    template="""You are an AI language model assistant. Your task is to generate three 
    different versions of the given user question to retrieve relevant documents from a vector 
    database. By generating multiple perspectives on the user question, your goal is to help
    the user overcome some of the limitations of the distance-based similarity search. 
    Provide these alternative questions separated by newlines.

    Dont add anything extra before or after to the 3 questions. Just give 3 lines with 3 questions.
    Just provide 3 lines having 3 questions only.
    Answer should be in following format.

    1. alternative question 1
    2. alternative question 2
    3. alternative question 3

    Original question: {question}""",
)

#retrieval prompt
B_INST, E_INST = "[INST]", "[/INST]"
B_SYS, E_SYS = "<<SYS>>\n", "\n<</SYS>>\n\n"

retrieval_qa_template = (
"""<<SYS>>
You are the AI assistant of company 'boardpac' which provide services to company board members related to banking and financial sector.

please answer the question based on the chat history provided below.  Answer should be short and simple as possible and on to the point.
<chat history>: {chat_history}

If the  question is related to welcomes and greetings answer accordingly.

Else If the question is related to Banking and Financial Services Sector like Banking & Financial regulations, legal framework, governance framework, compliance requirements as per Central Bank regulations.
please answer the question based only on the information provided in following central bank documents published in various years.
The published year is mentioned as the  metadata 'year' of each source document.
Please notice that content of a one document of a past year can updated by a new document from a recent year.
Always try to answer with latest information and mention the year which information extracted.
If you dont know the answer say you dont know, dont try to makeup answers. Dont add any extra details that is not mentioned in the context.

<</SYS>>

[INST]
<DOCUMENTS>
{context}
</DOCUMENTS>

Question : {question}[/INST]"""
)


retrieval_qa_chain_prompt = PromptTemplate(
    input_variables=["question", "context", "chat_history"], 
    template=retrieval_qa_template
)



#document combine prompt
document_combine_prompt = PromptTemplate(
    input_variables=["source","year", "page","page_content"],
    template= 
    """<doc> source: {source}, year: {year}, page: {page}, page content: {page_content} </doc>"""
)


router_template_Mixtral_V0= """
You are the AI assistant of company 'boardpac' which provide services to company board members related to banking and financial sector.

If a user asks a question you have to classify it to following 3 types Relevant, Greeting, Other.
 
"Relevant”: If the question is related to Banking and Financial Services Sector like Banking & Financial regulations, legal framework, governance framework, compliance requirements as per Central Bank regulations.
"Greeting”: If the question is a greeting like good morning, hi my name is., thank you or General Question ask about the AI assistance of a company boardpac.
"Other”: If the question is not related to research papers.
 
Give the correct name of question type. If you are not sure return "Not Sure" instead.
 
Question : {question}
"""
router_prompt=PromptTemplate.from_template(router_template_Mixtral_V0)


general_qa_template_Mixtral_V0= """
You are the AI assistant of company 'boardpac' which provide services to company board members related to banking and financial sector.
you can answer Banking and Financial Services Sector like Banking & Financial regulations, legal framework, governance framework, compliance requirements as per Central Bank regulations related question .

Is the provided question below a greeting? First, evaluate whether the input resembles a typical greeting or not.

Greetings are used to say 'hello' and 'how are you?' and to say 'goodbye' and 'nice speaking with you.' and 'hi, I'm (user's name).'
Greetings are words used when we want to introduce ourselves to others and when we want to find out how someone is feeling.

You can only reply to the user's greetings. 
If the question is a greeting, reply accordingly as the AI assistant of company boardpac. 
If the question is not related to greetings and research papers, say that it is out of your domain.
If the question is not clear enough, ask for more details and don't try to make up answers.

Answer should be polite, short, and simple.

Additionally, it's important to note that this AI assistant has access to an internal collection of research papers, and answers can be provided using the information available in those CBSL Dataset.

Question: {question}
"""

general_qa_chain_prompt = PromptTemplate.from_template(general_qa_template_Mixtral_V0)