Upgrade translator playground

This commit is contained in:
Kwaroran
2025-02-25 04:36:58 +09:00
parent 2af57d3c67
commit ed2957fc50
4 changed files with 123 additions and 9 deletions

View File

@@ -1371,6 +1371,7 @@ export async function exportCharacterCard(char:character, type:'png'|'json'|'cha
}
card.data.assets[i].uri = 'embeded://' + path
const imageType = checkImageType(rData)
const metaPath = `x_meta/${name}.json`
if(imageType === 'PNG' && writer instanceof CharXWriter){
const metadatas:Record<string,string> = {}
const gen = PngChunk.readGenerator(rData)
@@ -1380,10 +1381,20 @@ export async function exportCharacterCard(char:character, type:'png'|'json'|'cha
}
metadatas[chunk.key] = chunk.value
}
console.log(metadatas)
if(Object.keys(metadatas).length > 0){
const metaPath = `x_meta/${name}.json`
await writer.write(metaPath, Buffer.from(JSON.stringify(metadatas, null, 4)), 6)
}
else{
await writer.write(metaPath, Buffer.from(JSON.stringify({
'type': imageType
}), 'utf-8'), 6)
}
}
else{
await writer.write(metaPath, Buffer.from(JSON.stringify({
'type': imageType
}), 'utf-8'), 6)
}
await writer.write(path, Buffer.from(await convertImage(rData)))
}

View File

@@ -247,7 +247,7 @@ export async function requestChatData(arg:requestDataArgument, model:ModelModeEx
try{
const currentChar = getCurrentCharacter()
if(currentChar.type !== 'group'){
if(currentChar?.type !== 'group'){
const perf = performance.now()
const d = await runTrigger(currentChar, 'request', {
chat: getCurrentChat(),

View File

@@ -41,7 +41,7 @@ export async function translate(text:string, reverse:boolean) {
return runTranslator(text, reverse, db.translator,db.aiModel.startsWith('novellist') ? 'ja' : 'en')
}
export async function runTranslator(text:string, reverse:boolean, from:string,target:string) {
export async function runTranslator(text:string, reverse:boolean, from:string,target:string, exarg?:{translatorNote?:string}) {
const arg = {
from: reverse ? from : target,
@@ -50,6 +50,7 @@ export async function runTranslator(text:string, reverse:boolean, from:string,ta
host: 'translate.googleapis.com',
translatorNote: exarg?.translatorNote
}
const texts = text.split('\n')
let chunks:[string,boolean][] = [['', true]]
@@ -104,11 +105,11 @@ export async function runTranslator(text:string, reverse:boolean, from:string,ta
}
async function translateMain(text:string, arg:{from:string, to:string, host:string}){
async function translateMain(text:string, arg:{from:string, to:string, host:string, translatorNote?:string}){
let db = getDatabase()
if(db.translatorType === 'llm'){
const tr = arg.to || 'en'
return translateLLM(text, {to: tr, from: arg.from})
return translateLLM(text, {to: tr, from: arg.from, translatorNote: arg.translatorNote})
}
if(db.translatorType === 'deepl'){
const body = {
@@ -473,7 +474,7 @@ function needSuperChunkedTranslate(){
return getDatabase().translatorType === 'deeplX'
}
async function translateLLM(text:string, arg:{to:string, from:string, regenerate?:boolean}):Promise<string>{
async function translateLLM(text:string, arg:{to:string, from:string, regenerate?:boolean,translatorNote?:string}):Promise<string>{
if(!arg.regenerate){
const cacheMatch = await LLMCacheStorage.getItem(text)
if(cacheMatch){
@@ -490,12 +491,17 @@ async function translateLLM(text:string, arg:{to:string, from:string, regenerate
const db = getDatabase()
const charIndex = get(selectedCharID)
const currentChar = db.characters[charIndex]
let translatorNote
if (currentChar?.type === "character") {
let translatorNote = ""
console.log(arg.translatorNote)
if(arg.translatorNote){
translatorNote = arg.translatorNote
}
else if (currentChar?.type === "character") {
translatorNote = currentChar.translatorNote ?? ""
} else {
translatorNote = ""
}
console.log(translatorNote)
let formated:OpenAIChat[] = []
let prompt = db.translatorPrompt || `You are a translator. translate the following html or text into {{slot}}. do not output anything other than the translation.`