Change the "code-repair" task prompt to enforce the generated code to…

… be inside code fences, to ensure we can extract the code from the LLMs response Closes #257
symflower · Jul 11, 2024 · b8a3bb5 · b8a3bb5
1 parent 56930f7
commit b8a3bb5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/model/llm/llm.go b/model/llm/llm.go
@@ -108,7 +108,7 @@ type llmCodeRepairSourceFilePromptContext struct {
 // llmCodeRepairSourceFilePromptTemplate is the template for generating an LLM code repair prompt.
 var llmCodeRepairSourceFilePromptTemplate = template.Must(template.New("model-llm-code-repair-source-file-prompt").Parse(bytesutil.StringTrimIndentations(`
 	Given the following {{ .Language.Name }} code file "{{ .FilePath }}" with package "{{ .ImportPath }}" and a list of compilation errors, modify the code such that the errors are resolved.
-	The response must contain only the source code and nothing else.
+	The response must contain only the source code in a fenced code block and nothing else.
 
 	` + "```" + `{{ .Language.ID }}
 	{{ .Code }}

diff --git a/model/llm/llm_test.go b/model/llm/llm_test.go
@@ -324,7 +324,7 @@ func TestLLMCodeRepairSourceFilePrompt(t *testing.T) {
 
 		ExpectedMessage: bytesutil.StringTrimIndentations(`
 			Given the following Go code file "/path/to/foobar.go" with package "foobar" and a list of compilation errors, modify the code such that the errors are resolved.
-			The response must contain only the source code and nothing else.
+			The response must contain only the source code in a fenced code block and nothing else.
 
 			` + "```" + `golang
 			package foobar