update main.go

llama/main.go

@@ -47,6 +47,12 @@ func main() {
 		// Update the conversation context with the response
 		conversationContext = updatedContext
 
-		fmt.Println("Ollama's response:", extraction.Extract(response))
+		var generatedCode = extraction.Extract(response)
+
+		fmt.Println("Ollama's response:", generatedCode)
 	}
 }
+
+// Following will be the suffix string for fixing the errors:
+
+// \nFollowing are the errors, please fix the code. Write it again, and write only source code along with same test cases with no further explanation. The format should be ```rust <yourcode + testcases> ```"