@stoker-platform/system-functions 0.5.86 → 0.5.87
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/lib/chatFlow.js +2 -23
- package/lib/chatFlow.js.map +1 -1
- package/package.json +1 -1
package/lib/chatFlow.js
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
import { vertexAI } from "@genkit-ai/google-genai";
|
|
2
2
|
import { defineFirestoreRetriever, } from "@genkit-ai/firebase";
|
|
3
3
|
import { z } from "zod";
|
|
4
|
-
import { addRecord, getOne,
|
|
4
|
+
import { addRecord, getOne, getInputSchema, getStokerFirestore, updateRecord, initializeStoker, deserializeTimestampsWithoutUnderscores, tryPromise, } from "@stoker-platform/node-client";
|
|
5
5
|
import { HttpsError } from "firebase-functions/https";
|
|
6
6
|
import { join } from "node:path";
|
|
7
7
|
/* eslint-disable max-len */
|
|
@@ -109,27 +109,6 @@ export const chatFlow = (collection, schema, ai) => {
|
|
|
109
109
|
return "Error getting record";
|
|
110
110
|
}
|
|
111
111
|
});
|
|
112
|
-
const getSomeTool = ai.defineTool({
|
|
113
|
-
name: `get${labels.collection}`,
|
|
114
|
-
description: `Gets multiple records from the ${labels.collection} collection`,
|
|
115
|
-
inputSchema: z.object({}),
|
|
116
|
-
outputSchema: z.string(),
|
|
117
|
-
}, async (_input, { context }) => {
|
|
118
|
-
if (!context.auth)
|
|
119
|
-
throw new HttpsError("unauthenticated", `You are not authorized to get records from the ${labels.collection} collection.`);
|
|
120
|
-
const tenantId = context?.auth?.token?.tenant;
|
|
121
|
-
if (!tenantId) {
|
|
122
|
-
throw new HttpsError("unauthenticated", "Tenant ID not found in authentication token");
|
|
123
|
-
}
|
|
124
|
-
await initializeStoker("production", tenantId, join(process.cwd(), "lib", "system-custom", "main.js"), join(process.cwd(), "lib", "system-custom", "collections"), true);
|
|
125
|
-
try {
|
|
126
|
-
const records = await getSome([labels.collection], { userId: context.auth?.uid });
|
|
127
|
-
return JSON.stringify(records.records);
|
|
128
|
-
}
|
|
129
|
-
catch {
|
|
130
|
-
return "Error getting records";
|
|
131
|
-
}
|
|
132
|
-
});
|
|
133
112
|
return ai.defineFlow({
|
|
134
113
|
name: `${collection.labels.collection.toLowerCase()}_chat`,
|
|
135
114
|
inputSchema: z.object({
|
|
@@ -164,7 +143,7 @@ export const chatFlow = (collection, schema, ai) => {
|
|
|
164
143
|
query: input.messages.map((message) => message.content.map((part) => part.text).join("")).join("\n\n"),
|
|
165
144
|
}, {
|
|
166
145
|
docs,
|
|
167
|
-
tools: [addRecordTool, updateRecordTool, getOneTool
|
|
146
|
+
tools: [addRecordTool, updateRecordTool, getOneTool],
|
|
168
147
|
});
|
|
169
148
|
for await (const chunk of stream) {
|
|
170
149
|
sendChunk(chunk.text);
|
package/lib/chatFlow.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"chatFlow.js","sourceRoot":"","sources":["../src/chatFlow.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,QAAQ,EAAC,MAAM,yBAAyB,CAAC;AAEjD,OAAO,EACH,wBAAwB,GAC3B,MAAM,qBAAqB,CAAC;AAE7B,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AACtB,OAAO,EACH,SAAS,EACT,MAAM,EACN,
|
|
1
|
+
{"version":3,"file":"chatFlow.js","sourceRoot":"","sources":["../src/chatFlow.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,QAAQ,EAAC,MAAM,yBAAyB,CAAC;AAEjD,OAAO,EACH,wBAAwB,GAC3B,MAAM,qBAAqB,CAAC;AAE7B,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AACtB,OAAO,EACH,SAAS,EACT,MAAM,EACN,cAAc,EACd,kBAAkB,EAClB,YAAY,EACZ,gBAAgB,EAChB,uCAAuC,EACvC,UAAU,GACb,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAAC,UAAU,EAAC,MAAM,0BAA0B,CAAC;AAEpD,OAAO,EAAC,IAAI,EAAC,MAAM,WAAW,CAAC;AAE/B,4BAA4B;AAE5B,MAAM,QAAQ,GAAG,QAAQ,CAAC,QAAQ,CAAC,oBAAoB,CAAC,CAAC;AAEzD,MAAM,CAAC,MAAM,cAAc,GAAG,CAC1B,IAAqB,EACrB,UAA4B,EAC9B,EAAE;IACA,MAAM,IAAI,GAAG,IAAI,EAAE,GAAG,CAAC;IACvB,MAAM,KAAK,GAAG,IAAI,EAAE,KAAK,CAAC;IAC1B,IAAI,CAAC,CAAC,KAAK,EAAE,MAAM,IAAI,KAAK,EAAE,UAAU,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,EAAE,GAAG,CAAC,EAAE,CAAC;QACrE,OAAO,KAAK,CAAC;IACjB,CAAC;IACD,IAAI,CAAC,IAAI,EAAE,CAAC;QACR,OAAO,KAAK,CAAC;IACjB,CAAC;IACD,IAAI,CAAC,UAAU,CAAC,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;QACnD,OAAO,KAAK,CAAC;IACjB,CAAC;IACD,OAAO,IAAI,CAAC;AAChB,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,QAAQ,GAAG,CACpB,UAA4B,EAC5B,MAAyB,EACzB,EAAU,EACZ,EAAE;IACA,MAAM,EAAC,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,QAAQ,EAAC,GAAG,UAAU,CAAC;IACtD,IAAI,CAAC,QAAQ,EAAE,SAAS,IAAI,CAAC,QAAQ,EAAE,IAAI;QAAE,MAAM,IAAI,UAAU,CAAC,kBAAkB,EAAE,yDAAyD,CAAC,CAAC;IACjJ,MAAM,EAAC,IAAI,EAAC,GAAG,QAAQ,CAAC;IAExB,MAAM,SAAS,GAAG,kBAAkB,EAAE,CAAC;IAEvC,MAAM,SAAS,GAAG,cAAc,CAAC,UAAU,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC;IAEtE,MAAM,MAAM,GAAG,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IAEjC,MAAM,SAAS,GAAG,wBAAwB,CAAC,EAAE,EAAE;QAC3C,IAAI,EAAE,GAAG,MAAM,CAAC,UAAU,WAAW;QACrC,SAAS;QACT,UAAU,EAAE,EAAE;QACd,YAAY,EAAE,OAAO;QACrB,WAAW,EAAE,QAAQ;QACrB,QAAQ;QACR,eAAe,EAAE,QAAQ;KAC5B,CAAC,CAAC;IAEH,MAAM,aAAa,GAAG,EAAE,CAAC,UAAU,CAC/B;QACI,IAAI,EAAE,MAAM,MAAM,CAAC,UAAU,EAAE;QAC/B,WAAW,EAAE,wBAAwB,MAAM,CAAC,UAAU,aAAa;QACnE,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;YAClB,MAAM,EAAE,SAAS,CAAC,QAAQ,CAAC,4BAA4B,MAAM,CAAC,UAAU,aAAa,CAAC;SACzF,CAAC;QACF,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE;KAC3B,EACD,KAAK,EAAE,KAAK,EAAE,EAAC,OAAO,EAAC,EAAE,EAAE;QACvB,IAAI,CAAC,OAAO,CAAC,IAAI;YAAE,MAAM,IAAI,UAAU,CAAC,iBAAiB,EAAE,gDAAgD,MAAM,CAAC,UAAU,cAAc,CAAC,CAAC;QAC5I,MAAM,QAAQ,GAAI,OAAO,EAAE,IAAI,EAAE,KAA2B,EAAE,MAAM,CAAC;QACrE,IAAI,CAAC,QAAQ,EAAE,CAAC;YACZ,MAAM,IAAI,UAAU,CAAC,iBAAiB,EAAE,6CAA6C,CAAC,CAAC;QAC3F,CAAC;QACD,MAAM,gBAAgB,CAClB,YAAY,EACZ,QAAQ,EACR,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,eAAe,EAAE,SAAS,CAAC,EACtD,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,eAAe,EAAE,aAAa,CAAC,EAC1D,IAAI,CACP,CAAC;QACF,IAAI,CAAC;YACD,uCAAuC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YACtD,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,KAAK,CAAC,MAAM,EAAE,EAAC,MAAM,EAAE,OAAO,CAAC,IAAI,EAAE,GAAG,EAAC,CAAC,CAAC;YAC/F,OAAO,UAAU,MAAM,CAAC,EAAE,kCAAkC,MAAM,CAAC,UAAU,cAAc,CAAC;QAChG,CAAC;QAAC,MAAM,CAAC;YACL,OAAO,qBAAqB,CAAC;QACjC,CAAC;IACL,CAAC,CACJ,CAAC;IAEF,MAAM,gBAAgB,GAAG,EAAE,CAAC,UAAU,CAClC;QACI,IAAI,EAAE,SAAS,MAAM,CAAC,UAAU,EAAE;QAClC,WAAW,EAAE,2BAA2B,MAAM,CAAC,UAAU,aAAa;QACtE,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;YAClB,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,yCAAyC,MAAM,CAAC,UAAU,aAAa,CAAC;YACtG,MAAM,EAAE,SAAS,CAAC,QAAQ,CAAC,8BAA8B,MAAM,CAAC,MAAM,SAAS,CAAC;SACnF,CAAC;QACF,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE;KAC3B,EACD,KAAK,EAAE,KAAK,EAAE,EAAC,OAAO,EAAC,EAAE,EAAE;QACvB,IAAI,CAAC,OAAO,CAAC,IAAI;YAAE,MAAM,IAAI,UAAU,CAAC,iBAAiB,EAAE,mDAAmD,MAAM,CAAC,UAAU,cAAc,CAAC,CAAC;QAC/I,MAAM,QAAQ,GAAI,OAAO,EAAE,IAAI,EAAE,KAA2B,EAAE,MAAM,CAAC;QACrE,IAAI,CAAC,QAAQ,EAAE,CAAC;YACZ,MAAM,IAAI,UAAU,CAAC,iBAAiB,EAAE,6CAA6C,CAAC,CAAC;QAC3F,CAAC;QACD,MAAM,gBAAgB,CAClB,YAAY,EACZ,QAAQ,EACR,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,eAAe,EAAE,SAAS,CAAC,EACtD,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,eAAe,EAAE,aAAa,CAAC,EAC1D,IAAI,CACP,CAAC;QACF,IAAI,CAAC;YACD,uCAAuC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YACtD,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,MAAM,EAAE,EAAC,MAAM,EAAE,OAAO,CAAC,IAAI,EAAE,GAAG,EAAC,CAAC,CAAC;YAClH,OAAO,UAAU,MAAM,CAAC,EAAE,oCAAoC,MAAM,CAAC,UAAU,cAAc,CAAC;QAClG,CAAC;QAAC,MAAM,CAAC;YACL,OAAO,uBAAuB,CAAC;QACnC,CAAC;IACL,CAAC,CACJ,CAAC;IAEF,MAAM,UAAU,GAAG,EAAE,CAAC,UAAU,CAC5B;QACI,IAAI,EAAE,MAAM,MAAM,CAAC,MAAM,EAAE;QAC3B,WAAW,EAAE,iCAAiC,MAAM,CAAC,UAAU,aAAa;QAC5E,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;YAClB,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,wCAAwC,MAAM,CAAC,UAAU,aAAa,CAAC;SACxG,CAAC;QACF,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,0CAA0C,MAAM,CAAC,UAAU,aAAa,CAAC;KAC9G,EACD,KAAK,EAAE,KAAK,EAAE,EAAC,OAAO,EAAC,EAAE,EAAE;QACvB,IAAI,CAAC,OAAO,CAAC,IAAI;YAAE,MAAM,IAAI,UAAU,CAAC,iBAAiB,EAAE,kDAAkD,MAAM,CAAC,UAAU,cAAc,CAAC,CAAC;QAC9I,MAAM,QAAQ,GAAI,OAAO,EAAE,IAAI,EAAE,KAA2B,EAAE,MAAM,CAAC;QACrE,IAAI,CAAC,QAAQ,EAAE,CAAC;YACZ,MAAM,IAAI,UAAU,CAAC,iBAAiB,EAAE,6CAA6C,CAAC,CAAC;QAC3F,CAAC;QACD,MAAM,gBAAgB,CAClB,YAAY,EACZ,QAAQ,EACR,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,eAAe,EAAE,SAAS,CAAC,EACtD,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,eAAe,EAAE,aAAa,CAAC,EAC1D,IAAI,CACP,CAAC;QACF,IAAI,CAAC;YACD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,KAAK,CAAC,QAAQ,EAAE,EAAC,MAAM,EAAE,OAAO,CAAC,IAAI,EAAE,GAAG,EAAC,CAAC,CAAC;YAC9F,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAClC,CAAC;QAAC,MAAM,CAAC;YACL,OAAO,sBAAsB,CAAC;QAClC,CAAC;IACL,CAAC,CACJ,CAAC;IAEF,OAAO,EAAE,CAAC,UAAU,CAChB;QACI,IAAI,EAAE,GAAG,UAAU,CAAC,MAAM,CAAC,UAAU,CAAC,WAAW,EAAE,OAAO;QAC1D,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;YAClB,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC;gBACvB,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC;oBACtB,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC;oBACvB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;iBACnB,CAAC,CAAC;aACN,CAAC,CAAC;SACN,CAAC;KACL,EACD,KAAK,EAAE,KAA8D,EAAE,EAAC,SAAS,EAAE,OAAO,EAAC,EAAE,EAAE;QAC3F,MAAM,QAAQ,GAAI,OAAO,EAAE,IAAI,EAAE,KAA2B,EAAE,MAAM,CAAC;QACrE,IAAI,CAAC,QAAQ,EAAE,CAAC;YACZ,MAAM,IAAI,UAAU,CAAC,iBAAiB,EAAE,6CAA6C,CAAC,CAAC;QAC3F,CAAC;QACD,MAAM,KAAK,GAAG,UAAU,CAAC,CAAC;YACtB,EAAC,CAAC,UAAU,CAAC,aAAa,CAAC,EAAE,KAAK,EAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAEpD,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC;YAC3B,SAAS;YACT,KAAK,EAAE,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC;YACtG,OAAO,EAAE;gBACL,KAAK,EAAE,IAAI,CAAC,iBAAiB,IAAI,EAAE;gBACnC,KAAK;gBACL,UAAU,EAAE,WAAW,QAAQ,sBAAsB,MAAM,CAAC,UAAU,EAAE;aAC3E;SACJ,CAAC,CAAC;QAEH,MAAM,EAAC,oBAAoB,EAAC,GAAG,MAAM,gBAAgB,CACjD,YAAY,EACZ,QAAQ,EACR,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,eAAe,EAAE,SAAS,CAAC,EACtD,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,eAAe,EAAE,aAAa,CAAC,EAC1D,IAAI,CACP,CAAC;QAEF,MAAM,aAAa,GAAG,MAAM,oBAAoB,CAAC,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;QAC5E,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,aAAa,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAE7D,MAAM,EAAC,MAAM,EAAE,QAAQ,EAAC,GAAG,MAAM,CAAC,MAAM,CAAC;YACrC,UAAU,EAAE,MAAM,EAAE,UAAU,IAAI,MAAM,CAAC,UAAU;YACnD,KAAK,EAAE,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC;SACzG,EAAE;YACC,IAAI;YACJ,KAAK,EAAE,CAAC,aAAa,EAAE,gBAAgB,EAAE,UAAU,CAAC;SACvD,CAAC,CAAC;QACH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC/B,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC1B,CAAC;QACD,MAAM,aAAa,GAAG,MAAM,QAAQ,CAAC;QACrC,OAAO,aAAa,CAAC,IAAI,CAAC;IAC9B,CAAC,CACJ,CAAC;AACN,CAAC,CAAC"}
|