Fix format of user query during general conversation with Llama 2

This commit is contained in:
Debanjum Singh Solanky 2023-07-31 17:07:20 -07:00
parent 48e5ac0169
commit ded606c7cb

View file

@ -125,7 +125,7 @@ def converse_offline(
# Get Conversation Primer appropriate to Conversation Type
# TODO If compiled_references_message is too long, we need to truncate it.
if compiled_references_message == "":
conversation_primer = prompts.conversation_llamav2.format(query=user_query)
conversation_primer = user_query
else:
conversation_primer = prompts.notes_conversation_llamav2.format(
query=user_query, references=compiled_references_message