@sqlrooms/ai 0.1.0

package/dist/schemas.d.ts

@@ -0,0 +1,354 @@
+import { z } from 'zod';
+export declare const QueryToolParameters: z.ZodObject<{
+    type: z.ZodLiteral<"query">;
+    sqlQuery: z.ZodString;
+    reasoning: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    type: "query";
+    sqlQuery: string;
+    reasoning: string;
+}, {
+    type: "query";
+    sqlQuery: string;
+    reasoning: string;
+}>;
+export type QueryToolParameters = z.infer<typeof QueryToolParameters>;
+export declare const AnswerToolParameters: z.ZodObject<{
+    type: z.ZodLiteral<"answer">;
+    answer: z.ZodString;
+    chart: z.ZodUnion<[z.ZodObject<{
+        sqlQuery: z.ZodString;
+        vegaLiteSpec: z.ZodString;
+    }, "strip", z.ZodTypeAny, {
+        sqlQuery: string;
+        vegaLiteSpec: string;
+    }, {
+        sqlQuery: string;
+        vegaLiteSpec: string;
+    }>, z.ZodNull]>;
+}, "strip", z.ZodTypeAny, {
+    type: "answer";
+    answer: string;
+    chart: {
+        sqlQuery: string;
+        vegaLiteSpec: string;
+    } | null;
+}, {
+    type: "answer";
+    answer: string;
+    chart: {
+        sqlQuery: string;
+        vegaLiteSpec: string;
+    } | null;
+}>;
+export type AnswerToolParameters = z.infer<typeof AnswerToolParameters>;
+export declare const ToolCallSchema: z.ZodObject<{
+    toolName: z.ZodString;
+    toolCallId: z.ZodString;
+    args: z.ZodUnion<[z.ZodObject<{
+        type: z.ZodLiteral<"query">;
+        sqlQuery: z.ZodString;
+        reasoning: z.ZodString;
+    }, "strip", z.ZodTypeAny, {
+        type: "query";
+        sqlQuery: string;
+        reasoning: string;
+    }, {
+        type: "query";
+        sqlQuery: string;
+        reasoning: string;
+    }>, z.ZodObject<{
+        type: z.ZodLiteral<"answer">;
+        answer: z.ZodString;
+        chart: z.ZodUnion<[z.ZodObject<{
+            sqlQuery: z.ZodString;
+            vegaLiteSpec: z.ZodString;
+        }, "strip", z.ZodTypeAny, {
+            sqlQuery: string;
+            vegaLiteSpec: string;
+        }, {
+            sqlQuery: string;
+            vegaLiteSpec: string;
+        }>, z.ZodNull]>;
+    }, "strip", z.ZodTypeAny, {
+        type: "answer";
+        answer: string;
+        chart: {
+            sqlQuery: string;
+            vegaLiteSpec: string;
+        } | null;
+    }, {
+        type: "answer";
+        answer: string;
+        chart: {
+            sqlQuery: string;
+            vegaLiteSpec: string;
+        } | null;
+    }>]>;
+}, "strip", z.ZodTypeAny, {
+    toolName: string;
+    toolCallId: string;
+    args: {
+        type: "query";
+        sqlQuery: string;
+        reasoning: string;
+    } | {
+        type: "answer";
+        answer: string;
+        chart: {
+            sqlQuery: string;
+            vegaLiteSpec: string;
+        } | null;
+    };
+}, {
+    toolName: string;
+    toolCallId: string;
+    args: {
+        type: "query";
+        sqlQuery: string;
+        reasoning: string;
+    } | {
+        type: "answer";
+        answer: string;
+        chart: {
+            sqlQuery: string;
+            vegaLiteSpec: string;
+        } | null;
+    };
+}>;
+export type ToolCallSchema = z.infer<typeof ToolCallSchema>;
+export declare const ToolResultSchema: z.ZodObject<{
+    toolName: z.ZodString;
+    toolCallId: z.ZodString;
+    args: z.ZodRecord<z.ZodString, z.ZodAny>;
+    result: z.ZodUnion<[z.ZodObject<{
+        success: z.ZodLiteral<false>;
+        error: z.ZodString;
+    }, "strip", z.ZodTypeAny, {
+        success: false;
+        error: string;
+    }, {
+        success: false;
+        error: string;
+    }>, z.ZodObject<{
+        success: z.ZodLiteral<true>;
+        data: z.ZodRecord<z.ZodString, z.ZodAny>;
+    }, "strip", z.ZodTypeAny, {
+        success: true;
+        data: Record<string, any>;
+    }, {
+        success: true;
+        data: Record<string, any>;
+    }>]>;
+}, "strip", z.ZodTypeAny, {
+    toolName: string;
+    toolCallId: string;
+    args: Record<string, any>;
+    result: {
+        success: false;
+        error: string;
+    } | {
+        success: true;
+        data: Record<string, any>;
+    };
+}, {
+    toolName: string;
+    toolCallId: string;
+    args: Record<string, any>;
+    result: {
+        success: false;
+        error: string;
+    } | {
+        success: true;
+        data: Record<string, any>;
+    };
+}>;
+export type ToolResultSchema = z.infer<typeof ToolResultSchema>;
+export declare const AnalysisResultSchema: z.ZodObject<{
+    id: z.ZodString;
+    prompt: z.ZodString;
+    toolResults: z.ZodArray<z.ZodObject<{
+        toolName: z.ZodString;
+        toolCallId: z.ZodString;
+        args: z.ZodRecord<z.ZodString, z.ZodAny>;
+        result: z.ZodUnion<[z.ZodObject<{
+            success: z.ZodLiteral<false>;
+            error: z.ZodString;
+        }, "strip", z.ZodTypeAny, {
+            success: false;
+            error: string;
+        }, {
+            success: false;
+            error: string;
+        }>, z.ZodObject<{
+            success: z.ZodLiteral<true>;
+            data: z.ZodRecord<z.ZodString, z.ZodAny>;
+        }, "strip", z.ZodTypeAny, {
+            success: true;
+            data: Record<string, any>;
+        }, {
+            success: true;
+            data: Record<string, any>;
+        }>]>;
+    }, "strip", z.ZodTypeAny, {
+        toolName: string;
+        toolCallId: string;
+        args: Record<string, any>;
+        result: {
+            success: false;
+            error: string;
+        } | {
+            success: true;
+            data: Record<string, any>;
+        };
+    }, {
+        toolName: string;
+        toolCallId: string;
+        args: Record<string, any>;
+        result: {
+            success: false;
+            error: string;
+        } | {
+            success: true;
+            data: Record<string, any>;
+        };
+    }>, "many">;
+    toolCalls: z.ZodArray<z.ZodObject<{
+        toolName: z.ZodString;
+        toolCallId: z.ZodString;
+        args: z.ZodUnion<[z.ZodObject<{
+            type: z.ZodLiteral<"query">;
+            sqlQuery: z.ZodString;
+            reasoning: z.ZodString;
+        }, "strip", z.ZodTypeAny, {
+            type: "query";
+            sqlQuery: string;
+            reasoning: string;
+        }, {
+            type: "query";
+            sqlQuery: string;
+            reasoning: string;
+        }>, z.ZodObject<{
+            type: z.ZodLiteral<"answer">;
+            answer: z.ZodString;
+            chart: z.ZodUnion<[z.ZodObject<{
+                sqlQuery: z.ZodString;
+                vegaLiteSpec: z.ZodString;
+            }, "strip", z.ZodTypeAny, {
+                sqlQuery: string;
+                vegaLiteSpec: string;
+            }, {
+                sqlQuery: string;
+                vegaLiteSpec: string;
+            }>, z.ZodNull]>;
+        }, "strip", z.ZodTypeAny, {
+            type: "answer";
+            answer: string;
+            chart: {
+                sqlQuery: string;
+                vegaLiteSpec: string;
+            } | null;
+        }, {
+            type: "answer";
+            answer: string;
+            chart: {
+                sqlQuery: string;
+                vegaLiteSpec: string;
+            } | null;
+        }>]>;
+    }, "strip", z.ZodTypeAny, {
+        toolName: string;
+        toolCallId: string;
+        args: {
+            type: "query";
+            sqlQuery: string;
+            reasoning: string;
+        } | {
+            type: "answer";
+            answer: string;
+            chart: {
+                sqlQuery: string;
+                vegaLiteSpec: string;
+            } | null;
+        };
+    }, {
+        toolName: string;
+        toolCallId: string;
+        args: {
+            type: "query";
+            sqlQuery: string;
+            reasoning: string;
+        } | {
+            type: "answer";
+            answer: string;
+            chart: {
+                sqlQuery: string;
+                vegaLiteSpec: string;
+            } | null;
+        };
+    }>, "many">;
+}, "strip", z.ZodTypeAny, {
+    id: string;
+    prompt: string;
+    toolResults: {
+        toolName: string;
+        toolCallId: string;
+        args: Record<string, any>;
+        result: {
+            success: false;
+            error: string;
+        } | {
+            success: true;
+            data: Record<string, any>;
+        };
+    }[];
+    toolCalls: {
+        toolName: string;
+        toolCallId: string;
+        args: {
+            type: "query";
+            sqlQuery: string;
+            reasoning: string;
+        } | {
+            type: "answer";
+            answer: string;
+            chart: {
+                sqlQuery: string;
+                vegaLiteSpec: string;
+            } | null;
+        };
+    }[];
+}, {
+    id: string;
+    prompt: string;
+    toolResults: {
+        toolName: string;
+        toolCallId: string;
+        args: Record<string, any>;
+        result: {
+            success: false;
+            error: string;
+        } | {
+            success: true;
+            data: Record<string, any>;
+        };
+    }[];
+    toolCalls: {
+        toolName: string;
+        toolCallId: string;
+        args: {
+            type: "query";
+            sqlQuery: string;
+            reasoning: string;
+        } | {
+            type: "answer";
+            answer: string;
+            chart: {
+                sqlQuery: string;
+                vegaLiteSpec: string;
+            } | null;
+        };
+    }[];
+}>;
+export type AnalysisResultSchema = z.infer<typeof AnalysisResultSchema>;
+//# sourceMappingURL=schemas.d.ts.map

package/dist/schemas.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"schemas.d.ts","sourceRoot":"","sources":["../src/schemas.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AAEtB,eAAO,MAAM,mBAAmB;;;;;;;;;;;;EAI9B,CAAC;AACH,MAAM,MAAM,mBAAmB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,mBAAmB,CAAC,CAAC;AAEtE,eAAO,MAAM,oBAAoB;;;;;;;;;;;;;;;;;;;;;;;;;;;EAU/B,CAAC;AACH,MAAM,MAAM,oBAAoB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,oBAAoB,CAAC,CAAC;AAExE,eAAO,MAAM,cAAc;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAIzB,CAAC;AACH,MAAM,MAAM,cAAc,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,cAAc,CAAC,CAAC;AAE5D,eAAO,MAAM,gBAAgB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAc3B,CAAC;AACH,MAAM,MAAM,gBAAgB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,gBAAgB,CAAC,CAAC;AAEhE,eAAO,MAAM,oBAAoB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAK/B,CAAC;AACH,MAAM,MAAM,oBAAoB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,oBAAoB,CAAC,CAAC"}

package/dist/schemas.js

@@ -0,0 +1,43 @@
+import { z } from 'zod';
+export const QueryToolParameters = z.object({
+    type: z.literal('query'),
+    sqlQuery: z.string(),
+    reasoning: z.string(),
+});
+export const AnswerToolParameters = z.object({
+    type: z.literal('answer'),
+    answer: z.string(),
+    chart: z.union([
+        z.object({
+            sqlQuery: z.string(),
+            vegaLiteSpec: z.string(),
+        }),
+        z.null(),
+    ]),
+});
+export const ToolCallSchema = z.object({
+    toolName: z.string(),
+    toolCallId: z.string(),
+    args: z.union([QueryToolParameters, AnswerToolParameters]),
+});
+export const ToolResultSchema = z.object({
+    toolName: z.string(),
+    toolCallId: z.string(),
+    args: z.record(z.any()),
+    result: z.union([
+        z.object({
+            success: z.literal(false),
+            error: z.string(),
+        }),
+        z.object({
+            success: z.literal(true),
+            data: z.record(z.any()),
+        }),
+    ]),
+});
+export const AnalysisResultSchema = z.object({
+    id: z.string().cuid2(),
+    prompt: z.string(),
+    toolResults: z.array(ToolResultSchema),
+    toolCalls: z.array(ToolCallSchema),
+});

package/package.json

@@ -0,0 +1,47 @@
+{
+  "name": "@sqlrooms/ai",
+  "version": "0.1.0",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "module": "dist/index.js",
+  "type": "module",
+  "private": false,
+  "author": "Ilya Boyandin <ilya@boyandin.me>",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/sqlrooms/sqlrooms.git"
+  },
+  "files": [
+    "dist"
+  ],
+  "publishConfig": {
+    "access": "public"
+  },
+  "dependencies": {
+    "@ai-sdk/provider": "^1.0.7",
+    "@paralleldrive/cuid2": "^2.2.2",
+    "@sqlrooms/duckdb": "0.1.0",
+    "@sqlrooms/project-builder": "0.1.0",
+    "@sqlrooms/project-config": "0.1.0",
+    "@sqlrooms/ui": "0.1.0",
+    "@sqlrooms/vega": "0.1.0",
+    "ai": "^4.1.28",
+    "immer": "^10.1.1",
+    "lucide-react": "^0.475.0",
+    "react-markdown": "^9.0.3",
+    "react-vega": "^7.6.0",
+    "zod": "^3.24.1"
+  },
+  "peerDependencies": {
+    "react": "*",
+    "react-dom": "*"
+  },
+  "scripts": {
+    "dev": "tsc -w",
+    "build": "tsc",
+    "lint": "eslint .",
+    "typedoc": "typedoc"
+  },
+  "gitHead": "2d44f6636dbd53d18c32a422351e93caa182ada6"
+}