MCPcopy Create free account
hub / github.com/THUDM/AgentTuning / build_sql

Function build_sql

AgentBench.old/src/tasks/dbbench/__init__.py:32–52  ·  view source on GitHub ↗
(entry, conn)

Source from the content-addressed store, hash-verified

30
31
32def build_sql(entry, conn):
33 name = entry["table"]["table_name"]
34 columns = ",".join(
35 [f"`{escape(column['name'], conn)}` TEXT" for column in entry["table"]["table_info"]["columns"]])
36 column_names = ",".join(
37 [f"`{escape(column['name'], conn)}`" for column in entry["table"]["table_info"]["columns"]])
38 items = []
39 for row in entry["table"]["table_info"]["rows"]:
40 item = "("
41 for col in row:
42 item += f"'{escape(col, conn)}',"
43 item = item[:-1] + ")"
44 items.append(item)
45 items = ",".join(items)
46 sql = f"""CREATE DATABASE IF NOT EXISTS `{name}`;
47USE `{name}`;
48CREATE TABLE IF NOT EXISTS `{name}` ({columns});
49INSERT INTO `{name}` ({column_names}) VALUES {items};
50COMMIT;
51"""
52 return sql
53
54
55def escape(string: str, conn):

Callers 1

processFunction · 0.85

Calls 2

escapeFunction · 0.85
joinMethod · 0.80

Tested by

no test coverage detected