ensure system prompt obeys constraint

dmarx · Jul 21, 2023 · 14bffda · 14bffda
1 parent 62cd5a4
commit 14bffda
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/tag.py b/scripts/tag.py
@@ -163,7 +163,7 @@ def build_chat_prompt(doc: Document, prompt_head: str, max_content_len: int = 20
     user_prompt = f"<content-{i}>{doc.title}\n\n{content}</content-{i}>\n"
     user_prompt += f"<tags-{i}>"
     messages = [
-      {'role':'system', 'content':system_prompt},
+      {'role':'system', 'content':system_prompt[:max_content_len]},
       {'role':'user',   'content':user_prompt},
     ]
     return messages