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 # Get Conversation Primer appropriate to Conversation Type
# TODO If compiled_references_message is too long, we need to truncate it. # TODO If compiled_references_message is too long, we need to truncate it.
if compiled_references_message == "": if compiled_references_message == "":
conversation_primer = prompts.conversation_llamav2.format(query=user_query) conversation_primer = user_query
else: else:
conversation_primer = prompts.notes_conversation_llamav2.format( conversation_primer = prompts.notes_conversation_llamav2.format(
query=user_query, references=compiled_references_message query=user_query, references=compiled_references_message