human_sql 0.1.7 → 0.1.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9465407cb06d1cdf07d92a41df2fbb43f0e6b4fa1d05f5a7023d99ca4d4cc893
-  data.tar.gz: ca2584673baa3b67f227fcd44dd05fe0a5dc79f6025b7784377ad43da7a68ac4
+  metadata.gz: a61925cacbf601619d70681bbb66384d911689155fe732e23dc2f848003008e5
+  data.tar.gz: 9e69254fada3069ceab0c720b4f11f79ebff12047ce329058ca3c31829b9b6be
 SHA512:
-  metadata.gz: a645a5c3a1182492ffb0c6cefe84ebf3fe9d924dea1c31a464e09a68be6f965233a8d01596ee4223a065a48f6953c6479a906589e35fc8b40e2c2b48ab6e433f
-  data.tar.gz: d1dd7b6023d0fd76aa3f92d0f10e1a40bf1110d159c61908f7bf9b8b2707d0f6c6094dd520c4c970adf89856040a05e1a1bbbc2ef5f8b5427363ee14994d8054
+  metadata.gz: 62e3320021cc7aec5b26cc658cedd12aef67a9147219962f618065e067a8a4e962cfffb5354902c98644f3ea1b79e4cc42aba4ffbbfb2045db2f74eef8429ab7
+  data.tar.gz: daa62db81b58e300f8cb8fb073363ee7f2875ab52d45c792f07d41ef2ad484c8a207f123bea52e1989fc22ded04ac25f83584106f489e7e005aa40d01ae1eec3

data/lib/human_sql/version.rb

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

data/lib/human_sql.rb

@@ -7,10 +7,10 @@ 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'))
+      @schema_content = summarize_schema_with_ignored_tables
     end
 
     def self.run(user_input)
@@ -43,6 +43,54 @@ module HumanSQL
 
     private
 
+    def summarize_schema_with_ignored_tables
+      schema_content = File.read(Rails.root.join('db', 'schema.rb'))
+      ignored_tables = HumanSQLConfig[:ignored_tables] || []
+
+      schema_summary = {}
+      schema_content.split("\n").each do |line|
+        if line.strip.start_with?("create_table")
+          table_name = line.match(/"([^"]+)"/)[1]
+          next if ignored_tables.include?(table_name)
+
+          schema_summary[table_name] = { columns: [], has_many: [], belongs_to: [] }
+        end
+
+        current_table = schema_summary.keys.last if schema_summary.any?
+
+        if line.strip.match?(/^t\.(\w+)\s+"([^"]+)"/) && current_table
+          next if ignored_tables.include?(current_table)
+
+          column_type = line.strip.match(/^t\.(\w+)\s+"([^"]+)"/)[1]
+          column_name = line.strip.match(/^t\.(\w+)\s+"([^"]+)"/)[2]
+          schema_summary[current_table][:columns] << "#{column_name}=#{column_type}"
+        end
+
+        if line.strip.start_with?("add_foreign_key")
+          match_data = line.strip.match(/"([^"]+)", "([^"]+)"(?:, column: "([^"]+)")?/)
+          if match_data
+            parent_table = match_data[1]
+            child_table = match_data[2]
+            next if ignored_tables.include?(parent_table) || ignored_tables.include?(child_table)
+
+            schema_summary[child_table] ||= { columns: [], has_many: [], belongs_to: [] }
+            schema_summary[parent_table] ||= { columns: [], has_many: [], belongs_to: [] }
+            schema_summary[child_table][:belongs_to] << parent_table
+            schema_summary[parent_table][:has_many] << child_table
+          end
+        end
+      end
+
+      schema_summary.map do |table, details|
+        columns = details[:columns].join(",")
+        has_many = details[:has_many].join(",")
+        belongs_to = details[:belongs_to].join(",")
+        %Q{#{table}={#{columns},:has_many=>[#{has_many}],:belongs_to=>[#{belongs_to}]}}
+      end.join("\n")
+    rescue StandardError => e
+      raise "Error processing schema: #{e.message}"
+    end
+
     def build_query_prompt(user_input, schema_content)
       prompt = <<-PROMPT
         The user has requested: "#{user_input}".
@@ -51,8 +99,10 @@ module HumanSQL
         #{schema_content}
 
         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.
+        If it's necessary to access multiple related tables, prefer to use preload to optimize data loading.
         If the user indicates a date and time type field, only the value of the indicated date or time is saved without considering the time zone.
+
+        #{HumanSQLConfig[:additional_instructions]}
       PROMPT
       prompt
     end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: human_sql
 version: !ruby/object:Gem::Version
-  version: 0.1.7
+  version: 0.1.8
 platform: ruby
 authors:
 - Nicolas Bertin
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-09-09 00:00:00.000000000 Z
+date: 2024-12-31 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
@@ -58,6 +58,7 @@ files:
 - human_sql-0.1.4.gem
 - human_sql-0.1.5.gem
 - human_sql-0.1.6.gem
+- human_sql-0.1.7.gem
 - lib/human_sql.rb
 - lib/human_sql/version.rb
 - sig/human_sql.rbs
@@ -83,7 +84,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.18
+rubygems_version: 3.5.22
 signing_key:
 specification_version: 4
 summary: A gem to convert natural language to ActiveRecord queries using OpenAI