human_sql 0.1.4 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2105327e3f06a71f6a2e87920bd571bd82e6c2d7c891a9bf939f685a9796a2dc
-  data.tar.gz: 5012bb480c854b5b86a891130dd73ac938e7cf1138428a6918d76193836f8d9d
+  metadata.gz: d43fde91937997b59b2239bd8669ecdffb020a7424007b48f1504aa0b3384221
+  data.tar.gz: 8f219654ff6f0371c76a98aff8e13573b651fcd42d689b8f7556b11a1f7c500d
 SHA512:
-  metadata.gz: e39cece766a98e64e0f6bf59a2124348e15a6e970fbee4d25c6184a74508b839445cf9d00012d081240b6dfff4336ff19b12b1ac27c0a587ab212cdba57adbee
-  data.tar.gz: 66b6ee8fb138967521063913ab8891f36b1950f01ce4dd733555d7cc58da40a3b6f72a1cf843b38a85546cd8f53b13625d2a6cef96d608f9292cc0356d691a28
+  metadata.gz: '0489de0620281892772e3e26918655e68e349d81858b803355dc33309fc2750dd80419ef6f537b259697a2258848fc5d11a9f0c0fc61051bdb91236a893400cf'
+  data.tar.gz: 0cb6a1eab9988336932a800193fd733f2baf713a4d31604852fea8bc031e0774f461bb724f65b37e1ae5fad39ea8c01f3684952c022c0d23d3453fc7d9b13cbe

data/lib/human_sql/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module HumanSql
-  VERSION = "0.1.4"
+  VERSION = "0.1.5"
 end

data/lib/human_sql.rb

@@ -7,7 +7,7 @@ require_relative 'human_sql/version'
 module HumanSQL
   class QueryBuilder
     OPENAI_API_URL = 'https://api.openai.com/v1/chat/completions'
-
+    
     def initialize(user_input)
       @user_input = user_input
       @schema_content = File.read(Rails.root.join('db', 'schema.rb'))
@@ -19,8 +19,11 @@ module HumanSQL
 
     def generate_query
       prompt = build_query_prompt(@user_input, @schema_content)
+
       generated_query_response = call_openai_service(prompt)&.strip
-      extract_active_record_query(generated_query_response)
+      extracted_query = extract_active_record_query(generated_query_response)
+      
+      extracted_query
     end
 
     def get_results
@@ -32,6 +35,7 @@ module HumanSQL
 
       formatted_results = format_results_for_openai(results)
       natural_language_response = generate_natural_language_response(formatted_results, @user_input)
+
       natural_language_response
     rescue StandardError => e
       process_error_in_natural_language(e.message)
@@ -40,7 +44,7 @@ module HumanSQL
     private
 
     def build_query_prompt(user_input, schema_content)
-      <<-PROMPT
+      prompt = <<-PROMPT
         The user has requested: "#{user_input}".
         This is the database schema:
 
@@ -49,11 +53,12 @@ module HumanSQL
         Please generate an ActiveRecord query based on this schema. The query should be in a single line of code and return the result according to the user's request. 
         If it's necessary to access multiple related tables, prefer to use `includes` over `joins` to optimize data loading.
       PROMPT
+      prompt
     end
 
     def call_openai_service(prompt)
       body = {
-        model: "gpt-4",
+        model: "gpt-4o-mini",
         messages: [
           {
             role: "system",
@@ -84,12 +89,14 @@ module HumanSQL
         response_body = JSON.parse(response.body)
         response_body.dig('choices', 0, 'message', 'content')
       else
-        raise "Error in OpenAI API: #{response.code} - #{response.body}"
+        error_message = "Error in OpenAI API: #{response.code} - #{response.body}"
+        raise error_message
       end
     end
 
     def extract_active_record_query(response)
       return nil if response.blank?
+
       code_lines = response.lines.select { |line| line.strip.match(/^[A-Za-z]\w*\./) }
       return nil if code_lines.empty?
       code_lines.join("\n").strip
@@ -103,15 +110,17 @@ module HumanSQL
 
     def format_results_for_openai(results)
       if results.is_a?(ActiveRecord::Relation) || results.is_a?(Array)
-        results.map do |result|
+        formatted = results.map do |result|
           if result.respond_to?(:attributes)
             result.attributes.map { |key, value| "#{key}: #{value}" }.join(', ')
           else
             result.to_s
           end
         end.join("\\n")
+        formatted
       else
-        results.respond_to?(:attributes) ? results.attributes.map { |key, value| "#{key}: #{value}" }.join(', ') : results.to_s
+        formatted = results.respond_to?(:attributes) ? results.attributes.map { |key, value| "#{key}: #{value}" }.join(', ') : results.to_s
+        formatted
       end
     end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: human_sql
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.1.5
 platform: ruby
 authors:
 - Nicolas Bertin
@@ -55,6 +55,7 @@ files:
 - human_sql-0.1.1.gem
 - human_sql-0.1.2.gem
 - human_sql-0.1.3.gem
+- human_sql-0.1.4.gem
 - lib/human_sql.rb
 - lib/human_sql/version.rb
 - sig/human_sql.rbs