Change role to user for user prompts

vladoovtcharov · web-flow · commit 45b34dd507f0 · 2024-12-30T16:50:47.000-05:00
diff --git a/torchchat/generate.py b/torchchat/generate.py
@@ -1122,7 +1122,7 @@ def chat(
                     messages_to_encode.append(
                         {"role": "system", "content": self.system_prompt}
                     )
-                messages_to_encode.append({"role": "system", "content": prompt})
+                messages_to_encode.append({"role": "user", "content": prompt})
                 encoded = self.chat_formatter.encode_dialog_prompt(
                     messages_to_encode, add_generation_prompt=True,
                 )

Original file line number	Diff line number	Diff line change
`@@ -1122,7 +1122,7 @@ def chat(`
`1122`	`1122`	`messages_to_encode.append(`
`1123`	`1123`	`{"role": "system", "content": self.system_prompt}`
`1124`	`1124`	`)`
`1125`		`- messages_to_encode.append({"role": "system", "content": prompt})`
	`1125`	`+ messages_to_encode.append({"role": "user", "content": prompt})`
`1126`	`1126`	`encoded = self.chat_formatter.encode_dialog_prompt(`
`1127`	`1127`	`messages_to_encode, add_generation_prompt=True,`
`1128`	`1128`	`)`