Merge pull request rizethereum#2 from justi-blue/66ve70-codex/configure-openai-api-with-gem

Integrate OpenAI sentence expansion

Author: justi

Gemfile

@@ -18,6 +18,7 @@ gem "stimulus-rails"
 gem "tailwindcss-rails"
 # Build JSON APIs with ease [https://github.com/rails/jbuilder]
 gem "jbuilder"
+gem "ruby-openai"
 
 # Use Active Model has_secure_password [https://guides.rubyonrails.org/active_model_basics.html#securepassword]
 # gem "bcrypt", "~> 3.1.7"

Gemfile.lock

@@ -98,6 +98,15 @@ GEM
     erubi (1.13.1)
     et-orbi (1.2.11)
       tzinfo
+    event_stream_parser (1.0.0)
+    faraday (2.13.4)
+      faraday-net_http (>= 2.0, < 3.5)
+      json
+      logger
+    faraday-multipart (1.1.1)
+      multipart-post (~> 2.0)
+    faraday-net_http (3.4.1)
+      net-http (>= 0.5.0)
     fugit (1.11.1)
       et-orbi (~> 1, >= 1.2.11)
       raabro (~> 1.4)
@@ -144,6 +153,9 @@ GEM
     mini_mime (1.1.5)
     minitest (5.25.5)
     msgpack (1.8.0)
+    multipart-post (2.4.1)
+    net-http (0.6.0)
+      uri
     net-imap (0.5.9)
       date
       net-protocol
@@ -281,6 +293,10 @@ GEM
       rubocop (>= 1.72)
       rubocop-performance (>= 1.24)
       rubocop-rails (>= 2.30)
+    ruby-openai (8.1.0)
+      event_stream_parser (>= 0.3.0, < 2.0.0)
+      faraday (>= 1)
+      faraday-multipart (>= 1)
     ruby-progressbar (1.13.0)
     securerandom (0.4.1)
     solid_cable (3.0.11)
@@ -371,6 +387,7 @@ DEPENDENCIES
   rails (~> 8.0.2)
   rspec-rails
   rubocop-rails-omakase
+  ruby-openai
   solid_cable
   solid_cache
   solid_queue

app/services/sentence_expander.rb

@@ -27,4 +27,28 @@ def expand(base_sentence)
       "B2" => "#{subject} thoroughly enjoy spending time with #{object}, as their captivating nature fascinates #{object_pronoun}."
     }
   end
+
+  def expand_with_ai(base_sentence, model: "gpt-4o-mini")
+    client = OpenAI::Client.new
+    safe_sentence = base_sentence.to_s.gsub(/[\r\n]/, " ")
+    prompt = "Expand the sentence #{JSON.generate(safe_sentence)} into CEFR levels: #{LEVELS.join(', ')}. Respond as JSON with keys #{LEVELS.join(', ')}."
+    response = client.chat(
+      parameters: {
+        model: model,
+        messages: [
+          { role: "user", content: prompt }
+        ],
+        response_format: { type: "json_object" }
+      }
+    )
+
+    content = response.dig("choices", 0, "message", "content")
+    begin
+      raise "OpenAI API response missing expected 'content' field: #{response.inspect}" unless content.is_a?(String) && !content.strip.empty?
+      JSON.parse(content)
+    rescue JSON::ParserError, RuntimeError
+      # Log the error or handle it as needed; return empty hash for safety
+      {}
+    end
+  end
 end

config/initializers/openai.rb

@@ -0,0 +1,3 @@
+OpenAI.configure do |config|
+  config.access_token = ENV["OPENAI_API_KEY"]
+end

spec/services/sentence_expander_spec.rb

@@ -19,4 +19,70 @@
       expect(result["B2"]).to include("fascinates her.")
     end
   end
+
+  describe "#expand_with_ai" do
+    it "parses JSON response from OpenAI" do
+      fake_client = instance_double(OpenAI::Client)
+      allow(OpenAI::Client).to receive(:new).and_return(fake_client)
+      allow(fake_client).to receive(:chat).and_return(
+        {
+          "choices" => [
+            { "message" => { "content" => '{"A1":"Hello."}' } }
+          ]
+        }
+      )
+
+      expander = described_class.new
+      result = expander.expand_with_ai("Hello")
+
+      expect(result["A1"]).to eq("Hello.")
+    end
+
+    it "returns empty hash when OpenAI response is missing content" do
+      fake_client = instance_double(OpenAI::Client)
+      allow(OpenAI::Client).to receive(:new).and_return(fake_client)
+      allow(fake_client).to receive(:chat).and_return({})
+
+      expander = described_class.new
+
+      expect(expander.expand_with_ai("Hello")).to eq({})
+    end
+
+    it "returns empty hash when OpenAI response contains invalid JSON" do
+      fake_client = instance_double(OpenAI::Client)
+      allow(OpenAI::Client).to receive(:new).and_return(fake_client)
+      allow(fake_client).to receive(:chat).and_return(
+        {
+          "choices" => [
+            { "message" => { "content" => "not-json" } }
+          ]
+        }
+      )
+
+      expander = described_class.new
+
+      expect(expander.expand_with_ai("Hello")).to eq({})
+    end
+
+    it "sanitizes the base sentence before sending to OpenAI" do
+      fake_client = instance_double(OpenAI::Client)
+      allow(OpenAI::Client).to receive(:new).and_return(fake_client)
+
+      captured_prompt = nil
+      allow(fake_client).to receive(:chat) do |params|
+        captured_prompt = params.dig(:parameters, :messages, 0, :content)
+        {
+          "choices" => [
+            { "message" => { "content" => '{"A1":"Hi."}' } }
+          ]
+        }
+      end
+
+      expander = described_class.new
+      expander.expand_with_ai("O'Hara\n<script>")
+
+      expect(captured_prompt).not_to include("\n")
+      expect(captured_prompt).to include("\"O'Hara <script>\"")
+    end
+  end
 end