From 634072fc372152d9c549abc0cc1e689e6f0704ed Mon Sep 17 00:00:00 2001 From: Yolanda Robla Date: Thu, 16 Jan 2025 09:10:52 +0100 Subject: [PATCH 1/2] fix: solve some of the problems with snippets and languages Closes: #599 --- .../pipeline/extract_snippets/extract_snippets.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/src/codegate/pipeline/extract_snippets/extract_snippets.py b/src/codegate/pipeline/extract_snippets/extract_snippets.py index 9dc5a3a0..6f5923b9 100644 --- a/src/codegate/pipeline/extract_snippets/extract_snippets.py +++ b/src/codegate/pipeline/extract_snippets/extract_snippets.py @@ -98,6 +98,11 @@ def extract_snippets(message: str) -> List[CodeSnippet]: # format ` ```python ` in output snippets if filename and not matched_language and "." not in filename: lang = filename + if lang not in available_languages: + # try to get it from the extension + lang = ecosystem_from_message(filename) + if lang not in available_languages: + lang = None filename = None else: # Determine language from the message, either by the short @@ -118,6 +123,9 @@ def extract_snippets(message: str) -> List[CodeSnippet]: if lang not in available_languages: lang = None + #  just correct the typescript exception + if lang == "typescript": + lang = "javascript" snippets.append(CodeSnippet(filepath=filename, code=content, language=lang)) return snippets From b8a1f487eb0bb1adef84b0a460a55ba412f71859 Mon Sep 17 00:00:00 2001 From: Yolanda Robla Mota Date: Thu, 16 Jan 2025 10:06:48 +0100 Subject: [PATCH 2/2] Update extract_snippets.py --- src/codegate/pipeline/extract_snippets/extract_snippets.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/codegate/pipeline/extract_snippets/extract_snippets.py b/src/codegate/pipeline/extract_snippets/extract_snippets.py index 6f5923b9..a4c8b632 100644 --- a/src/codegate/pipeline/extract_snippets/extract_snippets.py +++ b/src/codegate/pipeline/extract_snippets/extract_snippets.py @@ -124,8 +124,10 @@ def extract_snippets(message: str) -> List[CodeSnippet]: lang = None #  just correct the typescript exception - if lang == "typescript": - lang = "javascript" + lang_map = { + "typescript": "javascript" + } + lang = lang_map.get(lang, lang) snippets.append(CodeSnippet(filepath=filename, code=content, language=lang)) return snippets