Code generation in Prompt Engineering / GenAI - Model Pipeline Trace

Model Pipeline - Code generation

This pipeline shows how a code generation model learns to write code from examples. It starts with raw code data, processes it, trains a model to predict code tokens, and finally generates new code snippets.

Data Flow - 4 Stages

1Raw code dataset

10000 code snippets x variable length→Collect raw code examples from repositories→10000 code snippets x variable length

def add(a, b): return a + b

↓

2Tokenization

10000 code snippets x variable length→Split code into tokens (words, symbols)→10000 sequences x 50 tokens (max length)

["def", "add", "(", "a", ",", "b", ")", ":", "return", "a", "+", "b"]

↓

3Train/test split

10000 sequences x 50 tokens→Split data into 8000 training and 2000 testing sequences→Training: 8000 x 50 tokens, Testing: 2000 x 50 tokens

Training example: ["def", "add", "(", "a", ",", "b", ")", ":", "return", "a"]

↓

4Model training

8000 sequences x 50 tokens→Train transformer-based model to predict next token→Trained model with learned token probabilities

Input: ["def", "add", "(", "a", ","] -> Output: "b"

Training Trace - Epoch by Epoch

Epoch 1: 2.3 *****
Epoch 2: 1.8 ****
Epoch 3: 1.4 ***
Epoch 4: 1.1 **
Epoch 5: 0.9 *
(Loss decreases over epochs)

Epoch	Loss ↓	Accuracy ↑	Observation
1	2.3	0.25	Model starts learning basic token patterns
2	1.8	0.40	Loss decreases, accuracy improves as model learns syntax
3	1.4	0.55	Model captures common code structures
4	1.1	0.65	Better prediction of tokens in code sequences
5	0.9	0.72	Model converges with good token prediction accuracy

Prediction Trace - 4 Layers

Layer 1: Input token embedding

Layer 2: Transformer layers

Layer 3: Output token probabilities

Layer 4: Token selection

Model Quiz - 3 Questions

What happens to the loss value as training progresses?

A It decreases steadily

B It increases steadily

C It stays the same

D It randomly jumps up and down

Key Insight

This visualization shows how a code generation model learns token patterns from code examples. The loss decreases and accuracy improves as the model better predicts the next code token, enabling it to generate meaningful code snippets.

Practice

1. What is the main purpose of code generation in AI?

easy

A. Manually write code faster

B. Automatically create code from instructions

C. Run code without errors

D. Delete unnecessary code

Solution

1. Step 1: Understand code generation meaning

   Code generation means creating code automatically from instructions or examples.

2. Step 2: Match purpose with options

   Automatically create code from instructions correctly states this purpose, others describe different tasks.

3. Final Answer:

   Automatically create code from instructions -> Option B

4. Quick Check:

   Code generation = automatic code creation [OK]

Hint: Code generation means automatic code writing [OK]

Common Mistakes:

• Confusing code generation with manual coding
• Thinking code generation fixes errors automatically
• Believing code generation deletes code

2. Which of the following is the correct Python syntax to define a function named generate_code?

easy

A. generate_code def():

B. function generate_code()

C. def generate_code[]:

D. def generate_code():

Solution

1. Step 1: Recall Python function syntax

   Python functions start with def, followed by name and parentheses, then colon.

2. Step 2: Check each option

   def generate_code(): matches correct syntax; A, B and D have syntax errors (A wrong order, B JavaScript style, D brackets).

3. Final Answer:

   def generate_code(): -> Option D

4. Quick Check:

   Python function = def name(): [OK]

Hint: Python functions start with def and parentheses [OK]

Common Mistakes:

• Using JavaScript function keyword in Python
• Missing parentheses after function name
• Using brackets instead of parentheses

3. What will be the output of this Python code generated by AI?

def add_numbers(a, b):
    return a + b

result = add_numbers(3, 4)
print(result)

medium

A. 7

B. 34

C. TypeError

D. None

Solution

1. Step 1: Understand function behavior

   The function adds two numbers and returns the sum.

2. Step 2: Calculate add_numbers(3, 4)

   3 + 4 equals 7, so result is 7 and printed.

3. Final Answer:

   7 -> Option A

4. Quick Check:

   3 + 4 = 7 [OK]

Hint: Adding numbers returns their sum [OK]

Common Mistakes:

• Thinking + concatenates numbers as strings
• Expecting error from simple addition
• Confusing return value with print output

4. Identify the error in this AI-generated Python code:

def multiply(x, y):
return x * y

print(multiply(2, 3))

medium

A. Missing indentation for return statement

B. Wrong function name

C. Missing parentheses in print

D. Using * instead of + operator

Solution

1. Step 1: Check Python indentation rules

   Python requires the return line inside function to be indented.

2. Step 2: Identify error in code

   Return is not indented, causing IndentationError; other options are incorrect.

3. Final Answer:

   Missing indentation for return statement -> Option A

4. Quick Check:

   Python needs indented blocks [OK]

Hint: Indent inside functions in Python [OK]

Common Mistakes:

• Ignoring indentation errors
• Thinking print needs no parentheses in Python 3
• Confusing operators without context

5. You want to generate Python code that creates a dictionary from a list of keys ["a", "b", "c"] with values as their lengths. Which code snippet correctly uses dictionary comprehension?

hard

A. result = {len(k): k for k in ["a", "b", "c"]}

B. result = [k: len(k) for k in ["a", "b", "c"]]

C. result = {k: len(k) for k in ["a", "b", "c"]}

D. result = {k, len(k) for k in ["a", "b", "c"]}

Solution

1. Step 1: Understand dictionary comprehension syntax

   It uses curly braces with key:value pairs inside a for loop.

2. Step 2: Check each option

   result = {k: len(k) for k in ["a", "b", "c"]} correctly creates dict with keys and their lengths; B uses list brackets wrongly; C swaps key and value; D uses comma instead of colon.

3. Final Answer:

   result = {k: len(k) for k in ["a", "b", "c"]} -> Option C

4. Quick Check:

   Dict comprehension = {key: value for item} [OK]

Hint: Dict comprehension uses {key: value for item} [OK]

Common Mistakes:

• Using list brackets [] instead of {}
• Swapping keys and values
• Using comma instead of colon in dict