mito-ai 0.1.55__py3-none-any.whl → 0.1.57__py3-none-any.whl

mito_ai/completions/prompt_builders/chart_conversion_prompt.py

@@ -0,0 +1,27 @@
+# Copyright (c) Saga Inc.
+# Distributed under the terms of the GNU Affero General Public License v3.0 License.
+
+from typing import List
+from mito_ai.completions.prompt_builders.prompt_section_registry import SG, Prompt
+from mito_ai.completions.prompt_builders.prompt_section_registry.base import PromptSection
+from mito_ai.completions.prompt_builders.prompt_constants import CHART_CONFIG_RULES
+
+def create_chart_conversion_prompt(code: str) -> str:
+    """
+    Create a prompt for converting matplotlib chart code to be used with the Chart Wizard.
+    
+    Args:
+        code: The matplotlib chart code to convert
+        
+    Returns:
+        A formatted prompt string
+    """
+    sections: List[PromptSection] = []
+
+    sections.append(SG.Generic("Instructions", "The following code contains a matplotlib chart. However, the chart must be converted to a specific format for use in our tool. Below you will find the rules used to create an acceptable chart; use these rules to reformat the code."))
+    
+    sections.append(SG.Generic("Chart Config Rules", CHART_CONFIG_RULES))
+    sections.append(SG.Generic("Code to Convert", f"```python\n{code}\n```"))
+    
+    prompt = Prompt(sections)
+    return str(prompt)

mito_ai/completions/prompt_builders/chat_prompt.py

@@ -2,18 +2,8 @@
 # Distributed under the terms of the GNU Affero General Public License v3.0 License.
 
 from typing import List, Optional, Dict
-from mito_ai.completions.prompt_builders.prompt_constants import (
-    ACTIVE_CELL_ID_SECTION_HEADING,
-    CHAT_CODE_FORMATTING_RULES,
-    FILES_SECTION_HEADING,
-    VARIABLES_SECTION_HEADING,
-    CODE_SECTION_HEADING,
-    get_active_cell_output_str,
-)
-from mito_ai.completions.prompt_builders.utils import (
-    get_rules_str,
-    get_selected_context_str,
-)
+from mito_ai.completions.prompt_builders.prompt_section_registry import SG, Prompt
+from mito_ai.completions.prompt_builders.prompt_section_registry.base import PromptSection
 
 
 def create_chat_prompt(
@@ -25,92 +15,17 @@ def create_chat_prompt(
     input: str,
     additional_context: Optional[List[Dict[str, str]]] = None,
 ) -> str:
-    variables_str = "\n".join([f"{variable}" for variable in variables])
-    files_str = "\n".join([f"{file}" for file in files])
-    selected_context_str = get_selected_context_str(additional_context)
-    rules_str = get_rules_str(additional_context)
-
-    prompt = f"""{rules_str}
+    sections: List[PromptSection] = [
+        SG.Rules(additional_context),
+        SG.Generic("Instructions", "Help me complete the following task. I will provide you with a set of variables, existing code, and a task to complete."),
+        SG.Files(files),
+        SG.Variables(variables),
+        SG.SelectedContext(additional_context),
+        SG.ActiveCellId(active_cell_id),
+        SG.ActiveCellCode(active_cell_code),
+        SG.ActiveCellOutput(has_active_cell_output),
+        SG.Task(input),
+    ]
     
-Help me complete the following task. I will provide you with a set of variables, existing code, and a task to complete.
-
-{CHAT_CODE_FORMATTING_RULES}
-
-<Example 1>
-
-{FILES_SECTION_HEADING}
-file_name: sales.csv
-
-{VARIABLES_SECTION_HEADING}
-{{
-    'loan_multiplier': 1.5,
-    'sales_df': pd.DataFrame({{
-        'transaction_date': ['2024-01-02', '2024-01-02', '2024-01-02', '2024-01-02', '2024-01-03'],
-        'price_per_unit': [10, 9.99, 13.99, 21.00, 100],
-        'units_sold': [1, 2, 1, 4, 5],
-        'total_price': [10, 19.98, 13.99, 84.00, 500]
-    }})
-}}
-
-{ACTIVE_CELL_ID_SECTION_HEADING}
-'9c0d5fda-2b16-4f52-a1c5-a48892f3e2e8'
-
-{CODE_SECTION_HEADING}
-```python
-import pandas as pd
-sales_df = pd.read_csv('./sales.csv')
-```
-
-Your task: convert the transaction_date column to datetime and then multiply the total_price column by the sales_multiplier.
-
-Output:
-```python
-import pandas as pd
-sales_df = pd.read_csv('./sales.csv')
-sales_df['transaction_date'] = pd.to_datetime(sales_df['transaction_date'])
-sales_df['total_price'] = sales_df['total_price'] * sales_multiplier
-```
-
-Applied datetime conversion to enable temporal analysis[MITO_CITATION:9c0d5fda-2b16-4f52-a1c5-a48892f3e2e8:2] and revenue adjustment using the 1.5x sales multiplier[MITO_CITATION:9c0d5fda-2b16-4f52-a1c5-a48892f3e2e8:3], scaling total revenue from $627.97 to $941.96.
-
-</Example 1>
-
-<Example 2>
-
-{ACTIVE_CELL_ID_SECTION_HEADING}
-'1a2b3c4d-5e6f-7g8h-9i0j-k1l2m3n4o5p6'
-
-{CODE_SECTION_HEADING}
-```python
-```
-
-Your task: Hello
-
-Output:
-Hey there! I'm Mito AI. How can I help you today? 
-
-</Example 2>
-
-
-{FILES_SECTION_HEADING}
-{files_str}
-
-{VARIABLES_SECTION_HEADING}
-{variables_str}
-
-{ACTIVE_CELL_ID_SECTION_HEADING}
-{active_cell_id}
-
-{CODE_SECTION_HEADING}
-```python
-{active_cell_code}
-```
-
-{selected_context_str}
-
-{get_active_cell_output_str(has_active_cell_output)}
-
-Your task: {input}
-"""
-
-    return prompt
+    prompt = Prompt(sections)
+    return str(prompt)

mito_ai/completions/prompt_builders/chat_system_message.py

@@ -1,17 +1,22 @@
 # Copyright (c) Saga Inc.
 # Distributed under the terms of the GNU Affero General Public License v3.0 License.
 
+from typing import List
+from mito_ai.completions.prompt_builders.prompt_section_registry import SG, Prompt
 from mito_ai.completions.prompt_builders.prompt_constants import (
+    CHART_CONFIG_RULES,
     CHAT_CODE_FORMATTING_RULES,
     CITATION_RULES,
     CELL_REFERENCE_RULES,
-    ACTIVE_CELL_ID_SECTION_HEADING, 
-    CODE_SECTION_HEADING,
     get_database_rules
 )
+from mito_ai.completions.prompt_builders.prompt_section_registry.base import PromptSection
 
 def create_chat_system_message_prompt() -> str:
-    return f"""You are Mito Data Copilot, an AI assistant for Jupyter. You're a great python programmer, a seasoned data scientist and a subject matter expert.
+    sections: List[PromptSection] = []
+    
+    # Add intro text
+    sections.append(SG.Generic("Instructions", """You are Mito Data Copilot, an AI assistant for Jupyter. You're a great python programmer, a seasoned data scientist and a subject matter expert.
 
 The user is going to ask you for help writing code, debugging code, explaining code, or drawing conclusions from their data/graphs. It is your job to help them accomplish their goal. 
 
@@ -25,77 +30,96 @@ There are three possible types of responses you might give:
 Other useful information:
 1. The user has two types of modes that they can collaborate with you in: Chat Mode (this mode) and agent mode. Chat mode gives the user more control over the edits made to the notebook and only edits the active cell. Agent mode gives you more autonomy over completing the user's task across mulitple messages. In agent mode, you can edit or create new cells, see the entire notebook, automatically run the code you write, and more.
 2. If the user asks you to generate a dashboard, app, or streamlit app for them, you should tell them that they must use Agent mode to complete the task. You are not able to automatically switch the user to agent mode, but they can switch to it themselves by using the Chat/Agent mode toggle in the bottom left corner of the Ai taskpane.
-
-====
-{CITATION_RULES}
-
-====
-{CELL_REFERENCE_RULES}
-
-<Example 1> 
-{ACTIVE_CELL_ID_SECTION_HEADING}
-'7b3a9e2c-5d14-4c83-b2f9-d67891e4a5f2'
-
-{CODE_SECTION_HEADING}
-```python
-sales_df = pd.read_csv('sales_data.csv')
-monthly_revenue = sales_df.groupby('month')['revenue'].sum()
-top_month = monthly_revenue.idxmax()
-peak_revenue = monthly_revenue.max()
-growth_rate = (monthly_revenue.iloc[-1] / monthly_revenue.iloc[0] - 1) * 100
-```
-
-Your task: What are the key revenue insights from this sales data?
-
-Output:
-Peak monthly revenue reached $847,392 in March[MITO_CITATION:7b3a9e2c-5d14-4c83-b2f9-d67891e4a5f2:2-3], representing a 23.8% year-over-year growth rate[MITO_CITATION:7b3a9e2c-5d14-4c83-b2f9-d67891e4a5f2:4]. The revenue aggregation analysis[MITO_CITATION:7b3a9e2c-5d14-4c83-b2f9-d67891e4a5f2:1-2] reveals strong seasonal performance patterns.
-
-</Example 1>
-
-Notice in the example above:
-- Citations support specific facts and numbers, not vague summaries
-- Single line citations reference specific calculations (e.g., :4 for growth rate)
-- Multiline citations reference broader analysis blocks (e.g., :1-2 for the groupby operation)
-- Language is information-dense with concrete metrics
-- All line numbers are 0-indexed
-
-<Example 2>
-
-{ACTIVE_CELL_ID_SECTION_HEADING}
-'1a2b3c4d-5e6f-7g8h-9i0j-k1l2m3n4o5p6'
-
-{CODE_SECTION_HEADING}
-```python
-```
-
-Your task: Hello
-
-Output:
-Hey there! I'm Mito AI. How can I help you today? 
-
-</Example 2>
-
-Notice in the example above that the user is just sending a friendly message, so we respond with a friendly message and do not return any code.
-
-===
-{get_database_rules()}
-
-==== 
-{CHAT_CODE_FORMATTING_RULES}
-
-====
-
-CODE STYLE
-
+"""))
+
+    sections.append(SG.Generic("Chart Config Rules", CHART_CONFIG_RULES))
+    sections.append(SG.Generic("DatabaseRules", get_database_rules()))
+    sections.append(SG.Generic("Citation Rules", CITATION_RULES))
+    sections.append(SG.Generic("Cell Reference Rules", CELL_REFERENCE_RULES))
+    
+    # Example 1
+    sections.append(SG.Example("Example 1", f"""
+                               
+    Notice in this example:
+    - Citations support specific facts and numbers, not vague summaries
+    - Single line citations reference specific calculations (e.g., :4 for growth rate)
+    - Multiline citations reference broader analysis blocks (e.g., :1-2 for the groupby operation)
+    - Language is information-dense with concrete metrics
+    - All line numbers are 0-indexed
+    
+    Active Cell ID: '7b3a9e2c-5d14-4c83-b2f9-d67891e4a5f2'
+
+    Active Cell Code: 
+    ```python
+    sales_df = pd.read_csv('sales_data.csv')\nmonthly_revenue = sales_df.groupby('month')['revenue'].sum()\ntop_month = monthly_revenue.idxmax()\npeak_revenue = monthly_revenue.max()\ngrowth_rate = (monthly_revenue.iloc[-1] / monthly_revenue.iloc[0] - 1) * 100
+    ```
+    
+    Your Task: What are the key revenue insights from this sales data?
+
+    Output:
+    Peak monthly revenue reached $847,392 in March[MITO_CITATION:7b3a9e2c-5d14-4c83-b2f9-d67891e4a5f2:2-3], representing a 23.8% year-over-year growth rate[MITO_CITATION:7b3a9e2c-5d14-4c83-b2f9-d67891e4a5f2:4]. The revenue aggregation analysis[MITO_CITATION:7b3a9e2c-5d14-4c83-b2f9-d67891e4a5f2:1-2] reveals strong seasonal performance patterns."""))
+    
+    sections.append(SG.Example("Example 2", """
+
+    Notice in this example that the user is just sending a friendly message, so we respond with a friendly message and do not return any code.            
+                               
+    Active Cell ID: 
+    '1a2b3c4d-5e6f-7g8h-9i0j-k1l2m3n4o5p6'
+
+    Active Cell Code: 
+    ```python
+    ```
+    
+    Your task: Hello
+
+    Output:
+    Hey there! I'm Mito AI. How can I help you today?"""))
+    
+    
+    sections.append(SG.Example("Example 3", """
+    Files:
+    "file_name: sales.csv"
+
+    Variables:
+    {{
+        'loan_multiplier': 1.5,
+        'sales_df': pd.DataFrame({
+            'transaction_date': ['2024-01-02', '2024-01-02', '2024-01-02', '2024-01-02', '2024-01-03'],
+            'price_per_unit': [10, 9.99, 13.99, 21.00, 100],
+            'units_sold': [1, 2, 1, 4, 5],
+            'total_price': [10, 19.98, 13.99, 84.00, 500]
+        })
+    }}
+
+    Active Cell ID: '9c0d5fda-2b16-4f52-a1c5-a48892f3e2e8'
+
+    Active Cell Code: import pandas as pd\nsales_df = pd.read_csv('./sales.csv')
+
+    Your Task: convert the transaction_date column to datetime and then multiply the total_price column by the sales_multiplier.
+
+    Output:
+    ```python
+    import pandas as pd
+    sales_df = pd.read_csv('./sales.csv')
+    sales_df['transaction_date'] = pd.to_datetime(sales_df['transaction_date'])
+    sales_df['total_price'] = sales_df['total_price'] * sales_multiplier
+    ```
+
+    Applied datetime conversion to enable temporal analysis[MITO_CITATION:9c0d5fda-2b16-4f52-a1c5-a48892f3e2e8:2] and revenue adjustment using the 1.5x sales multiplier[MITO_CITATION:9c0d5fda-2b16-4f52-a1c5-a48892f3e2e8:3], scaling total revenue from $627.97 to $941.96."""))
+
+    # Add code formatting rules
+    sections.append(SG.Generic("CODE FORMATTING RULES", CHAT_CODE_FORMATTING_RULES))
+    
+    # Add code style
+    sections.append(SG.Generic("CODE STYLE", """
 - Avoid using try/except blocks and other defensive programming patterns (like checking if files exist before reading them, verifying variables are defined before using them, etc.) unless there is a really good reason. In Jupyter notebooks, errors should surface immediately so users can identify and fix issues. When errors are caught and suppressed or when defensive checks hide problems, users continue running broken code without realizing it, and the agent's auto-error-fix loop cannot trigger. If a column doesn't exist, a file is missing, a variable isn't defined, or a module isn't installed, let it error. The user needs to know.
 - Write code that preserves the intent of the original code shared with you and the task to complete.
 - Make the solution as simple as possible.
 - Do not add temporary comments like '# Fixed the typo here' or '# Added this line to fix the error'
-- When importing matplotlib, write the code `%matplotlib inline` to make sure the graphs render in Jupyter.
-
-====
-
-IMPORTANT RULES:
+- When importing matplotlib, write the code `%matplotlib inline` to make sure the graphs render in Jupyter."""))
+    
+    # Add important rules
+    sections.append(SG.Generic("IMPORTANT RULES", """
 - Do not recreate variables that already exist
 - Keep as much of the original code as possible
 - When updating an existing code cell, return the full code cell with the update applied. Do not only return part of the code cell with a comment like "# Updated code starts here", etc.
@@ -103,5 +127,7 @@ IMPORTANT RULES:
 - Write code that preserves the intent of the original code shared with you and the task to complete.
 - Make the solution as simple as possible.
 - Reuse as much of the existing code as possible.
-- Whenever writing Python code, it should be a python code block starting with ```python and ending with ```
-"""
+- Whenever writing Python code, it should be a python code block starting with ```python and ending with ```"""))
+
+    prompt = Prompt(sections)
+    return str(prompt)

mito_ai/completions/prompt_builders/explain_code_prompt.py

@@ -1,32 +1,30 @@
 # Copyright (c) Saga Inc.
 # Distributed under the terms of the GNU Affero General Public License v3.0 License.
 
-from mito_ai.completions.prompt_builders.prompt_constants import CODE_SECTION_HEADING
+from typing import List
+from mito_ai.completions.prompt_builders.prompt_section_registry import SG, Prompt
+from mito_ai.completions.prompt_builders.prompt_section_registry.base import PromptSection
 
-def create_explain_code_prompt(active_cell_code: str) -> str:
-    prompt = f"""Explain the code in the active code cell to me like I have a basic understanding of Python. Don't explain each line, but instead explain the overall logic of the code.
-
-<Example>
 
-{CODE_SECTION_HEADING}
-
-```python
-def multiply(x, y):
-    return x * y
-```
+def create_explain_code_prompt(active_cell_code: str) -> str:
+    sections: List[PromptSection] = []
+    # Add intro text
+    sections.append(SG.Generic("Instructions", "Explain the code in the active code cell to me like I have a basic understanding of Python. Don't explain each line, but instead explain the overall logic of the code."))
+    
+    # Add example
+    example_content = f"""{SG.ActiveCellCode('''def multiply(x, y):
+    return x * y''')}
 
 Output:
 
-This code creates a function called `multiply` that takes two arguments `x` and `y`, and returns the product of `x` and `y`.
-
-</Example>
-
-{CODE_SECTION_HEADING}
-
-```python
-{active_cell_code}
-```
-
-Output: 
-"""
-    return prompt
+This code creates a function called `multiply` that takes two arguments `x` and `y`, and returns the product of `x` and `y`."""
+    sections.append(SG.Example("Example", example_content))
+    
+    # Add actual code section
+    sections.append(SG.ActiveCellCode(active_cell_code))
+    
+    # Add output prompt
+    sections.append(SG.Task("Output:"))
+
+    prompt = Prompt(sections)
+    return str(prompt)

mito_ai/completions/prompt_builders/inline_completer_prompt.py

@@ -2,11 +2,8 @@
 # Distributed under the terms of the GNU Affero General Public License v3.0 License.
 
 from typing import List
-from mito_ai.completions.prompt_builders.prompt_constants import (
-    FILES_SECTION_HEADING,
-    VARIABLES_SECTION_HEADING,
-    CODE_SECTION_HEADING
-)
+from mito_ai.completions.prompt_builders.prompt_section_registry import SG, Prompt
+from mito_ai.completions.prompt_builders.prompt_section_registry.base import PromptSection
 
 
 def create_inline_prompt(
@@ -15,10 +12,10 @@ def create_inline_prompt(
     variables: List[str],
     files: List[str]
 ) -> str:
-    variables_str = '\n'.join([f"{variable}" for variable in variables])
-    files_str = '\n'.join([f"file_name: {file}" for file in files])
+    sections: List[PromptSection] = []
     
-    prompt = f"""You are a coding assistant that lives inside of JupyterLab. Your job is to help the user write code.
+    # Add intro text
+    sections.append(SG.Task("""You are a coding assistant that lives inside of JupyterLab. Your job is to help the user write code.
 
 You're given the current code cell, the user's cursor position, and the variables defined in the notebook. The user's cursor is signified by the symbol <cursor>.
 
@@ -27,28 +24,30 @@ CRITICAL FORMATTING RULES:
 2. If you are finishing a line of code that the user started, return the full line of code with no newline character at the start or end.
 3. Your response must preserve correct Python indentation and spacing. For example, if you're completing a line of indented code, you must preserve the indentation.
 
-Your job is to complete the code that matches the user's intent. Write the minimal code to achieve the user's intent. Don't expand upon the user's intent.
-
-<Example 1>
-{FILES_SECTION_HEADING}
-file_name: sales.csv
+Your job is to complete the code that matches the user's intent. Write the minimal code to achieve the user's intent. Don't expand upon the user's intent."""))
+    
+    # Example 1
+    example1_content = f"""
+Files:
+"file_name: sales.csv"
 
-{VARIABLES_SECTION_HEADING} 
+Variables:
 {{
     'loan_multiplier': 1.5,
-    'sales_df': pd.DataFrame({{
+    'sales_df': pd.DataFrame({
         'transaction_date': ['2024-01-02', '2024-01-02', '2024-01-02', '2024-01-02', '2024-01-03'],
         'price_per_unit': [10, 9.99, 13.99, 21.00, 100],
         'units_sold': [1, 2, 1, 4, 5],
         'total_price': [10, 19.98, 13.99, 84.00, 500]
-    }})
+    })
 }}
 
-{CODE_SECTION_HEADING}
+Active Cell Code:
 ```python
 import pandas as pd
 sales_df = pd.read_csv('./sales.csv')
 
+
 # Multiply the total_price column by the loan_multiplier<cursor>
 ```
 
@@ -57,23 +56,22 @@ Output:
 
 sales_df['total_price'] = sales_df['total_price'] * loan_multiplier
 ```
-</Example 1>
 
-IMPORTANT: Notice in Example 1 that the output starts with a newline because the cursor was at the end of a comment. This newline is REQUIRED to maintain proper Python formatting.
-
-<Example 2>
-{FILES_SECTION_HEADING}
+Notice in Example 1 that the output starts with a newline because the cursor was at the end of a comment. This newline is REQUIRED to maintain proper Python formatting."""
 
+    sections.append(SG.Example("Example 1", example1_content))
+    
+    # Example 2
+    example2_content = """
+Files:
 
-{VARIABLES_SECTION_HEADING} 
-{{
-    df: pd.DataFrame({{
-        'age': [20, 25, 22, 23, 29],
-        'name': ['Nawaz', 'Aaron', 'Charlie', 'Tamir', 'Eve'],
-    }})
-}}
+Variables:
+df: pd.DataFrame({
+    'age': [20, 25, 22, 23, 29],
+    'name': ['Nawaz', 'Aaron', 'Charlie', 'Tamir', 'Eve'],
+})
 
-{CODE_SECTION_HEADING}
+Active Cell Code:
 ```python
 df['age'] = df[<cursor>['age'] > 23]
 ```
@@ -82,18 +80,20 @@ Output:
 ```python
 df['age'] = df[df['age'] > 23]
 ```
-</Example 2>
-
-IMPORTANT: Notice in Example 2 that the output does NOT start with a newline because the cursor is in the middle of existing code.
 
-<Example 3>
-{FILES_SECTION_HEADING}
-file_name: voters.csv
+IMPORTANT: Notice in Example 2 that the output does NOT start with a newline because the cursor is in the middle of existing code."
+"""
+    sections.append(SG.Example("Example 2", example2_content))
+    
+    # Example 3
+    example3_content = f"""
+Files:
+"file_name: voters.csv"
 
-{VARIABLES_SECTION_HEADING} 
+Variables:
 {{}}
 
-{CODE_SECTION_HEADING}
+Active Cell Code:
 ```python
 voters = pd.read_csv('./voters.csv')
 
@@ -108,90 +108,61 @@ ohio_voters = voters[voters['state'] == 'OH']
 ca_voters = voters[voters['state'] == 'CA']
 tx_voters = voters[voters['state'] == 'TX']
 ```
-</Example 3>
-
-IMPORTANT: Notice in Example 3 that output does not start with a newline character because it wasnts to continue the line of code that the user started. Also notice the output contains three lines of code because that is the minimal code to achieve the user's intent.
 
-<Example 4>
-{FILES_SECTION_HEADING}
-file_name: july_2025.xlsx
-file_name: august_2025.xlsx
+IMPORTANT: Notice in Example 3 that output does not start with a newline character because it wasnts to continue the line of code that the user started. Also notice the output contains three lines of code because that is the minimal code to achieve the user's intent."""
+    sections.append(SG.Example("Example 3", example3_content))
+    
+    # Example 4
+    example4_content = f"""
+Files:
+"file_name: july_2025.xlsx"
+"file_name: august_2025.xlsx"
 
-{VARIABLES_SECTION_HEADING} 
+Variables:
 {{}}
 
-{CODE_SECTION_HEADING}
+Active Cell Code:
 ```python
-# Display the first 5 rows of the dataframe
-df.head()
-<cursor>
-```
-
-Output:
-```python
-```
-</Example 4>
-
-IMPORTANT: Notice in Example 4 that the output is empty becuase the user's intent is already complete.
-
-<Example 5>
-{FILES_SECTION_HEADING}
-
-
-{VARIABLES_SECTION_HEADING} 
+```"""
+    sections.append(SG.Example("Example 4", example4_content))
+    
+    # Example 5
+    example5_content = f"""
+Files:
 {{}}
 
-{CODE_SECTION_HEADING}
+Active Cell Code:
 ```python
-def even_and_odd():
-    for i in range(10):
-        if i % 2 == 0:
-            print(f"Even: {{i}}")
-        else:
-            pri<cursor>
+def even_and_odd():\n    for i in range(10):\n        if i % 2 == 0:\n            print(f\"Even: {{i}}\")\n        else:\n            pri<cursor>
 ```
 
 Output:
-```python
-            print(f"Odd: {{i}}")
-```
-</Example 5>
-
-IMPORTANT: Notice in Example 5 that the output is indented several times because the code must be executed as part of the else block.
-
-<Example 6>
-{FILES_SECTION_HEADING}
-
-
-{VARIABLES_SECTION_HEADING} 
+```python"""
+    sections.append(SG.Example("Example 5", example5_content))
+    
+    # Example 6
+    example6_content = f"""
+Files:
 {{}}
 
-{CODE_SECTION_HEADING}
-```python
-days_in_week <cursor>
-```
-
-Output:
+Active Cell Code:
 ```python
 days_in_week = 7
 ```
-</Example 6>
-
-IMPORTANT: Notice in Example 6 that inorder to finish the variable declaration, the output continues the existing line of code and does not start with a new line character.
 
-Your Task:
-
-{FILES_SECTION_HEADING}
-{files_str}
-
-{VARIABLES_SECTION_HEADING} 
-{variables_str}
-
-{CODE_SECTION_HEADING}
-```python
-{prefix}<cursor>{suffix}
-```
+IMPORTANT: Notice in Example 6 that inorder to finish the variable declaration, the output continues the existing line of code and does not start with a new line character."""
+    sections.append(SG.Example("Example 6", example6_content))
+    
+    # Add task sections
+    sections.append(SG.Task("Your Task:"))
+    
+    sections.append(SG.Files(files))
+    sections.append(SG.Variables(variables))
+    
+    code_content = f"{prefix}<cursor>{suffix}\n"
+    sections.append(SG.ActiveCellCode(code_content))
+    
+    sections.append(SG.Task("Output:"))
 
-Output:
-"""
-    return prompt
+    prompt = Prompt(sections)
+    return str(prompt)