op_init: add extra files prompts for implement operation

Author: DarkCaster

config/config_keys.go

@@ -58,6 +58,9 @@ const K_ImplementTaskStage3PlanningJsonModePrompt = "stage3_task_planning_json_m
 const K_ImplementStage3PlanningLitePrompt = "stage3_planning_lite_prompt"
 const K_ImplementStage3PlanningLiteJsonModePrompt = "stage3_planning_lite_json_mode_prompt"
 
+// Extra prompt when adding unexpected source file to the prompts on late stage 3
+const K_ImplementTaskStage3ExtraFilesPrompt = "stage3_task_extra_files_prompt"
+
 const K_ImplementStage4ChangesDonePrompt = "stage4_changes_done_prompt"
 const K_ImplementStage4ChangesDoneResponse = "stage4_changes_done_response"
 const K_ImplementStage4ProcessPrompt = "stage4_process_prompt"

config/config_templates.go

@@ -60,6 +60,7 @@ func GetImplementConfigTemplate() map[string]interface{} {
 	result[K_ImplementStage3PlanningJsonModePrompt] = templateString
 	result[K_ImplementTaskStage3PlanningPrompt] = templateString
 	result[K_ImplementTaskStage3PlanningJsonModePrompt] = templateString
+	result[K_ImplementTaskStage3ExtraFilesPrompt] = templateString
 	result[K_ImplementStage3PlanningLitePrompt] = templateString
 	result[K_ImplementStage3PlanningLiteJsonModePrompt] = templateString
 	result[K_Stage3OutputSchema] = templateObject

op_init/prompt_templates.go

@@ -202,6 +202,7 @@ func getDefaultImplementConfigTemplate() map[string]interface{} {
 	result[config.K_ImplementTaskStage3PlanningJsonModePrompt] = "Below are the tasks that need to be implemented. Study all the source code provided to you and create a list of files that will be changed or created by you as a result of implementing the tasks. The tasks are:"
 	result[config.K_ImplementStage3PlanningLitePrompt] = "Now create a list of file names that will be changed or created by you as a result of implementing the tasks according to your work plan. Place each filename in <filename></filename> tags."
 	result[config.K_ImplementStage3PlanningLiteJsonModePrompt] = "Now create a list of files that will be changed or created by you as a result of implementing the tasks according to your work plan."
+	result[config.K_ImplementTaskStage3ExtraFilesPrompt] = "Below are the contents of additional source code files that may be relevant to the tasks."
 
 	result[config.K_Stage3OutputSchema] = getDefaultListOfFilesOutputSchema()
 	result[config.K_Stage3OutputKey] = defaultListOfFilesOutputKey