-
Notifications
You must be signed in to change notification settings - Fork 2
/
function-insert-action.js
152 lines (149 loc) · 4.54 KB
/
function-insert-action.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
const Workflow = require("@saltcorn/data/models/workflow");
const Form = require("@saltcorn/data/models/form");
const FieldRepeat = require("@saltcorn/data/models/fieldrepeat");
const Field = require("@saltcorn/data/models/field");
const Table = require("@saltcorn/data/models/table");
const { getState } = require("@saltcorn/data/db/state");
const { interpolate } = require("@saltcorn/data/utils");
const { getCompletion, getEmbedding } = require("./generate");
const { eval_expression } = require("@saltcorn/data/models/expression");
const noSpaces = (s) => s.replaceAll(" ", "");
module.exports = (config) => ({
description: "Use LLM function call to insert rows in tables",
requireRow: true,
disableInList: true,
disableInBuilder: true,
configFields: async ({ table }) => {
const tables = await Table.find();
return [
{
name: "prompt_template",
label: "Prompt",
type: "String",
fieldview: "textarea",
sublabel: `Use interpolations {{ }} to access variables in ${table.name} table.`,
},
{
name: "function_name",
label: "Function name",
type: "String",
},
{
name: "function_description",
label: "Function description",
sublabel: "Describe what you are trying to achieve in general terms",
type: "String",
},
new FieldRepeat({
name: "columns",
fields: [
{
name: "target_table",
label: "Target table",
type: "String",
required: true,
attributes: { options: tables.map((t) => t.name) },
},
{
label: "Fixed values",
name: "fixed_values",
type: "String",
fieldview: "textarea",
},
{
name: "cardinality",
label: "Cardinality",
type: "String",
required: true,
attributes: {
options: ["One", /*"Zero or one",*/ "Zero to many"],
},
},
],
}),
];
},
run: async ({
row,
table,
configuration: {
prompt_template,
columns,
function_description,
function_name,
},
user,
}) => {
const prompt = interpolate(prompt_template, row, user);
let args = {};
const json_type = (ty) => {
if (ty?.js_type) return ty?.js_type;
};
for (const col of columns) {
const target_table = Table.findOne({ name: col.target_table });
const tableArgs = {};
const fixed = eval_expression(
col.fixed_values,
row,
user,
"llm_function_call fixed values"
);
for (const field of target_table.fields) {
if (field.primary_key) continue;
if (typeof fixed[field.name] !== "undefined") continue;
tableArgs[field.name] = {
type: json_type(field.type),
description: field.description,
};
}
const argObj = { type: "object", properties: tableArgs };
args[noSpaces(target_table.name)] =
col.cardinality == "One" ? argObj : { type: "array", items: argObj };
}
if (columns.length === 1) {
//args = args[Object]
}
const expert_function = {
type: "function",
function: {
name: function_name,
description: function_description,
parameters: {
type: "object",
properties: args,
},
},
};
const toolargs = {
tools: [expert_function],
tool_choice: { type: "function", function: { name: function_name } },
};
//console.log(JSON.stringify(expert_function, null, 2));
const compl = await getCompletion(config, { prompt, ...toolargs });
getState().log(6, `llm_function_call completion: ${JSON.stringify(compl)}`);
const response = JSON.parse(compl.tool_calls[0].function.arguments);
//console.log("response: ", JSON.stringify(response, null, 2));
for (const col of columns) {
const target_table = Table.findOne({ name: col.target_table });
const fixed = eval_expression(
col.fixed_values,
row,
user,
"llm_function_call fixed values"
);
if (col.cardinality == "One") {
const row = {
...(response[noSpaces(target_table.name)] || {}),
...fixed,
};
await target_table.insertRow(row, user);
} else {
for (const resp of response[noSpaces(target_table.name)] || []) {
const row = { ...resp, ...fixed };
await target_table.insertRow(row, user);
}
}
}
return {};
},
});