Skip to content

Commit d96e46a

Browse files
committed
feat(ai-completions): add model options
1 parent 9e31d1e commit d96e46a

2 files changed

Lines changed: 12 additions & 4 deletions

File tree

packages/common/js/completion.js

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
*
1111
*/
1212
import { ref } from 'vue'
13-
import { useCanvas, useResource } from '@opentiny/tiny-engine-meta-register'
13+
import { useCanvas, useResource, getMergeMeta } from '@opentiny/tiny-engine-meta-register'
1414

1515
const keyWords = [
1616
'state',
@@ -224,14 +224,15 @@ const generateBaseReference = () => {
224224

225225
const fetchAiInlineCompletion = (codeBeforeCursor, codeAfterCursor) => {
226226
const referenceContext = generateBaseReference()
227-
return fetch('https://agent.opentiny.design/api/v1/ai/chat/completions', {
227+
const { modelName, apiKey, url } = getMergeMeta('engine.plugins.pagecontroller')?.options?.AIModel || {}
228+
return fetch(`${url ?? 'https://agent.opentiny.design/api/v1/ai/chat/completions'}`, {
228229
method: 'POST',
229230
headers: {
230231
'Content-Type': 'application/json',
231-
Authorization: 'Bearer sk-1234'
232+
Authorization: `Bearer ${apiKey ?? 'sk-1234'}`
232233
},
233234
body: JSON.stringify({
234-
model: 'internvl3-14b',
235+
model: modelName ?? 'internvl3-14b',
235236
messages: [
236237
{
237238
role: 'user',

packages/plugins/script/meta.js

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,5 +5,12 @@ export default {
55
icon: 'plugin-icon-js',
66
width: 600,
77
widthResizable: true,
8+
options: {
9+
AIModel: {
10+
modelName: '',
11+
apiKey: '',
12+
url: ''
13+
}
14+
},
815
confirm: 'close' // 当点击插件栏切换或关闭前是否需要确认, 会调用插件中confirm值指定的方法,e.g. 此处指向 close方法,会调用插件的close方法执行确认逻辑
916
}

0 commit comments

Comments
 (0)