1
- // @version 0.0.2
1
+ // @version 0.0.3 新增了环境变量,以解决aircode 的时不时抽风问题
2
2
const aircode = require ( 'aircode' ) ;
3
3
const lark = require ( '@larksuiteoapi/node-sdk' ) ;
4
4
var axios = require ( 'axios' ) ;
5
5
const EventDB = aircode . db . table ( 'event' ) ;
6
6
7
+ // 如果你不想配置环境变量,或环境变量不生效,则可以把结果填写在每一行最后的 "" 内部
8
+ const FEISHU_APP_ID = process . env . APPID || "" ; // 飞书的应用 ID
9
+ const FEISHU_APP_SECRET = process . env . SECRET || "" ; // 飞书的应用的 Secret
10
+ const FEISHU_BOTNAME = process . env . BOTNAME || "" ; // 飞书机器人的名字
11
+ const OPENAI_KEY = process . env . KEY || "" ; // OpenAI 的 Key
12
+ const OPENAI_MODEL = process . env . MODEL || "text-davinci-003" ; // 使用的模型
13
+ const OPENAI_MAX_TOKEN = process . env . MAX_TOKEN || 1024 ; // 最大 token 的值
14
+
7
15
const client = new lark . Client ( {
8
- appId : process . env . APPID ,
9
- appSecret : process . env . SECRET ,
16
+ appId : FEISHU_APP_ID ,
17
+ appSecret : FEISHU_APP_SECRET ,
10
18
disableTokenCache : false
11
19
} ) ;
12
20
@@ -43,9 +51,9 @@ async function getOpenAIReply(content) {
43
51
var prompt = getPrompt ( content . trim ( ) ) ;
44
52
45
53
var data = JSON . stringify ( {
46
- "model" : process . env . MODEL || "text-davinci-003" ,
54
+ "model" : OPENAI_MODEL ,
47
55
"prompt" : prompt ,
48
- "max_tokens" : process . env . MAX_TOKEN || 1024 ,
56
+ "max_tokens" : OPENAI_MAX_TOKEN ,
49
57
"temperature" : 0.9 ,
50
58
"frequency_penalty" : 0.0 ,
51
59
"presence_penalty" : 0.0 ,
@@ -58,7 +66,7 @@ async function getOpenAIReply(content) {
58
66
maxBodyLength : Infinity ,
59
67
url : 'https://api.openai.com/v1/completions' ,
60
68
headers : {
61
- 'Authorization' : `Bearer ${ process . env . KEY } ` ,
69
+ 'Authorization' : `Bearer ${ OPENAI_KEY } ` ,
62
70
'Content-Type' : 'application/json'
63
71
} ,
64
72
data : data
@@ -111,7 +119,7 @@ module.exports = async function (params, context) {
111
119
return { "code" : 0 }
112
120
}
113
121
// 没有 mention 机器人,则退出。
114
- if ( params . event . message . mentions [ 0 ] . name != process . env . BOTNAME ) {
122
+ if ( params . event . message . mentions [ 0 ] . name != FEISHU_BOTNAME ) {
115
123
return { "code" : 0 }
116
124
}
117
125
const userInput = JSON . parse ( params . event . message . content ) ;
0 commit comments