@kwirthmagnify/kwirth-common-ai 0.5.10 → 0.5.12

package/dist/back.d.ts

@@ -4,6 +4,6 @@ import { LanguageModel } from 'ai';
 import { z } from 'zod';
 export declare const buildModel: (llm: ILlm, providers: ILlmProvider[]) => LanguageModel | null;
 export declare const loadModels: (providers: ILlmProvider[], log: IBackChannelObject) => Promise<void>;
-export { generateText, Output, stepCountIs } from 'ai';
+export { generateText, Output, stepCountIs, tool } from 'ai';
 export { z } from 'zod';
 export declare const zodFromExample: (example: Record<string, unknown>) => z.ZodObject<Record<string, z.ZodTypeAny>>;

package/dist/back.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.zodFromExample = exports.z = exports.stepCountIs = exports.Output = exports.generateText = exports.loadModels = exports.buildModel = void 0;
+exports.zodFromExample = exports.z = exports.tool = exports.stepCountIs = exports.Output = exports.generateText = exports.loadModels = exports.buildModel = void 0;
 const zod_1 = require("zod");
 const openai_1 = require("@ai-sdk/openai");
 const groq_1 = require("@ai-sdk/groq");
@@ -124,6 +124,7 @@ var ai_1 = require("ai");
 Object.defineProperty(exports, "generateText", { enumerable: true, get: function () { return ai_1.generateText; } });
 Object.defineProperty(exports, "Output", { enumerable: true, get: function () { return ai_1.Output; } });
 Object.defineProperty(exports, "stepCountIs", { enumerable: true, get: function () { return ai_1.stepCountIs; } });
+Object.defineProperty(exports, "tool", { enumerable: true, get: function () { return ai_1.tool; } });
 var zod_2 = require("zod");
 Object.defineProperty(exports, "z", { enumerable: true, get: function () { return zod_2.z; } });
 const zodFromExample = (example) => {

package/dist/front.js

@@ -37,6 +37,24 @@ exports.AiConfigProvider = exports.AiConfigLlm = exports.LlmSelector = void 0;
 const react_1 = __importStar(require("react"));
 const material_1 = require("@mui/material");
 const icons_material_1 = require("@mui/icons-material");
+const downloadJson = async (data, filename) => {
+    var _a, _b;
+    const json = JSON.stringify(data, null, 2);
+    const tauri = window.__TAURI__;
+    if (((_a = tauri === null || tauri === void 0 ? void 0 : tauri.dialog) === null || _a === void 0 ? void 0 : _a.save) && ((_b = tauri === null || tauri === void 0 ? void 0 : tauri.fs) === null || _b === void 0 ? void 0 : _b.writeTextFile)) {
+        const path = await tauri.dialog.save({ defaultPath: filename, filters: [{ name: 'JSON', extensions: ['json'] }] });
+        if (path)
+            await tauri.fs.writeTextFile(path, json);
+        return;
+    }
+    const blob = new Blob([json], { type: 'application/json' });
+    const url = URL.createObjectURL(blob);
+    const a = document.createElement('a');
+    a.href = url;
+    a.download = filename;
+    a.click();
+    URL.revokeObjectURL(url);
+};
 const LlmSelector = ({ llms, value, onChange, label = 'LLM', size = 'small', fullWidth = true }) => {
     return (react_1.default.createElement(material_1.FormControl, { size: size, fullWidth: fullWidth },
         react_1.default.createElement(material_1.InputLabel, null, label),
@@ -52,6 +70,7 @@ exports.LlmSelector = LlmSelector;
 const AiConfigLlm = (props) => {
     var _a;
     const [llms, setLlms] = (0, react_1.useState)(JSON.parse(JSON.stringify(props.llms)));
+    const importLlmRef = (0, react_1.useRef)(null);
     const [selectedIndex, setSelectedIndex] = (0, react_1.useState)(null);
     const [showPassword, setShowPassword] = (0, react_1.useState)(false);
     const [id, setId] = (0, react_1.useState)('');
@@ -140,6 +159,22 @@ const AiConfigLlm = (props) => {
                         react_1.default.createElement(material_1.Button, { color: 'error', onClick: onRemove, disabled: selectedIndex === null }, "Remove"),
                         react_1.default.createElement(material_1.Button, { variant: 'contained', onClick: onAdd, disabled: !id || !model }, selectedIndex !== null ? 'Update' : 'Add'))))),
         react_1.default.createElement(material_1.DialogActions, null,
+            react_1.default.createElement("input", { ref: importLlmRef, type: 'file', accept: '.json', style: { display: 'none' }, onChange: e => {
+                    var _a;
+                    const f = (_a = e.target.files) === null || _a === void 0 ? void 0 : _a[0];
+                    if (!f)
+                        return;
+                    const reader = new FileReader();
+                    reader.onload = ev => { try {
+                        setLlms(JSON.parse(ev.target.result));
+                    }
+                    catch (_a) { } };
+                    reader.readAsText(f);
+                    e.target.value = '';
+                } }),
+            react_1.default.createElement(material_1.Button, { startIcon: react_1.default.createElement(icons_material_1.FileUpload, { fontSize: 'small' }), onClick: () => { var _a; return (_a = importLlmRef.current) === null || _a === void 0 ? void 0 : _a.click(); } }, "Import"),
+            react_1.default.createElement(material_1.Button, { startIcon: react_1.default.createElement(icons_material_1.FileDownload, { fontSize: 'small' }), onClick: () => downloadJson(llms, 'kwirth-llms.json') }, "Export"),
+            react_1.default.createElement(material_1.Box, { flex: 1 }),
             react_1.default.createElement(material_1.Button, { onClick: () => props.onClose(llms), variant: 'contained' }, "OK"),
             react_1.default.createElement(material_1.Button, { onClick: () => props.onClose(undefined), color: 'inherit' }, "Cancel"))));
 };
@@ -147,6 +182,7 @@ exports.AiConfigLlm = AiConfigLlm;
 const AiConfigProvider = (props) => {
     var _a;
     const [providers, setProviders] = (0, react_1.useState)(JSON.parse(JSON.stringify(props.providers)));
+    const importProvRef = (0, react_1.useRef)(null);
     const [selectedIndex, setSelectedIndex] = (0, react_1.useState)(null);
     const [showPassword, setShowPassword] = (0, react_1.useState)(false);
     const [providerName, setProviderName] = (0, react_1.useState)('');
@@ -204,6 +240,22 @@ const AiConfigProvider = (props) => {
                         react_1.default.createElement(material_1.Button, { variant: 'text', color: 'error', onClick: onRemove, disabled: selectedIndex === null }, "Remove"),
                         react_1.default.createElement(material_1.Button, { variant: 'contained', onClick: onAdd, disabled: !providerName }, selectedIndex !== null ? 'Update' : 'Add'))))),
         react_1.default.createElement(material_1.DialogActions, { sx: { p: 2 } },
+            react_1.default.createElement("input", { ref: importProvRef, type: 'file', accept: '.json', style: { display: 'none' }, onChange: e => {
+                    var _a;
+                    const f = (_a = e.target.files) === null || _a === void 0 ? void 0 : _a[0];
+                    if (!f)
+                        return;
+                    const reader = new FileReader();
+                    reader.onload = ev => { try {
+                        setProviders(JSON.parse(ev.target.result));
+                    }
+                    catch (_a) { } };
+                    reader.readAsText(f);
+                    e.target.value = '';
+                } }),
+            react_1.default.createElement(material_1.Button, { startIcon: react_1.default.createElement(icons_material_1.FileUpload, { fontSize: 'small' }), onClick: () => { var _a; return (_a = importProvRef.current) === null || _a === void 0 ? void 0 : _a.click(); } }, "Import"),
+            react_1.default.createElement(material_1.Button, { startIcon: react_1.default.createElement(icons_material_1.FileDownload, { fontSize: 'small' }), onClick: () => downloadJson(providers.map(p => ({ name: p.name, key: p.key })), 'kwirth-providers.json') }, "Export"),
+            react_1.default.createElement(material_1.Box, { flex: 1 }),
             react_1.default.createElement(material_1.Button, { onClick: () => props.onClose(providers), color: 'primary', variant: 'contained' }, "Save"),
             react_1.default.createElement(material_1.Button, { onClick: () => props.onClose(undefined), color: 'inherit' }, "Cancel"))));
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kwirthmagnify/kwirth-common-ai",
-  "version": "0.5.10",
+  "version": "0.5.12",
   "description": "Shared AI/LLM types and utilities for Kwirth AI plugins",
   "scripts": {
     "build": "del .\\dist\\* /s /q 2>nul & tsc"