1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
from typing import Dict, List, Union
from ...models.filesystem import RangeInFileWithContents
from ...models.filesystem_edit import FileEdit
class MarkdownStyleEncoderDecoder:
# Filename -> the part of the file you care about
range_in_files: List[RangeInFileWithContents]
def __init__(self, range_in_files: List[RangeInFileWithContents]):
self.range_in_files = range_in_files
def encode(self) -> str:
return "\n\n".join(
[
f"File ({rif.filepath})\n```\n{rif.contents}\n```"
for rif in self.range_in_files
]
)
def _suggestions_to_file_edits(self, suggestions: Dict[str, str]) -> List[FileEdit]:
file_edits: List[FileEdit] = []
for suggestion_filepath, suggestion in suggestions.items():
matching_rifs = list(
filter(lambda r: r.filepath == suggestion_filepath, self.range_in_files)
)
if len(matching_rifs) > 0:
range_in_file = matching_rifs[0]
file_edits.append(
FileEdit(
range=range_in_file.range,
filepath=range_in_file.filepath,
replacement=suggestion,
)
)
return file_edits
def _decode_to_suggestions(self, completion: str) -> Dict[str, str]:
if len(self.range_in_files) == 0:
return {}
if "```" not in completion:
completion = "```\n" + completion + "\n```"
if completion.strip().splitlines()[0].strip() == "```":
first_filepath = self.range_in_files[0].filepath
completion = f"File ({first_filepath})\n" + completion
suggestions: Dict[str, str] = {}
current_file_lines: List[str] = []
current_filepath: Union[str, None] = None
last_was_file = False
inside_file = False
for line in completion.splitlines():
if line.strip().startswith("File ("):
last_was_file = True
current_filepath = line.strip()[6:-1]
elif last_was_file and line.startswith("```"):
last_was_file = False
inside_file = True
elif inside_file:
if line.startswith("```"):
inside_file = False
suggestions[current_filepath] = "\n".join(current_file_lines)
current_file_lines = []
current_filepath = None
else:
current_file_lines.append(line)
return suggestions
def decode(self, completion: str) -> List[FileEdit]:
suggestions = self._decode_to_suggestions(completion)
file_edits = self._suggestions_to_file_edits(suggestions)
return file_edits
|
