update haiku and llama prompts

tigergraph · Jun 5, 2024 · d8a5480 · d8a5480
1 parent a061ffa
commit d8a5480
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 1 deletion.
diff --git a/app/prompts/aws_bedrock_claude3haiku/generate_function.txt b/app/prompts/aws_bedrock_claude3haiku/generate_function.txt
@@ -18,5 +18,10 @@ Sixth Docstring: {doc6}
 Seventh Docstring: {doc7}
 Eighth Docstring: {doc8}
 
+If the output of this function answers the user's question, immediately return that answer.
+
+Follow the output directions below on how to structure your response
+Only include valid JSON do not include any other texts which would render the response invalid JSON: 
+
 Follow the output directions below on how to structure your response: 
 {format_instructions} 
diff --git a/app/prompts/aws_bedrock_claude3haiku/map_question_to_schema.txt b/app/prompts/aws_bedrock_claude3haiku/map_question_to_schema.txt
@@ -7,6 +7,9 @@ If there are words that are synonyms with the entities or relationships above, m
 Generate the complete question with the appropriate replacements. Keep the case of the schema elements the same. 
 Don't generate target_vertex_ids if there is no the term 'id' explicitly mentioned in the question.
 
-Respond in JSON (and only JSON)
+Follow the output directions below on how to structure your response
+Only include valid JSON do not include any other texts which would render the response invalid JSON: 
+
+Respond in JSON (and only JSON). Follow the format instructions below:
 {format_instructions}
 QUESTION: {question}
diff --git a/app/prompts/llama_70b/generate_function.txt b/app/prompts/llama_70b/generate_function.txt
@@ -18,5 +18,10 @@ Sixth Docstring: {doc6}
 Seventh Docstring: {doc7}
 Eighth Docstring: {doc8}
 
+If the output of this function answers the user's question, immediately return that answer.
+
+Follow the output directions below on how to structure your response
+Only include valid JSON do not include any other texts which would render the response invalid JSON: 
+
 Follow the output directions below on how to structure your response: 
 {format_instructions} 
diff --git a/app/prompts/llama_70b/map_question_to_schema.txt b/app/prompts/llama_70b/map_question_to_schema.txt
@@ -6,5 +6,10 @@ When certain entities are mapped to vertex attributes, may consider to generate
 If there are words that are synonyms with the entities or relationships above, make sure to output the cannonical form found in the choices above.
 Generate the complete question with the appropriate replacements. Keep the case of the schema elements the same. 
 Don't generate target_vertex_ids if there is no the term 'id' explicitly mentioned in the question.
+
+Follow the output directions below on how to structure your response
+Only include valid JSON do not include any other texts which would render the response invalid JSON: 
+
+Respond in JSON (and only JSON). Follow the format instructions below:
 {format_instructions}
 QUESTION: {question}