@sqlrooms/vega 0.26.0-rc.2 → 0.26.0-rc.4
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/dist/VegaChartTool.d.ts
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
import { z } from 'zod';
|
|
2
|
-
import {
|
|
2
|
+
import { OpenAssistantTool } from '@openassistant/utils';
|
|
3
3
|
/**
|
|
4
4
|
* Zod schema for the VegaChart tool parameters
|
|
5
5
|
*/
|
|
@@ -7,15 +7,7 @@ export declare const VegaChartToolParameters: z.ZodObject<{
|
|
|
7
7
|
sqlQuery: z.ZodString;
|
|
8
8
|
vegaLiteSpec: z.ZodString;
|
|
9
9
|
reasoning: z.ZodString;
|
|
10
|
-
},
|
|
11
|
-
sqlQuery: string;
|
|
12
|
-
vegaLiteSpec: string;
|
|
13
|
-
reasoning: string;
|
|
14
|
-
}, {
|
|
15
|
-
sqlQuery: string;
|
|
16
|
-
vegaLiteSpec: string;
|
|
17
|
-
reasoning: string;
|
|
18
|
-
}>;
|
|
10
|
+
}, z.core.$strip>;
|
|
19
11
|
export type VegaChartToolParameters = z.infer<typeof VegaChartToolParameters>;
|
|
20
12
|
export type VegaChartToolArgs = z.ZodObject<{
|
|
21
13
|
sqlQuery: z.ZodString;
|
|
@@ -43,5 +35,5 @@ export declare const DEFAULT_VEGA_CHART_DESCRIPTION = "A tool for creating VegaL
|
|
|
43
35
|
*/
|
|
44
36
|
export declare function createVegaChartTool({ description, }?: {
|
|
45
37
|
description?: string;
|
|
46
|
-
}):
|
|
38
|
+
}): OpenAssistantTool<typeof VegaChartToolParameters, VegaChartToolLlmResult, VegaChartToolAdditionalData, VegaChartToolContext>;
|
|
47
39
|
//# sourceMappingURL=VegaChartTool.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"VegaChartTool.d.ts","sourceRoot":"","sources":["../src/VegaChartTool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AAEtB,OAAO,EAAC,
|
|
1
|
+
{"version":3,"file":"VegaChartTool.d.ts","sourceRoot":"","sources":["../src/VegaChartTool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AAEtB,OAAO,EAAC,iBAAiB,EAAC,MAAM,sBAAsB,CAAC;AAEvD;;GAEG;AACH,eAAO,MAAM,uBAAuB;;;;iBAIlC,CAAC;AAEH,MAAM,MAAM,uBAAuB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,uBAAuB,CAAC,CAAC;AAE9E,MAAM,MAAM,iBAAiB,GAAG,CAAC,CAAC,SAAS,CAAC;IAC1C,QAAQ,EAAE,CAAC,CAAC,SAAS,CAAC;IACtB,YAAY,EAAE,CAAC,CAAC,SAAS,CAAC;IAC1B,SAAS,EAAE,CAAC,CAAC,SAAS,CAAC;CACxB,CAAC,CAAC;AAEH,MAAM,MAAM,sBAAsB,GAAG;IACnC,OAAO,EAAE,OAAO,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;CACjB,CAAC;AAEF,MAAM,MAAM,2BAA2B,GAAG;IACxC,QAAQ,EAAE,MAAM,CAAC;IACjB,YAAY,EAAE,MAAM,CAAC;CACtB,CAAC;AAEF,MAAM,MAAM,oBAAoB,GAAG,OAAO,CAAC;AAE3C;;GAEG;AACH,eAAO,MAAM,8BAA8B,mNAGC,CAAC;AAE7C;;;;;GAKG;AACH,wBAAgB,mBAAmB,CAAC,EAClC,WAA4C,GAC7C,GAAE;IACD,WAAW,CAAC,EAAE,MAAM,CAAC;CACjB,GAAG,iBAAiB,CACxB,OAAO,uBAAuB,EAC9B,sBAAsB,EACtB,2BAA2B,EAC3B,oBAAoB,CACrB,CAgCA"}
|
package/dist/VegaChartTool.js
CHANGED
|
@@ -1,6 +1,5 @@
|
|
|
1
1
|
import { z } from 'zod';
|
|
2
2
|
import { VegaChartToolResult } from './VegaChartToolResult';
|
|
3
|
-
import { extendedTool } from '@openassistant/utils';
|
|
4
3
|
/**
|
|
5
4
|
* Zod schema for the VegaChart tool parameters
|
|
6
5
|
*/
|
|
@@ -23,10 +22,12 @@ In the response:
|
|
|
23
22
|
* @returns A tool that can be used with the AI assistant
|
|
24
23
|
*/
|
|
25
24
|
export function createVegaChartTool({ description = DEFAULT_VEGA_CHART_DESCRIPTION, } = {}) {
|
|
26
|
-
return
|
|
25
|
+
return {
|
|
26
|
+
name: 'chart',
|
|
27
27
|
description,
|
|
28
28
|
parameters: VegaChartToolParameters,
|
|
29
|
-
execute: async (
|
|
29
|
+
execute: async (params) => {
|
|
30
|
+
const { sqlQuery, vegaLiteSpec } = params;
|
|
30
31
|
try {
|
|
31
32
|
const parsedVegaLiteSpec = JSON.parse(vegaLiteSpec);
|
|
32
33
|
// data object of the vegaLiteSpec and sqlQuery
|
|
@@ -52,6 +53,6 @@ export function createVegaChartTool({ description = DEFAULT_VEGA_CHART_DESCRIPTI
|
|
|
52
53
|
}
|
|
53
54
|
},
|
|
54
55
|
component: VegaChartToolResult,
|
|
55
|
-
}
|
|
56
|
+
};
|
|
56
57
|
}
|
|
57
58
|
//# sourceMappingURL=VegaChartTool.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"VegaChartTool.js","sourceRoot":"","sources":["../src/VegaChartTool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AACtB,OAAO,EAAC,mBAAmB,EAAC,MAAM,uBAAuB,CAAC;
|
|
1
|
+
{"version":3,"file":"VegaChartTool.js","sourceRoot":"","sources":["../src/VegaChartTool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AACtB,OAAO,EAAC,mBAAmB,EAAC,MAAM,uBAAuB,CAAC;AAG1D;;GAEG;AACH,MAAM,CAAC,MAAM,uBAAuB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC9C,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;IACpB,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE;IACxB,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;CACtB,CAAC,CAAC;AAsBH;;GAEG;AACH,MAAM,CAAC,MAAM,8BAA8B,GAAG;;;4CAGF,CAAC;AAE7C;;;;;GAKG;AACH,MAAM,UAAU,mBAAmB,CAAC,EAClC,WAAW,GAAG,8BAA8B,MAG1C,EAAE;IAMJ,OAAO;QACL,IAAI,EAAE,OAAO;QACb,WAAW;QACX,UAAU,EAAE,uBAAuB;QACnC,OAAO,EAAE,KAAK,EAAE,MAA+B,EAAE,EAAE;YACjD,MAAM,EAAC,QAAQ,EAAE,YAAY,EAAC,GAAG,MAAM,CAAC;YACxC,IAAI,CAAC;gBACH,MAAM,kBAAkB,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;gBACpD,+CAA+C;gBAC/C,+HAA+H;gBAC/H,OAAO;oBACL,SAAS,EAAE;wBACT,OAAO,EAAE,IAAI;wBACb,OAAO,EAAE,6BAA6B;qBACvC;oBACD,cAAc,EAAE;wBACd,QAAQ;wBACR,YAAY,EAAE,kBAAkB;qBACjC;iBACF,CAAC;YACJ,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO;oBACL,SAAS,EAAE;wBACT,OAAO,EAAE,KAAK;wBACd,OAAO,EAAE,4BAA4B,KAAK,EAAE;qBAC7C;iBACF,CAAC;YACJ,CAAC;QACH,CAAC;QACD,SAAS,EAAE,mBAAmB;KAC/B,CAAC;AACJ,CAAC","sourcesContent":["import {z} from 'zod';\nimport {VegaChartToolResult} from './VegaChartToolResult';\nimport {OpenAssistantTool} from '@openassistant/utils';\n\n/**\n * Zod schema for the VegaChart tool parameters\n */\nexport const VegaChartToolParameters = z.object({\n sqlQuery: z.string(),\n vegaLiteSpec: z.string(),\n reasoning: z.string(),\n});\n\nexport type VegaChartToolParameters = z.infer<typeof VegaChartToolParameters>;\n\nexport type VegaChartToolArgs = z.ZodObject<{\n sqlQuery: z.ZodString;\n vegaLiteSpec: z.ZodString;\n reasoning: z.ZodString;\n}>;\n\nexport type VegaChartToolLlmResult = {\n success: boolean;\n details: string;\n};\n\nexport type VegaChartToolAdditionalData = {\n sqlQuery: string;\n vegaLiteSpec: object;\n};\n\nexport type VegaChartToolContext = unknown;\n\n/**\n * Default description for the VegaChart tool\n */\nexport const DEFAULT_VEGA_CHART_DESCRIPTION = `A tool for creating VegaLite charts based on the schema of the SQL query result from the \"query\" tool.\nIn the response:\n- omit the data from the vegaLiteSpec\n- provide an sql query in sqlQuery instead.`;\n\n/**\n * Creates a VegaLite chart visualization tool for AI assistants\n * @param options - Configuration options for the VegaChart tool\n * @param options.description - Custom description for the tool (defaults to a standard description)\n * @returns A tool that can be used with the AI assistant\n */\nexport function createVegaChartTool({\n description = DEFAULT_VEGA_CHART_DESCRIPTION,\n}: {\n description?: string;\n} = {}): OpenAssistantTool<\n typeof VegaChartToolParameters,\n VegaChartToolLlmResult,\n VegaChartToolAdditionalData,\n VegaChartToolContext\n> {\n return {\n name: 'chart',\n description,\n parameters: VegaChartToolParameters,\n execute: async (params: VegaChartToolParameters) => {\n const {sqlQuery, vegaLiteSpec} = params;\n try {\n const parsedVegaLiteSpec = JSON.parse(vegaLiteSpec);\n // data object of the vegaLiteSpec and sqlQuery\n // it is not used yet, but we can use it to create a JSON editor for user to edit the vegaLiteSpec so that chart can be updated\n return {\n llmResult: {\n success: true,\n details: 'Chart created successfully.',\n },\n additionalData: {\n sqlQuery,\n vegaLiteSpec: parsedVegaLiteSpec,\n },\n };\n } catch (error) {\n return {\n llmResult: {\n success: false,\n details: `Not a valid JSON object: ${error}`,\n },\n };\n }\n },\n component: VegaChartToolResult,\n };\n}\n"]}
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@sqlrooms/vega",
|
|
3
|
-
"version": "0.26.0-rc.
|
|
3
|
+
"version": "0.26.0-rc.4",
|
|
4
4
|
"main": "dist/index.js",
|
|
5
5
|
"types": "dist/index.d.ts",
|
|
6
6
|
"module": "dist/index.js",
|
|
@@ -18,16 +18,16 @@
|
|
|
18
18
|
"access": "public"
|
|
19
19
|
},
|
|
20
20
|
"dependencies": {
|
|
21
|
-
"@openassistant/utils": "
|
|
22
|
-
"@sqlrooms/ai": "0.26.0-rc.
|
|
23
|
-
"@sqlrooms/duckdb": "0.26.0-rc.
|
|
24
|
-
"@sqlrooms/ui": "0.26.0-rc.
|
|
25
|
-
"@sqlrooms/utils": "0.26.0-rc.
|
|
21
|
+
"@openassistant/utils": "1.0.0-alpha.0",
|
|
22
|
+
"@sqlrooms/ai": "0.26.0-rc.4",
|
|
23
|
+
"@sqlrooms/duckdb": "0.26.0-rc.4",
|
|
24
|
+
"@sqlrooms/ui": "0.26.0-rc.4",
|
|
25
|
+
"@sqlrooms/utils": "0.26.0-rc.4",
|
|
26
26
|
"lucide-react": "^0.475.0",
|
|
27
27
|
"react-vega": "^7.7.1",
|
|
28
28
|
"vega": "^6.2.0",
|
|
29
29
|
"vega-lite": "^6.4.0",
|
|
30
|
-
"zod": "^
|
|
30
|
+
"zod": "^4.1.8"
|
|
31
31
|
},
|
|
32
32
|
"peerDependencies": {
|
|
33
33
|
"apache-arrow": "17.0.0",
|
|
@@ -41,5 +41,5 @@
|
|
|
41
41
|
"typecheck": "tsc --noEmit",
|
|
42
42
|
"typedoc": "typedoc"
|
|
43
43
|
},
|
|
44
|
-
"gitHead": "
|
|
44
|
+
"gitHead": "9e23926085d8852178792d6d98d8091e9a5f71d5"
|
|
45
45
|
}
|