Skip to content

Commit b8a1f48

Browse files
authored
Update extract_snippets.py
1 parent 634072f commit b8a1f48

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/codegate/pipeline/extract_snippets/extract_snippets.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -124,8 +124,10 @@ def extract_snippets(message: str) -> List[CodeSnippet]:
124124
lang = None
125125

126126
#  just correct the typescript exception
127-
if lang == "typescript":
128-
lang = "javascript"
127+
lang_map = {
128+
"typescript": "javascript"
129+
}
130+
lang = lang_map.get(lang, lang)
129131
snippets.append(CodeSnippet(filepath=filename, code=content, language=lang))
130132

131133
return snippets

0 commit comments

Comments
 (0)