fix: calling generate functions with openai

Signed-off-by: Matthew Peveler <mpeveler@timescale.com>
timescale · Feb 7, 2025 · 3bf9b38 · 3bf9b38
1 parent 59b9e6c
commit 3bf9b38
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/projects/extension/ai/semantic_catalog.py b/projects/extension/ai/semantic_catalog.py
@@ -161,7 +161,7 @@ def generate_description(
                     $1
                     , $2
                     , tools => $3
-                    , tool_choice => '{"type": "function", "function": {"name": "generate_description"}}'::jsonb
+                    , tool_choice => '{"type": "function", "function": {"name": "generate_description"}}'
                 )
             """,
             ["text", "jsonb", "jsonb"],
@@ -293,7 +293,7 @@ def generate_column_descriptions(
                     $1
                     , $2
                     , tools => $3
-                    , tool_choice => '{"type": "function", "function": {"name": "generate_description"}}'::jsonb
+                    , tool_choice => '{"type": "function", "function": {"name": "generate_description"}}'
                 )
             """,
             ["text", "jsonb", "jsonb"],
@@ -414,7 +414,7 @@ def generate_function_description(
                     $1
                     , $2
                     , tools => $3
-                    , tool_choice => '{"type": "function", "function": {"name": "generate_description"}}'::jsonb
+                    , tool_choice => '{"type": "function", "function": {"name": "generate_description"}}'
                 )
             """,
             ["text", "jsonb", "jsonb"],