diff --git a/src/components/content/index.vue b/src/components/content/index.vue
index 09d8c20af825bdca618bcacbf65a3dac3c2ae6b8..04134933d1563077086c6e8aeeb7f07983b55e50 100644
--- a/src/components/content/index.vue
+++ b/src/components/content/index.vue
@@ -39,6 +39,7 @@
+
|
diff --git a/src/js/config.js b/src/js/config.js
index 9ec479aa139e2fd93b84526cfe4c3ccec203203c..ea1d72c7e1cd7179dc2ab76293bed97f67a2a043 100644
--- a/src/js/config.js
+++ b/src/js/config.js
@@ -9,8 +9,6 @@ const prompt_template = `
以下为用户后续问题相关的内容:
{context}
-
-请根据以上相关信息,回答用户问题。
`
// 欢迎语,用户打开时自动发送给用户
@@ -46,7 +44,7 @@ export default {
"default_prompt": default_prompt,
"max_request_len": "2048",
"prompt_template": prompt_template,
- "api_prompt_prefix":"结合上下文回答问题,如果问题不是技术问题,请直接回答不知道。\n问题为:",
+ "api_prompt_prefix":"结合上下文回答问题。\n问题为:",
"show_profile_setting":false,
"index_url": index_url,
diff --git a/src/js/inscodeai.js b/src/js/inscodeai.js
index 9a578c0d178aa7556f998ffd05f46fe1a2a358c0..cad39d73e800ae161c29c5a107483c12c0873cfc 100644
--- a/src/js/inscodeai.js
+++ b/src/js/inscodeai.js
@@ -31,23 +31,28 @@ class InsCodeAI {
const model = config?.model??'vicuna-13b-all-v1.1'
const temperature = this.temperature??0.1
const top_p = config?.top_p??1.0
+ const prefix = config?.prompt_prefix??''
let stop_key = config?.stop_key??null
if (stop_key !== null && stop_key !== '') {
stop_key = stop_key.split(';;')
}
const data = {
- model: model,
+ // model: model,
max_tokens: parseInt(max_tokens),
temperature: parseFloat(temperature),
top_p: parseFloat(top_p),
stream: true,
stop: stop_key,
apikey: apiKey,
- // prefix: prefix
+ bizParam: {
+ support: "ai"
+ },
+ prefix: prefix
}
if (mode === 'chat') {
- data.messages = Prompt.getPromptByChatMode(config, context, history)
+
+ data.messages = JSON.stringify(Prompt.getPromptByChatMode(config, context, history))
} else {
data.prompt = Prompt.getPromptByTemplate(config, context, prompt)
}
@@ -56,6 +61,7 @@ class InsCodeAI {
method: 'POST',
signal: signal,
headers: {
+ 'accept': 'text/event-stream',
'Content-Type': 'application/json',
},
body: JSON.stringify(data),
@@ -63,7 +69,12 @@ class InsCodeAI {
// if the server emits an error message, throw an exception
// so it gets handled by the onerror callback below:
if (msg && msg?.data) {
- if (msg?.data === stop) {
+ const aiResult = decodeURIComponent(msg?.data??'')
+
+ // console.info(aiResult)
+
+
+ if (aiResult === stop) {
if (callback?.onclose) {
callback?.onclose()
}
@@ -72,7 +83,6 @@ class InsCodeAI {
} else {
// console.info(msg.data)
- const jsonData = JSON.parse(msg.data)
// 和上面重复触发,只留一个
// if (jsonData.choices[0].finish_reason === 'stop') {
@@ -81,20 +91,20 @@ class InsCodeAI {
// }
// return
// }
- let message = null
- if (mode === 'chat') {
- message = jsonData?.choices[0]?.message?.content
- if (typeof message === 'undefined') {
- message = jsonData?.choices[0]?.delta?.content
- }
- if (typeof message === 'undefined') {
- message = ''
- }
- } else {
- message = jsonData?.choices[0]?.text
- }
+ // let message = msg?.data??''
+ // if (mode === 'chat') {
+ // message = jsonData?.choices[0]?.message?.content
+ // if (typeof message === 'undefined') {
+ // message = jsonData?.choices[0]?.delta?.content
+ // }
+ // if (typeof message === 'undefined') {
+ // message = ''
+ // }
+ // } else {
+ // message = jsonData?.choices[0]?.text
+ // }
- callback?.onmessage(message, true)
+ callback?.onmessage(aiResult, false)
}
}