@kwirthmagnify/kwirth-common-ai 0.5.6 → 0.5.8

package/dist/front.js

@@ -60,7 +60,10 @@ const AiConfigLlm = (props) => {
     const [temperature, setTemperature] = (0, react_1.useState)(0);
     const [useProviderKey, setUseProviderKey] = (0, react_1.useState)(true);
     const [key, setKey] = (0, react_1.useState)('');
+    const [inputCostPerMillion, setInputCostPerMillion] = (0, react_1.useState)(0);
+    const [outputCostPerMillion, setOutputCostPerMillion] = (0, react_1.useState)(0);
     const onLlmSelected = (index) => {
+        var _a, _b;
         const l = llms[index];
         if (l) {
             setId(l.id);
@@ -69,6 +72,8 @@ const AiConfigLlm = (props) => {
             setTemperature(l.temperature);
             setUseProviderKey(l.useProviderKey);
             setKey(l.key);
+            setInputCostPerMillion((_a = l.inputCostPerMillion) !== null && _a !== void 0 ? _a : 0);
+            setOutputCostPerMillion((_b = l.outputCostPerMillion) !== null && _b !== void 0 ? _b : 0);
             setSelectedIndex(index);
         }
     };
@@ -80,9 +85,11 @@ const AiConfigLlm = (props) => {
         setTemperature(0);
         setUseProviderKey(false);
         setKey('');
+        setInputCostPerMillion(0);
+        setOutputCostPerMillion(0);
     };
     const onAdd = () => {
-        const llm = { id, provider, model, temperature, useProviderKey, key };
+        const llm = { id, provider, model, temperature, useProviderKey, key, inputCostPerMillion: inputCostPerMillion === '' ? 0 : inputCostPerMillion, outputCostPerMillion: outputCostPerMillion === '' ? 0 : outputCostPerMillion };
         const updated = [...llms];
         if (selectedIndex !== null)
             updated[selectedIndex] = llm;
@@ -97,7 +104,7 @@ const AiConfigLlm = (props) => {
         setLlms(llms.filter((_, i) => i !== selectedIndex));
         onNew();
     };
-    return (react_1.default.createElement(material_1.Dialog, { open: true, onClose: () => props.onClose(undefined), PaperProps: { sx: { width: '80vw', maxWidth: '800px', height: '55vh' } } },
+    return (react_1.default.createElement(material_1.Dialog, { open: true, onClose: () => props.onClose(undefined), PaperProps: { sx: { width: '80vw', maxWidth: '800px', height: '78vh' } } },
         react_1.default.createElement(material_1.DialogTitle, null, "AI \u2014 LLM config"),
         react_1.default.createElement(material_1.DialogContent, { style: { display: 'flex', height: '100%' } },
             react_1.default.createElement(material_1.Box, { sx: { flex: 1, display: 'flex', flexDirection: 'column', boxSizing: 'border-box', maxWidth: '40%' } },
@@ -117,6 +124,9 @@ const AiConfigLlm = (props) => {
                             react_1.default.createElement(material_1.InputLabel, null, "Model"),
                             react_1.default.createElement(material_1.Select, { value: model, onChange: e => setModel(e.target.value), variant: 'standard', fullWidth: true, displayEmpty: true }, (_a = props.providers.find(p => p.name === provider)) === null || _a === void 0 ? void 0 : _a.models.map((m, i) => (react_1.default.createElement(material_1.MenuItem, { key: i, value: m.id }, m.name))))),
                         react_1.default.createElement(material_1.TextField, { value: temperature, onChange: e => setTemperature(+e.target.value), label: 'Model temperature', variant: 'standard', type: 'number', fullWidth: true }),
+                        react_1.default.createElement(material_1.Stack, { direction: 'row', spacing: 1 },
+                            react_1.default.createElement(material_1.TextField, { value: inputCostPerMillion, onChange: e => setInputCostPerMillion(e.target.value === '' ? '' : +e.target.value), label: 'Input cost / M tokens (\u20AC/$)', variant: 'standard', type: 'number', fullWidth: true, inputProps: { min: 0, step: 0.01 } }),
+                            react_1.default.createElement(material_1.TextField, { value: outputCostPerMillion, onChange: e => setOutputCostPerMillion(e.target.value === '' ? '' : +e.target.value), label: 'Output cost / M tokens (\u20AC/$)', variant: 'standard', type: 'number', fullWidth: true, inputProps: { min: 0, step: 0.01 } })),
                         react_1.default.createElement(material_1.Stack, { direction: 'row', alignItems: 'center' },
                             react_1.default.createElement(material_1.Typography, { flex: 1 }, "Use provider API Key (or enter a specific one)"),
                             react_1.default.createElement(material_1.Checkbox, { checked: useProviderKey, onChange: e => setUseProviderKey(e.target.checked) })),

package/dist/index.d.ts

@@ -19,5 +19,7 @@ export interface ILlm {
     temperature: number;
     useProviderKey: boolean;
     key: string;
+    inputCostPerMillion?: number;
+    outputCostPerMillion?: number;
     data?: unknown;
 }

package/package.json

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