Langchain::Tool::Database#describe_table with COMMENT

Allows LLM to make use of columns' comments. Possible since Sequel 5.87.0.
patterns-ai-core · Dec 1, 2024 · 4a3d3bc · 4a3d3bc
1 parent 4b447e3
commit 4a3d3bc
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 4 deletions.
diff --git a/Gemfile.lock b/Gemfile.lock
@@ -381,7 +381,7 @@ GEM
     rubyzip (2.3.2)
     safe_ruby (1.0.4)
       childprocess (>= 0.3.9)
-    sequel (5.68.0)
+    sequel (5.87.0)
     signet (0.19.0)
       addressable (~> 2.8)
       faraday (>= 0.17.5, < 3.a)
@@ -480,7 +480,7 @@ DEPENDENCIES
   rubocop
   ruby-openai (~> 7.1.0)
   safe_ruby (~> 1.0.4)
-  sequel (~> 5.68.0)
+  sequel (~> 5.87.0)
   standard (>= 1.35.1)
   vcr
   weaviate-ruby (~> 0.9.2)

diff --git a/langchain.gemspec b/langchain.gemspec
@@ -71,7 +71,7 @@ Gem::Specification.new do |spec|
   spec.add_development_dependency "roo-xls", "~> 1.2.0"
   spec.add_development_dependency "ruby-openai", "~> 7.1.0"
   spec.add_development_dependency "safe_ruby", "~> 1.0.4"
-  spec.add_development_dependency "sequel", "~> 5.68.0"
+  spec.add_development_dependency "sequel", "~> 5.87.0"
   spec.add_development_dependency "weaviate-ruby", "~> 0.9.2"
   spec.add_development_dependency "wikipedia-client", "~> 1.17.0"
   spec.add_development_dependency "power_point_pptx", "~> 0.1.0"

diff --git a/lib/langchain/tool/database.rb b/lib/langchain/tool/database.rb
@@ -115,6 +115,7 @@ def describe_table(table)
         else
           primary_key_columns << column[0]
         end
+        schema << " COMMENT '#{column[1][:comment]}'" if column[1][:comment]
         schema << ",\n" unless column == db.schema(table).last && primary_key_column_count == 1
       end
       if primary_key_column_count > 1

diff --git a/lib/langchain/vectorsearch/pgvector.rb b/lib/langchain/vectorsearch/pgvector.rb
@@ -6,7 +6,7 @@ class Pgvector < Base
     # The PostgreSQL vector search adapter
     #
     # Gem requirements:
-    #     gem "sequel", "~> 5.68.0"
+    #     gem "sequel", "~> 5.87.0"
     #     gem "pgvector", "~> 0.2"
     #
     # Usage: