mirror of
https://github.com/klxf/eda-copilot.git
synced 2025-12-07 20:34:14 +08:00
首次提交
This commit is contained in:
389
iframe/js/AI.js
Normal file
389
iframe/js/AI.js
Normal file
@@ -0,0 +1,389 @@
|
||||
// eslint-disable-next-line @typescript-eslint/no-unused-vars
|
||||
async function getAIResponse(contents) {
|
||||
let response;
|
||||
let thinking = document.createElement('div');
|
||||
thinking.className = 'message-bubble model';
|
||||
thinking.id = 'thinking';
|
||||
thinking.innerHTML = '<div><div class="loading"></div> 思考中...</div>';
|
||||
document.querySelector('.chat-body').appendChild(thinking);
|
||||
document.querySelector('.chat-body').scrollTop = document.querySelector('.chat-body').scrollHeight;
|
||||
|
||||
if (MODEL_GROUP === "Gemini") {
|
||||
response = await getGeminiResponse(contents);
|
||||
} else if (MODEL_GROUP === "DeepSeek") {
|
||||
response = await getDeepSeekResponse(contents);
|
||||
} else if (MODEL_GROUP === "QWen") {
|
||||
response = await getQWenResponse(contents);
|
||||
} else if (MODEL_GROUP === "Custom") {
|
||||
response = await getCustomResponse(contents);
|
||||
} else {
|
||||
response = { err_code: 1, error: "模型不被支持" };
|
||||
}
|
||||
|
||||
console.log(response);
|
||||
|
||||
if (response.err_code === 0) {
|
||||
document.getElementById('thinking').remove();
|
||||
let content = response.response;
|
||||
content = content.replace(/\u003cthink\u003e/, '<div class="think">').replace(/\u003c\/think\u003e/, '</div>');
|
||||
createBubble('model', content, undefined);
|
||||
|
||||
if (MODEL_GROUP === "Gemini") {
|
||||
chatHistory.push({ role: 'model', parts: [{ text: content }] });
|
||||
} else if (MODEL_GROUP === "DeepSeek" || MODEL_GROUP === "Qwen" || MODEL_GROUP === "Custom") {
|
||||
chatHistory.push({ role: 'assistant', content: content });
|
||||
}
|
||||
|
||||
if (CHAT_TITLE === "新的对话") {
|
||||
let temp = chatHistory.slice();
|
||||
if (MODEL_GROUP === "Gemini") {
|
||||
temp.push({ role: 'user', parts: [{ text: "用几个字总结这个对话，将其作为对话的标题" }] });
|
||||
|
||||
getGeminiResponse(temp).then(response => {
|
||||
if (response.err_code === 0) {
|
||||
const summary = response.response;
|
||||
CHAT_TITLE = summary.replace('*', '');
|
||||
document.getElementById('title').innerText = summary;
|
||||
}
|
||||
});
|
||||
} else if (MODEL_GROUP === "DeepSeek") {
|
||||
temp.push({ role: 'user', content: "用几个字总结这个对话，将其作为对话的标题，除此以外不要有任何其他文字" });
|
||||
|
||||
getDeepSeekResponse(temp).then(response => {
|
||||
if (response.err_code === 0) {
|
||||
const summary = response.response;
|
||||
CHAT_TITLE = summary.replace(/\u003cthink\u003e.*\u003c\/think\u003e/ms, '').replace('*', '');
|
||||
document.getElementById('title').innerText = CHAT_TITLE;
|
||||
}
|
||||
});
|
||||
} else if (MODEL_GROUP === "QWen") {
|
||||
temp.push({ role: 'user', content: "用几个字总结这个对话，将其作为对话的标题，除此以外不要有任何其他文字" });
|
||||
|
||||
getQWenResponse(temp).then(response => {
|
||||
if (response.err_code === 0) {
|
||||
const summary = response.response;
|
||||
CHAT_TITLE = summary.replace(/\u003cthink\u003e.*\u003c\/think\u003e/ms, '').replace('*', '');
|
||||
document.getElementById('title').innerText = CHAT_TITLE;
|
||||
}
|
||||
});
|
||||
} else if (MODEL_GROUP === "Custom") {
|
||||
temp.push({ role: 'user', content: "用几个字总结这个对话，将其作为对话的标题，除此以外不要有任何其他文字" });
|
||||
|
||||
getCustomResponse(temp).then((response) => {
|
||||
if (response.err_code === 0) {
|
||||
const summary = response.response;
|
||||
CHAT_TITLE = summary.replace(/<think>.*<\/think>/ms, '').replaceAll('*', '').replaceAll('\n', '');
|
||||
document.getElementById('title').innerText = CHAT_TITLE;
|
||||
}
|
||||
});
|
||||
}
|
||||
}
|
||||
} else {
|
||||
document.getElementById('thinking').remove();
|
||||
createBubble('model', response.error, "异常");
|
||||
}
|
||||
}
|
||||
|
||||
// eslint-disable-next-line @typescript-eslint/no-unused-vars
|
||||
async function setAICache(cache) {
|
||||
if (MODEL_GROUP === "Gemini" && MODEL_NAME !== "gemini-2.0-flash-001") {
|
||||
return await setGeminiCache(cache);
|
||||
} else if (MODEL_GROUP === "QWen" || MODEL_GROUP === "DeepSeek" || MODEL_GROUP === "Custom") {
|
||||
return await setOtherCache(decodeURIComponent(atob(cache)));
|
||||
} else if (MODEL_GROUP === "Gemini" && MODEL_NAME === "gemini-2.0-flash-001") {
|
||||
return await setOtherCache(decodeURIComponent(atob(cache)));
|
||||
} else {
|
||||
createBubble('model', "抱歉，该模型不支持缓存长上下文", "异常");
|
||||
}
|
||||
}
|
||||
|
||||
async function getGeminiResponse(contents) {
|
||||
const API = GOOGLE_SETTINGS.HOST + "/v1beta/models/" + MODEL_NAME + ":generateContent?key=" + GOOGLE_SETTINGS.API_KEY;
|
||||
console.log(chatHistory);
|
||||
const data = {
|
||||
contents: [
|
||||
contents
|
||||
],
|
||||
safetySettings: [
|
||||
[
|
||||
{
|
||||
"category": "HARM_CATEGORY_HARASSMENT",
|
||||
"threshold": "BLOCK_ONLY_HIGH"
|
||||
},
|
||||
{
|
||||
"category": "HARM_CATEGORY_HATE_SPEECH",
|
||||
"threshold": "BLOCK_ONLY_HIGH"
|
||||
},
|
||||
{
|
||||
"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
|
||||
"threshold": "BLOCK_ONLY_HIGH"
|
||||
},
|
||||
{
|
||||
"category": "HARM_CATEGORY_DANGEROUS_CONTENT",
|
||||
"threshold": "BLOCK_ONLY_HIGH"
|
||||
}
|
||||
]
|
||||
]
|
||||
};
|
||||
|
||||
if (CACHE_NAME !== null) {
|
||||
data.cachedContent = CACHE_NAME;
|
||||
} else {
|
||||
data.systemInstruction = { parts: [{ "text": "You are an electronic engineer. Use Chinese to answer." }] }
|
||||
}
|
||||
|
||||
try {
|
||||
const response = await eda.sys_ClientUrl.request(API, "POST", JSON.stringify(data), { headers: GOOGLE_SETTINGS.HEADER });
|
||||
if (response.status !== 200) {
|
||||
throw new Error(response.statusText);
|
||||
}
|
||||
const result = await response.json();
|
||||
return {
|
||||
response: result.candidates[0].content.parts[0].text,
|
||||
err_code: 0
|
||||
};
|
||||
} catch (error) {
|
||||
return {
|
||||
error: error.message || "请求失败",
|
||||
err_code: 1
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
async function getDeepSeekResponse(contents) {
|
||||
const API = DS_SETTINGS.HOST + "/chat/completions";
|
||||
let HEADERS = DS_SETTINGS.HEADER;
|
||||
|
||||
HEADERS['Authorization'] = 'Bearer ' + DS_SETTINGS.API_KEY;
|
||||
|
||||
const data = {
|
||||
model: MODEL_NAME,
|
||||
messages: contents,
|
||||
stream: false
|
||||
}
|
||||
|
||||
try {
|
||||
const response = await eda.sys_ClientUrl.request(API, "POST", JSON.stringify(data), { headers: HEADERS });
|
||||
if (response.status !== 200) {
|
||||
throw new Error(response.statusText);
|
||||
}
|
||||
const result = await response.json();
|
||||
return {
|
||||
response: result.choices[0].message.content,
|
||||
err_code: 0
|
||||
};
|
||||
} catch (error) {
|
||||
return {
|
||||
error: error.message || "请求失败",
|
||||
err_code: 1
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
async function getQWenResponse(contents) {
|
||||
const API = QWEN_SETTINGS.HOST + "/compatible-mode/v1/chat/completions";
|
||||
let HEADERS = QWEN_SETTINGS.HEADER;
|
||||
|
||||
HEADERS['Authorization'] = 'Bearer ' + QWEN_SETTINGS.API_KEY;
|
||||
|
||||
const data = {
|
||||
model: MODEL_NAME,
|
||||
messages: contents,
|
||||
}
|
||||
|
||||
try {
|
||||
const response = await eda.sys_ClientUrl.request(API, "POST", JSON.stringify(data), { headers: HEADERS });
|
||||
if (response.status !== 200) {
|
||||
throw new Error(response.statusText);
|
||||
}
|
||||
const result = await response.json();
|
||||
return {
|
||||
response: result.choices[0].message.content,
|
||||
err_code: 0
|
||||
};
|
||||
} catch (error) {
|
||||
return {
|
||||
error: error.message || "请求失败",
|
||||
err_code: 1
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
async function getCustomResponse(contents) {
|
||||
const API = CUSTOM_SETTINGS.HOST + CUSTOM_SETTINGS.ENDPOINT;
|
||||
|
||||
const data = {
|
||||
model: CUSTOM_SETTINGS.MODEL,
|
||||
messages: contents,
|
||||
stream: false
|
||||
}
|
||||
|
||||
try {
|
||||
const response = await eda.sys_ClientUrl.request(API, "POST", JSON.stringify(data), { headers: CUSTOM_SETTINGS.HEADER });
|
||||
if (response.status !== 200) {
|
||||
throw new Error(response.statusText);
|
||||
}
|
||||
const result = await response.json();
|
||||
return {
|
||||
response: result.choices[0].message.content,
|
||||
err_code: 0
|
||||
};
|
||||
} catch (error) {
|
||||
return {
|
||||
error: error.message || "请求失败",
|
||||
err_code: 1
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
async function setGeminiCache(cache) {
|
||||
let API = GOOGLE_SETTINGS.HOST + "/v1beta/cachedContents?key=" + GOOGLE_SETTINGS.API_KEY;
|
||||
|
||||
const data = {
|
||||
model: "models/" + MODEL_NAME,
|
||||
contents: [
|
||||
{ role: 'user', parts: [{ inline_data: { mime_type: "text/plain", data: cache} }, {text: "Here's a netlist file describing the circuit diagram, and I'm going to ask you questions about it."}] }
|
||||
],
|
||||
systemInstruction: {
|
||||
parts: [
|
||||
{
|
||||
"text": "You are an electronic engineer. The text describes a netlist of circuit diagrams. When asked a question about a component, it is displayed if there is a URL in the netlist, otherwise it is not. Use Chinese to answer."
|
||||
}
|
||||
]
|
||||
},
|
||||
}
|
||||
|
||||
let thinking = document.createElement('div');
|
||||
thinking.className = 'message-bubble model';
|
||||
thinking.id = 'thinking';
|
||||
thinking.innerHTML = '<div><div class="loading"></div> 解读中...</div>';
|
||||
document.querySelector('.chat-body').appendChild(thinking);
|
||||
document.querySelector('.chat-body').scrollTop = document.querySelector('.chat-body').scrollHeight;
|
||||
|
||||
eda.sys_ClientUrl.request(API, "POST", JSON.stringify(data), { headers: GOOGLE_SETTINGS.HEADER }).then(response => {
|
||||
if (response.status !== 200) {
|
||||
throw new Error(response.statusText);
|
||||
}
|
||||
return response.json();
|
||||
}).then(result => {
|
||||
document.getElementById('thinking').remove();
|
||||
createBubble('model', "好的，接下来你可以围绕这张原理图向我提问", undefined);
|
||||
CACHE_NAME = result.name;
|
||||
}).catch(error => {
|
||||
document.getElementById('thinking').remove();
|
||||
createBubble('model', error.toString(), "异常");
|
||||
});
|
||||
}
|
||||
|
||||
async function setOtherCache(cache) {
|
||||
chatHistory.push({ role: 'user', content: '这是一个电路图的网表，后续回答依照这个网表回答：' + cache });
|
||||
chatHistory.push({ role: 'assistant', content: '好的，我了解了'});
|
||||
}
|
||||
|
||||
// eslint-disable-next-line @typescript-eslint/no-unused-vars
|
||||
function getNetlist() {
|
||||
let netlist = '';
|
||||
eda.sch_Netlist.getNetlist().then(data => {
|
||||
netlist = data[0].toString();
|
||||
const cache = btoa(encodeURIComponent(netlist));
|
||||
// 获取 cache 的字节数
|
||||
let size = cache.length;
|
||||
let unit = 'B';
|
||||
if (size > 1024) {
|
||||
size /= 1024;
|
||||
unit = 'KB';
|
||||
}
|
||||
if (size > 1024) {
|
||||
size /= 1024;
|
||||
unit = 'MB';
|
||||
}
|
||||
let fileBubble = document.createElement('div');
|
||||
fileBubble.classList.add('file-bubble');
|
||||
fileBubble.innerHTML = `<img src="data:image/svg+xml;base64,PD94bWwgdmVyc2lvbj0iMS4wIiBlbmNvZGluZz0iaXNvLTg4NTktMSI/Pgo8IS0tIEdlbmVyYXRvcjogQWRvYmUgSWxsdXN0cmF0b3IgMjMuMC4wLCBTVkcgRXhwb3J0IFBsdWctSW4gLiBTVkcgVmVyc2lvbjogNi4wMCBCdWlsZCAwKSAgLS0+CjxzdmcgdmVyc2lvbj0iMS4xIiBpZD0iJiN4NTZGRTsmI3g1QzQyO18xIiB4bWxucz0iaHR0cDovL3d3dy53My5vcmcvMjAwMC9zdmciIHhtbG5zOnhsaW5rPSJodHRwOi8vd3d3LnczLm9yZy8xOTk5L3hsaW5rIiB4PSIwcHgiCgkgeT0iMHB4IiB3aWR0aD0iMTZweCIgaGVpZ2h0PSIxNnB4IiB2aWV3Qm94PSIwIDAgMTYgMTYiIHN0eWxlPSJlbmFibGUtYmFja2dyb3VuZDpuZXcgMCAwIDE2IDE2OyIgeG1sOnNwYWNlPSJwcmVzZXJ2ZSI+CjxyZWN0IHN0eWxlPSJmaWxsOm5vbmU7IiB3aWR0aD0iMTYiIGhlaWdodD0iMTYiLz4KPHJlY3Qgc3R5bGU9ImZpbGw6bm9uZTsiIHdpZHRoPSIxNiIgaGVpZ2h0PSIxNiIvPgo8cGF0aCBkPSJNMTUsNGgtMVYyYzAtMC41NS0wLjQ1LTEtMS0xSDRDMy40NSwxLDMsMS40NSwzLDJ2NEgxLjVDMS4yMiw2LDEsNi4yMiwxLDYuNUMxLDYuNzgsMS4yMiw3LDEuNSw3SDNoMWgxLjUKCUM1Ljc4LDcsNiw2Ljc4LDYsNi41QzYsNi4yMiw1Ljc4LDYsNS41LDZINFYyaDl2MmgtMWMtMC41NSwwLTEsMC40NS0xLDF2N2MwLDAuNTUsMC40NSwxLDEsMWgxdjJINHYtNGgxLjVDNS43OCwxMSw2LDEwLjc4LDYsMTAuNQoJQzYsMTAuMjIsNS43OCwxMCw1LjUsMTBINEgzSDEuNUMxLjIyLDEwLDEsMTAuMjIsMSwxMC41QzEsMTAuNzgsMS4yMiwxMSwxLjUsMTFIM3Y0YzAsMC41NSwwLjQ1LDEsMSwxaDljMC41NSwwLDEtMC40NSwxLTF2LTJoMQoJYzAuNTUsMCwxLTAuNDUsMS0xVjVDMTYsNC40NSwxNS41NSw0LDE1LDR6IE0xNSwxMmgtMWgtMWgtMVY1aDFoMWgxVjEyeiIvPgo8L3N2Zz4K" alt="File Icon" class="file-icon">
|
||||
<div class="file-info">
|
||||
<div class="file-name">网表文件</div>
|
||||
<div class="file-size">${size} ${unit}</div>
|
||||
</div>`;
|
||||
document.querySelector('.chat-body').appendChild(fileBubble);
|
||||
setAICache(cache);
|
||||
})
|
||||
|
||||
}
|
||||
|
||||
// eslint-disable-next-line @typescript-eslint/no-unused-vars
|
||||
async function setAIFile() {
|
||||
if (MODEL_GROUP !== "Gemini") {
|
||||
eda.sys_Message.showToastMessage('当前模型尚未支持该功能', 'warn');
|
||||
return;
|
||||
}
|
||||
const file = await eda.sys_FileSystem.openReadFileDialog("pdf");
|
||||
const reader = new FileReader();
|
||||
reader.onload = function (event) {
|
||||
const base64 = event.target.result;
|
||||
const title = file.name;
|
||||
// 判断是不是 pdf 文件
|
||||
if (file.type !== 'application/pdf') {
|
||||
eda.sys_Message.showToastMessage('仅可上传 PDF 格式的数据手册', 'warn');
|
||||
return;
|
||||
}
|
||||
FILE_BASE64 = base64.replaceAll('data:application/pdf;base64,', '');
|
||||
// 如果 chat-body 中最后一个元素是 file-bubble，替换之
|
||||
const chatBody = document.querySelector('.chat-body');
|
||||
const lastElement = chatBody.lastElementChild;
|
||||
if (lastElement && lastElement.classList.contains('file-bubble')) {
|
||||
lastElement.remove();
|
||||
}
|
||||
// 创建 file bubble
|
||||
let size = FILE_BASE64.length;
|
||||
let unit = 'B';
|
||||
if (size > 1024) {
|
||||
size /= 1024;
|
||||
unit = 'KB';
|
||||
}
|
||||
if (size > 1024) {
|
||||
size /= 1024;
|
||||
unit = 'MB';
|
||||
}
|
||||
let fileBubble = document.createElement('div');
|
||||
fileBubble.classList.add('file-bubble');
|
||||
fileBubble.innerHTML = `<img src="data:image/svg+xml;base64,PD94bWwgdmVyc2lvbj0iMS4wIiBlbmNvZGluZz0iaXNvLTg4NTktMSI/Pgo8IS0tIEdlbmVyYXRvcjogQWRvYmUgSWxsdXN0cmF0b3IgMjMuMC4wLCBTVkcgRXhwb3J0IFBsdWctSW4gLiBTVkcgVmVyc2lvbjogNi4wMCBCdWlsZCAwKSAgLS0+CjxzdmcgdmVyc2lvbj0iMS4xIiBpZD0iJiN4NTZGRTsmI3g1QzQyO18xIiB4bWxucz0iaHR0cDovL3d3dy53My5vcmcvMjAwMC9zdmciIHhtbG5zOnhsaW5rPSJodHRwOi8vd3d3LnczLm9yZy8xOTk5L3hsaW5rIiB4PSIwcHgiCgkgeT0iMHB4IiB3aWR0aD0iMTZweCIgaGVpZ2h0PSIxNnB4IiB2aWV3Qm94PSIwIDAgMTYgMTYiIHN0eWxlPSJlbmFibGUtYmFja2dyb3VuZDpuZXcgMCAwIDE2IDE2OyIgeG1sOnNwYWNlPSJwcmVzZXJ2ZSI+CjxwYXRoIGQ9Ik04LjI5NiwxSDNDMi40NDgsMSwyLDEuNDQ4LDIsMnYxM2MwLDAuNTUyLDAuNDQ4LDEsMSwxaDEwYzAuNTUyLDAsMS0wLjQ0OCwxLTFWNS43MDRjMC0wLjI2NS0wLjEwNS0wLjUyLTAuMjkzLTAuNzA3CgljLTAuMzYyLTAuMjg1LTAuNzI1LTAuNTcxLTEuMDg3LTAuODU2Yy0wLjcwNi0wLjU1Ni0xLjQxMi0xLjExMi0yLjExOC0xLjY2OGMtMC4zNjgtMC4yOS0wLjc0NC0wLjU3MS0xLjExLTAuODY0CglDOS4wNTcsMS4zNDIsOC43NTcsMSw4LjI5NiwxeiBNMTIuMjksNUg5VjIuNzFMMTIuMjksNXogTTMsMTVWMmg1djNjMCwwLjU1MiwwLjQ0OCwxLDEsMWg0djlIM3oiLz4KPHJlY3Qgc3R5bGU9ImZpbGw6bm9uZTsiIHdpZHRoPSIxNiIgaGVpZ2h0PSIxNiIvPgo8cmVjdCBzdHlsZT0iZmlsbDpub25lOyIgd2lkdGg9IjE2IiBoZWlnaHQ9IjE2Ii8+Cjwvc3ZnPgo=" alt="File Icon" class="file-icon">
|
||||
<div class="file-info">
|
||||
<div class="file-name">${title}</div>
|
||||
<div class="file-size">${size} ${unit}</div>
|
||||
</div>`;
|
||||
document.querySelector('.chat-body').appendChild(fileBubble);
|
||||
};
|
||||
reader.readAsDataURL(file);
|
||||
}
|
||||
|
||||
/* 废弃 */
|
||||
// eslint-disable-next-line @typescript-eslint/no-unused-vars
|
||||
async function getDatasheetUrl() {
|
||||
try {
|
||||
const selectedPrimitives = await eda.sch_SelectControl.getAllSelectedPrimitives_PrimitiveId();
|
||||
const primitive = await eda.sch_PrimitiveComponent.get(selectedPrimitives[0]);
|
||||
let url = primitive[0].otherProperty.Datasheet;
|
||||
// atta.szlcsc.com -> atta-szlcsc.mirror.soraharu.com
|
||||
url = url.replace('atta.szlcsc.com', 'atta-szlcsc.mirror.soraharu.com');
|
||||
console.log(url);
|
||||
return url;
|
||||
} catch (error) {
|
||||
console.error(error);
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
// eslint-disable-next-line @typescript-eslint/no-unused-vars
|
||||
async function getDatasheetData(url) {
|
||||
console.log(url);
|
||||
try {
|
||||
const response = await eda.sys_ClientUrl.request(url);
|
||||
const data = await response.text();
|
||||
console.log(data);
|
||||
return data;
|
||||
} catch (error) {
|
||||
console.error(error);
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
213
iframe/js/main.js
Normal file
213
iframe/js/main.js
Normal file
@@ -0,0 +1,213 @@
|
||||
let CACHE_NAME = null;
|
||||
let chatHistory = [];
|
||||
let CUSTOM_SETTINGS = {
|
||||
HOST: "http://localhost:11434",
|
||||
ENDPOINT: "/v1/chat/completions",
|
||||
MODEL: "",
|
||||
HEADER: {
|
||||
"Content-Type": "application/json"
|
||||
}
|
||||
};
|
||||
let GOOGLE_SETTINGS = {
|
||||
HOST: "https://generativelanguage.googleapis.com",
|
||||
API_KEY: "",
|
||||
HEADER: {
|
||||
"Content-Type": "application/json"
|
||||
}
|
||||
};
|
||||
let DS_SETTINGS = {
|
||||
HOST: "https://api.deepseek.com",
|
||||
API_KEY: "",
|
||||
HEADER: {
|
||||
"Content-Type": "application/json"
|
||||
}
|
||||
};
|
||||
let QWEN_SETTINGS = {
|
||||
HOST: "https://dashscope.aliyuncs.com",
|
||||
API_KEY: "",
|
||||
HEADER: {
|
||||
"Content-Type": "application/json"
|
||||
}
|
||||
};
|
||||
let MODEL_NAME = "gemini-1.5-flash-002";
|
||||
let MODEL_GROUP = "Gemini";
|
||||
let CHAT_TITLE = "新的对话";
|
||||
let FILE_BASE64 = null;
|
||||
|
||||
console.log(eda.sys_Storage.getExtensionAllUserConfigs());
|
||||
|
||||
if(eda.sys_Storage.getExtensionUserConfig("GOOGLE-SETTINGS") !== undefined) {
|
||||
GOOGLE_SETTINGS = eda.sys_Storage.getExtensionUserConfig("GOOGLE-SETTINGS");
|
||||
}
|
||||
|
||||
if (eda.sys_Storage.getExtensionUserConfig("DS-SETTINGS") !== undefined) {
|
||||
DS_SETTINGS = eda.sys_Storage.getExtensionUserConfig("DS-SETTINGS");
|
||||
}
|
||||
|
||||
if (eda.sys_Storage.getExtensionUserConfig("QWEN-SETTINGS") !== undefined) {
|
||||
QWEN_SETTINGS = eda.sys_Storage.getExtensionUserConfig("QWEN-SETTINGS");
|
||||
}
|
||||
|
||||
if (eda.sys_Storage.getExtensionUserConfig("CUSTOM-SETTINGS") !== undefined) {
|
||||
CUSTOM_SETTINGS = eda.sys_Storage.getExtensionUserConfig("CUSTOM-SETTINGS");
|
||||
}
|
||||
|
||||
if(GOOGLE_SETTINGS.HOST !== null && GOOGLE_SETTINGS.HOST !== "" && GOOGLE_SETTINGS.HOST !== undefined) {
|
||||
document.getElementById("custom-google-host").value = GOOGLE_SETTINGS.HOST;
|
||||
}
|
||||
|
||||
if (GOOGLE_SETTINGS.API_KEY !== null && GOOGLE_SETTINGS.API_KEY !== "" && GOOGLE_SETTINGS.API_KEY !== undefined) {
|
||||
document.getElementById("google-api-key").value = GOOGLE_SETTINGS.API_KEY;
|
||||
}
|
||||
|
||||
if (GOOGLE_SETTINGS.HEADER !== null && GOOGLE_SETTINGS.HEADER !== "" && GOOGLE_SETTINGS.HEADER !== undefined) {
|
||||
document.getElementById("custom-google-header").value = JSON.stringify(GOOGLE_SETTINGS.HEADER);
|
||||
}
|
||||
|
||||
if (DS_SETTINGS.HEADER !== null && DS_SETTINGS.HEADER !== "" && DS_SETTINGS.HEADER !== undefined) {
|
||||
document.getElementById("custom-ds-header").value = JSON.stringify(DS_SETTINGS.HEADER);
|
||||
}
|
||||
|
||||
if (DS_SETTINGS.HOST !== null && DS_SETTINGS.HOST !== "" && DS_SETTINGS.HOST !== undefined) {
|
||||
document.getElementById("custom-ds-host").value = DS_SETTINGS.HOST;
|
||||
}
|
||||
|
||||
if (DS_SETTINGS.API_KEY !== null && DS_SETTINGS.API_KEY !== "" && DS_SETTINGS.API_KEY !== undefined) {
|
||||
document.getElementById("ds-api-key").value = DS_SETTINGS.API_KEY;
|
||||
}
|
||||
|
||||
if (QWEN_SETTINGS.HEADER !== null && QWEN_SETTINGS.HEADER !== "" && QWEN_SETTINGS.HEADER !== undefined) {
|
||||
document.getElementById("custom-qwen-header").value = JSON.stringify(QWEN_SETTINGS.HEADER);
|
||||
}
|
||||
|
||||
if (QWEN_SETTINGS.HOST !== null && QWEN_SETTINGS.HOST !== "" && QWEN_SETTINGS.HOST !== undefined) {
|
||||
document.getElementById("custom-qwen-host").value = QWEN_SETTINGS.HOST;
|
||||
}
|
||||
|
||||
if (QWEN_SETTINGS.API_KEY !== null && QWEN_SETTINGS.API_KEY !== "" && QWEN_SETTINGS.API_KEY !== undefined) {
|
||||
document.getElementById("qwen-api-key").value = QWEN_SETTINGS.API_KEY;
|
||||
}
|
||||
|
||||
if (CUSTOM_SETTINGS.HOST !== null && CUSTOM_SETTINGS.HOST !== "" && CUSTOM_SETTINGS.HOST !== undefined) {
|
||||
document.getElementById("custom-host").value = CUSTOM_SETTINGS.HOST;
|
||||
}
|
||||
|
||||
if (CUSTOM_SETTINGS.ENDPOINT !== null && CUSTOM_SETTINGS.ENDPOINT !== "" && CUSTOM_SETTINGS.ENDPOINT !== undefined) {
|
||||
document.getElementById("custom-endpoint").value = CUSTOM_SETTINGS.ENDPOINT;
|
||||
}
|
||||
|
||||
if (CUSTOM_SETTINGS.HEADER !== null && CUSTOM_SETTINGS.HEADER !== "" && CUSTOM_SETTINGS.HEADER !== undefined) {
|
||||
document.getElementById("custom-header").value = JSON.stringify(CUSTOM_SETTINGS.HEADER);
|
||||
}
|
||||
|
||||
if (CUSTOM_SETTINGS.MODEL !== null && CUSTOM_SETTINGS.MODEL !== "" && CUSTOM_SETTINGS.MODEL !== undefined) {
|
||||
document.getElementById("custom-model").value = CUSTOM_SETTINGS.MODEL;
|
||||
}
|
||||
|
||||
document.getElementById('subtitle').innerText = MODEL_NAME;
|
||||
|
||||
|
||||
|
||||
function openSettings() {
|
||||
document.getElementById('settingsModal').style.display = 'block';
|
||||
}
|
||||
|
||||
function closeSettings() {
|
||||
document.getElementById('settingsModal').style.display = 'none';
|
||||
}
|
||||
|
||||
function saveSettings() {
|
||||
let customSettings = {
|
||||
HOST: document.getElementById('custom-host').value,
|
||||
ENDPOINT: document.getElementById('custom-endpoint').value,
|
||||
MODEL: document.getElementById('custom-model').value,
|
||||
HEADER: JSON.parse(document.getElementById('custom-header').value)
|
||||
};
|
||||
let googleSettings = {
|
||||
HOST: document.getElementById('custom-google-host').value,
|
||||
API_KEY: document.getElementById('google-api-key').value,
|
||||
HEADER: JSON.parse(document.getElementById('custom-google-header').value)
|
||||
};
|
||||
|
||||
let dsSettings = {
|
||||
HOST: document.getElementById('custom-ds-host').value,
|
||||
API_KEY: document.getElementById('ds-api-key').value,
|
||||
HEADER: JSON.parse(document.getElementById('custom-ds-header').value)
|
||||
};
|
||||
|
||||
let qwenSettings = {
|
||||
HOST: document.getElementById('custom-qwen-host').value,
|
||||
API_KEY: document.getElementById('qwen-api-key').value,
|
||||
HEADER: JSON.parse(document.getElementById('custom-qwen-header').value)
|
||||
};
|
||||
eda.sys_Storage.setExtensionUserConfig("CUSTOM-SETTINGS", customSettings);
|
||||
eda.sys_Storage.setExtensionUserConfig("GOOGLE-SETTINGS", googleSettings);
|
||||
eda.sys_Storage.setExtensionUserConfig("DS-SETTINGS", dsSettings);
|
||||
eda.sys_Storage.setExtensionUserConfig("QWEN-SETTINGS", qwenSettings);
|
||||
CUSTOM_SETTINGS = customSettings;
|
||||
GOOGLE_SETTINGS = googleSettings;
|
||||
DS_SETTINGS = dsSettings;
|
||||
QWEN_SETTINGS = qwenSettings;
|
||||
closeSettings();
|
||||
}
|
||||
|
||||
function openModels() {
|
||||
document.getElementById('modelsModal').style.display = 'block';
|
||||
}
|
||||
|
||||
function closeModels() {
|
||||
document.getElementById('modelsModal').style.display = 'none';
|
||||
}
|
||||
|
||||
function saveModels() {
|
||||
MODEL_NAME = document.getElementById('model-name').value;
|
||||
document.getElementById('subtitle').innerText = MODEL_NAME;
|
||||
|
||||
const select = document.getElementById('model-name');
|
||||
const selectedOption = select.options[select.selectedIndex];
|
||||
const selectedOptgroup = selectedOption.parentNode;
|
||||
|
||||
if (selectedOptgroup.tagName === 'OPTGROUP') {
|
||||
console.log('选中的 optgroup 标签:', selectedOptgroup.label);
|
||||
MODEL_GROUP = selectedOptgroup.label;
|
||||
if (MODEL_GROUP === 'Gemini') {
|
||||
document.getElementById('google-settings').style.display = 'block';
|
||||
document.getElementById('ds-settings').style.display = 'none';
|
||||
document.getElementById('qwen-settings').style.display = 'none';
|
||||
document.getElementById('custom-settings').style.display = 'none';
|
||||
} else if (MODEL_GROUP === 'DeepSeek') {
|
||||
document.getElementById('ds-settings').style.display = 'block';
|
||||
document.getElementById('google-settings').style.display = 'none';
|
||||
document.getElementById('qwen-settings').style.display = 'none';
|
||||
document.getElementById('custom-settings').style.display = 'none';
|
||||
} else if (MODEL_GROUP === 'QWen') {
|
||||
document.getElementById('qwen-settings').style.display = 'block';
|
||||
document.getElementById('google-settings').style.display = 'none';
|
||||
document.getElementById('ds-settings').style.display = 'none';
|
||||
document.getElementById('custom-settings').style.display = 'none';
|
||||
} else {
|
||||
document.getElementById('custom-settings').style.display = 'block';
|
||||
document.getElementById('google-settings').style.display = 'none';
|
||||
document.getElementById('ds-settings').style.display = 'none';
|
||||
document.getElementById('qwen-settings').style.display = 'none';
|
||||
}
|
||||
}
|
||||
|
||||
closeModels();
|
||||
}
|
||||
|
||||
function createBubble(type, message, meta) {
|
||||
let bubble = document.createElement('div');
|
||||
bubble.className = 'message-bubble ' + type;
|
||||
console.log(message);
|
||||
bubble.innerHTML = '<div>' + marked.parse(message) + '</div>';
|
||||
if (!meta) {
|
||||
meta = new Date().toLocaleTimeString();
|
||||
}
|
||||
let metaElement = document.createElement('div');
|
||||
metaElement.className = 'message-meta';
|
||||
metaElement.innerHTML = meta;
|
||||
bubble.appendChild(metaElement);
|
||||
document.querySelector('.chat-body').appendChild(bubble);
|
||||
document.querySelector('.chat-body').scrollTop = document.querySelector('.chat-body').scrollHeight;
|
||||
}
|
||||
2592
iframe/js/marked.umd.js
Normal file
2592
iframe/js/marked.umd.js
Normal file
File diff suppressed because it is too large
Load Diff
Reference in New Issue
Block a user