[feat] added translator api to plugin

This commit is contained in:
kwaroran
2023-05-07 22:03:43 +09:00
parent acbf4b0cc2
commit 10863cee51
3 changed files with 202 additions and 132 deletions

View File

@@ -1,30 +1,31 @@
interface risuPlugin{ (() => {
interface risuPlugin{
providers: {name:string, func:(arg:providerArgument) => Promise<{success:boolean,content:string}>}[] providers: {name:string, func:(arg:providerArgument) => Promise<{success:boolean,content:string}>}[]
fetchResponseQueue:{id:string,data:any}[] fetchResponseQueue:{id:string,data:any}[]
} }
let __risuPlugin__:risuPlugin = { let __risuPlugin__:risuPlugin = {
providers: [], providers: [],
fetchResponseQueue: [] fetchResponseQueue: []
} }
const sleep = (ms: number) => new Promise((r) => setTimeout(r, ms)); const sleep = (ms: number) => new Promise((r) => setTimeout(r, ms));
interface OpenAIChat{ interface OpenAIChat{
role: 'system'|'user'|'assistant' role: 'system'|'user'|'assistant'
content: string content: string
} }
interface providerArgument{ interface providerArgument{
prompt_chat?: OpenAIChat, prompt_chat?: OpenAIChat,
temperature?: number, temperature?: number,
max_tokens?: number, max_tokens?: number,
presence_penalty?: number presence_penalty?: number
frequency_penalty?: number frequency_penalty?: number
bias?: {[key:string]:string} bias?: {[key:string]:string}
} }
async function risuFetch(url:string, arg:{body:any,headers?:{[key:string]:string}}){ async function risuFetch(url:string, arg:{body:any,headers?:{[key:string]:string}}){
const id = `${Date.now()}_${Math.random()}` const id = `${Date.now()}_${Math.random()}`
postMessage({ postMessage({
type: 'fetch', type: 'fetch',
@@ -48,9 +49,9 @@ async function risuFetch(url:string, arg:{body:any,headers?:{[key:string]:string
} }
} }
} }
} }
async function getArg(arg:string){ async function getArg(arg:string){
const id = `${Date.now()}_${Math.random()}` const id = `${Date.now()}_${Math.random()}`
postMessage({ postMessage({
type: 'getArg', type: 'getArg',
@@ -69,9 +70,9 @@ async function getArg(arg:string){
} }
} }
} }
} }
function addProvider(name:string, func:(arg:providerArgument) => Promise<{success:boolean,content:string}>){ function addProvider(name:string, func:(arg:providerArgument) => Promise<{success:boolean,content:string}>){
postMessage({ postMessage({
type: 'addProvider', type: 'addProvider',
body: name body: name
@@ -80,16 +81,16 @@ function addProvider(name:string, func:(arg:providerArgument) => Promise<{succes
name: name, name: name,
func: func func: func
}) })
} }
function printLog(data:any){ function printLog(data:any){
postMessage({ postMessage({
type: 'log', type: 'log',
body: data body: data
}) })
} }
async function handleOnmessage(data:{type:string,body:any}) { async function handleOnmessage(data:{type:string,body:any}) {
if(!data.type){ if(!data.type){
return return
} }
@@ -135,9 +136,12 @@ async function handleOnmessage(data:{type:string,body:any}) {
break break
} }
} }
} }
onmessage = (ev) => { onmessage = (ev) => {
handleOnmessage(ev.data) handleOnmessage(ev.data)
const data:{type:string,body:any} = ev.data const data:{type:string,body:any} = ev.data
} }
//{{placeholder}}
})()

View File

@@ -107,6 +107,7 @@ export function getCurrentPluginMax(prov:string){
let pluginWorker:Worker = null let pluginWorker:Worker = null
let providerRes:{success:boolean, content:string} = null let providerRes:{success:boolean, content:string} = null
let translatorRes:{success:boolean, content:string} = null
function postMsgPluginWorker(type:string, body:any){ function postMsgPluginWorker(type:string, body:any){
const bod = { const bod = {
@@ -116,6 +117,8 @@ function postMsgPluginWorker(type:string, body:any){
pluginWorker.postMessage(bod) pluginWorker.postMessage(bod)
} }
let pluginTranslator = false
export async function loadPlugins() { export async function loadPlugins() {
let db = get(DataBase) let db = get(DataBase)
if(pluginWorker){ if(pluginWorker){
@@ -127,10 +130,12 @@ export async function loadPlugins() {
const da = await fetch("/pluginApi.js") const da = await fetch("/pluginApi.js")
const pluginApiString = await da.text() const pluginApiString = await da.text()
let pluginjs = `${pluginApiString}\n` let pluginjs = `${pluginApiString}\n`
let pluginLoadedJs = ''
for(const plug of db.plugins){ for(const plug of db.plugins){
pluginjs += `(() => {${plug.script}})()` pluginLoadedJs += `(() => {${plug.script}})()`
} }
pluginjs = pluginjs.replace('//{{placeholder}}',pluginLoadedJs)
const blob = new Blob([pluginjs], {type: 'application/javascript'}); const blob = new Blob([pluginjs], {type: 'application/javascript'});
pluginWorker = new Worker(URL.createObjectURL(blob)); pluginWorker = new Worker(URL.createObjectURL(blob));
@@ -167,6 +172,31 @@ export async function loadPlugins() {
} }
break break
} }
case "resTrans":{
const provres:{success:boolean, content:string} = data.body
if(checkNullish(provres.success) || checkNullish(provres.content)){
translatorRes = {
success: false,
content :"plugin didn't respond 'success' or 'content' in response object"
}
}
else if(typeof(provres.content) !== 'string'){
translatorRes = {
success: false,
content :"plugin didn't respond 'content' in response object in string"
}
}
else{
translatorRes = {
success: !!provres.success,
content: provres.content
}
}
break
}
case "useTranslator": {
pluginTranslator = true
}
case "fetch": { case "fetch": {
postMsgPluginWorker('fetchData',{ postMsgPluginWorker('fetchData',{
id: data.body.id, id: data.body.id,
@@ -208,6 +238,33 @@ export async function loadPlugins() {
} }
} }
export async function translatorPlugin(text:string, from:string, to:string) {
if(!pluginTranslator){
return false
}
else{
try {
translatorRes = null
postMsgPluginWorker("requestTrans", {text, from, to})
while(true){
await sleep(50)
if(providerRes){
break
}
}
return {
success: translatorRes.success,
content: translatorRes.content
}
} catch (error) {
return {
success: false,
content: "unknownError"
}
}
}
}
export async function pluginProcess(arg:{ export async function pluginProcess(arg:{
prompt_chat: OpenAIChat, prompt_chat: OpenAIChat,
temperature: number, temperature: number,

View File

@@ -1,28 +1,34 @@
import { Body,fetch,ResponseType } from "@tauri-apps/api/http" import { Body,fetch,ResponseType } from "@tauri-apps/api/http"
import { isTauri } from "../globalApi" import { isTauri } from "../globalApi"
import { translatorPlugin } from "../process/plugins"
let cache={ let cache={
origin: [''], origin: [''],
trans: [''] trans: ['']
} }
export async function translate(params:string, reverse:boolean) { export async function translate(text:string, reverse:boolean) {
if(!isTauri){ if(!isTauri){
return params return text
}
const plug = await translatorPlugin(text, reverse ? 'ko' : 'en', reverse ? 'en' : 'ko')
if(plug){
return plug.content
} }
if(!reverse){ if(!reverse){
const ind = cache.origin.indexOf(params) const ind = cache.origin.indexOf(text)
if(ind !== -1){ if(ind !== -1){
return cache.trans[ind] return cache.trans[ind]
} }
} }
else{ else{
const ind = cache.trans.indexOf(params) const ind = cache.trans.indexOf(text)
if(ind !== -1){ if(ind !== -1){
return cache.origin[ind] return cache.origin[ind]
} }
} }
return googleTrans(params, reverse)
return googleTrans(text, reverse)
} }
async function googleTrans(text:string, reverse:boolean) { async function googleTrans(text:string, reverse:boolean) {
@@ -45,5 +51,8 @@ async function googleTrans(text:string, reverse:boolean) {
}) })
const res = f.data as {sentences:{trans?:string}[]} const res = f.data as {sentences:{trans?:string}[]}
if(typeof(f.data) === 'string'){
return res
}
return res.sentences.filter((s) => 'trans' in s).map((s) => s.trans).join(''); return res.sentences.filter((s) => 'trans' in s).map((s) => s.trans).join('');
} }