Fix bug related to alternative instruction block identifiers

stanford-oval · Jun 2, 2024 · 5ec78c1 · 5ec78c1
1 parent bb80d6f
commit 5ec78c1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/chainlite/load_prompt.py b/chainlite/load_prompt.py
@@ -16,7 +16,7 @@
 prompt_block_identifiers = {
     "input": ["# input\n", "# Input\n", "# INPUT\n"],
     "output": ["# output\n", "# Output\n", "# OUTPUT\n"],
-    "instruction": ["# instruction\n", "# Instruction\n", "# Instruction\n"],
+    "instruction": ["# instruction\n", "# Instruction\n", "# INSTRUCTION\n"],
     "distillation_instruction": [
         "# distillation instruction\n",
         "# distillation_instruction\n",

diff --git a/tests/joke.prompt b/tests/joke.prompt
@@ -1,4 +1,4 @@
-# instruction
+# Instruction
 Tell a joke about the input topic. The format of the joke should be a question and response, separated by a line break.
 {# This is a comment, and will be ignored anywhere in a .prompt file. Other than block definitions and comments, '#' is allowed and is treated as a normal character.  #}