+
- 1
+ 2
Input your data in text format
-
spec:
-
- {/*
{JSON.stringify(props.spec, null, 4)}*/}
+ {props.insights.map((insight, index) => {
+ const { plainText = '' } = insight?.textContent || {};
+ return
{plainText}
;
+ })}
) : null}
diff --git a/packages/vmind/__tests__/browser/src/pages/Insight/DataInput.tsx b/packages/vmind/__tests__/browser/src/pages/Insight/DataInput.tsx
index 8b89b781..d995cdc7 100644
--- a/packages/vmind/__tests__/browser/src/pages/Insight/DataInput.tsx
+++ b/packages/vmind/__tests__/browser/src/pages/Insight/DataInput.tsx
@@ -1,20 +1,8 @@
/* eslint-disable no-console */
-import React, { useState, useCallback, useMemo } from 'react';
-import './index.scss';
-import {
- Avatar,
- Input,
- Divider,
- Button,
- InputNumber,
- Upload,
- Message,
- Select,
- Radio,
- Checkbox,
- Modal
-} from '@arco-design/web-react';
-import VMind, { ArcoTheme } from '../../../../../src/index';
+import React, { useState, useCallback, useEffect } from 'react';
+import '../index.scss';
+import { Avatar, Input, Divider, Button, InputNumber, Select, Radio, Modal } from '@arco-design/web-react';
+import VMind from '../../../../../src/index';
import { Model } from '../../../../../src/index';
import {
ChangePointChart,
@@ -26,18 +14,19 @@ import {
SalesLineChart,
SalesLineChart2,
SalesLineChart3,
+ SalesScatterChart,
ScatterClusterChart,
- ScatterPlotChart
+ ScatterPlotChart,
+ ScatterIrisData
} from './data';
import JSON5 from 'json5';
-import { InsightLanguage } from '../../../../../src/applications/types';
const TextArea = Input.TextArea;
const Option = Select.Option;
const RadioGroup = Radio.Group;
type IPropsType = {
- onInsightGenerate: (insights: any, costTime: number) => void;
+ onInsightGenerate: (insights: any, spec: any, costTime: number) => void;
onSpecChange: (spec: any) => void;
};
const demoDataList: { [key: string]: any } = {
@@ -51,68 +40,81 @@ const demoDataList: { [key: string]: any } = {
ChangePointChart: ChangePointChart,
MultiLineChart2: MultiLineChart2,
ScatterPlotChart: ScatterPlotChart,
- ScatterClusterChart: ScatterClusterChart
+ ScatterClusterChart: ScatterClusterChart,
+ ScatterSalesChart: SalesScatterChart,
+ ScatterIrisData: ScatterIrisData
};
const globalVariables = (import.meta as any).env;
-const ModelConfigMap: any = {
- [Model.SKYLARK2]: { url: globalVariables.VITE_SKYLARK_URL, key: globalVariables.VITE_SKYLARK_KEY },
- [Model.SKYLARK2_v1_2]: { url: globalVariables.VITE_SKYLARK_URL, key: globalVariables.VITE_SKYLARK_KEY },
- [Model.GPT3_5]: { url: globalVariables.VITE_GPT_URL, key: globalVariables.VITE_GPT_KEY },
- [Model.GPT4]: { url: globalVariables.VITE_GPT_URL, key: globalVariables.VITE_GPT_KEY }
+const ModelConfigMap: Record
= {
+ [Model.DOUBAO_PRO]: { url: globalVariables.VITE_DOUBAO_URL, key: globalVariables.VITE_DOUBAO_KEY },
+ [Model.GPT4]: { url: globalVariables.VITE_GPT_URL, key: globalVariables.VITE_GPT_KEY },
+ [Model.GPT_4o]: { url: globalVariables.VITE_GPT_URL, key: globalVariables.VITE_GPT_KEY },
+ [Model.DEEPSEEK_R1]: { url: globalVariables.VITE_DEEPSEEK_URL, key: globalVariables.VITE_DEEPSEEK_KEY },
+ [Model.DEEPSEEK_V3]: { url: globalVariables.VITE_DEEPSEEK_URL, key: globalVariables.VITE_DEEPSEEK_KEY },
+ Custom: { url: globalVariables.VITE_CUSTOM_URL, key: globalVariables.VITE_CUSTOM_KEY }
};
const OPENAI_API_URL = 'https://api.openai.com/v1/chat/completions';
export function DataInput(props: IPropsType) {
const defaultDataKey = Object.keys(demoDataList)[0];
const [spec, setSpec] = useState(JSON.stringify(demoDataList[defaultDataKey].spec));
- const [fieldInfo, setFieldInfo] = useState(demoDataList[defaultDataKey].fieldInfo);
//const [spec, setSpec] = useState('');
//const [time, setTime] = useState(1000);
- const [model, setModel] = useState(Model.GPT3_5);
- const [cache, setCache] = useState(true);
- const [showThoughts, setShowThoughts] = useState(false);
+ const [model, setModel] = useState(Model.GPT_4o);
+ const [numLimits, setNumLimits] = useState(8);
const [visible, setVisible] = React.useState(false);
const [url, setUrl] = React.useState(ModelConfigMap[model]?.url ?? OPENAI_API_URL);
const [apiKey, setApiKey] = React.useState(ModelConfigMap[model]?.key);
const [loading, setLoading] = useState(false);
- const vmind: VMind = useMemo(() => {
- if (!url || !apiKey) {
- Message.error('Please set your LLM URL and API Key!!!');
- return null as unknown as VMind;
- }
- return new VMind({
+ const vmind = React.useRef(
+ new VMind({
url,
+ headers: {
+ 'api-key': apiKey,
+ Authorization: `Bearer ${apiKey}`
+ },
model,
- cache,
- showThoughts: showThoughts,
+ maxTokens: 2048
+ })
+ );
+ useEffect(() => {
+ vmind.current.updateOptions({
+ url,
headers: {
- 'api-key': apiKey
- }
+ 'api-key': apiKey,
+ Authorization: `Bearer ${apiKey}`
+ },
+ model
});
- }, [apiKey, cache, model, showThoughts, url]);
+ }, [url, model, apiKey]);
const getInsight = useCallback(async () => {
const startTime = new Date().getTime();
const specJson = JSON5.parse(spec);
- const insights = await vmind.intelligentInsight(specJson, {
- //insightNumberLimit: 2,
- generateText: true
- //language: InsightLanguage.EN
+ const { insights } = await vmind.current.getInsights(specJson, {
+ maxNum: numLimits,
+ detailMaxNum: [
+ { types: ['outlier', 'pair_outlier', 'extreme_value', 'turning_point', 'majority_value'], maxNum: 3 },
+ { types: ['abnormal_band'], maxNum: 3 },
+ { types: ['correlation'], maxNum: 2 },
+ { types: ['overall_trend'], maxNum: 2 },
+ { types: ['abnormal_trend'], maxNum: 3 }
+ ] as any
});
const endTime = new Date().getTime();
const costTime = endTime - startTime;
- props.onInsightGenerate(insights, costTime);
+ props.onInsightGenerate(insights, specJson, costTime);
console.log(costTime);
console.log(insights);
setLoading(false);
- }, [props, spec, vmind]);
+ }, [numLimits, props, spec]);
return (
@@ -143,7 +145,6 @@ export function DataInput(props: IPropsType) {
const dataObj = demoDataList[v];
setSpec(JSON.stringify(dataObj.spec));
props.onSpecChange(dataObj.spec);
- setFieldInfo(dataObj.fieldInfo);
}}
>
{Object.keys(demoDataList).map(name => (
@@ -154,57 +155,62 @@ export function DataInput(props: IPropsType) {
-
-
-
- 2
-
- Input your spec
-
-
+ {/*
{JSON.stringify(props.spec, null, 4)}*/}
+
+ );
+}
diff --git a/packages/vmind/__tests__/browser/src/pages/NewChartGeneration/DataInput.tsx b/packages/vmind/__tests__/browser/src/pages/NewChartGeneration/DataInput.tsx
new file mode 100644
index 00000000..80eadff1
--- /dev/null
+++ b/packages/vmind/__tests__/browser/src/pages/NewChartGeneration/DataInput.tsx
@@ -0,0 +1,276 @@
+/* eslint-disable no-console */
+import React, { useState, useCallback, useMemo } from 'react';
+import '../index.scss';
+import { Avatar, Input, Divider, Button, Message, Select, Radio, Modal } from '@arco-design/web-react';
+import { chartGenerationMockData } from '../../constants/chartGeneratorData';
+import type { DataTable, FieldInfo } from '../../../../../src/index';
+import VMind from '../../../../../src/index';
+import { Model, AtomName, Schedule, LLMManage, DataType } from '../../../../../src/index';
+
+const TextArea = Input.TextArea;
+const Option = Select.Option;
+const RadioGroup = Radio.Group;
+
+type IPropsType = {
+ onSpecGenerate: (spec: any, command: string, costTime: number) => void;
+ onSpecListGenerate?: any;
+};
+
+const globalVariables = (import.meta as any).env;
+const ModelConfigMap: any = {
+ [Model.DOUBAO_PRO]: { url: globalVariables.VITE_DOUBAO_URL, key: globalVariables.VITE_DOUBAO_KEY },
+ [Model.GPT_4o]: { url: globalVariables.VITE_GPT_URL, key: globalVariables.VITE_GPT_KEY }
+};
+const OPENAI_API_URL = 'https://api.openai.com/v1/chat/completions';
+
+const specTemplateTest = false;
+
+function dataTypeTransfer(dataType: string): DataType {
+ switch (dataType) {
+ case 'date':
+ case 'time':
+ return DataType.DATE;
+ case 'count':
+ return DataType.INT;
+ case 'numerical':
+ case 'ratio':
+ return DataType.FLOAT;
+ default:
+ return DataType.STRING;
+ }
+}
+
+function transferFieldInfoInSimpleFieldInfo(fieldInfo: FieldInfo[]): FieldInfo[] {
+ return fieldInfo.map(item => ({
+ fieldName: item.fieldName,
+ description: item.description,
+ type: dataTypeTransfer(item.type),
+ role: item.role
+ }));
+}
+
+export function DataInput(props: IPropsType) {
+ const [dataTableIndex, setdataTableIndex] = useState
-
-
- 
-
-