"""
"""
from langchain_google_genai import ChatGoogleGenerativeAI
from langchain_core.prompts import ChatPromptTemplate

class GeminiAI:
    """
    Google Gemini AI class.
    """

    def __init__(self, llm_model_name: str) -> None:
        self.llm = ChatGoogleGenerativeAI(model=llm_model_name)

    def getMainTemplate(self) -> ChatPromptTemplate:
        """
        Devuelve el system prompt principal.
        """

        prompt = ChatPromptTemplate.from_messages(
            [
                ("system",
                "Eres un asesor experto en la Resolucion 1650/24 de la DGCyE de la Provincia de Buenos Aires.\n"
                "Tu tarea es utiliza la información de la conversación y el contexto disponible para responder las consultas del usuario.\n"),
                ("placeholder", "{history}"),
                ("human", "{message}"),
            ]
        )

        return prompt



if __name__ == "__main__":
    llm = GeminiAI("gemini-1.5-flash")
    response = llm.llm.invoke("Hola")
    print(response)