From 43b236e42508136948644e2be7bbe0124711da6c Mon Sep 17 00:00:00 2001
From: Vinny <vinnyglennon@gmail.com>
Date: Thu, 5 Sep 2024 09:42:27 +0100
Subject: [PATCH 1/2] Lock down json response
 https://github.com/alexrudall/ruby-openai?tab=readme-ov-file#json-mode

FormatOutput can be argued as not needed, but the class could come in handy if Claude or other LLMs have backticks in their output, hence I left it there
---
 lib/dotcodegen/test_code_generator.rb | 1 +
 lib/dotcodegen/version.rb             | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/lib/dotcodegen/test_code_generator.rb b/lib/dotcodegen/test_code_generator.rb
index 12fbf98..4006eef 100644
--- a/lib/dotcodegen/test_code_generator.rb
+++ b/lib/dotcodegen/test_code_generator.rb
@@ -19,6 +19,7 @@ def generate_test_code
         parameters: {
           model: 'gpt-4-turbo-preview',
           messages: [{ role: 'user', content: test_prompt_text }], # Required.
+          response_format: { type: "json_object" },
           temperature: 0.7
         }
       )
diff --git a/lib/dotcodegen/version.rb b/lib/dotcodegen/version.rb
index bc7553b..a13e5bc 100644
--- a/lib/dotcodegen/version.rb
+++ b/lib/dotcodegen/version.rb
@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Dotcodegen
-  VERSION = '0.1.5'
+  VERSION = '0.1.6'
 end

From c1ae6448266d99e5f17ef842132eade44eddf01d Mon Sep 17 00:00:00 2001
From: Vinny <vinnyglennon@gmail.com>
Date: Thu, 5 Sep 2024 09:55:55 +0100
Subject: [PATCH 2/2] gpt-4o model

---
 Gemfile.lock                          | 2 +-
 lib/dotcodegen/lint_code.rb           | 1 -
 lib/dotcodegen/test_code_generator.rb | 2 +-
 3 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/Gemfile.lock b/Gemfile.lock
index 087ab77..e0da9d9 100644
--- a/Gemfile.lock
+++ b/Gemfile.lock
@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    dotcodegen (0.1.5)
+    dotcodegen (0.1.6)
       dotenv
       front_matter_parser
       ruby-openai
diff --git a/lib/dotcodegen/lint_code.rb b/lib/dotcodegen/lint_code.rb
index 9ca1307..3d0376f 100644
--- a/lib/dotcodegen/lint_code.rb
+++ b/lib/dotcodegen/lint_code.rb
@@ -23,7 +23,6 @@ def run
       end
     end
 
-
     def standardrb_code
       puts "Linting: StandardRB"
       system("standardrb --fix-unsafely #{@file_path}")
diff --git a/lib/dotcodegen/test_code_generator.rb b/lib/dotcodegen/test_code_generator.rb
index 4006eef..a226e57 100644
--- a/lib/dotcodegen/test_code_generator.rb
+++ b/lib/dotcodegen/test_code_generator.rb
@@ -17,7 +17,7 @@ def initialize(config:, file_to_test_path:, openai_key:, openai_org_id: nil)
     def generate_test_code
       response = openai_client.chat(
         parameters: {
-          model: 'gpt-4-turbo-preview',
+          model: 'gpt-4o',
           messages: [{ role: 'user', content: test_prompt_text }], # Required.
           response_format: { type: "json_object" },
           temperature: 0.7