@sqlrooms/ai 0.5.0 → 0.6.0

package/dist/AnalysisResult.js.map

@@ -1 +1 @@
-{"version":3,"file":"AnalysisResult.js","sourceRoot":"","sources":["../src/AnalysisResult.tsx"],"names":[],"mappings":";AACA,OAAO,EAAC,QAAQ,EAAC,MAAM,YAAY,CAAC;AACpC,OAAO,EAAC,UAAU,EAAC,MAAM,cAAc,CAAC;AACxC,OAAO,EAAC,kBAAkB,EAAE,QAAQ,EAAC,MAAM,cAAc,CAAC;AAC1D,OAAO,EAAC,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,cAAc,EAAC,MAAM,cAAc,CAAC;AAM7E,MAAM,CAAC,MAAM,cAAc,GAAkC,CAAC,EAAC,MAAM,EAAC,EAAE,EAAE;IACxE,OAAO,CACL,eAAK,SAAS,EAAC,gEAAgE,aAC7E,eAAK,SAAS,EAAC,8EAA8E,aAC3F,KAAC,kBAAkB,IAAC,SAAS,EAAC,SAAS,GAAG,EAC1C,cAAK,SAAS,EAAC,gBAAgB,YAAE,MAAM,CAAC,MAAM,GAAO,EACrD,MAAC,OAAO,eACN,KAAC,cAAc,IAAC,OAAO,kBACrB,KAAC,MAAM,IAAC,OAAO,EAAC,OAAO,EAAC,IAAI,EAAC,MAAM,EAAC,SAAS,EAAC,SAAS,YACrD,KAAC,QAAQ,IAAC,SAAS,EAAC,SAAS,GAAG,GACzB,GACM,EACjB,KAAC,cAAc,IACb,SAAS,EAAC,uCAAuC,EACjD,KAAK,EAAC,KAAK,EACX,IAAI,EAAC,OAAO,YAEZ,cAAK,SAAS,EAAC,SAAS,YAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,GAAO,GACjD,IACT,IACN,EACN,cAAK,SAAS,EAAC,0BAA0B,YACtC,MAAM,CAAC,WAAW;qBAChB,MAAM,CACL,CAAC,UAAU,EAAE,EAAE,CACb,UAAU,CAAC,QAAQ,KAAK,QAAQ,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,OAAO,CACjE;qBACA,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,CACnB,KAAC,UAAU,IAA6B,UAAU,EAAE,UAAU,IAA7C,UAAU,CAAC,UAAU,CAA4B,CACnE,CAAC,GACA,EACN,cAAK,SAAS,EAAC,0BAA0B,YACtC,MAAM,CAAC,SAAS;qBACd,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,QAAQ,KAAK,QAAQ,CAAC;qBACpD,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CACjB,KAAC,QAAQ,IAA2B,QAAQ,EAAE,QAAQ,IAAvC,QAAQ,CAAC,UAAU,CAAwB,CAC3D,CAAC,GACA,IACF,CACP,CAAC;AACJ,CAAC,CAAC"}
+{"version":3,"file":"AnalysisResult.js","sourceRoot":"","sources":["../src/AnalysisResult.tsx"],"names":[],"mappings":";AACA,OAAO,EAAC,kBAAkB,EAAE,QAAQ,EAAC,MAAM,cAAc,CAAC;AAC1D,OAAO,EAAC,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,cAAc,EAAC,MAAM,cAAc,CAAC;AAC7E,OAAO,EAAC,UAAU,EAAC,MAAM,cAAc,CAAC;AAUxC;;;;;;GAMG;AACH,MAAM,eAAe,GAAG,CAAC,MAA4B,EAAE,EAAE;IACvD,0CAA0C;IAC1C,6DAA6D;IAC7D,MAAM,EAAC,gBAAgB,EAAE,GAAG,IAAI,EAAC,GAAG,MAAM,CAAC;IAC3C,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;AAC9B,CAAC,CAAC;AAEF;;;;;GAKG;AACH,MAAM,iBAAiB,GAAG,CACxB,gBAA0D,EAC1D,UAAkB,EAClB,EAAE;IACF,OAAO,gBAAgB,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,UAAU,KAAK,UAAU,CAAC;QAC1E,EAAE,OAAO,CAAC;AACd,CAAC,CAAC;AAEF;;;;;;;;;GASG;AACH,MAAM,CAAC,MAAM,cAAc,GAAkC,CAAC,EAAC,MAAM,EAAC,EAAE,EAAE;IACxE,oFAAoF;IACpF,iDAAiD;IACjD,MAAM,mBAAmB,GAAG,MAAM,CAAC,WAAW,CAAC;IAE/C,OAAO,CACL,eAAK,SAAS,EAAC,gEAAgE,aAC7E,eAAK,SAAS,EAAC,8EAA8E,aAC3F,KAAC,kBAAkB,IAAC,SAAS,EAAC,SAAS,GAAG,EAC1C,cAAK,SAAS,EAAC,gBAAgB,YAAE,MAAM,CAAC,MAAM,GAAO,EACrD,MAAC,OAAO,eACN,KAAC,cAAc,IAAC,OAAO,kBACrB,KAAC,MAAM,IAAC,OAAO,EAAC,OAAO,EAAC,IAAI,EAAC,MAAM,EAAC,SAAS,EAAC,SAAS,YACrD,KAAC,QAAQ,IAAC,SAAS,EAAC,SAAS,GAAG,GACzB,GACM,EACjB,KAAC,cAAc,IACb,SAAS,EAAC,uCAAuC,EACjD,KAAK,EAAC,KAAK,EACX,IAAI,EAAC,OAAO,YAEZ,cAAK,SAAS,EAAC,SAAS,YAAE,eAAe,CAAC,MAAM,CAAC,GAAO,GACzC,IACT,IACN,EACL,mBAAmB,CAAC,MAAM,GAAG,CAAC,IAAI,CACjC,cAAK,SAAS,EAAC,0BAA0B,YACtC,mBAAmB,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,CACvC,KAAC,UAAU,IAET,UAAU,EAAE,UAAU,EACtB,aAAa,EAAE,iBAAiB,CAC9B,MAAM,CAAC,gBAAgB,EACvB,UAAU,CAAC,UAAU,CACtB,IALI,UAAU,CAAC,UAAU,CAM1B,CACH,CAAC,GACE,CACP,EACA,MAAM,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,IAAI,CAC7B,cAAK,SAAS,EAAC,0BAA0B,YACvC,KAAC,UAAU,IAET,UAAU,EAAE;wBACV,QAAQ,EAAE,QAAQ;wBAClB,UAAU,EAAE,MAAM,CAAC,EAAE;wBACrB,IAAI,EAAE,EAAE;wBACR,MAAM,EAAE,EAAC,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,EAAC,QAAQ,EAAE,MAAM,CAAC,QAAQ,EAAC,EAAC;qBAC3D,IANI,MAAM,CAAC,EAAE,GAAG,mBAAmB,CAOpC,GACE,CACP,IACG,CACP,CAAC;AACJ,CAAC,CAAC","sourcesContent":["import {AnalysisResultSchema} from './schemas';\nimport {SquareTerminalIcon, CodeIcon} from 'lucide-react';\nimport {Button, Popover, PopoverContent, PopoverTrigger} from '@sqlrooms/ui';\nimport {ToolResult} from './ToolResult';\n\n/**\n * Props for the AnalysisResult component\n * @property {AnalysisResultSchema} result - The result of the analysis containing prompt, tool calls, and analysis data\n */\ntype AnalysisResultProps = {\n  result: AnalysisResultSchema;\n};\n\n/**\n * Stringify the result of the analysis, excluding toolCallMessages.\n * Used to display raw result data in a code view.\n *\n * @param result - The complete analysis result\n * @returns A JSON string representation of the result without toolCallMessages\n */\nconst stringifyResult = (result: AnalysisResultSchema) => {\n  // remove toolCallMessages from the result\n  // eslint-disable-next-line @typescript-eslint/no-unused-vars\n  const {toolCallMessages, ...rest} = result;\n  return JSON.stringify(rest);\n};\n\n/**\n * Find a custom message element for a given tool call ID\n * @param toolCallMessages - Array of tool call messages\n * @param toolCallId - The ID of the tool call to find a message for\n * @returns The custom message element if found, undefined otherwise\n */\nconst findCustomMessage = (\n  toolCallMessages: AnalysisResultSchema['toolCallMessages'],\n  toolCallId: string,\n) => {\n  return toolCallMessages.find((message) => message.toolCallId === toolCallId)\n    ?.element;\n};\n\n/**\n * Component that displays the results of an AI analysis.\n * Shows the original prompt, intermediate tool calls, final analysis text,\n * and any tool results.\n *\n * @component\n * @param props - Component props\n * @param props.result - The analysis result data to display\n * @returns A React component displaying the analysis results\n */\nexport const AnalysisResult: React.FC<AnalysisResultProps> = ({result}) => {\n  // the toolResults are reasoning steps that the LLM took to achieve the final result\n  // by calling function tools to answer the prompt\n  const analysisToolResults = result.toolResults;\n\n  return (\n    <div className=\"flex flex-col w-full text-sm gap-5 border py-6 px-4 rounded-md\">\n      <div className=\"p-2 mb-2 rounded-md text-gray-700 dark:text-gray-100 flex items-center gap-2\">\n        <SquareTerminalIcon className=\"w-4 h-4\" />\n        <div className=\"text-sm flex-1\">{result.prompt}</div>\n        <Popover>\n          <PopoverTrigger asChild>\n            <Button variant=\"ghost\" size=\"icon\" className=\"w-6 h-6\">\n              <CodeIcon className=\"w-4 h-4\" />\n            </Button>\n          </PopoverTrigger>\n          <PopoverContent\n            className=\"w-[400px] max-h-[400px] overflow-auto\"\n            align=\"end\"\n            side=\"right\"\n          >\n            <pre className=\"text-xs\">{stringifyResult(result)}</pre>\n          </PopoverContent>\n        </Popover>\n      </div>\n      {analysisToolResults.length > 0 && (\n        <div className=\"flex flex-col gap-5 px-4\">\n          {analysisToolResults.map((toolResult) => (\n            <ToolResult\n              key={toolResult.toolCallId}\n              toolResult={toolResult}\n              customMessage={findCustomMessage(\n                result.toolCallMessages,\n                toolResult.toolCallId,\n              )}\n            />\n          ))}\n        </div>\n      )}\n      {result.analysis.length > 0 && (\n        <div className=\"flex flex-col gap-5 px-4\">\n          <ToolResult\n            key={result.id + '-streaming-result'}\n            toolResult={{\n              toolName: 'answer',\n              toolCallId: result.id,\n              args: {},\n              result: {success: true, data: {analysis: result.analysis}},\n            }}\n          />\n        </div>\n      )}\n    </div>\n  );\n};\n"]}

package/dist/AnalysisResultsContainer.js.map

@@ -1 +1 @@
-{"version":3,"file":"AnalysisResultsContainer.js","sourceRoot":"","sources":["../src/AnalysisResultsContainer.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;;AACb,OAAO,EAAC,EAAE,EAAE,YAAY,EAAC,MAAM,cAAc,CAAC;AAC9C,OAAO,EAAC,WAAW,EAAC,MAAM,cAAc,CAAC;AACzC,OAAO,EAAC,cAAc,EAAC,MAAM,WAAW,CAAC;AACzC,OAAO,EAAC,cAAc,EAAC,MAAM,kBAAkB,CAAC;AAChD,OAAO,EACL,iBAAiB,EACjB,uBAAuB,GACxB,MAAM,8BAA8B,CAAC;AAEtC,MAAM,CAAC,MAAM,wBAAwB,GAAa,GAAG,EAAE;IACrD,MAAM,iBAAiB,GAAG,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,iBAAiB,CAAC,CAAC;IACxE,MAAM,eAAe,GAAG,cAAc,CACpC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,eAAe,CAC3C,CAAC;IACF,MAAM,CAAC,oBAAoB,EAAE,cAAc,CAAC,GAC1C,iBAAiB,EAAkB,CAAC;IACtC,MAAM,EAAC,UAAU,EAAE,cAAc,EAAC,GAChC,uBAAuB,CAAC,oBAAoB,CAAC,CAAC;IAEhD,OAAO,CACL,eAAK,SAAS,EAAC,6CAA6C,aAC1D,eACE,SAAS,EAAC,yDAAyD,EACnE,GAAG,EAAE,oBAAoB,aAExB,eAAe,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAC/B,KAAC,cAAc,IAAiB,MAAM,EAAE,MAAM,IAAzB,MAAM,CAAC,EAAE,CAAoB,CACnD,CAAC,EACD,iBAAiB,IAAI,KAAC,YAAY,IAAC,SAAS,EAAC,KAAK,GAAG,EACtD,cAAK,SAAS,EAAC,MAAM,GAAG,EACvB,cAAc,IAAI,CACjB,cACE,GAAG,EAAE,cAAc,EACnB,SAAS,EAAC,oCAAoC,GAC9C,CACH,IACG,EACN,cAAK,SAAS,EAAC,qEAAqE,YAClF,iBACE,OAAO,EAAE,cAAc,EACvB,SAAS,EAAE,EAAE,CACX,iFAAiF,EACjF,qFAAqF,EACrF,UAAU,IAAI,2BAA2B,CAC1C,gBACU,kBAAkB,YAE7B,KAAC,WAAW,IAAC,SAAS,EAAC,SAAS,GAAG,GAC5B,GACL,IACF,CACP,CAAC;AACJ,CAAC,CAAC"}
+{"version":3,"file":"AnalysisResultsContainer.js","sourceRoot":"","sources":["../src/AnalysisResultsContainer.tsx"],"names":[],"mappings":"AAAA,YAAY,CAAC;;AACb,OAAO,EAAC,EAAE,EAAE,YAAY,EAAC,MAAM,cAAc,CAAC;AAC9C,OAAO,EAAC,WAAW,EAAC,MAAM,cAAc,CAAC;AACzC,OAAO,EAAC,cAAc,EAAC,MAAM,WAAW,CAAC;AACzC,OAAO,EAAC,cAAc,EAAC,MAAM,kBAAkB,CAAC;AAChD,OAAO,EACL,iBAAiB,EACjB,uBAAuB,GACxB,MAAM,8BAA8B,CAAC;AAEtC,MAAM,CAAC,MAAM,wBAAwB,GAAa,GAAG,EAAE;IACrD,MAAM,iBAAiB,GAAG,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,iBAAiB,CAAC,CAAC;IACxE,MAAM,eAAe,GAAG,cAAc,CACpC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,eAAe,CAC3C,CAAC;IACF,MAAM,CAAC,oBAAoB,EAAE,cAAc,CAAC,GAC1C,iBAAiB,EAAkB,CAAC;IACtC,MAAM,EAAC,UAAU,EAAE,cAAc,EAAC,GAChC,uBAAuB,CAAC,oBAAoB,CAAC,CAAC;IAEhD,OAAO,CACL,eAAK,SAAS,EAAC,6CAA6C,aAC1D,eACE,SAAS,EAAC,yDAAyD,EACnE,GAAG,EAAE,oBAAoB,aAExB,eAAe,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAC/B,KAAC,cAAc,IAAiB,MAAM,EAAE,MAAM,IAAzB,MAAM,CAAC,EAAE,CAAoB,CACnD,CAAC,EACD,iBAAiB,IAAI,KAAC,YAAY,IAAC,SAAS,EAAC,KAAK,GAAG,EACtD,cAAK,SAAS,EAAC,MAAM,GAAG,EACvB,cAAc,IAAI,CACjB,cACE,GAAG,EAAE,cAAc,EACnB,SAAS,EAAC,oCAAoC,GAC9C,CACH,IACG,EACN,cAAK,SAAS,EAAC,qEAAqE,YAClF,iBACE,OAAO,EAAE,cAAc,EACvB,SAAS,EAAE,EAAE,CACX,iFAAiF,EACjF,qFAAqF,EACrF,UAAU,IAAI,2BAA2B,CAC1C,gBACU,kBAAkB,YAE7B,KAAC,WAAW,IAAC,SAAS,EAAC,SAAS,GAAG,GAC5B,GACL,IACF,CACP,CAAC;AACJ,CAAC,CAAC","sourcesContent":["'use client';\nimport {cn, SkeletonPane} from '@sqlrooms/ui';\nimport {ChevronDown} from 'lucide-react';\nimport {useStoreWithAi} from './AiSlice';\nimport {AnalysisResult} from './AnalysisResult';\nimport {\n  useScrollToBottom,\n  useScrollToBottomButton,\n} from './hooks/use-scroll-to-bottom';\n\nexport const AnalysisResultsContainer: React.FC = () => {\n  const isRunningAnalysis = useStoreWithAi((s) => s.ai.isRunningAnalysis);\n  const analysisResults = useStoreWithAi(\n    (s) => s.project.config.ai.analysisResults,\n  );\n  const [messagesContainerRef, messagesEndRef] =\n    useScrollToBottom<HTMLDivElement>();\n  const {showButton, scrollToBottom} =\n    useScrollToBottomButton(messagesContainerRef);\n\n  return (\n    <div className=\"flex relative h-full w-full overflow-hidden\">\n      <div\n        className=\"flex flex-grow flex-col w-full gap-10 p-4 overflow-auto\"\n        ref={messagesContainerRef}\n      >\n        {analysisResults.map((result) => (\n          <AnalysisResult key={result.id} result={result} />\n        ))}\n        {isRunningAnalysis && <SkeletonPane className=\"p-4\" />}\n        <div className=\"h-20\" />\n        {messagesEndRef && (\n          <div\n            ref={messagesEndRef}\n            className=\"shrink-0 min-w-[24px] min-h-[24px]\"\n          />\n        )}\n      </div>\n      <div className=\"absolute inset-x-0 bottom-0 pointer-events-none flex justify-center\">\n        <button\n          onClick={scrollToBottom}\n          className={cn(\n            'pointer-events-auto bg-primary hover:bg-primary/90 text-primary-foreground z-50',\n            'rounded-full p-2 shadow-md transition-all duration-200 opacity-0 translate-y-4 mb-6',\n            showButton && 'opacity-100 translate-y-0',\n          )}\n          aria-label=\"Scroll to bottom\"\n        >\n          <ChevronDown className=\"w-5 h-5\" />\n        </button>\n      </div>\n    </div>\n  );\n};\n"]}

package/dist/QueryControls.js.map

@@ -1 +1 @@
-{"version":3,"file":"QueryControls.js","sourceRoot":"","sources":["../src/QueryControls.tsx"],"names":[],"mappings":";AAAA,OAAO,EAAC,MAAM,EAAE,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAC,MAAM,cAAc,CAAC;AAC3D,OAAO,EAAC,YAAY,EAAC,MAAM,cAAc,CAAC;AAC1C,OAAO,EAAC,WAAW,EAAC,MAAM,OAAO,CAAC;AAClC,OAAO,EAAC,cAAc,EAAC,MAAM,WAAW,CAAC;AAMzC,MAAM,CAAC,MAAM,aAAa,GAAiC,CAAC,EAAC,SAAS,EAAC,EAAE,EAAE;IACzE,MAAM,iBAAiB,GAAG,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,iBAAiB,CAAC,CAAC;IACxE,MAAM,WAAW,GAAG,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,aAAa,CAAC,CAAC;IAC9D,MAAM,cAAc,GAAG,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,cAAc,CAAC,CAAC;IAClE,MAAM,cAAc,GAAG,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,cAAc,CAAC,CAAC;IAClE,MAAM,iBAAiB,GAAG,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,iBAAiB,CAAC,CAAC;IACxE,MAAM,KAAK,GAAG,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC;IAE/D,MAAM,aAAa,GAAG,WAAW,CAC/B,CAAC,CAA2C,EAAE,EAAE;QAC9C,IACE,CAAC,CAAC,GAAG,KAAK,OAAO;YACjB,CAAC,CAAC,CAAC,QAAQ;YACX,CAAC,CAAC,CAAC,OAAO;YACV,CAAC,CAAC,CAAC,MAAM;YACT,CAAC,CAAC,CAAC,OAAO,EACV,CAAC;YACD,CAAC,CAAC,cAAc,EAAE,CAAC;YACnB,IAAI,CAAC,iBAAiB,IAAI,KAAK,IAAI,cAAc,CAAC,IAAI,EAAE,CAAC,MAAM,EAAE,CAAC;gBAChE,WAAW,EAAE,CAAC;YAChB,CAAC;QACH,CAAC;IACH,CAAC,EACD,CAAC,iBAAiB,EAAE,KAAK,EAAE,cAAc,EAAE,WAAW,CAAC,CACxD,CAAC;IAEF,MAAM,QAAQ,GAAG,OAAO,CACtB,CAAC,iBAAiB,IAAI,KAAK,IAAI,cAAc,CAAC,IAAI,EAAE,CAAC,MAAM,CAC5D,CAAC;IACF,OAAO,CACL,eACE,SAAS,EAAE,EAAE,CACX,4DAA4D,EAC5D,SAAS,CACV,aAED,cAAK,SAAS,EAAC,gDAAgD,YAC7D,cAAK,SAAS,EAAC,4BAA4B,6DAErC,GACF,EAEN,eAAK,SAAS,EAAC,iBAAiB,aAC9B,KAAC,QAAQ,IACP,QAAQ,EAAE,iBAAiB,EAC3B,SAAS,EAAC,WAAW,EACrB,KAAK,EAAE,cAAc,EACrB,QAAQ,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAClD,SAAS,EAAE,aAAa,GACxB,EACF,eAAK,SAAS,EAAC,gDAAgD,aAC5D,iBAAiB,IAAI,CACpB,MAAC,MAAM,IAAC,OAAO,EAAC,SAAS,EAAC,OAAO,EAAE,cAAc,aAC/C,KAAC,YAAY,IAAC,SAAS,EAAC,SAAS,GAAG,YAE7B,CACV,EACD,KAAC,MAAM,IAAC,OAAO,EAAC,SAAS,EAAC,OAAO,EAAE,WAAW,EAAE,QAAQ,EAAE,CAAC,QAAQ,YAChE,iBAAiB,CAAC,CAAC,CAAC,CACnB,eAAK,SAAS,EAAC,yBAAyB,aACtC,KAAC,OAAO,IAAC,SAAS,EAAC,SAAS,GAAG,sBAC3B,CACP,CAAC,CAAC,CAAC,CACF,gBAAgB,CACjB,GACM,IACL,IACF,IACF,CACP,CAAC;AACJ,CAAC,CAAC"}
+{"version":3,"file":"QueryControls.js","sourceRoot":"","sources":["../src/QueryControls.tsx"],"names":[],"mappings":";AAAA,OAAO,EAAC,MAAM,EAAE,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAC,MAAM,cAAc,CAAC;AAC3D,OAAO,EAAC,YAAY,EAAC,MAAM,cAAc,CAAC;AAC1C,OAAO,EAAC,WAAW,EAAC,MAAM,OAAO,CAAC;AAClC,OAAO,EAAC,cAAc,EAAC,MAAM,WAAW,CAAC;AAMzC,MAAM,CAAC,MAAM,aAAa,GAAiC,CAAC,EAAC,SAAS,EAAC,EAAE,EAAE;IACzE,MAAM,iBAAiB,GAAG,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,iBAAiB,CAAC,CAAC;IACxE,MAAM,WAAW,GAAG,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,aAAa,CAAC,CAAC;IAC9D,MAAM,cAAc,GAAG,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,cAAc,CAAC,CAAC;IAClE,MAAM,cAAc,GAAG,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,cAAc,CAAC,CAAC;IAClE,MAAM,iBAAiB,GAAG,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,iBAAiB,CAAC,CAAC;IACxE,MAAM,KAAK,GAAG,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC;IAE/D,MAAM,aAAa,GAAG,WAAW,CAC/B,CAAC,CAA2C,EAAE,EAAE;QAC9C,IACE,CAAC,CAAC,GAAG,KAAK,OAAO;YACjB,CAAC,CAAC,CAAC,QAAQ;YACX,CAAC,CAAC,CAAC,OAAO;YACV,CAAC,CAAC,CAAC,MAAM;YACT,CAAC,CAAC,CAAC,OAAO,EACV,CAAC;YACD,CAAC,CAAC,cAAc,EAAE,CAAC;YACnB,IAAI,CAAC,iBAAiB,IAAI,KAAK,IAAI,cAAc,CAAC,IAAI,EAAE,CAAC,MAAM,EAAE,CAAC;gBAChE,WAAW,EAAE,CAAC;YAChB,CAAC;QACH,CAAC;IACH,CAAC,EACD,CAAC,iBAAiB,EAAE,KAAK,EAAE,cAAc,EAAE,WAAW,CAAC,CACxD,CAAC;IAEF,MAAM,QAAQ,GAAG,OAAO,CACtB,CAAC,iBAAiB,IAAI,KAAK,IAAI,cAAc,CAAC,IAAI,EAAE,CAAC,MAAM,CAC5D,CAAC;IACF,OAAO,CACL,eACE,SAAS,EAAE,EAAE,CACX,4DAA4D,EAC5D,SAAS,CACV,aAED,cAAK,SAAS,EAAC,gDAAgD,YAC7D,cAAK,SAAS,EAAC,4BAA4B,6DAErC,GACF,EAEN,eAAK,SAAS,EAAC,iBAAiB,aAC9B,KAAC,QAAQ,IACP,QAAQ,EAAE,iBAAiB,EAC3B,SAAS,EAAC,WAAW,EACrB,KAAK,EAAE,cAAc,EACrB,QAAQ,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAClD,SAAS,EAAE,aAAa,GACxB,EACF,eAAK,SAAS,EAAC,gDAAgD,aAC5D,iBAAiB,IAAI,CACpB,MAAC,MAAM,IAAC,OAAO,EAAC,SAAS,EAAC,OAAO,EAAE,cAAc,aAC/C,KAAC,YAAY,IAAC,SAAS,EAAC,SAAS,GAAG,YAE7B,CACV,EACD,KAAC,MAAM,IAAC,OAAO,EAAC,SAAS,EAAC,OAAO,EAAE,WAAW,EAAE,QAAQ,EAAE,CAAC,QAAQ,YAChE,iBAAiB,CAAC,CAAC,CAAC,CACnB,eAAK,SAAS,EAAC,yBAAyB,aACtC,KAAC,OAAO,IAAC,SAAS,EAAC,SAAS,GAAG,sBAC3B,CACP,CAAC,CAAC,CAAC,CACF,gBAAgB,CACjB,GACM,IACL,IACF,IACF,CACP,CAAC;AACJ,CAAC,CAAC","sourcesContent":["import {Button, cn, Spinner, Textarea} from '@sqlrooms/ui';\nimport {OctagonXIcon} from 'lucide-react';\nimport {useCallback} from 'react';\nimport {useStoreWithAi} from './AiSlice';\n\ninterface QueryControlsProps {\n  className?: string;\n}\n\nexport const QueryControls: React.FC<QueryControlsProps> = ({className}) => {\n  const isRunningAnalysis = useStoreWithAi((s) => s.ai.isRunningAnalysis);\n  const runAnalysis = useStoreWithAi((s) => s.ai.startAnalysis);\n  const cancelAnalysis = useStoreWithAi((s) => s.ai.cancelAnalysis);\n  const analysisPrompt = useStoreWithAi((s) => s.ai.analysisPrompt);\n  const setAnalysisPrompt = useStoreWithAi((s) => s.ai.setAnalysisPrompt);\n  const model = useStoreWithAi((s) => s.project.config.ai.model);\n\n  const handleKeyDown = useCallback(\n    (e: React.KeyboardEvent<HTMLTextAreaElement>) => {\n      if (\n        e.key === 'Enter' &&\n        !e.shiftKey &&\n        !e.ctrlKey &&\n        !e.altKey &&\n        !e.metaKey\n      ) {\n        e.preventDefault();\n        if (!isRunningAnalysis && model && analysisPrompt.trim().length) {\n          runAnalysis();\n        }\n      }\n    },\n    [isRunningAnalysis, model, analysisPrompt, runAnalysis],\n  );\n\n  const canStart = Boolean(\n    !isRunningAnalysis && model && analysisPrompt.trim().length,\n  );\n  return (\n    <div\n      className={cn(\n        'flex flex-col items-center justify-center gap-4 w-full p-4',\n        className,\n      )}\n    >\n      <div className=\"flex items-center justify-between gap-4 w-full\">\n        <div className=\"font-semibold text-lg pl-1\">\n          What would you like to learn about the data?\n        </div>\n      </div>\n\n      <div className=\"relative w-full\">\n        <Textarea\n          disabled={isRunningAnalysis}\n          className=\"h-[100px]\"\n          value={analysisPrompt}\n          onChange={(e) => setAnalysisPrompt(e.target.value)}\n          onKeyDown={handleKeyDown}\n        />\n        <div className=\"absolute top-1 right-1 flex items-center gap-2\">\n          {isRunningAnalysis && (\n            <Button variant=\"outline\" onClick={cancelAnalysis}>\n              <OctagonXIcon className=\"w-4 h-4\" />\n              Stop\n            </Button>\n          )}\n          <Button variant=\"outline\" onClick={runAnalysis} disabled={!canStart}>\n            {isRunningAnalysis ? (\n              <div className=\"flex items-center gap-2\">\n                <Spinner className=\"w-4 h-4\" /> Running…\n              </div>\n            ) : (\n              'Start Analysis'\n            )}\n          </Button>\n        </div>\n      </div>\n    </div>\n  );\n};\n"]}

package/dist/QueryResult.d.ts

@@ -0,0 +1,9 @@
+import { CustomFunctionCall } from '@openassistant/core';
+/**
+ * Creates a query result message component based on a custom function call.
+ *
+ * @param {CustomFunctionCall} props - The custom function call properties
+ * @returns {JSX.Element | null} A query result table modal component or null if no data is available
+ */
+export declare function queryMessage(props: CustomFunctionCall): import("react/jsx-runtime").JSX.Element | null;
+//# sourceMappingURL=QueryResult.d.ts.map

package/dist/QueryResult.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryResult.d.ts","sourceRoot":"","sources":["../src/QueryResult.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAC,kBAAkB,EAAC,MAAM,qBAAqB,CAAC;AA6DvD;;;;;GAKG;AACH,wBAAgB,YAAY,CAAC,KAAK,EAAE,kBAAkB,kDAarD"}

package/dist/QueryResult.js

@@ -0,0 +1,46 @@
+import { jsx as _jsx, jsxs as _jsxs, Fragment as _Fragment } from "react/jsx-runtime";
+import { DataTableModal } from '@sqlrooms/data-table';
+import { Expand, TableIcon } from 'lucide-react';
+import { useState } from 'react';
+/**
+ * A modal component that displays a data table with a title.
+ *
+ * @component
+ * @param {Object} props - The component props
+ * @param {string} props.title - The title to display in the modal header
+ * @param {string} props.sqlQuery - The SQL query that generated the data
+ * @returns {JSX.Element} A modal component containing a data table
+ */
+function QueryResultTableModal({ title, sqlQuery, }) {
+    const [isModalOpen, setIsModalOpen] = useState(false);
+    const handleClick = () => {
+        setIsModalOpen(true);
+    };
+    const onClose = () => {
+        setIsModalOpen(false);
+    };
+    return (_jsxs(_Fragment, { children: [_jsxs("div", { className: "flex gap-2 flex-row items-center text-muted-foreground", onClick: handleClick, children: [_jsx(TableIcon, { className: "h-4 w-4" }), _jsx("h3", { className: "ml-1 text-xs uppercase", children: "Query Result" }), _jsx(Expand, { className: "h-3 w-3 shrink-0 text-muted-foreground transition-transform duration-200" })] }), _jsx(DataTableModal, { title: title, query: sqlQuery, tableModal: {
+                    isOpen: isModalOpen,
+                    onClose: onClose,
+                } })] }));
+}
+function isQueryToolOutputData(data) {
+    return typeof data === 'object' && data !== null && 'sqlQuery' in data;
+}
+/**
+ * Creates a query result message component based on a custom function call.
+ *
+ * @param {CustomFunctionCall} props - The custom function call properties
+ * @returns {JSX.Element | null} A query result table modal component or null if no data is available
+ */
+export function queryMessage(props) {
+    const data = props.output?.data;
+    if (!isQueryToolOutputData(data)) {
+        throw new Error('Invalid query tool output data');
+    }
+    if (!data.sqlQuery) {
+        return null;
+    }
+    return (_jsx(QueryResultTableModal, { title: data.sqlQuery, sqlQuery: data.sqlQuery }));
+}
+//# sourceMappingURL=QueryResult.js.map

package/dist/QueryResult.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QueryResult.js","sourceRoot":"","sources":["../src/QueryResult.tsx"],"names":[],"mappings":";AACA,OAAO,EAAC,cAAc,EAAC,MAAM,sBAAsB,CAAC;AACpD,OAAO,EAAC,MAAM,EAAE,SAAS,EAAC,MAAM,cAAc,CAAC;AAC/C,OAAO,EAAC,QAAQ,EAAC,MAAM,OAAO,CAAC;AAE/B;;;;;;;;GAQG;AACH,SAAS,qBAAqB,CAAC,EAC7B,KAAK,EACL,QAAQ,GAIT;IACC,MAAM,CAAC,WAAW,EAAE,cAAc,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IAEtD,MAAM,WAAW,GAAG,GAAG,EAAE;QACvB,cAAc,CAAC,IAAI,CAAC,CAAC;IACvB,CAAC,CAAC;IAEF,MAAM,OAAO,GAAG,GAAG,EAAE;QACnB,cAAc,CAAC,KAAK,CAAC,CAAC;IACxB,CAAC,CAAC;IAEF,OAAO,CACL,8BACE,eACE,SAAS,EAAC,wDAAwD,EAClE,OAAO,EAAE,WAAW,aAEpB,KAAC,SAAS,IAAC,SAAS,EAAC,SAAS,GAAG,EACjC,aAAI,SAAS,EAAC,wBAAwB,6BAAkB,EACxD,KAAC,MAAM,IAAC,SAAS,EAAC,0EAA0E,GAAG,IAC3F,EAEN,KAAC,cAAc,IACb,KAAK,EAAE,KAAK,EACZ,KAAK,EAAE,QAAQ,EACf,UAAU,EAAE;oBACV,MAAM,EAAE,WAAW;oBACnB,OAAO,EAAE,OAAO;iBACjB,GACD,IACD,CACJ,CAAC;AACJ,CAAC;AAMD,SAAS,qBAAqB,CAAC,IAAa;IAC1C,OAAO,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,IAAI,IAAI,UAAU,IAAI,IAAI,CAAC;AACzE,CAAC;AACD;;;;;GAKG;AACH,MAAM,UAAU,YAAY,CAAC,KAAyB;IACpD,MAAM,IAAI,GAAG,KAAK,CAAC,MAAM,EAAE,IAAI,CAAC;IAEhC,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,EAAE,CAAC;QACjC,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;IACpD,CAAC;IACD,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;QACnB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,CACL,KAAC,qBAAqB,IAAC,KAAK,EAAE,IAAI,CAAC,QAAQ,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,GAAI,CACzE,CAAC;AACJ,CAAC","sourcesContent":["import {CustomFunctionCall} from '@openassistant/core';\nimport {DataTableModal} from '@sqlrooms/data-table';\nimport {Expand, TableIcon} from 'lucide-react';\nimport {useState} from 'react';\n\n/**\n * A modal component that displays a data table with a title.\n *\n * @component\n * @param {Object} props - The component props\n * @param {string} props.title - The title to display in the modal header\n * @param {string} props.sqlQuery - The SQL query that generated the data\n * @returns {JSX.Element} A modal component containing a data table\n */\nfunction QueryResultTableModal({\n  title,\n  sqlQuery,\n}: {\n  title: string;\n  sqlQuery: string;\n}) {\n  const [isModalOpen, setIsModalOpen] = useState(false);\n\n  const handleClick = () => {\n    setIsModalOpen(true);\n  };\n\n  const onClose = () => {\n    setIsModalOpen(false);\n  };\n\n  return (\n    <>\n      <div\n        className=\"flex gap-2 flex-row items-center text-muted-foreground\"\n        onClick={handleClick}\n      >\n        <TableIcon className=\"h-4 w-4\" />\n        <h3 className=\"ml-1 text-xs uppercase\">Query Result</h3>\n        <Expand className=\"h-3 w-3 shrink-0 text-muted-foreground transition-transform duration-200\" />\n      </div>\n\n      <DataTableModal\n        title={title}\n        query={sqlQuery}\n        tableModal={{\n          isOpen: isModalOpen,\n          onClose: onClose,\n        }}\n      />\n    </>\n  );\n}\n\ntype QueryToolOutputData = {\n  sqlQuery: string;\n};\n\nfunction isQueryToolOutputData(data: unknown): data is QueryToolOutputData {\n  return typeof data === 'object' && data !== null && 'sqlQuery' in data;\n}\n/**\n * Creates a query result message component based on a custom function call.\n *\n * @param {CustomFunctionCall} props - The custom function call properties\n * @returns {JSX.Element | null} A query result table modal component or null if no data is available\n */\nexport function queryMessage(props: CustomFunctionCall) {\n  const data = props.output?.data;\n\n  if (!isQueryToolOutputData(data)) {\n    throw new Error('Invalid query tool output data');\n  }\n  if (!data.sqlQuery) {\n    return null;\n  }\n\n  return (\n    <QueryResultTableModal title={data.sqlQuery} sqlQuery={data.sqlQuery} />\n  );\n}\n"]}

package/dist/ToolCall.d.ts

@@ -1,7 +1,66 @@
-import { ToolCallSchema } from './schemas';
-interface ToolCallProps {
-    toolCall: ToolCallSchema;
-}
-export declare function ToolCall({ toolCall }: ToolCallProps): import("react/jsx-runtime").JSX.Element;
+import { ChartToolParameters, QueryToolParameters, ToolResultSchema } from './schemas';
+import { ReactNode } from 'react';
+import React from 'react';
+/**
+ * Props for the QueryToolCall component
+ * @interface QueryToolCallProps
+ * @extends {QueryToolParameters}
+ * @property {ReactNode} [customMessage] - Optional custom message to display with the query
+ */
+type QueryToolCallProps = QueryToolParameters & {
+    customMessage?: ReactNode;
+};
+/**
+ * Type guard to check if an argument object matches QueryToolParameters
+ * @param {unknown} args - The arguments to check
+ * @returns {boolean} True if args matches QueryToolParameters structure
+ */
+export declare function isQueryToolParameters(args: unknown): args is QueryToolParameters;
+/**
+ * Type guard to check if an argument object matches ChartToolParameters
+ * @param {unknown} args - The arguments to check
+ * @returns {boolean} True if args matches ChartToolParameters structure
+ */
+export declare function isChartToolParameters(args: unknown): args is ChartToolParameters;
+/**
+ * Renders a SQL query tool call with reasoning and query text
+ * @param {QueryToolCallProps} props - The component props
+ * @returns {JSX.Element} The rendered query tool call
+ */
+export declare const QueryToolCall: React.NamedExoticComponent<QueryToolCallProps>;
+/**
+ * Renders a chart tool call with visualization using Vega-Lite
+ * @param {ChartToolParameters} props - The component props
+ * @returns {JSX.Element} The rendered chart tool call
+ */
+export declare const ChartToolCall: React.NamedExoticComponent<{
+    sqlQuery: string;
+    reasoning: string;
+    vegaLiteSpec: string;
+}>;
+/**
+ * Type guard to check if a result object matches AnalysisAnswerProps
+ * @param {unknown} result - The result to check
+ * @returns {boolean} True if result matches AnalysisAnswerProps structure
+ */
+type AnalysisAnswerProps = {
+    success: true;
+    data: {
+        analysis: string;
+    };
+};
+/**
+ * Type guard to check if a result object matches AnalysisAnswerProps
+ * @param {unknown} result - The result to check
+ * @returns {boolean} True if result matches AnalysisAnswerProps structure
+ */
+export declare function isAnalysisAnswer(result: ToolResultSchema['result']): result is AnalysisAnswerProps;
+/**
+ * Renders an analysis answer with markdown content of the final streaming response.
+ *
+ * @param {AnalysisAnswerProps} props - The component props. See {@link AnalysisAnswerProps} for more details.
+ * @returns {JSX.Element} The rendered answer tool call
+ */
+export declare const AnalysisAnswer: React.NamedExoticComponent<AnalysisAnswerProps>;
 export {};
 //# sourceMappingURL=ToolCall.d.ts.map

package/dist/ToolCall.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ToolCall.d.ts","sourceRoot":"","sources":["../src/ToolCall.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAC,cAAc,EAAC,MAAM,WAAW,CAAC;AAezC,UAAU,aAAa;IACrB,QAAQ,EAAE,cAAc,CAAC;CAC1B;AAED,wBAAgB,QAAQ,CAAC,EAAC,QAAQ,EAAC,EAAE,aAAa,2CAiFjD"}
+{"version":3,"file":"ToolCall.d.ts","sourceRoot":"","sources":["../src/ToolCall.tsx"],"names":[],"mappings":"AAAA,OAAO,EACL,mBAAmB,EACnB,mBAAmB,EACnB,gBAAgB,EACjB,MAAM,WAAW,CAAC;AAEnB,OAAO,EAAW,SAAS,EAAC,MAAM,OAAO,CAAC;AAG1C,OAAO,KAAK,MAAM,OAAO,CAAC;AAE1B;;;;;GAKG;AACH,KAAK,kBAAkB,GAAG,mBAAmB,GAAG;IAC9C,aAAa,CAAC,EAAE,SAAS,CAAC;CAC3B,CAAC;AAEF;;;;GAIG;AACH,wBAAgB,qBAAqB,CACnC,IAAI,EAAE,OAAO,GACZ,IAAI,IAAI,mBAAmB,CAO7B;AAED;;;;GAIG;AACH,wBAAgB,qBAAqB,CACnC,IAAI,EAAE,OAAO,GACZ,IAAI,IAAI,mBAAmB,CAE7B;AAED;;;;GAIG;AACH,eAAO,MAAM,aAAa,gDAYxB,CAAC;AAEH;;;;GAIG;AACH,eAAO,MAAM,aAAa;;;;EAiCxB,CAAC;AAEH;;;;GAIG;AACH,KAAK,mBAAmB,GAAG;IACzB,OAAO,EAAE,IAAI,CAAC;IACd,IAAI,EAAE;QACJ,QAAQ,EAAE,MAAM,CAAC;KAClB,CAAC;CACH,CAAC;AAEF;;;;GAIG;AACH,wBAAgB,gBAAgB,CAC9B,MAAM,EAAE,gBAAgB,CAAC,QAAQ,CAAC,GACjC,MAAM,IAAI,mBAAmB,CAE/B;AAED;;;;;GAKG;AACH,eAAO,MAAM,cAAc,iDAYzB,CAAC"}

package/dist/ToolCall.js

@@ -1,16 +1,59 @@
 import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
-import { Badge, Button, Popover, PopoverContent, PopoverTrigger, Spinner, cn, } from '@sqlrooms/ui';
-import { CodeIcon } from 'lucide-react';
+import { Spinner } from '@sqlrooms/ui';
 import { Suspense } from 'react';
 import Markdown from 'react-markdown';
 import { VegaLiteChart } from '@sqlrooms/vega';
-export function ToolCall({ toolCall }) {
-    const { args, toolName, toolCallId } = toolCall;
-    const { type } = args;
-    return (_jsxs("div", { className: cn('border-2 relative px-5 py-6 rounded-md dark:bg-gray-900 dark:text-gray-300 bg-gray-100 text-gray-700 text-xs', {
-            ' border-blue-500': toolName === 'answer',
-        }), children: [_jsx(Badge, { variant: "secondary", className: cn('text-xs absolute top-[-12px] left-2 dark:text-gray-100 text-gray-700', {
-                    ' border-blue-500': toolName === 'answer',
-                }), children: toolName }), _jsx("div", { className: "absolute top-2 right-2", children: _jsxs(Popover, { children: [_jsx(PopoverTrigger, { asChild: true, children: _jsx(Button, { variant: "outline", size: "icon", className: "w-6 h-6", children: _jsx(CodeIcon, { className: "w-4 h-4" }) }) }), _jsx(PopoverContent, { className: "w-[400px] max-h-[400px] overflow-auto", align: "start", side: "right", children: _jsx("pre", { className: "text-xs", children: JSON.stringify(toolCall, null, 2) }) })] }) }), type === 'query' ? (_jsxs("div", { className: "flex flex-col gap-5", children: [_jsx("div", { className: "text-xs text-gray-700 dark:text-gray-400", children: args.reasoning }), _jsx("div", { className: "font-mono", children: args.sqlQuery })] })) : type === 'answer' ? (_jsxs("div", { className: "flex flex-col gap-5", children: [_jsx("div", { className: "text-sm", children: _jsx(Markdown, { children: args.answer }) }), args.chart && (_jsxs("div", { className: "flex flex-col gap-2 w-full", children: [_jsx("div", { className: "text-xs text-muted-foreground font-mono", children: args.chart.sqlQuery }), _jsx(Suspense, { fallback: _jsx("div", { className: "w-full h-full flex items-center justify-center", children: _jsx(Spinner, { className: "w-4 h-4" }) }), children: _jsx(VegaLiteChart, { className: "max-w-[600px]", aspectRatio: 16 / 9, sqlQuery: args.chart.sqlQuery, spec: args.chart.vegaLiteSpec }) })] }))] })) : (_jsx("pre", { children: JSON.stringify(args, null, 2) }))] }, toolCallId));
+import React from 'react';
+/**
+ * Type guard to check if an argument object matches QueryToolParameters
+ * @param {unknown} args - The arguments to check
+ * @returns {boolean} True if args matches QueryToolParameters structure
+ */
+export function isQueryToolParameters(args) {
+    return (typeof args === 'object' &&
+        args !== null &&
+        'reasoning' in args &&
+        'sqlQuery' in args);
 }
+/**
+ * Type guard to check if an argument object matches ChartToolParameters
+ * @param {unknown} args - The arguments to check
+ * @returns {boolean} True if args matches ChartToolParameters structure
+ */
+export function isChartToolParameters(args) {
+    return typeof args === 'object' && args !== null && 'vegaLiteSpec' in args;
+}
+/**
+ * Renders a SQL query tool call with reasoning and query text
+ * @param {QueryToolCallProps} props - The component props
+ * @returns {JSX.Element} The rendered query tool call
+ */
+export const QueryToolCall = React.memo(function QueryToolCall({ reasoning, sqlQuery, customMessage, }) {
+    return (_jsxs("div", { className: "flex flex-col gap-5", children: [_jsx("div", { className: "text-xs text-gray-400", children: reasoning }), _jsx("div", { className: "font-mono", children: sqlQuery }), customMessage] }));
+});
+/**
+ * Renders a chart tool call with visualization using Vega-Lite
+ * @param {ChartToolParameters} props - The component props
+ * @returns {JSX.Element} The rendered chart tool call
+ */
+export const ChartToolCall = React.memo(function ChartToolCall({ reasoning, sqlQuery, vegaLiteSpec, }) {
+    return (_jsxs("div", { className: "flex flex-col gap-5", children: [_jsx("div", { className: "text-sm", children: _jsx(Markdown, { children: reasoning }) }), vegaLiteSpec && (_jsxs("div", { className: "flex flex-col gap-2", children: [_jsx("div", { className: "text-xs text-muted-foreground font-mono", children: sqlQuery }), _jsx(Suspense, { fallback: _jsx("div", { className: "w-full h-full flex items-center justify-center", children: _jsx(Spinner, { className: "w-4 h-4" }) }), children: _jsx(VegaLiteChart, { width: 400, height: 250, sqlQuery: sqlQuery, spec: vegaLiteSpec }) })] }))] }));
+});
+/**
+ * Type guard to check if a result object matches AnalysisAnswerProps
+ * @param {unknown} result - The result to check
+ * @returns {boolean} True if result matches AnalysisAnswerProps structure
+ */
+export function isAnalysisAnswer(result) {
+    return result.success && result.data.analysis !== undefined;
+}
+/**
+ * Renders an analysis answer with markdown content of the final streaming response.
+ *
+ * @param {AnalysisAnswerProps} props - The component props. See {@link AnalysisAnswerProps} for more details.
+ * @returns {JSX.Element} The rendered answer tool call
+ */
+export const AnalysisAnswer = React.memo(function AnalysisAnswer(props) {
+    return (_jsx("div", { className: "flex flex-col gap-5", children: _jsx("div", { className: "text-xs", children: _jsx(Markdown, { className: "whitespace-pre-wrap break-words", children: props.data.analysis }) }) }));
+});
 //# sourceMappingURL=ToolCall.js.map

package/dist/ToolCall.js.map

@@ -1 +1 @@
-{"version":3,"file":"ToolCall.js","sourceRoot":"","sources":["../src/ToolCall.tsx"],"names":[],"mappings":";AACA,OAAO,EACL,KAAK,EACL,MAAM,EACN,OAAO,EACP,cAAc,EACd,cAAc,EACd,OAAO,EACP,EAAE,GACH,MAAM,cAAc,CAAC;AACtB,OAAO,EAAC,QAAQ,EAAC,MAAM,cAAc,CAAC;AACtC,OAAO,EAAC,QAAQ,EAAC,MAAM,OAAO,CAAC;AAC/B,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AACtC,OAAO,EAAC,aAAa,EAAC,MAAM,gBAAgB,CAAC;AAM7C,MAAM,UAAU,QAAQ,CAAC,EAAC,QAAQ,EAAgB;IAChD,MAAM,EAAC,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAC,GAAG,QAAQ,CAAC;IAC9C,MAAM,EAAC,IAAI,EAAC,GAAG,IAAI,CAAC;IAEpB,OAAO,CACL,eAEE,SAAS,EAAE,EAAE,CACX,8GAA8G,EAC9G;YACE,kBAAkB,EAAE,QAAQ,KAAK,QAAQ;SAC1C,CACF,aAED,KAAC,KAAK,IACJ,OAAO,EAAC,WAAW,EACnB,SAAS,EAAE,EAAE,CACX,sEAAsE,EACtE;oBACE,kBAAkB,EAAE,QAAQ,KAAK,QAAQ;iBAC1C,CACF,YAEA,QAAQ,GACH,EAER,cAAK,SAAS,EAAC,wBAAwB,YACrC,MAAC,OAAO,eACN,KAAC,cAAc,IAAC,OAAO,kBACrB,KAAC,MAAM,IAAC,OAAO,EAAC,SAAS,EAAC,IAAI,EAAC,MAAM,EAAC,SAAS,EAAC,SAAS,YACvD,KAAC,QAAQ,IAAC,SAAS,EAAC,SAAS,GAAG,GACzB,GACM,EACjB,KAAC,cAAc,IACb,SAAS,EAAC,uCAAuC,EACjD,KAAK,EAAC,OAAO,EACb,IAAI,EAAC,OAAO,YAEZ,cAAK,SAAS,EAAC,SAAS,YAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,GAAO,GACnD,IACT,GACN,EACL,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,CAClB,eAAK,SAAS,EAAC,qBAAqB,aAClC,cAAK,SAAS,EAAC,0CAA0C,YACtD,IAAI,CAAC,SAAS,GACX,EACN,cAAK,SAAS,EAAC,WAAW,YAAE,IAAI,CAAC,QAAQ,GAAO,IAC5C,CACP,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,CACtB,eAAK,SAAS,EAAC,qBAAqB,aAClC,cAAK,SAAS,EAAC,SAAS,YACtB,KAAC,QAAQ,cAAE,IAAI,CAAC,MAAM,GAAY,GAC9B,EACL,IAAI,CAAC,KAAK,IAAI,CACb,eAAK,SAAS,EAAC,4BAA4B,aACzC,cAAK,SAAS,EAAC,yCAAyC,YACrD,IAAI,CAAC,KAAK,CAAC,QAAQ,GAChB,EACN,KAAC,QAAQ,IACP,QAAQ,EACN,cAAK,SAAS,EAAC,gDAAgD,YAC7D,KAAC,OAAO,IAAC,SAAS,EAAC,SAAS,GAAG,GAC3B,YAGR,KAAC,aAAa,IACZ,SAAS,EAAC,eAAe,EACzB,WAAW,EAAE,EAAE,GAAG,CAAC,EACnB,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ,EAC7B,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY,GAC7B,GACO,IACP,CACP,IACG,CACP,CAAC,CAAC,CAAC,CACF,wBAAM,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,GAAO,CAC3C,KAxEI,UAAU,CAyEX,CACP,CAAC;AACJ,CAAC"}
+{"version":3,"file":"ToolCall.js","sourceRoot":"","sources":["../src/ToolCall.tsx"],"names":[],"mappings":";AAKA,OAAO,EAAC,OAAO,EAAC,MAAM,cAAc,CAAC;AACrC,OAAO,EAAC,QAAQ,EAAY,MAAM,OAAO,CAAC;AAC1C,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AACtC,OAAO,EAAC,aAAa,EAAC,MAAM,gBAAgB,CAAC;AAC7C,OAAO,KAAK,MAAM,OAAO,CAAC;AAY1B;;;;GAIG;AACH,MAAM,UAAU,qBAAqB,CACnC,IAAa;IAEb,OAAO,CACL,OAAO,IAAI,KAAK,QAAQ;QACxB,IAAI,KAAK,IAAI;QACb,WAAW,IAAI,IAAI;QACnB,UAAU,IAAI,IAAI,CACnB,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,qBAAqB,CACnC,IAAa;IAEb,OAAO,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,IAAI,IAAI,cAAc,IAAI,IAAI,CAAC;AAC7E,CAAC;AAED;;;;GAIG;AACH,MAAM,CAAC,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,SAAS,aAAa,CAAC,EAC7D,SAAS,EACT,QAAQ,EACR,aAAa,GACM;IACnB,OAAO,CACL,eAAK,SAAS,EAAC,qBAAqB,aAClC,cAAK,SAAS,EAAC,uBAAuB,YAAE,SAAS,GAAO,EACxD,cAAK,SAAS,EAAC,WAAW,YAAE,QAAQ,GAAO,EAC1C,aAAa,IACV,CACP,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH;;;;GAIG;AACH,MAAM,CAAC,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,SAAS,aAAa,CAAC,EAC7D,SAAS,EACT,QAAQ,EACR,YAAY,GACQ;IACpB,OAAO,CACL,eAAK,SAAS,EAAC,qBAAqB,aAClC,cAAK,SAAS,EAAC,SAAS,YACtB,KAAC,QAAQ,cAAE,SAAS,GAAY,GAC5B,EACL,YAAY,IAAI,CACf,eAAK,SAAS,EAAC,qBAAqB,aAClC,cAAK,SAAS,EAAC,yCAAyC,YACrD,QAAQ,GACL,EACN,KAAC,QAAQ,IACP,QAAQ,EACN,cAAK,SAAS,EAAC,gDAAgD,YAC7D,KAAC,OAAO,IAAC,SAAS,EAAC,SAAS,GAAG,GAC3B,YAGR,KAAC,aAAa,IACZ,KAAK,EAAE,GAAG,EACV,MAAM,EAAE,GAAG,EACX,QAAQ,EAAE,QAAQ,EAClB,IAAI,EAAE,YAAY,GAClB,GACO,IACP,CACP,IACG,CACP,CAAC;AACJ,CAAC,CAAC,CAAC;AAcH;;;;GAIG;AACH,MAAM,UAAU,gBAAgB,CAC9B,MAAkC;IAElC,OAAO,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,QAAQ,KAAK,SAAS,CAAC;AAC9D,CAAC;AAED;;;;;GAKG;AACH,MAAM,CAAC,MAAM,cAAc,GAAG,KAAK,CAAC,IAAI,CAAC,SAAS,cAAc,CAC9D,KAA0B;IAE1B,OAAO,CACL,cAAK,SAAS,EAAC,qBAAqB,YAClC,cAAK,SAAS,EAAC,SAAS,YACtB,KAAC,QAAQ,IAAC,SAAS,EAAC,iCAAiC,YAClD,KAAK,CAAC,IAAI,CAAC,QAAQ,GACX,GACP,GACF,CACP,CAAC;AACJ,CAAC,CAAC,CAAC","sourcesContent":["import {\n  ChartToolParameters,\n  QueryToolParameters,\n  ToolResultSchema,\n} from './schemas';\nimport {Spinner} from '@sqlrooms/ui';\nimport {Suspense, ReactNode} from 'react';\nimport Markdown from 'react-markdown';\nimport {VegaLiteChart} from '@sqlrooms/vega';\nimport React from 'react';\n\n/**\n * Props for the QueryToolCall component\n * @interface QueryToolCallProps\n * @extends {QueryToolParameters}\n * @property {ReactNode} [customMessage] - Optional custom message to display with the query\n */\ntype QueryToolCallProps = QueryToolParameters & {\n  customMessage?: ReactNode;\n};\n\n/**\n * Type guard to check if an argument object matches QueryToolParameters\n * @param {unknown} args - The arguments to check\n * @returns {boolean} True if args matches QueryToolParameters structure\n */\nexport function isQueryToolParameters(\n  args: unknown,\n): args is QueryToolParameters {\n  return (\n    typeof args === 'object' &&\n    args !== null &&\n    'reasoning' in args &&\n    'sqlQuery' in args\n  );\n}\n\n/**\n * Type guard to check if an argument object matches ChartToolParameters\n * @param {unknown} args - The arguments to check\n * @returns {boolean} True if args matches ChartToolParameters structure\n */\nexport function isChartToolParameters(\n  args: unknown,\n): args is ChartToolParameters {\n  return typeof args === 'object' && args !== null && 'vegaLiteSpec' in args;\n}\n\n/**\n * Renders a SQL query tool call with reasoning and query text\n * @param {QueryToolCallProps} props - The component props\n * @returns {JSX.Element} The rendered query tool call\n */\nexport const QueryToolCall = React.memo(function QueryToolCall({\n  reasoning,\n  sqlQuery,\n  customMessage,\n}: QueryToolCallProps) {\n  return (\n    <div className=\"flex flex-col gap-5\">\n      <div className=\"text-xs text-gray-400\">{reasoning}</div>\n      <div className=\"font-mono\">{sqlQuery}</div>\n      {customMessage}\n    </div>\n  );\n});\n\n/**\n * Renders a chart tool call with visualization using Vega-Lite\n * @param {ChartToolParameters} props - The component props\n * @returns {JSX.Element} The rendered chart tool call\n */\nexport const ChartToolCall = React.memo(function ChartToolCall({\n  reasoning,\n  sqlQuery,\n  vegaLiteSpec,\n}: ChartToolParameters) {\n  return (\n    <div className=\"flex flex-col gap-5\">\n      <div className=\"text-sm\">\n        <Markdown>{reasoning}</Markdown>\n      </div>\n      {vegaLiteSpec && (\n        <div className=\"flex flex-col gap-2\">\n          <div className=\"text-xs text-muted-foreground font-mono\">\n            {sqlQuery}\n          </div>\n          <Suspense\n            fallback={\n              <div className=\"w-full h-full flex items-center justify-center\">\n                <Spinner className=\"w-4 h-4\" />\n              </div>\n            }\n          >\n            <VegaLiteChart\n              width={400}\n              height={250}\n              sqlQuery={sqlQuery}\n              spec={vegaLiteSpec}\n            />\n          </Suspense>\n        </div>\n      )}\n    </div>\n  );\n});\n\n/**\n * Type guard to check if a result object matches AnalysisAnswerProps\n * @param {unknown} result - The result to check\n * @returns {boolean} True if result matches AnalysisAnswerProps structure\n */\ntype AnalysisAnswerProps = {\n  success: true;\n  data: {\n    analysis: string;\n  };\n};\n\n/**\n * Type guard to check if a result object matches AnalysisAnswerProps\n * @param {unknown} result - The result to check\n * @returns {boolean} True if result matches AnalysisAnswerProps structure\n */\nexport function isAnalysisAnswer(\n  result: ToolResultSchema['result'],\n): result is AnalysisAnswerProps {\n  return result.success && result.data.analysis !== undefined;\n}\n\n/**\n * Renders an analysis answer with markdown content of the final streaming response.\n *\n * @param {AnalysisAnswerProps} props - The component props. See {@link AnalysisAnswerProps} for more details.\n * @returns {JSX.Element} The rendered answer tool call\n */\nexport const AnalysisAnswer = React.memo(function AnalysisAnswer(\n  props: AnalysisAnswerProps,\n) {\n  return (\n    <div className=\"flex flex-col gap-5\">\n      <div className=\"text-xs\">\n        <Markdown className=\"whitespace-pre-wrap break-words\">\n          {props.data.analysis}\n        </Markdown>\n      </div>\n    </div>\n  );\n});\n"]}

package/dist/ToolResult.d.ts

@@ -1,6 +1,9 @@
 import { ToolResultSchema } from './schemas';
+import { ReactNode } from 'react';
+import React from 'react';
 interface ToolResultProps {
     toolResult: ToolResultSchema;
+    customMessage?: ReactNode;
 }
 export declare const ToolResult: React.FC<ToolResultProps>;
 export {};

package/dist/ToolResult.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ToolResult.d.ts","sourceRoot":"","sources":["../src/ToolResult.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAC,gBAAgB,EAAC,MAAM,WAAW,CAAC;AAW3C,UAAU,eAAe;IACvB,UAAU,EAAE,gBAAgB,CAAC;CAC9B;AAED,eAAO,MAAM,UAAU,EAAE,KAAK,CAAC,EAAE,CAAC,eAAe,CA2DhD,CAAC"}
+{"version":3,"file":"ToolResult.d.ts","sourceRoot":"","sources":["../src/ToolResult.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAC,gBAAgB,EAAC,MAAM,WAAW,CAAC;AAU3C,OAAO,EAAC,SAAS,EAAC,MAAM,OAAO,CAAC;AAQhC,OAAO,KAAK,MAAM,OAAO,CAAC;AAE1B,UAAU,eAAe;IACvB,UAAU,EAAE,gBAAgB,CAAC;IAC7B,aAAa,CAAC,EAAE,SAAS,CAAC;CAC3B;AAkBD,eAAO,MAAM,UAAU,EAAE,KAAK,CAAC,EAAE,CAAC,eAAe,CAqEhD,CAAC"}

package/dist/ToolResult.js

@@ -1,9 +1,26 @@
 import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
 import { Badge, Button, Popover, PopoverContent, PopoverTrigger, cn, } from '@sqlrooms/ui';
 import { CheckCircle2Icon, CodeIcon, XCircleIcon } from 'lucide-react';
-export const ToolResult = ({ toolResult }) => {
+import { AnalysisAnswer, ChartToolCall, isAnalysisAnswer, isChartToolParameters, } from './ToolCall';
+import { isQueryToolParameters, QueryToolCall } from './ToolCall';
+function getBorderColor(isSuccess, toolName) {
+    if (!isSuccess) {
+        return 'border-red-500';
+    }
+    switch (toolName) {
+        case 'query':
+            return 'border-green-500';
+        case 'chart':
+            return 'border-blue-500';
+        case 'answer':
+            return 'border-gray-500';
+        default:
+            return 'border-gray-500';
+    }
+}
+export const ToolResult = ({ toolResult, customMessage, }) => {
     const { toolName, args, result } = toolResult;
     const isSuccess = result.success;
-    return (_jsxs("div", { className: cn('border-2 relative bg-gray-100 dark:bg-gray-900 px-5 py-6 rounded-md text-gray-700 dark:text-gray-300 text-xs', isSuccess ? 'border-green-500' : 'border-red-500'), children: [_jsxs(Badge, { variant: "secondary", className: cn('text-xs absolute top-[-12px] left-2 dark:text-gray-100 text-gray-700 flex items-center gap-1 border', isSuccess ? 'border-green-500' : 'border-red-500'), children: [isSuccess ? (_jsx(CheckCircle2Icon, { className: "w-3 h-3 text-green-500" })) : (_jsx(XCircleIcon, { className: "w-3 h-3 text-red-500" })), toolName] }), _jsx("div", { className: "absolute top-2 right-2", children: _jsxs(Popover, { children: [_jsx(PopoverTrigger, { asChild: true, children: _jsx(Button, { variant: "outline", size: "icon", className: "w-6 h-6", children: _jsx(CodeIcon, { className: "w-4 h-4" }) }) }), _jsx(PopoverContent, { className: "w-[400px] max-h-[300px] overflow-auto p-4", side: "right", align: "start", children: _jsx("pre", { className: "whitespace-no-wrap text-xs", children: JSON.stringify(toolResult, null, 2) }) })] }) }), _jsxs("div", { className: "flex flex-col gap-5", children: [args.reasoning && (_jsx("div", { className: "text-xs text-gray-400", children: args.reasoning })), args.sqlQuery && _jsx("div", { className: "font-mono", children: args.sqlQuery }), !result.success && (_jsxs("div", { className: "text-red-500 gap-2 flex flex-col", children: [_jsx("p", { className: "text-sm font-bold", children: "Oops! Something went wrong..." }), _jsx("p", { className: "text-xs", children: result.error })] }))] })] }));
+    return (_jsxs("div", { className: cn('border-2 relative bg-gray-100 dark:bg-gray-900 px-5 py-6 rounded-md text-gray-700 dark:text-gray-300 text-xs', getBorderColor(isSuccess, toolName)), children: [_jsxs(Badge, { variant: "secondary", className: cn('text-xs absolute top-[-12px] left-2 dark:text-gray-100 text-gray-700 flex items-center gap-1 border', getBorderColor(isSuccess, toolName)), children: [isSuccess ? (_jsx(CheckCircle2Icon, { className: "w-3 h-3 text-green-500" })) : (_jsx(XCircleIcon, { className: "w-3 h-3 text-red-500" })), toolName] }), _jsx("div", { className: "absolute top-2 right-2", children: _jsxs(Popover, { children: [_jsx(PopoverTrigger, { asChild: true, children: _jsx(Button, { variant: "outline", size: "icon", className: "w-6 h-6", children: _jsx(CodeIcon, { className: "w-4 h-4" }) }) }), _jsx(PopoverContent, { className: "w-[400px] max-h-[300px] overflow-auto p-4", side: "right", align: "start", children: _jsx("pre", { className: "whitespace-no-wrap text-xs", children: JSON.stringify(toolResult, null, 2) }) })] }) }), _jsxs("div", { className: "flex flex-col gap-5", children: [toolName === 'query' && isQueryToolParameters(args) ? (_jsx(QueryToolCall, { ...args, customMessage: customMessage })) : toolName === 'chart' && isChartToolParameters(args) ? (_jsx(ChartToolCall, { ...args })) : toolName === 'answer' && isAnalysisAnswer(result) ? (_jsx(AnalysisAnswer, { ...result })) : (Object.keys(args).length > 0 && (_jsx("pre", { children: JSON.stringify(args, null, 2) }))), !result.success && (_jsxs("div", { className: "text-red-500 gap-2 flex flex-col", children: [_jsx("p", { className: "text-sm font-bold", children: "Oops! Something went wrong..." }), _jsx("p", { className: "text-xs", children: result.error })] }))] })] }));
 };
 //# sourceMappingURL=ToolResult.js.map

package/dist/ToolResult.js.map

@@ -1 +1 @@
-{"version":3,"file":"ToolResult.js","sourceRoot":"","sources":["../src/ToolResult.tsx"],"names":[],"mappings":";AACA,OAAO,EACL,KAAK,EACL,MAAM,EACN,OAAO,EACP,cAAc,EACd,cAAc,EACd,EAAE,GACH,MAAM,cAAc,CAAC;AACtB,OAAO,EAAC,gBAAgB,EAAE,QAAQ,EAAE,WAAW,EAAC,MAAM,cAAc,CAAC;AAMrE,MAAM,CAAC,MAAM,UAAU,GAA8B,CAAC,EAAC,UAAU,EAAC,EAAE,EAAE;IACpE,MAAM,EAAC,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAC,GAAG,UAAU,CAAC;IAC5C,MAAM,SAAS,GAAG,MAAM,CAAC,OAAO,CAAC;IAEjC,OAAO,CACL,eACE,SAAS,EAAE,EAAE,CACX,8GAA8G,EAC9G,SAAS,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,gBAAgB,CAClD,aAED,MAAC,KAAK,IACJ,OAAO,EAAC,WAAW,EACnB,SAAS,EAAE,EAAE,CACX,qGAAqG,EACrG,SAAS,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,gBAAgB,CAClD,aAEA,SAAS,CAAC,CAAC,CAAC,CACX,KAAC,gBAAgB,IAAC,SAAS,EAAC,wBAAwB,GAAG,CACxD,CAAC,CAAC,CAAC,CACF,KAAC,WAAW,IAAC,SAAS,EAAC,sBAAsB,GAAG,CACjD,EACA,QAAQ,IACH,EAER,cAAK,SAAS,EAAC,wBAAwB,YACrC,MAAC,OAAO,eACN,KAAC,cAAc,IAAC,OAAO,kBACrB,KAAC,MAAM,IAAC,OAAO,EAAC,SAAS,EAAC,IAAI,EAAC,MAAM,EAAC,SAAS,EAAC,SAAS,YACvD,KAAC,QAAQ,IAAC,SAAS,EAAC,SAAS,GAAG,GACzB,GACM,EACjB,KAAC,cAAc,IACb,SAAS,EAAC,2CAA2C,EACrD,IAAI,EAAC,OAAO,EACZ,KAAK,EAAC,OAAO,YAEb,cAAK,SAAS,EAAC,4BAA4B,YACxC,IAAI,CAAC,SAAS,CAAC,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,GAChC,GACS,IACT,GACN,EAEN,eAAK,SAAS,EAAC,qBAAqB,aACjC,IAAI,CAAC,SAAS,IAAI,CACjB,cAAK,SAAS,EAAC,uBAAuB,YAAE,IAAI,CAAC,SAAS,GAAO,CAC9D,EACA,IAAI,CAAC,QAAQ,IAAI,cAAK,SAAS,EAAC,WAAW,YAAE,IAAI,CAAC,QAAQ,GAAO,EACjE,CAAC,MAAM,CAAC,OAAO,IAAI,CAClB,eAAK,SAAS,EAAC,kCAAkC,aAC/C,YAAG,SAAS,EAAC,mBAAmB,8CAAkC,EAClE,YAAG,SAAS,EAAC,SAAS,YAAE,MAAM,CAAC,KAAK,GAAK,IACrC,CACP,IACG,IACF,CACP,CAAC;AACJ,CAAC,CAAC"}
+{"version":3,"file":"ToolResult.js","sourceRoot":"","sources":["../src/ToolResult.tsx"],"names":[],"mappings":";AACA,OAAO,EACL,KAAK,EACL,MAAM,EACN,OAAO,EACP,cAAc,EACd,cAAc,EACd,EAAE,GACH,MAAM,cAAc,CAAC;AACtB,OAAO,EAAC,gBAAgB,EAAE,QAAQ,EAAE,WAAW,EAAC,MAAM,cAAc,CAAC;AAErE,OAAO,EACL,cAAc,EACd,aAAa,EACb,gBAAgB,EAChB,qBAAqB,GACtB,MAAM,YAAY,CAAC;AACpB,OAAO,EAAC,qBAAqB,EAAE,aAAa,EAAC,MAAM,YAAY,CAAC;AAQhE,SAAS,cAAc,CAAC,SAAkB,EAAE,QAAgB;IAC1D,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,OAAO,gBAAgB,CAAC;IAC1B,CAAC;IACD,QAAQ,QAAQ,EAAE,CAAC;QACjB,KAAK,OAAO;YACV,OAAO,kBAAkB,CAAC;QAC5B,KAAK,OAAO;YACV,OAAO,iBAAiB,CAAC;QAC3B,KAAK,QAAQ;YACX,OAAO,iBAAiB,CAAC;QAC3B;YACE,OAAO,iBAAiB,CAAC;IAC7B,CAAC;AACH,CAAC;AAED,MAAM,CAAC,MAAM,UAAU,GAA8B,CAAC,EACpD,UAAU,EACV,aAAa,GACd,EAAE,EAAE;IACH,MAAM,EAAC,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAC,GAAG,UAAU,CAAC;IAC5C,MAAM,SAAS,GAAG,MAAM,CAAC,OAAO,CAAC;IAEjC,OAAO,CACL,eACE,SAAS,EAAE,EAAE,CACX,8GAA8G,EAC9G,cAAc,CAAC,SAAS,EAAE,QAAQ,CAAC,CACpC,aAED,MAAC,KAAK,IACJ,OAAO,EAAC,WAAW,EACnB,SAAS,EAAE,EAAE,CACX,qGAAqG,EACrG,cAAc,CAAC,SAAS,EAAE,QAAQ,CAAC,CACpC,aAEA,SAAS,CAAC,CAAC,CAAC,CACX,KAAC,gBAAgB,IAAC,SAAS,EAAC,wBAAwB,GAAG,CACxD,CAAC,CAAC,CAAC,CACF,KAAC,WAAW,IAAC,SAAS,EAAC,sBAAsB,GAAG,CACjD,EACA,QAAQ,IACH,EAER,cAAK,SAAS,EAAC,wBAAwB,YACrC,MAAC,OAAO,eACN,KAAC,cAAc,IAAC,OAAO,kBACrB,KAAC,MAAM,IAAC,OAAO,EAAC,SAAS,EAAC,IAAI,EAAC,MAAM,EAAC,SAAS,EAAC,SAAS,YACvD,KAAC,QAAQ,IAAC,SAAS,EAAC,SAAS,GAAG,GACzB,GACM,EACjB,KAAC,cAAc,IACb,SAAS,EAAC,2CAA2C,EACrD,IAAI,EAAC,OAAO,EACZ,KAAK,EAAC,OAAO,YAEb,cAAK,SAAS,EAAC,4BAA4B,YACxC,IAAI,CAAC,SAAS,CAAC,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,GAChC,GACS,IACT,GACN,EAEN,eAAK,SAAS,EAAC,qBAAqB,aACjC,QAAQ,KAAK,OAAO,IAAI,qBAAqB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CACrD,KAAC,aAAa,OAAK,IAAI,EAAE,aAAa,EAAE,aAAa,GAAI,CAC1D,CAAC,CAAC,CAAC,QAAQ,KAAK,OAAO,IAAI,qBAAqB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CACxD,KAAC,aAAa,OAAK,IAAI,GAAI,CAC5B,CAAC,CAAC,CAAC,QAAQ,KAAK,QAAQ,IAAI,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CACtD,KAAC,cAAc,OAAK,MAAM,GAAI,CAC/B,CAAC,CAAC,CAAC,CACF,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC,IAAI,CAC9B,wBAAM,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,GAAO,CAC3C,CACF,EACA,CAAC,MAAM,CAAC,OAAO,IAAI,CAClB,eAAK,SAAS,EAAC,kCAAkC,aAC/C,YAAG,SAAS,EAAC,mBAAmB,8CAAkC,EAClE,YAAG,SAAS,EAAC,SAAS,YAAE,MAAM,CAAC,KAAK,GAAK,IACrC,CACP,IACG,IACF,CACP,CAAC;AACJ,CAAC,CAAC","sourcesContent":["import {ToolResultSchema} from './schemas';\nimport {\n  Badge,\n  Button,\n  Popover,\n  PopoverContent,\n  PopoverTrigger,\n  cn,\n} from '@sqlrooms/ui';\nimport {CheckCircle2Icon, CodeIcon, XCircleIcon} from 'lucide-react';\nimport {ReactNode} from 'react';\nimport {\n  AnalysisAnswer,\n  ChartToolCall,\n  isAnalysisAnswer,\n  isChartToolParameters,\n} from './ToolCall';\nimport {isQueryToolParameters, QueryToolCall} from './ToolCall';\nimport React from 'react';\n\ninterface ToolResultProps {\n  toolResult: ToolResultSchema;\n  customMessage?: ReactNode;\n}\n\nfunction getBorderColor(isSuccess: boolean, toolName: string) {\n  if (!isSuccess) {\n    return 'border-red-500';\n  }\n  switch (toolName) {\n    case 'query':\n      return 'border-green-500';\n    case 'chart':\n      return 'border-blue-500';\n    case 'answer':\n      return 'border-gray-500';\n    default:\n      return 'border-gray-500';\n  }\n}\n\nexport const ToolResult: React.FC<ToolResultProps> = ({\n  toolResult,\n  customMessage,\n}) => {\n  const {toolName, args, result} = toolResult;\n  const isSuccess = result.success;\n\n  return (\n    <div\n      className={cn(\n        'border-2 relative bg-gray-100 dark:bg-gray-900 px-5 py-6 rounded-md text-gray-700 dark:text-gray-300 text-xs',\n        getBorderColor(isSuccess, toolName),\n      )}\n    >\n      <Badge\n        variant=\"secondary\"\n        className={cn(\n          'text-xs absolute top-[-12px] left-2 dark:text-gray-100 text-gray-700 flex items-center gap-1 border',\n          getBorderColor(isSuccess, toolName),\n        )}\n      >\n        {isSuccess ? (\n          <CheckCircle2Icon className=\"w-3 h-3 text-green-500\" />\n        ) : (\n          <XCircleIcon className=\"w-3 h-3 text-red-500\" />\n        )}\n        {toolName}\n      </Badge>\n\n      <div className=\"absolute top-2 right-2\">\n        <Popover>\n          <PopoverTrigger asChild>\n            <Button variant=\"outline\" size=\"icon\" className=\"w-6 h-6\">\n              <CodeIcon className=\"w-4 h-4\" />\n            </Button>\n          </PopoverTrigger>\n          <PopoverContent\n            className=\"w-[400px] max-h-[300px] overflow-auto p-4\"\n            side=\"right\"\n            align=\"start\"\n          >\n            <pre className=\"whitespace-no-wrap text-xs\">\n              {JSON.stringify(toolResult, null, 2)}\n            </pre>\n          </PopoverContent>\n        </Popover>\n      </div>\n\n      <div className=\"flex flex-col gap-5\">\n        {toolName === 'query' && isQueryToolParameters(args) ? (\n          <QueryToolCall {...args} customMessage={customMessage} />\n        ) : toolName === 'chart' && isChartToolParameters(args) ? (\n          <ChartToolCall {...args} />\n        ) : toolName === 'answer' && isAnalysisAnswer(result) ? (\n          <AnalysisAnswer {...result} />\n        ) : (\n          Object.keys(args).length > 0 && (\n            <pre>{JSON.stringify(args, null, 2)}</pre>\n          )\n        )}\n        {!result.success && (\n          <div className=\"text-red-500 gap-2 flex flex-col\">\n            <p className=\"text-sm font-bold\">Oops! Something went wrong...</p>\n            <p className=\"text-xs\">{result.error}</p>\n          </div>\n        )}\n      </div>\n    </div>\n  );\n};\n"]}

package/dist/analysis.d.ts

@@ -1,180 +1,54 @@
-import type { LanguageModelV1 } from '@ai-sdk/provider';
-import { StepResult, ToolExecutionOptions } from 'ai';
+import { StepResult } from 'ai';
+import { ToolCallMessage, VercelToolSet } from '@openassistant/core';
 /**
- * Run analysis on the project data
- * @param prompt - The prompt for the analysis
- * @param abortSignal - An optional abort signal to cancel the analysis
- * @returns The tool calls and the final answer
+ * Configuration options for running an AI analysis session
  */
-export declare function runAnalysis({ model, abortSignal, onStepFinish, maxSteps, messages, }: {
-    abortSignal?: AbortSignal;
-    onStepFinish?: (event: StepResult<typeof TOOLS>) => Promise<void> | void;
-    model: LanguageModelV1;
+export type AnalysisConfig = {
+    /** Assistant instance identifier (default: 'sqlrooms-ai') */
+    name?: string;
+    /** AI model provider (e.g., 'openai', 'anthropic') */
+    modelProvider: string;
+    /** Model identifier (e.g., 'gpt-4', 'claude-3') */
+    model: string;
+    /** Authentication key for the model provider's API */
+    apiKey: string;
+    /** Analysis prompt or question to be processed */
+    prompt: string;
+    /** Optional controller for canceling the analysis operation */
+    abortController?: AbortController;
+    /**
+     * Callback fired after each analysis step completion
+     * @param event - Current step result containing tool execution details. See Vercel AI SDK documentation for more details.
+     * Specifically, it contains the array of tool calls and the results of the tool calls (toolResults).
+     * @param toolCallMessages - Collection of messages generated during tool calls. They are linked to the tool call by the toolCallId.
+     */
+    onStepFinish?: (event: StepResult<typeof TOOLS>, toolCallMessages: ToolCallMessage[]) => Promise<void> | void;
+    /** Maximum number of analysis steps allowed (default: 100) */
     maxSteps?: number;
-    messages?: any[];
-}): Promise<import("ai").GenerateTextResult<{
-    query: import("ai").Tool<import("zod").ZodObject<{
-        type: import("zod").ZodLiteral<"query">;
-        sqlQuery: import("zod").ZodString;
-        reasoning: import("zod").ZodString;
-    }, "strip", import("zod").ZodTypeAny, {
-        type: "query";
-        sqlQuery: string;
-        reasoning: string;
-    }, {
-        type: "query";
-        sqlQuery: string;
-        reasoning: string;
-    }>, {
-        success: false;
-        error: string;
-    } | {
-        success: true;
-        data: Record<string, any>;
-    }> & {
-        execute: (args: {
-            type: "query";
-            sqlQuery: string;
-            reasoning: string;
-        }, options: ToolExecutionOptions) => PromiseLike<{
-            success: false;
-            error: string;
-        } | {
-            success: true;
-            data: Record<string, any>;
-        }>;
-    };
-    answer: import("ai").Tool<import("zod").ZodObject<{
-        type: import("zod").ZodLiteral<"answer">;
-        answer: import("zod").ZodString;
-        chart: import("zod").ZodUnion<[import("zod").ZodObject<{
-            sqlQuery: import("zod").ZodString;
-            vegaLiteSpec: import("zod").ZodString;
-        }, "strip", import("zod").ZodTypeAny, {
-            sqlQuery: string;
-            vegaLiteSpec: string;
-        }, {
-            sqlQuery: string;
-            vegaLiteSpec: string;
-        }>, import("zod").ZodNull]>;
-    }, "strip", import("zod").ZodTypeAny, {
-        type: "answer";
-        answer: string;
-        chart: {
-            sqlQuery: string;
-            vegaLiteSpec: string;
-        } | null;
-    }, {
-        type: "answer";
-        answer: string;
-        chart: {
-            sqlQuery: string;
-            vegaLiteSpec: string;
-        } | null;
-    }>, {
-        success: false;
-        error: string;
-    } | {
-        success: true;
-        data: Record<string, any>;
-    }> & {
-        execute: (args: {
-            type: "answer";
-            answer: string;
-            chart: {
-                sqlQuery: string;
-                vegaLiteSpec: string;
-            } | null;
-        }, options: ToolExecutionOptions) => PromiseLike<{
-            success: false;
-            error: string;
-        } | {
-            success: true;
-            data: Record<string, any>;
-        }>;
-    };
-}, never>>;
-declare const TOOLS: {
-    query: import("ai").Tool<import("zod").ZodObject<{
-        type: import("zod").ZodLiteral<"query">;
-        sqlQuery: import("zod").ZodString;
-        reasoning: import("zod").ZodString;
-    }, "strip", import("zod").ZodTypeAny, {
-        type: "query";
-        sqlQuery: string;
-        reasoning: string;
-    }, {
-        type: "query";
-        sqlQuery: string;
-        reasoning: string;
-    }>, {
-        success: false;
-        error: string;
-    } | {
-        success: true;
-        data: Record<string, any>;
-    }> & {
-        execute: (args: {
-            type: "query";
-            sqlQuery: string;
-            reasoning: string;
-        }, options: ToolExecutionOptions) => PromiseLike<{
-            success: false;
-            error: string;
-        } | {
-            success: true;
-            data: Record<string, any>;
-        }>;
-    };
-    answer: import("ai").Tool<import("zod").ZodObject<{
-        type: import("zod").ZodLiteral<"answer">;
-        answer: import("zod").ZodString;
-        chart: import("zod").ZodUnion<[import("zod").ZodObject<{
-            sqlQuery: import("zod").ZodString;
-            vegaLiteSpec: import("zod").ZodString;
-        }, "strip", import("zod").ZodTypeAny, {
-            sqlQuery: string;
-            vegaLiteSpec: string;
-        }, {
-            sqlQuery: string;
-            vegaLiteSpec: string;
-        }>, import("zod").ZodNull]>;
-    }, "strip", import("zod").ZodTypeAny, {
-        type: "answer";
-        answer: string;
-        chart: {
-            sqlQuery: string;
-            vegaLiteSpec: string;
-        } | null;
-    }, {
-        type: "answer";
-        answer: string;
-        chart: {
-            sqlQuery: string;
-            vegaLiteSpec: string;
-        } | null;
-    }>, {
-        success: false;
-        error: string;
-    } | {
-        success: true;
-        data: Record<string, any>;
-    }> & {
-        execute: (args: {
-            type: "answer";
-            answer: string;
-            chart: {
-                sqlQuery: string;
-                vegaLiteSpec: string;
-            } | null;
-        }, options: ToolExecutionOptions) => PromiseLike<{
-            success: false;
-            error: string;
-        } | {
-            success: true;
-            data: Record<string, any>;
-        }>;
-    };
+    /**
+     * Callback for handling streaming results
+     * @param message - Current message content being streamed
+     * @param isCompleted - Indicates if this is the final message in the stream
+     */
+    onStreamResult: (message: string, isCompleted: boolean) => void;
 };
+/**
+ * Executes an AI analysis session on the project data
+ *
+ * @param config - Analysis configuration options. See {@link AnalysisConfig} for more details.
+ * @returns Object containing tool calls executed and the final analysis result
+ */
+export declare function runAnalysis({ name, modelProvider, model, apiKey, prompt, abortController, onStepFinish, onStreamResult, maxSteps, }: AnalysisConfig): Promise<{
+    messages: import("ai").Message[];
+    outputToolResults: never[][] | undefined;
+    outputToolCalls: import("ai").ToolCall<string, unknown>[] | undefined;
+}>;
+/**
+ * Collection of tools available to the AI assistant for data analysis
+ * Includes:
+ * - query: Executes SQL queries against DuckDB
+ * - chart: Creates VegaLite visualizations
+ */
+declare const TOOLS: VercelToolSet;
 export {};
 //# sourceMappingURL=analysis.d.ts.map

package/dist/analysis.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"analysis.d.ts","sourceRoot":"","sources":["../src/analysis.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAC,eAAe,EAAC,MAAM,kBAAkB,CAAC;AAEtD,OAAO,EAAe,UAAU,EAAQ,oBAAoB,EAAC,MAAM,IAAI,CAAC;AAOxE;;;;;GAKG;AACH,wBAAsB,WAAW,CAAC,EAChC,KAAK,EAEL,WAAW,EACX,YAAY,EACZ,QAAc,EACd,QAAQ,GACT,EAAE;IAED,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B,YAAY,CAAC,EAAE,CAAC,KAAK,EAAE,UAAU,CAAC,OAAO,KAAK,CAAC,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IACzE,KAAK,EAAE,eAAe,CAAC;IACvB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,GAAG,EAAE,CAAC;CAClB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;WA+BA;AAED,QAAA,MAAM,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA0DV,CAAC"}
+{"version":3,"file":"analysis.d.ts","sourceRoot":"","sources":["../src/analysis.ts"],"names":[],"mappings":"AAKA,OAAO,EAAC,UAAU,EAAC,MAAM,IAAI,CAAC;AAE9B,OAAO,EAGL,eAAe,EACf,aAAa,EACd,MAAM,qBAAqB,CAAC;AAsF7B;;GAEG;AACH,MAAM,MAAM,cAAc,GAAG;IAC3B,6DAA6D;IAC7D,IAAI,CAAC,EAAE,MAAM,CAAC;IAEd,sDAAsD;IACtD,aAAa,EAAE,MAAM,CAAC;IAEtB,mDAAmD;IACnD,KAAK,EAAE,MAAM,CAAC;IAEd,sDAAsD;IACtD,MAAM,EAAE,MAAM,CAAC;IAEf,kDAAkD;IAClD,MAAM,EAAE,MAAM,CAAC;IAEf,+DAA+D;IAC/D,eAAe,CAAC,EAAE,eAAe,CAAC;IAElC;;;;;OAKG;IACH,YAAY,CAAC,EAAE,CACb,KAAK,EAAE,UAAU,CAAC,OAAO,KAAK,CAAC,EAC/B,gBAAgB,EAAE,eAAe,EAAE,KAChC,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;IAE1B,8DAA8D;IAC9D,QAAQ,CAAC,EAAE,MAAM,CAAC;IAElB;;;;OAIG;IACH,cAAc,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,OAAO,KAAK,IAAI,CAAC;CACjE,CAAC;AAEF;;;;;GAKG;AACH,wBAAsB,WAAW,CAAC,EAChC,IAAoB,EACpB,aAAa,EACb,KAAK,EACL,MAAM,EACN,MAAM,EACN,eAAe,EACf,YAAY,EACZ,cAAc,EACd,QAAY,GACb,EAAE,cAAc;;;;GA6BhB;AAiBD;;;;;GAKG;AACH,QAAA,MAAM,KAAK,EAAE,aAwHZ,CAAC"}