update finetuning example to ensure asst messages are the last to appear

nerdai · Sep 23, 2024 · 7a80e4c · 7a80e4c
1 parent dacb06f
commit 7a80e4c
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/arc_finetuning_st/finetuning/finetuning_example.py b/arc_finetuning_st/finetuning/finetuning_example.py
@@ -65,6 +65,20 @@ def from_attempts(
                     ),
                 ]
             )
+
+        # always end with an asst message or else openai finetuning job will failt
+        if a.critique == "This predicted output is correct.":
+            final_asst_message = ChatMessage(
+                role=MessageRole.ASSISTANT,
+                content="Glad, we were able to solve the puzzle!",
+            )
+        else:
+            final_asst_message = ChatMessage(
+                role=MessageRole.ASSISTANT,
+                content="Thanks for the feedback. I'll incorporate this into my next prediction.",
+            )
+
+        messages.append(final_asst_message)
         return cls(messages=messages, task_name=task_name)
 
     def to_json(self) -> str: