From 3acf46964bf2628d41341d2cd77e6ed468292325 Mon Sep 17 00:00:00 2001 From: narawat Date: Wed, 15 Jul 2026 14:25:04 +0700 Subject: [PATCH] update --- src/llmfunction.jl | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/llmfunction.jl b/src/llmfunction.jl index 9a2dad4..f9f7adf 100644 --- a/src/llmfunction.jl +++ b/src/llmfunction.jl @@ -530,7 +530,8 @@ function generatesql(a::T, searchterm::String, related_tables = a.context.find_related_tables_for_user_question(searchterm) table_schema = "" for table in related_tables - table_schema_df = GeneralUtils.get_db_table_schema(a.context.pg_conn_str, table) + _table_schema_df = GeneralUtils.get_db_table_schema(a.context.pg_conn_str, table) + table_schema_df = _table_schema_df[:, [:column_name, :data_type, :constraint_type]] table_schema_str = sprint(show, table_schema_df) * "\n" table_schema = table_schema * table_schema_str end