add rewrite file prompt

mentat/resources/prompts/two_step_edit_prompt_rewrite_file.txt

@@ -0,0 +1,11 @@
+You are part of an expert AI coding system.
+
+In the next message you will be given the contents of a code file. The user will then specify some edits to be made to the file.
+
+Your response should:
+    - rewrite the entire file, including all the requested edits
+    - wrap your entire response in ```python ... ``` to format it as Python code
+    - do not include anything else in your response other than the wrapped code
+    - do not make any other changes to the code other than the requested edits
+    - even formatting changes should not be made unless explicitly requested by the user
+    - if a change is not fully specified, do your best to follow the spirit of what was asked

mentat/stream_model_response.py

@@ -23,6 +23,9 @@
 
 two_step_edit_prompt_filename = Path("two_step_edit_prompt.txt")
 two_step_edit_prompt_list_files_filename = Path("two_step_edit_prompt_list_files.txt")
+two_step_edit_prompt_rewrite_file_filename = Path(
+    "two_step_edit_prompt_rewrite_file.txt"
+)
 
 
 async def stream_model_response(
@@ -96,6 +99,10 @@ def get_two_step_list_files_prompt() -> str:
     return read_prompt(two_step_edit_prompt_list_files_filename)
 
 
+def get_two_step_rewrite_file_prompt() -> str:
+    return read_prompt(two_step_edit_prompt_rewrite_file_filename)
+
+
 async def stream_model_response_two_step(
     messages: list[ChatCompletionMessageParam],
 ) -> ParsedLLMResponse:
@@ -167,7 +174,10 @@ async def stream_model_response_two_step(
         stream.send("Error processing model response: Invalid JSON", style="error")
         # TODO: handle error
 
-    stream.send(f"\n\n{response_json}\n\n")
+    stream.send(f"\n{response_json}\n")
+
+    # for file_path in response_json["files"]:
+    # rewrite_file_messages: list[ChatCompletionMessageParam] = [
 
     # async with parser.interrupt_catcher():
     #     parsed_llm_response = await parser.stream_and_parse_llm_response(