Function extract_code_language

utils/code_utils.py:212–237

Try to determine the programming language of a code snippet using simple heuristics. Args: code: The code snippet string. Returns: A string representing the detected language (e.g., "python") or "unknown".

(code: str)

def extract_code_language(code: str) -> str:
    """
    Try to determine the programming language of a code snippet using simple heuristics.

    Args:
        code: The code snippet string.

    Returns:
        A string representing the detected language (e.g., "python") or "unknown".
    """
    # Look for common language-specific keywords or patterns at the start of lines
    if re.search(r"^(import|from|def|class)\s", code, re.MULTILINE):
        return "python"
    elif re.search(r"^(package|import java|public class)", code, re.MULTILINE):
        return "java"
    elif re.search(r"^(#include|int main|void main)", code, re.MULTILINE):
        return "cpp"
    elif re.search(r"^(function|var|let|const|console\.log)", code, re.MULTILINE):
        return "javascript"
    elif re.search(r"^(module|fn|let mut|impl)", code, re.MULTILINE):
        return "rust"
    elif re.search(r"^(SELECT|CREATE TABLE|INSERT INTO)", code, re.MULTILINE):
        return "sql"

    # If no specific patterns are matched, return "unknown"
    return "unknown"