change citations

api/src/stampy_chat/chat.py

@@ -302,7 +302,7 @@ def merge_history(history):
     messages = []
     current_message = history[0]
     for message in history[1:]:
-        if message.get('role') == 'deleted':
+        if message.get('role') in ['deleted', 'error']:
             continue
         if message.get('role') != current_message.get('role'):
             messages.append(current_message)

api/src/stampy_chat/citations.py

@@ -56,7 +56,7 @@ class Config:
     @staticmethod
     def make_reference(i: int) -> str:
         """Make the reference used in citations - basically translate i -> 'a + i'"""
-        return chr(i + 97)
+        return str(i + 1)
 
     def fetch_docs(self, input_variables) -> List:
         ### Copied from parent - for some reason they ignore the ids of the returned items, so

api/src/stampy_chat/settings.py

@@ -10,8 +10,8 @@
 SOURCE_PROMPT = (
     "You are a helpful assistant knowledgeable about AI Alignment and Safety. "
     "Please give a clear and coherent answer to the user's questions. (written after \"Q:\") "
-    "using the following sources. Each source is labeled with a letter. Feel free to "
-    "use the sources in any order, and try to use multiple sources in your answers.\n\n"
+    "using the following sources. Each source is labeled with a number. Feel free to "
+    "use the sources in any order, and try to reference up to 8 sources in your answers.\n\n"
 )
 HISTORY_PROMPT = (
     "\n\n"
@@ -29,8 +29,8 @@
 
 QUESTION_PROMPT = (
     "In your answer, please cite any claims you make back to each source "
-    "using the format: [a], [b], etc. If you use multiple sources to make a claim "
-    "cite all of them. For example: \"AGI is concerning [c, d, e].\"\n\n"
+    "using the format: [1], [2], etc. If you use multiple sources to make a claim "
+    "cite all of them. For example: \"AGI is concerning [1, 3, 8].\"\n\n"
 )
 PROMPT_MODES = {
     'default': "",