@@ -22,8 +22,13 @@ import {
22
22
mockUserInput12 ,
23
23
mockUserInput13 ,
24
24
mockUserInput14 ,
25
- mockUserInput16
25
+ mockUserInput16 ,
26
+ mockUserInput17
26
27
} from '../browser/src/constants/mockData' ;
28
+
29
+ const TEST_GPT = false ;
30
+ const TEST_SKYLARK = true ;
31
+
27
32
const demoDataList : { [ key : string ] : any } = {
28
33
pie : mockUserInput2 ,
29
34
'dynamic bar zh_cn' : mockUserInput6 ,
@@ -44,7 +49,8 @@ const demoDataList: { [key: string]: any } = {
44
49
'College entrance examination' : acceptRatioData ,
45
50
'Shopping Mall Sales Performance' : mallSalesData ,
46
51
'Global GDP' : mockUserInput6Eng ,
47
- 'Sales of different drinkings' : mockUserInput3Eng
52
+ 'Sales of different drinkings' : mockUserInput3Eng ,
53
+ 'Multi measure' : mockUserInput17
48
54
} ;
49
55
const CHART_GENERATION_AVERAGE_TIME = 10000 ;
50
56
const QPM_LIMIT = 10 ; //qpm limit of your llm service
@@ -53,7 +59,6 @@ const START_INDEX = 0;
53
59
54
60
const modelResultMap = {
55
61
[ Model . GPT3_5 ] : { totalCount : 0 , successCount : 0 , totalTime : 0 } ,
56
- [ Model . SKYLARK ] : { totalCount : 0 , successCount : 0 , totalTime : 0 } ,
57
62
[ Model . SKYLARK2 ] : { totalCount : 0 , successCount : 0 , totalTime : 0 }
58
63
} ;
59
64
@@ -88,48 +93,34 @@ const dataList = Object.keys(demoDataList);
88
93
89
94
const gptKey = process . env . VITE_GPT_KEY ;
90
95
const gptURL = process . env . VITE_GPT_JEST_URL ;
91
- if ( gptKey && gptURL ) {
96
+ if ( gptKey && gptURL && TEST_GPT ) {
92
97
const vmind = new VMind ( {
93
98
url : gptURL ,
94
99
model : Model . GPT3_5 ,
95
- cache : false ,
100
+ cache : true ,
101
+ showThoughts : false ,
96
102
headers : {
97
103
'api-key' : gptKey
98
104
}
99
105
} ) ;
100
106
testPerformance ( Model . GPT3_5 , vmind ) ;
101
107
}
102
108
103
- //const skylarkKey = process.env.VITE_SKYLARK_KEY;
104
- //const skylarkURL = process.env.VITE_SKYLARK_JEST_URL;
105
-
106
- //if (skylarkKey && skylarkURL) {
107
- // const vmind = new VMind({
108
- // url: skylarkURL,
109
- // model: Model.SKYLARK,
110
- // cache: false,
111
- // headers: {
112
- // 'api-key': skylarkKey
113
- // }
114
- // });
115
- // testPerformance(Model.SKYLARK, vmind);
116
- //}
117
-
118
109
const skylark2Key = process . env . VITE_SKYLARK_KEY ;
119
110
const skylark2URL = process . env . VITE_SKYLARK_JEST_URL ;
120
111
121
- // if (skylark2Key && skylark2URL) {
122
- // const vmind = new VMind({
123
- // url: skylark2URL,
124
- // model: Model.SKYLARK2,
125
- // cache: false ,
126
- // showThoughts: false,
127
- // headers: {
128
- // 'api-key': skylark2Key
129
- // }
130
- // });
131
- // testPerformance(Model.SKYLARK2, vmind);
132
- // }
112
+ if ( skylark2Key && skylark2URL && TEST_SKYLARK ) {
113
+ const vmind = new VMind ( {
114
+ url : skylark2URL ,
115
+ model : Model . SKYLARK2 ,
116
+ cache : true ,
117
+ showThoughts : false ,
118
+ headers : {
119
+ 'api-key' : skylark2Key
120
+ }
121
+ } ) ;
122
+ testPerformance ( Model . SKYLARK2 , vmind ) ;
123
+ }
133
124
134
125
afterAll ( ( ) => {
135
126
log ( '---------------VMind performance test---------------' ) ;
0 commit comments