INNER CODE UNIT · Python

get_seed_prompt

KodCode-AI/kodcode · pipeline/step1.1_gen_questions.py:70

def get_seed_prompt(input, mode):
    # Load prompt template from markdown file
    env = Environment(loader=FileSystemLoader('prompts'))
    
    if mode in ["leetcode", "codeforces"]:
        template = env.get_template('genq_from_seed.md').render()
        
        language = language_map.get(args.language)
        if language is None:
            print(f"Warning: Invalid language '{args.language}', using default 'Python'")
            language = "Python"
            
        template = template.replace("{LANGUAGE}", language)

        # Add sample questions
        for i, problem in enumerate(input, 1):
            template += f"[Question {i}]: {problem['description']}\n\n"
        

View source record →