Add more example blueprints and update system prompt

README.md

@@ -48,11 +48,11 @@ pinned: false
 
 ### Finetuning the GPT model
 
-1. Download `data/finetuning_data_cot_v8.jsonl` in the GitHub repository.
+1. Download `data/finetuning_data_cot_v9.jsonl` in the GitHub repository.
 2. Visit the [OpenAI Fine-tuning page](https://platform.openai.com/finetune) and click `+ Create` in the top right corner.
 3. Fill in the form with the following information, and leave the rest as default:
    - `Base Model`: `gpt-3.5-turbo-0125` (or any other model you prefer)
-   - `Training data`: Upload the `finetuning_data_cot_v8.jsonl` file.
+   - `Training data`: Upload the `finetuning_data_cot_v9.jsonl` file.
    - `Suffix`: NotionGPT (or any other name you prefer)
 4. Click `Create` and wait for the model to finish training.
 5. Once the model is trained, copy the model name and save it for later. (It should look something like `ft:gpt-3.5-turbo-0125:personal:notiongpt:<ID>`)

blueprints/architect.py

@@ -91,7 +91,7 @@ def process_blueprint(parent_id, block_json):
 def generate_blueprint(description, force_json=False, temperature=0.8, top_p=0.3, error=None, failed_response=None):
     system_prompt = """You are NotionGPT, a state-of-the-art template designer for Notion, programmed to create custom JSON blueprints that represent detailed, organized, and highly functional Notion templates. Your templates should be ready for users to use immediately and should meet their specific organizational needs, allowing users to customize them to suit their needs.
 
-    Please respond ONLY with valid json that conforms to the `OpenAIResponse(BaseModel)` class as defined by pydantic: 
+    Please respond ONLY with valid json that conforms to the `OpenAIResponse(BaseModel)` class as defined by pydantic in the Python code below:
 
     ```
     from __future__ import annotations
@@ -338,8 +338,7 @@ def generate_blueprint(description, force_json=False, temperature=0.8, top_p=0.3
     Quote.update_forward_refs()
     ```
 
-    Do not include any additional text other than the object json as we will load this object with json.loads() and pydantic."""
-
+    Do not include any additional text other than the object json as we will load this object with json.loads() and pydantic. Additionally, make sure the Notion page structure you generate is complete and fully represents the rough outline described in the response."""
     if error and failed_response:
         response = client.chat.completions.create(
             model=MODEL_NAME,

data/process_data.py

@@ -24,7 +24,7 @@ blueprints_df = pd.read_csv("./example_blueprints.csv", encoding="utf-8-sig")
 
 system_prompt = """You are NotionGPT, a state-of-the-art template designer for Notion, programmed to create custom JSON blueprints that represent detailed, organized, and highly functional Notion templates. Your templates should be ready for users to use immediately and should meet their specific organizational needs, allowing users to customize them to suit their needs.
 
-Please respond ONLY with valid json that conforms to the `OpenAIResponse(BaseModel)` class as defined by pydantic: 
+Please respond ONLY with valid json that conforms to the `OpenAIResponse(BaseModel)` class as defined by pydantic in the Python code below:
 
 ```
 from __future__ import annotations
@@ -271,7 +271,7 @@ Callout.update_forward_refs()
 Quote.update_forward_refs()
 ```
 
-Do not include any additional text other than the object json as we will load this object with json.loads() and pydantic."""
+Do not include any additional text other than the object json as we will load this object with json.loads() and pydantic. Additionally, make sure the Notion page structure you generate is complete and fully represents the rough outline described in the response."""
 
 messages = []
 for _, row in blueprints_df.iterrows():
@@ -292,6 +292,6 @@ for _, row in blueprints_df.iterrows():
     }
     messages.append(entry)
 
-with open("finetuning_data_cot_v8.jsonl", "w") as f:
+with open("finetuning_data_cot_v9.jsonl", "w") as f:
     for message in messages:
         print(json.dumps(message, ensure_ascii=False), file=f)