Skip to content

Commit b16ba86

Browse files
authored
Langchain::Tool::Database#describe_table with COMMENT (#891)
* Langchain::Tool::Database#describe_table with COMMENT Allows LLM to make use of columns' comments. Possible since Sequel 5.87.0. * New sequel dependency in Gemfile.lock
1 parent ce93ea5 commit b16ba86

File tree

4 files changed

+7
-5
lines changed

4 files changed

+7
-5
lines changed

Gemfile.lock

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -381,7 +381,8 @@ GEM
381381
rubyzip (2.3.2)
382382
safe_ruby (1.0.4)
383383
childprocess (>= 0.3.9)
384-
sequel (5.68.0)
384+
sequel (5.87.0)
385+
bigdecimal
385386
signet (0.19.0)
386387
addressable (~> 2.8)
387388
faraday (>= 0.17.5, < 3.a)
@@ -480,7 +481,7 @@ DEPENDENCIES
480481
rubocop
481482
ruby-openai (~> 7.1.0)
482483
safe_ruby (~> 1.0.4)
483-
sequel (~> 5.68.0)
484+
sequel (~> 5.87.0)
484485
standard (>= 1.35.1)
485486
vcr
486487
weaviate-ruby (~> 0.9.2)

langchain.gemspec

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ Gem::Specification.new do |spec|
7171
spec.add_development_dependency "roo-xls", "~> 1.2.0"
7272
spec.add_development_dependency "ruby-openai", "~> 7.1.0"
7373
spec.add_development_dependency "safe_ruby", "~> 1.0.4"
74-
spec.add_development_dependency "sequel", "~> 5.68.0"
74+
spec.add_development_dependency "sequel", "~> 5.87.0"
7575
spec.add_development_dependency "weaviate-ruby", "~> 0.9.2"
7676
spec.add_development_dependency "wikipedia-client", "~> 1.17.0"
7777
spec.add_development_dependency "power_point_pptx", "~> 0.1.0"

lib/langchain/tool/database.rb

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ module Langchain::Tool
55
# Connects to a SQL database, executes SQL queries, and outputs DB schema for Agents to use
66
#
77
# Gem requirements:
8-
# gem "sequel", "~> 5.68.0"
8+
# gem "sequel", "~> 5.87.0"
99
#
1010
# Usage:
1111
# database = Langchain::Tool::Database.new(connection_string: "postgres://user:password@localhost:5432/db_name")
@@ -115,6 +115,7 @@ def describe_table(table)
115115
else
116116
primary_key_columns << column[0]
117117
end
118+
schema << " COMMENT '#{column[1][:comment]}'" if column[1][:comment]
118119
schema << ",\n" unless column == db.schema(table).last && primary_key_column_count == 1
119120
end
120121
if primary_key_column_count > 1

lib/langchain/vectorsearch/pgvector.rb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ class Pgvector < Base
66
# The PostgreSQL vector search adapter
77
#
88
# Gem requirements:
9-
# gem "sequel", "~> 5.68.0"
9+
# gem "sequel", "~> 5.87.0"
1010
# gem "pgvector", "~> 0.2"
1111
#
1212
# Usage:

0 commit comments

Comments
 (0)