zaclys/searxng
1
0
Fork 1
mirror of https://github.com/searxng/searxng synced 2024-01-01 19:24:07 +01:00
Code Issues Projects Releases Packages Wiki Activity

Update webapp.py

Browse source
This commit is contained in:
Joseph Cheung 2023-02-23 12:38:50 +08:00
parent 75139c2e10
commit 5059a1ae1e
1 changed files with 141 additions and 124 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

265
searx/webapp.py
View file

@ -1012,135 +1012,152 @@ Authorization: "Bearer '''+ os.environ['GPTKEY'] + r'''",
}; };
let prompt = JSON.parse(document.querySelector("#prompt").textContent); let prompt = JSON.parse(document.querySelector("#prompt").textContent);
chatTextRawIntro = ""; chatTextRawIntro = "";
text_offset = -1; text_offset = -1;
const optionsIntro = { const optionsIntro = {
method: "POST", method: "POST",
headers: headers, headers: headers,
body: JSON.stringify({ body: JSON.stringify({
"prompt": "你是一个叫Charles的搜索引擎机器人，用户搜索的是“''' + search_query.query + r'''”有关的信息。\n不要假定搜索结果。\n用简体中文写一句语言幽默的、含有emoji的引入语：\n", "prompt": "你是一个叫Charles的搜索引擎机器人，用户搜索的是“''' + search_query.query + r'''”有关的信息。\n不要假定搜索结果。\n用简体中文写一句语言幽默的、含有emoji的引入语：\n",
"max_tokens": 256, "max_tokens": 256,
"temperature": 0, "temperature": 0,
"top_p": 1, "top_p": 1,
"frequency_penalty": 0, "frequency_penalty": 0,
"presence_penalty": 2, "presence_penalty": 2,
"best_of": 1, "best_of": 1,
"echo": false, "echo": false,
"logprobs": 0, "logprobs": 0,
"stream": true "stream": true
}) })
}; };
fetch("https://api.openai.com/v1/engines/text-davinci-003/completions", optionsIntro) fetch("https://api.openai.com/v1/engines/text-davinci-003/completions", optionsIntro)
.then((responseIntro) => { .then((responseIntro) => {
const readerIntro = responseIntro.body.getReader(); const readerIntro = responseIntro.body.getReader();
let result = ''; let result = '';
readerIntro.read().then(function processText({ done, value }) { readerIntro.read().then(function processText({ done, value }) {
if (done) return; if (done) return;
const textIntro = new TextDecoder('utf-8').decode(value); const textIntro = new TextDecoder('utf-8').decode(value);
textIntro.trim().split('\n').forEach(function(v) { textIntro.trim().split('\n').forEach(function(v) {
if(v.length>6) result = v.slice(6); if(v.length>6) result = v.slice(6);
if(result == "[DONE]") if(result == "[DONE]")
{ {
return;
}
const { choices } = JSON.parse(result);
if(choices[0].logprobs.text_offset[0] > text_offset)
{
chatTextRawIntro+=choices[0].text text_offset = -1;
text_offset = choices[0].logprobs.text_offset[choices[0].logprobs.text_offset.length - 1] const options = {
} method: "POST",
markdownToHtml(beautify(chatTextRawIntro+'\n'), document.getElementById('chat_intro')); headers: headers,
body: JSON.stringify(prompt.data)
};
fetch("https://api.openai.com/v1/engines/text-davinci-003/completions", options)
.then((response) => {
const reader = response.body.getReader();
let result = '';
reader.read().then(function processText({ done, value }) {
if (done) return;
const text = new TextDecoder('utf-8').decode(value);
text.trim().split('\n').forEach(function(v) {
if(v.length>6) result = v.slice(6);
if(result == "[DONE]")
{
chatTextRawPlusComment = chatTextRaw+"\n\n";
text_offset = -1;
const optionsPlus = {
method: "POST",
headers: headers,
body: JSON.stringify({
"prompt": "围绕关键词 ''' + search_query.query + r'''，结合你的知识总结归纳发表评论，不得重复提及已有内容：\n" + document.querySelector("#chat > p").innerHTML.replace(/<.*?>.*?<\/.*?>/g, '') +"\n",
"max_tokens": 1500,
"temperature": 0.7,
"top_p": 1,
"frequency_penalty": 0,
"presence_penalty": 2,
"best_of": 1,
"echo": false,
"logprobs": 0,
"stream": true
})
};
fetch("https://api.openai.com/v1/engines/text-davinci-003/completions", optionsPlus)
.then((responsePlusComment) => {
const readerPlusComment = responsePlusComment.body.getReader();
let result = '';
readerPlusComment.read().then(function processText({ done, value }) {
if (done) return;
const textPlusComment = new TextDecoder('utf-8').decode(value);
textPlusComment.trim().split('\n').forEach(function(v) {
if(v.length>6) result = v.slice(6);
if(result == "[DONE]")
{
lock_chat = 0;
document.getElementById('chat_continue').style.display="";
return;
}
const { choices } = JSON.parse(result);
if(choices[0].logprobs.text_offset[0] > text_offset)
{
chatTextRawPlusComment+=choices[0].text
text_offset = choices[0].logprobs.text_offset[choices[0].logprobs.text_offset.length - 1]
}
markdownToHtml(beautify(chatTextRawPlusComment), document.getElementById('chat'));
})
return readerPlusComment.read().then(processText);
});
})
.catch((error) => {
console.error('Error:', error);
});
return;
}
const { choices } = JSON.parse(result);
if(choices[0].logprobs.text_offset[0] > text_offset)
{
chatTextRaw+=choices[0].text
text_offset = choices[0].logprobs.text_offset[choices[0].logprobs.text_offset.length - 1]
}
markdownToHtml(beautify(chatTextRaw), document.getElementById('chat'));
})
return readerIntro.read().then(processText);
});
}) })
.catch((error) => { return reader.read().then(processText);
console.error('Error:', error);
}); });
})
text_offset = -1; .catch((error) => {
const options = { console.error('Error:', error);
method: "POST", });
headers: headers,
body: JSON.stringify(prompt.data)
};
fetch("https://api.openai.com/v1/engines/text-davinci-003/completions", options)
.then((response) => {
const reader = response.body.getReader();
let result = '';
reader.read().then(function processText({ done, value }) {
if (done) return;
const text = new TextDecoder('utf-8').decode(value);
text.trim().split('\n').forEach(function(v) {
if(v.length>6) result = v.slice(6); return;
if(result == "[DONE]") }
{ const { choices } = JSON.parse(result);
chatTextRawPlusComment = chatTextRaw+"\n"; if(choices[0].logprobs.text_offset[0] > text_offset)
text_offset = -1; {
const optionsPlus = { chatTextRawIntro+=choices[0].text
method: "POST", text_offset = choices[0].logprobs.text_offset[choices[0].logprobs.text_offset.length - 1]
headers: headers, }
body: JSON.stringify({ markdownToHtml(beautify(chatTextRawIntro+'\n'), document.getElementById('chat_intro'));
"prompt": "围绕关键词 ''' + search_query.query + r'''，结合你的知识总结归纳发表评论，不得重复提及已有内容：\n" + document.querySelector("#chat > p").innerHTML.replace(/<.*?>.*?<\/.*?>/g, '') +"\n",
"max_tokens": 1500, })
"temperature": 0.7, return readerIntro.read().then(processText);
"top_p": 1, });
"frequency_penalty": 0, })
"presence_penalty": 2, .catch((error) => {
"best_of": 1, console.error('Error:', error);
"echo": false, });
"logprobs": 0,
"stream": true
})
};
fetch("https://api.openai.com/v1/engines/text-davinci-003/completions", optionsPlus)
.then((responsePlusComment) => {
const readerPlusComment = responsePlusComment.body.getReader();
let result = '';
readerPlusComment.read().then(function processText({ done, value }) {
if (done) return;
const textPlusComment = new TextDecoder('utf-8').decode(value);
textPlusComment.trim().split('\n').forEach(function(v) {
if(v.length>6) result = v.slice(6);
if(result == "[DONE]")
{
lock_chat = 0;
document.getElementById('chat_continue').style.display="";
return;
}
const { choices } = JSON.parse(result);
if(choices[0].logprobs.text_offset[0] > text_offset)
{
chatTextRawPlusComment+=choices[0].text
text_offset = choices[0].logprobs.text_offset[choices[0].logprobs.text_offset.length - 1]
}
markdownToHtml(beautify(chatTextRawPlusComment), document.getElementById('chat'));
})
return readerPlusComment.read().then(processText);
});
})
.catch((error) => {
console.error('Error:', error);
});
return;
}
const { choices } = JSON.parse(result);
if(choices[0].logprobs.text_offset[0] > text_offset)
{
chatTextRaw+=choices[0].text
text_offset = choices[0].logprobs.text_offset[choices[0].logprobs.text_offset.length - 1]
}
markdownToHtml(beautify(chatTextRaw), document.getElementById('chat'));
})
return reader.read().then(processText);
});
})
.catch((error) => {
console.error('Error:', error);
});
</script> </script>
''' '''