pengjunjie/onlyoffice.github.io
1
0
Fork 1
mirror of https://github.com/ONLYOFFICE/onlyoffice.github.io.git synced 2026-04-07 14:04:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Full worked version

Browse Source
This commit is contained in:
Oleg Korshul
2025-02-11 15:46:49 +03:00
parent dff84387af
commit f1ca359c33
7 changed files with 172 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

83
sdkjs-plugins/content/ai/scripts/engine/engine.js
View File

@ -117,6 +117,12 @@
return provider.getRequestHeaderOptions();
};
AI._getModelsSync = function(_provider) {
let provider = _provider.createInstance ? _provider : AI.Storage.getProvider(_provider.name);
if (!provider) provider = new AI.Provider();
return provider.getModels();
};
AI._extendBody = function(_provider, body) {
let provider = _provider.createInstance ? _provider : AI.Storage.getProvider(_provider.name);
if (!provider) provider = new AI.Provider();
@ -137,7 +143,7 @@
let provider = _provider.createInstance ? _provider : AI.Storage.getProvider(_provider.name);
if (!provider) provider = new AI.Provider(_provider.name, _provider.url, _provider.key);
if (_provider.key && !provider.key)
if (_provider.key)
provider.key = _provider.key;
let url = provider.url;
@ -158,12 +164,8 @@
{
AI.TmpProviderForModels = null;
return new Promise(function (resolve, reject) {
let headers = AI._getHeaders(provider);
requestWrapper({
url : AI._getEndpointUrl(provider, AI.Endpoints.Types.v1.Models),
headers : headers,
method : "GET"
}).then(function(data) {
function resolveRequest(data) {
if (data.error)
resolve({
error : 1,
@ -201,6 +203,25 @@
models : AI.TmpProviderForModels.modelsUI
});
}
}
let syncModels = AI._getModelsSync(provider);
if (Array.isArray(syncModels))
{
resolveRequest({
error : 0,
data : syncModels
});
return;
}
let headers = AI._getHeaders(provider);
requestWrapper({
url : AI._getEndpointUrl(provider, AI.Endpoints.Types.v1.Models),
headers : headers,
method : "GET"
}).then(function(data) {
resolveRequest(data);
});
});
};
@ -221,7 +242,9 @@
if (provider) {
for (let i = 0, len = provider.models.length; i < len; i++) {
if (model.id === provider.models[i].id) {
if (model.id === provider.models[i].id ||
model.id === provider.models[i].name)
{
this.model = provider.models[i];
}
}
@ -363,40 +386,27 @@
AI.Endpoints.Types.v1.Chat_Completions;
objRequest.url = AI._getEndpointUrl(provider, endpointType, this.model);
objRequest.body = {
model : this.modelUI.id
};
objRequest.isUseProxy = AI._extendBody(provider, objRequest.body);
let requestBody = {};
let processResult = function(data) {
let arrResult = data.data.choices || data.data.content;
if (!arrResult)
let result = provider.getChatCompletionsResult(data, this.model);
if (result.content.length === 0)
return "";
let choice = arrResult[0];
if (!choice)
return "";
let text = "";
if (choice.message)
text = choice.message.content;
if (choice.text)
text = choice.text;
let i = 0; let trimStartCh = "\n".charCodeAt(0);
while (text.charCodeAt(i) === trimStartCh)
i++;
if (i > 0)
text = text.substring(i);
return text;
return result.content[0];
};
if (1 === messages.length) {
if (!isUseCompletionsInsteadChat) {
objRequest.body.messages = isMessages ? messages[0] : [{role:"user",content:messages[0]}];
if (isMessages)
requestBody.messages = messages[0];
else
requestBody.messages = [{role:"user",content:messages[0]}];
objRequest.body = provider.getChatCompletions(requestBody, this.model);
} else {
objRequest.body.prompt = messages[0];
objRequest.body = provider.getCompletions({ text : messages[0] });
}
objRequest.isUseProxy = AI._extendBody(provider, objRequest.body);
let result = await requestWrapper(objRequest);
if (result.error) {
throw {
@ -434,16 +444,17 @@
return footer;
}
let isBadAI = false;
for (let i = 0, len = messages.length; i < len; i++) {
let message = getHeader(i + 1, len) + messages[i] + getFooter(i + 1, len);
if (!isUseCompletionsInsteadChat) {
objRequest.body.messages = [{role:"user",content:message}];
objRequest.body = provider.getChatCompletions({ messages : [{role:"user",content:message}] });
} else {
objRequest.body.prompt = message;
objRequest.body = provider.getCompletions( { text : message });
}
objRequest.isUseProxy = AI._extendBody(provider, objRequest.body);
let result = await requestWrapper(objRequest);
if (result.error) {
throw {

2
sdkjs-plugins/content/ai/scripts/engine/local_storage.js
View File

@ -78,7 +78,7 @@
for (let i in obj.providers) {
let pr = obj.providers[i];
AI.Providers[i] = AI.createProviderInstance(pr.name, pr.url, pr.key);
AI.Providers[i] = AI.createProviderInstance(pr.name, pr.url, pr.key, pr.addon);
AI.Providers[i].models = pr.models || [];
if (fixVersion2) {

4
sdkjs-plugins/content/ai/scripts/engine/providers/base.js
View File

@ -116,10 +116,10 @@
AI.CapabilitiesUI.All = capabilitiesAll;
AI.InternalProviders = [];
AI.createProviderInstance = function(name, url, key) {
AI.createProviderInstance = function(name, url, key, addon) {
for (let i = 0, len = window.AI.InternalProviders.length; i < len; i++) {
if (name === AI.InternalProviders[i].name)
return AI.InternalProviders[i].createInstance(name, url, key, AI.InternalProviders[i].addon);
return AI.InternalProviders[i].createInstance(name, url, key, addon || AI.InternalProviders[i].addon);
}
return new Provider(name, url, key);
};

8
sdkjs-plugins/content/ai/scripts/engine/providers/internal/anthropic.js
View File

@ -29,7 +29,13 @@ class Provider extends AI.Provider {
getEndpointUrl(endpoint, model) {
if (AI.Endpoints.Types.v1.Chat_Completions === endpoint)
return "/messages";
return super.getEndpointUrl();
return super.getEndpointUrl(endpoint, model);
}
getRequestBodyOptions() {
return {
max_tokens : 4096
};
}
getRequestHeaderOptions() {

14
sdkjs-plugins/content/ai/scripts/engine/providers/internal/google-gemini.js
View File

@ -73,4 +73,18 @@ class Provider extends AI.Provider {
return headers;
}
getChatCompletions(message, model) {
let body = { contents : [] };
for (let i = 0, len = message.messages.length; i < len; i++) {
let rec = {
role : message.messages[i].role,
parts : [ { text : message.messages[i].content } ]
};
if (rec.role === "assistant")
rec.role = "model";
body.contents.push(rec);
}
return body;
}
}

20
sdkjs-plugins/content/ai/scripts/engine/providers/internal/ollama.js
View File

@ -3,25 +3,7 @@
class Provider extends AI.Provider {
constructor() {
super("Ollama", "http://localhost:11434/api", "", "");
super("Ollama", "http://localhost:11434", "", "v1");
}
getEndpointUrl(endpoint, model) {
let Types = AI.Endpoints.Types;
switch (endpoint)
{
case Types.v1.Models:
return "/tags";
case Types.v1.Chat_Completions:
return "/chat";
case Types.v1.Completions:
return "/generate";
default:
break;
}
return super.getEndpointUrl();
}
}

100
sdkjs-plugins/content/ai/scripts/engine/providers/provider.js
View File

@ -20,10 +20,22 @@
this.modelsUI = [];
}
/**
* If you add an implementation here, then no request will be made to the service.
* @returns {Object[] | undefined}
*/
getModels() {
return undefined;
}
/**
* Correct received (*models* endpoint) model object.
*/
correctModelInfo(model) {
if (undefined === model.id && model.name) {
model.id = model.name;
return;
}
model.name = model.id;
}
@ -136,6 +148,94 @@
return false;
}
/**
* Get request body object by message.
* @param {Object} message
* *message* is in folowing format:
* {
* messages: [
* { role: "developer", content: "You are a helpful assistant." },
* { role: "system", content: "You are a helpful assistant." },
* { role: "user", content: "Hello" },
* { role: "assistant", content: "Hey!" },
* { role: "user", content: "Hello" },
* { role: "assistant", content: "Hey again!" }
* ]
* }
*/
getChatCompletions(message, model) {
return {
model : model.id,
messages : message.messages
}
}
/**
* Get request body object by message.
* @param {Object} message
* *message* is in folowing format:
* {
* text: "Please, calculate 2+2."
* }
*/
getCompletions(message, model) {
return {
model : model.id,
prompt : message.text
}
}
/**
* Convert *getChatCompletions* and *getCompletions* answer to result simple message.
* @returns {Object} result
* *result* is in folowing format:
* {
* content: ["Hello", "Hi"]
* }
*/
getChatCompletionsResult(message, model) {
let result = {
content : []
};
let arrResult = message.data.choices || message.data.content || message.data.candidates;
if (!arrResult)
return result;
let choice = arrResult[0];
if (!choice)
return result;
if (choice.message && choice.message.content)
result.content.push(choice.message.content);
if (choice.text)
result.content.push(choice.message.text);
if (choice.content) {
if (typeof(choice.content) === "string")
result.content.push(choice.content);
else if (Array.isArray(choice.content.parts)) {
for (let i = 0, len = choice.content.parts.length; i < len; i++) {
result.content.push(choice.content.parts[i].text);
}
}
}
let trimArray = ["\n".charCodeAt(0)];
for (let i = 0, len = result.content.length; i < len; i++) {
let iEnd = result.content[i].length - 1;
let iStart = 0;
while (iStart < iEnd && trimArray.includes(result.content[i].charCodeAt(iStart)))
iStart++;
while (iEnd > iStart && trimArray.includes(result.content[i].charCodeAt(iEnd)))
iEnd--;
if (iEnd > iStart && ((0 !== iStart) || ((result.content[i].length - 1) !== iEnd)))
result.content[i] = result.content[i].substring(iStart, iEnd + 1);
}
return result;
}
/**
* ========================================================================================
* The following are methods for internal work. There is no need to overload these methods.