-
Notifications
You must be signed in to change notification settings - Fork 5
Expand file tree
/
Copy pathinput.ts
More file actions
156 lines (143 loc) · 4.64 KB
/
input.ts
File metadata and controls
156 lines (143 loc) · 4.64 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
import { cli } from 'cleye'
import { input, password, select } from '@inquirer/prompts'
import { logger } from './utils/logger.js'
import { existsSync, readFileSync, writeFileSync } from 'node:fs'
import path from 'node:path'
import { dirname } from 'path'
import { fileURLToPath } from 'url'
// * Set __dirname and __filename
const __filename = fileURLToPath(import.meta.url)
const __dirname = dirname(__filename)
// * Parse CLI arguments
const argv = cli({
name: 'friday',
version: '0.0.1',
flags: {
prompt: {
type: String,
description: 'Prompt to run',
alias: 'p'
}
}
})
let userRequest = argv._.join(' ')
// * Load config file
const configPath = path.join(__dirname, '../', '.fridayconfig.json')
let isConfigFileExist = existsSync(configPath)
if (isConfigFileExist) {
try {
const config = JSON.parse(readFileSync(configPath, 'utf-8'))
if (config.OPENAI_API_KEY)
process.env.OPENAI_API_KEY = config.OPENAI_API_KEY
if (config.OPENAI_CHAT_MODEL)
process.env.OPENAI_CHAT_MODEL = config.OPENAI_CHAT_MODEL
if (config.AZURE_SPEECH_KEY)
process.env.AZURE_SPEECH_KEY = config.AZURE_SPEECH_KEY
if (config.AZURE_SPEECH_REGION)
process.env.AZURE_SPEECH_REGION = config.AZURE_SPEECH_REGION
if (config.AZURE_SPEECH_VOICE)
process.env.AZURE_SPEECH_VOICE = config.AZURE_SPEECH_VOICE
} catch (e) {}
}
// * If OPENAI_API_KEY is empty, ask for it
if (!process.env.OPENAI_API_KEY || process.env.OPENAI_API_KEY.length === 0) {
logger(`Friday GPT 를 사용하기 위해선 Open A.I API 키를 입력해야합니다.`)
logger(`키 발급 페이지: https://platform.openai.com/account/api-keys`)
process.env.OPENAI_API_KEY = await password({
message: 'Open A.I API 키를 입력해주세요:'
})
if (!process.env.OPENAI_API_KEY || process.env.OPENAI_API_KEY.length === 0) {
logger('API 키가 입력되지 않았습니다. 프로그램을 종료합니다.')
process.exit(0)
}
}
// * If OPENAI_CHAT_MODEL is empty, ask for it
if (
!process.env.OPENAI_CHAT_MODEL ||
process.env.OPENAI_CHAT_MODEL.length === 0
) {
process.env.OPENAI_CHAT_MODEL = await input({
message: '사용하려는 ChatGPT 모델을 입력해주세요:',
default: 'gpt-4'
})
if (
!process.env.OPENAI_CHAT_MODEL ||
process.env.OPENAI_CHAT_MODEL.length === 0
) {
logger('모델이 입력되지 않았습니다. 프로그램을 종료합니다.')
process.exit(0)
}
}
// * Optional asking section
if (!isConfigFileExist) {
// * If AZURE_SPEECH_KEY is empty, ask for it
if (
process.env.AZURE_SPEECH_KEY === undefined ||
process.env.AZURE_SPEECH_KEY === ''
) {
logger(`Azure Speech API 키를 입력하시면 인공지능이 음성으로 답변합니다.`)
logger(
`키 발급 페이지: https://portal.azure.com/?quickstart=true#view/Microsoft_Azure_ProjectOxford/CognitiveServicesHub/~/SpeechServices`
)
const isNeedAzureSpeech = await select({
message: 'Azure Speech API 키를 입력하고 TTS를 사용하시겠습니까?',
choices: [
{
name: '네',
value: true
},
{
name: '아니오',
value: false
}
]
})
if (isNeedAzureSpeech) {
process.env.AZURE_SPEECH_KEY = await password({
message: 'Azure Speech API 키를 입력해주세요:'
})
process.env.AZURE_SPEECH_REGION = await input({
message: 'Azure Speech API 지역을 입력해주세요:',
default: 'koreacentral'
})
process.env.AZURE_SPEECH_VOICE = await input({
message: 'Azure Speech API 음성을 입력해주세요:',
default: 'ko-KR-SeoHyeonNeural'
})
}
}
}
// * If userRequest is empty, use the provided default value
if (userRequest === '') {
userRequest = await input({
message: 'A.I가 수행할 개발 업무 요청을 입력해주세요:'
})
if (userRequest === '') {
logger('요청이 없습니다. 프로그램을 종료합니다.')
process.exit(0)
}
}
// * Save config file
writeFileSync(
configPath,
JSON.stringify(
{
OPENAI_API_KEY: process.env.OPENAI_API_KEY,
OPENAI_CHAT_MODEL: process.env.OPENAI_CHAT_MODEL,
AZURE_SPEECH_KEY: process.env.AZURE_SPEECH_KEY,
AZURE_SPEECH_REGION: process.env.AZURE_SPEECH_REGION,
AZURE_SPEECH_VOICE: process.env.AZURE_SPEECH_VOICE
},
null,
2
)
)
export { userRequest }
export const countConfig = {
architect: {
// * 초기 기획서를 몇 개까지 만들고 고민할지 설정합니다.
draftCount: 3,
// * 기획서 보완을 위한 회의를 최대 몇 번까지 진행 할지 설정합니다.
maxMeetingCount: 10
}
}