mincomk/risuai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[feat] added translator api to plugin

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

46
public/pluginApi.ts
View File

@@ -1,30 +1,31 @@
interface risuPlugin{
(() => {
interface risuPlugin{
providers: {name:string, func:(arg:providerArgument) => Promise<{success:boolean,content:string}>}[]
fetchResponseQueue:{id:string,data:any}[]
}
}
let __risuPlugin__:risuPlugin = {
let __risuPlugin__:risuPlugin = {
providers: [],
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'
content: string
}
}
interface providerArgument{
interface providerArgument{
prompt_chat?: OpenAIChat,
temperature?: number,
max_tokens?: number,
presence_penalty?: number
frequency_penalty?: number
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()}`
postMessage({
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()}`
postMessage({
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({
type: 'addProvider',
body: name
@@ -80,16 +81,16 @@ function addProvider(name:string, func:(arg:providerArgument) => Promise<{succes
name: name,
func: func
})
}
}
function printLog(data:any){
function printLog(data:any){
postMessage({
type: 'log',
body: data
})
}
}
async function handleOnmessage(data:{type:string,body:any}) {
async function handleOnmessage(data:{type:string,body:any}) {
if(!data.type){
return
}
@@ -135,9 +136,12 @@ async function handleOnmessage(data:{type:string,body:any}) {
break
}
}
}
}
onmessage = (ev) => {
onmessage = (ev) => {
handleOnmessage(ev.data)
const data:{type:string,body:any} = ev.data
}
}
//{{placeholder}}
})()

59
src/ts/process/plugins.ts
View File

@@ -107,6 +107,7 @@ export function getCurrentPluginMax(prov:string){
let pluginWorker:Worker = null
let providerRes:{success:boolean, content:string} = null
let translatorRes:{success:boolean, content:string} = null
function postMsgPluginWorker(type:string, body:any){
const bod = {
@@ -116,6 +117,8 @@ function postMsgPluginWorker(type:string, body:any){
pluginWorker.postMessage(bod)
}
let pluginTranslator = false
export async function loadPlugins() {
let db = get(DataBase)
if(pluginWorker){
@@ -127,10 +130,12 @@ export async function loadPlugins() {
const da = await fetch("/pluginApi.js")
const pluginApiString = await da.text()
let pluginjs = `${pluginApiString}\n`
let pluginLoadedJs = ''
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'});
pluginWorker = new Worker(URL.createObjectURL(blob));
@@ -167,6 +172,31 @@ export async function loadPlugins() {
}
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": {
postMsgPluginWorker('fetchData',{
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:{
prompt_chat: OpenAIChat,
temperature: number,

19
src/ts/translator/translator.ts
View File

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