From 4a3d3bcb32cc85c41b53506d5c95f8e67cf3709f Mon Sep 17 00:00:00 2001
From: Julien <364924+Bahanix@users.noreply.github.com>
Date: Sun, 1 Dec 2024 20:53:56 +0100
Subject: [PATCH] Langchain::Tool::Database#describe_table with COMMENT

Allows LLM to make use of columns' comments.
Possible since Sequel 5.87.0.
---
 Gemfile.lock                           | 4 ++--
 langchain.gemspec                      | 2 +-
 lib/langchain/tool/database.rb         | 1 +
 lib/langchain/vectorsearch/pgvector.rb | 2 +-
 4 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/Gemfile.lock b/Gemfile.lock
index 8bc496e52..7583b8bb6 100644
--- 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
index f38500d0a..fec9e2ea8 100644
--- 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
index ff38107ff..a40744244 100644
--- 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
index 3d7b9ad68..de29224cc 100644
--- 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: