bldemo / components /promptengine.py
Druso's picture
Update components/promptengine.py
ab1e150
raw
history blame contribute delete
908 Bytes
from components.configs_prompts import system_message, maintemplate
#costruisce il prompt inserendo i pagecontent e gli elementi del template
def querybuilder (relevantcontents, user_message, template=maintemplate):
return template.format(relevantcontents=relevantcontents, user_message=user_message)
#monta l'intera sequenza di messaggi per chatopenai, aggiunge a chatmemory il sys message (forse da svuotare e mettere tutto in user message) e aggiunge l'ultima domanda dell'utente
def promptenginefinalize(chatmemory,relevantcontents, user_message, system_message=system_message):
finalprompt = chatmemory.copy()
finalprompt.insert(0, {"role": "system", "content": system_message})
query= querybuilder(relevantcontents, user_message)
finalprompt.append({"role": "user", "content": query})
return finalprompt #it's what we're going to pass to the llm to generate the next response