Skip to content

Commit 3706593

Browse files
return create functions in resp
1 parent 960cefe commit 3706593

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

singlestoredb/functions/ext/asgi.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -964,6 +964,13 @@ def get_function_info(
964964
functions = {}
965965
no_default = object()
966966

967+
# Generate CREATE FUNCTION SQL for each function using get_create_functions
968+
create_sqls = self.get_create_functions(replace=True)
969+
sql_map = {}
970+
for key, (_, info), sql in zip(self.endpoints.keys(), self.endpoints.values(), create_sqls):
971+
sig = info['signature']
972+
sql_map[sig['name']] = sql
973+
967974
for key, (_, info) in self.endpoints.items():
968975
if not func_name or key == func_name:
969976
sig = info['signature']
@@ -1002,7 +1009,10 @@ def get_function_info(
10021009
returns[-1]['default'] = a['default']
10031010

10041011
functions[sig['name']] = dict(
1005-
args=args, returns=returns, function_type=info['function_type'],
1012+
args=args,
1013+
returns=returns,
1014+
function_type=info['function_type'],
1015+
sql_statement=sql_map[sig['name']] if sig['name'] in sql_map else None,
10061016
)
10071017

10081018
return functions

0 commit comments

Comments
 (0)