This commit is contained in:
2026-04-20 12:45:00 -04:00
parent fbd1728aee
commit bdad27d6ac
3 changed files with 276 additions and 189 deletions

View File

@@ -14,13 +14,19 @@ import {
type Participant
} from '../stores/settings';
import { cn } from '../lib/utils';
import { invoke } from '@tauri-apps/api/core';
import { listen } from '@tauri-apps/api/event';
import { useClipboard } from '../composables/useClipboard';
import {
executeTranslationRequest,
extractAssistantContent,
resolveModelConfig,
tryParseEvaluationResult,
type TranslationChunkEvent,
type TranslationPayload,
} from '../lib/translation-service';
const settings = useSettingsStore();
const { activeCopyId, copyWithFeedback } = useClipboard();
interface TranslationChunkEvent { request_id: string; chunk: string; }
// UI States
const searchQuery = ref('');
@@ -137,12 +143,6 @@ onMounted(async () => {
});
onUnmounted(() => { if (unlisten) unlisten(); });
const generateCurl = (baseUrl: string, key: string, body: any) => {
const fullUrl = `${baseUrl.replace(/\/$/, '')}/chat/completions`;
const maskedKey = key ? `${key.slice(0, 6)}...${key.slice(-4)}` : 'YOUR_API_KEY';
return `curl "${fullUrl}" \\\n -H "Content-Type: application/json" \\\n -H "Authorization: Bearer ${maskedKey}" \\\n -d '${JSON.stringify(body, null, 2)}'`;
};
const translateMessage = async (sender: 'me' | 'partner', retranslateId?: string) => {
if (!activeSession.value || isTranslating.value) return;
@@ -205,7 +205,7 @@ const translateMessage = async (sender: 'me' | 'partner', retranslateId?: string
.replace(/{TO_LANG}/g, toLang.englishName)
.replace(/{TARGET_TONE}/g, targetTone);
const requestBody = {
const requestBody: TranslationPayload = {
model: settings.modelName,
messages: [
{ role: "system", content: systemPrompt },
@@ -213,31 +213,24 @@ const translateMessage = async (sender: 'me' | 'partner', retranslateId?: string
],
stream: settings.enableStreaming
};
const requestId = crypto.randomUUID();
settings.addLog('request', { type: retranslateId ? 'conversation-retranslate' : 'conversation', ...requestBody }, generateCurl(settings.apiBaseUrl, settings.apiKey, requestBody));
try {
if (settings.enableStreaming) activeStreamRequestId.value = requestId;
const response = await invoke<string>('translate', {
apiAddress: settings.apiBaseUrl,
apiKey: settings.apiKey,
const response = await executeTranslationRequest({
apiAddress: settings.apiBaseUrl,
apiKey: settings.apiKey,
payload: requestBody,
requestId
logger: settings,
logType: retranslateId ? 'conversation-retranslate' : 'conversation',
onStreamStart: (requestId) => {
activeStreamRequestId.value = requestId;
},
});
if (!settings.enableStreaming) {
const fullResponseJson = JSON.parse(response);
settings.addLog('response', fullResponseJson);
const translatedText = fullResponseJson.choices?.[0]?.message?.content || response;
settings.updateChatMessage(activeSession.value.id, messageId, { translated: translatedText });
} else {
settings.addLog('response', response);
settings.updateChatMessage(activeSession.value.id, messageId, { translated: extractAssistantContent(response) });
}
} catch (err: any) {
const errorMsg = String(err);
settings.addLog('error', errorMsg);
settings.updateChatMessage(activeSession.value.id, messageId, { translated: `Error: ${errorMsg}` });
settings.updateChatMessage(activeSession.value.id, messageId, { translated: `Error: ${String(err)}` });
} finally {
isTranslating.value = false;
currentStreamingMessageId.value = null;
@@ -305,45 +298,31 @@ const evaluateMessage = async (messageId: string, force = false) => {
const userPrompt = `[Source Text]\n${msg.original}\n\n[Current Translation]\n${msg.translated}`;
// 使用审计专用配置
let evalApiBaseUrl = settings.apiBaseUrl;
let evalApiKey = settings.apiKey;
let evalModelName = settings.modelName;
const modelConfig = resolveModelConfig({
apiBaseUrl: settings.apiBaseUrl,
apiKey: settings.apiKey,
modelName: settings.modelName,
}, settings.profiles, settings.evaluationProfileId);
if (settings.evaluationProfileId) {
const profile = settings.profiles.find(p => p.id === settings.evaluationProfileId);
if (profile) {
evalApiBaseUrl = profile.apiBaseUrl;
evalApiKey = profile.apiKey;
evalModelName = profile.modelName;
}
}
const requestBody = {
model: evalModelName,
const requestBody: TranslationPayload = {
model: modelConfig.modelName,
messages: [
{ role: "system", content: systemPrompt },
{ role: "user", content: userPrompt }
],
stream: false
};
const requestId = crypto.randomUUID();
settings.addLog('request', { type: 'conversation-eval', ...requestBody }, generateCurl(evalApiBaseUrl, evalApiKey, requestBody));
try {
const response = await invoke<string>('translate', {
apiAddress: evalApiBaseUrl,
apiKey: evalApiKey,
const response = await executeTranslationRequest({
apiAddress: modelConfig.apiBaseUrl,
apiKey: modelConfig.apiKey,
payload: requestBody,
requestId
logger: settings,
logType: 'conversation-eval',
});
const fullResponseJson = JSON.parse(response);
settings.addLog('response', fullResponseJson);
const evaluationJson = fullResponseJson.choices?.[0]?.message?.content || response;
settings.updateChatMessage(activeSession.value.id, messageId, { evaluation: evaluationJson });
} catch (err: any) {
settings.addLog('error', String(err));
settings.updateChatMessage(activeSession.value.id, messageId, { evaluation: extractAssistantContent(response) });
} catch {
} finally {
settings.updateChatMessage(activeSession.value.id, messageId, { isEvaluating: false });
}
@@ -360,10 +339,8 @@ const refineMessage = async (messageId: string) => {
const msg = activeSession.value.messages.find(m => m.id === messageId);
if (!msg || !msg.evaluation || msg.isRefining) return;
let evalData;
try {
evalData = parseEvaluation(msg.evaluation);
} catch (e) { return; }
const parsedEvaluation = tryParseEvaluationResult(msg.evaluation);
const evalData = parsedEvaluation.result;
// 仅使用选中的建议
const selectedSuggestions = evalData.suggestions.filter((s: any) => selectedSuggestionIds.value.includes(s.id));
@@ -410,51 +387,37 @@ const refineMessage = async (messageId: string) => {
.replace(/{FROM_LANG}/g, fromLang.englishName)
.replace(/{TO_LANG}/g, toLang.englishName);
// 使用审计专用配置
let refineApiBaseUrl = settings.apiBaseUrl;
let refineApiKey = settings.apiKey;
let refineModelName = settings.modelName;
const modelConfig = resolveModelConfig({
apiBaseUrl: settings.apiBaseUrl,
apiKey: settings.apiKey,
modelName: settings.modelName,
}, settings.profiles, settings.evaluationProfileId);
if (settings.evaluationProfileId) {
const profile = settings.profiles.find(p => p.id === settings.evaluationProfileId);
if (profile) {
refineApiBaseUrl = profile.apiBaseUrl;
refineApiKey = profile.apiKey;
refineModelName = profile.modelName;
}
}
const requestBody = {
model: refineModelName,
const requestBody: TranslationPayload = {
model: modelConfig.modelName,
messages: [
{ role: "system", content: systemPrompt },
{ role: "user", content: `[Source Text]\n${msg.original}\n\n[Current Translation]\n${currentTranslation}\n\n[Suggestions]\n${suggestionsText}` }
],
stream: settings.enableStreaming
};
const requestId = crypto.randomUUID();
settings.addLog('request', { type: 'conversation-refine', ...requestBody }, generateCurl(refineApiBaseUrl, refineApiKey, requestBody));
try {
if (settings.enableStreaming) activeStreamRequestId.value = requestId;
const response = await invoke<string>('translate', {
apiAddress: refineApiBaseUrl,
apiKey: refineApiKey,
const response = await executeTranslationRequest({
apiAddress: modelConfig.apiBaseUrl,
apiKey: modelConfig.apiKey,
payload: requestBody,
requestId
logger: settings,
logType: 'conversation-refine',
onStreamStart: (requestId) => {
activeStreamRequestId.value = requestId;
},
});
if (!settings.enableStreaming) {
const fullResponseJson = JSON.parse(response);
settings.addLog('response', fullResponseJson);
const refinedText = fullResponseJson.choices?.[0]?.message?.content || response;
settings.updateChatMessage(activeSession.value.id, messageId, { translated: refinedText });
} else {
settings.addLog('response', response);
settings.updateChatMessage(activeSession.value.id, messageId, { translated: extractAssistantContent(response) });
}
} catch (err: any) {
settings.addLog('error', String(err));
} catch {
} finally {
settings.updateChatMessage(activeSession.value.id, messageId, { isRefining: false, evaluation: undefined });
currentStreamingMessageId.value = null;
@@ -468,20 +431,7 @@ const refineMessage = async (messageId: string) => {
}
};
const parseEvaluation = (evalStr?: string) => {
if (!evalStr) return null;
try {
// 1. 清洗数据:剔除 Markdown 代码块标记 (如 ```json ... ``` 或 ``` ...)
let cleanStr = evalStr.trim();
if (cleanStr.startsWith('```')) {
cleanStr = cleanStr.replace(/^```[a-zA-Z]*\n?/, '').replace(/\n?```$/, '').trim();
}
return JSON.parse(cleanStr);
} catch (e) {
console.error('Failed to parse evaluation JSON:', e, 'Raw string:', evalStr);
return { score: 0, analysis: '无法解析审计结果,请查看日志', suggestions: [] };
}
};
const parseEvaluation = (evalStr?: string) => tryParseEvaluationResult(evalStr).result;
const handleGlobalClick = () => {
myToneDropdownOpen.value = false;

View File

@@ -1,11 +1,19 @@
<script setup lang="ts">
import { ref, computed, onMounted, onUnmounted } from 'vue';
import { ChevronDown, Check, ArrowRightLeft, Trash2, FileText, Plus, Loader2, Send, User, Type, Copy, Save } from 'lucide-vue-next';
import { invoke } from '@tauri-apps/api/core';
import { listen } from '@tauri-apps/api/event';
import { useSettingsStore, LANGUAGES, SPEAKER_IDENTITY_OPTIONS, TONE_REGISTER_OPTIONS } from '../stores/settings';
import { cn } from '../lib/utils';
import { useClipboard } from '../composables/useClipboard';
import {
executeTranslationRequest,
extractAssistantContent,
resolveModelConfig,
tryParseEvaluationResult,
type EvaluationResult,
type TranslationChunkEvent,
type TranslationPayload,
} from '../lib/translation-service';
const settings = useSettingsStore();
const { activeCopyId, copyWithFeedback } = useClipboard();
@@ -51,10 +59,6 @@ const targetText = ref('');
const isTranslating = ref(false);
const currentHistoryId = ref<string | null>(null);
interface Suggestion { id: number; text: string; importance: number; }
interface EvaluationResult { score: number; analysis: string; suggestions?: Suggestion[]; }
interface TranslationChunkEvent { request_id: string; chunk: string; }
const evaluationResult = ref<EvaluationResult | null>(null);
const isEvaluating = ref(false);
const isRefining = ref(false);
@@ -107,15 +111,6 @@ const clearSource = () => {
evaluationResult.value = null;
};
const generateCurl = (apiBaseUrl: string, apiKey: string, body: any) => {
const fullUrl = `${apiBaseUrl.replace(/\/$/, '')}/chat/completions`;
const maskedKey = apiKey ? `${apiKey.slice(0, 6)}...${apiKey.slice(-4)}` : 'YOUR_API_KEY';
return `curl "${fullUrl}" \\
-H "Content-Type: application/json" \\
-H "Authorization: Bearer ${maskedKey}" \\
-d '${JSON.stringify(body, null, 2)}'`;
};
const evaluateTranslation = async () => {
if (!targetText.value) return;
isEvaluating.value = true;
@@ -123,18 +118,11 @@ const evaluateTranslation = async () => {
selectedSuggestionIds.value = [];
appliedSuggestionIds.value = [];
let apiBaseUrl = settings.apiBaseUrl;
let apiKey = settings.apiKey;
let modelName = settings.modelName;
if (settings.evaluationProfileId) {
const profile = settings.profiles.find(p => p.id === settings.evaluationProfileId);
if (profile) {
apiBaseUrl = profile.apiBaseUrl;
apiKey = profile.apiKey;
modelName = profile.modelName;
}
}
const modelConfig = resolveModelConfig({
apiBaseUrl: settings.apiBaseUrl,
apiKey: settings.apiKey,
modelName: settings.modelName,
}, settings.profiles, settings.evaluationProfileId);
const evaluationSystemPrompt = settings.evaluationPromptTemplate
.replace(/{SOURCE_LANG}/g, sourceLang.value.englishName)
@@ -145,31 +133,27 @@ const evaluateTranslation = async () => {
const evaluationUserPrompt = `[Source Text]\n${sourceText.value}\n\n[Translated Text]\n${targetText.value}`;
const requestBody = {
model: modelName,
const requestBody: TranslationPayload = {
model: modelConfig.modelName,
messages: [ { role: "system", content: evaluationSystemPrompt }, { role: "user", content: evaluationUserPrompt } ],
stream: false
};
const requestId = crypto.randomUUID();
settings.addLog('request', { type: 'evaluation', ...requestBody }, generateCurl(apiBaseUrl, apiKey, requestBody));
try {
const response = await invoke<string>('translate', { apiAddress: apiBaseUrl, apiKey: apiKey, payload: requestBody, requestId });
try {
// 解析 API 的原始响应 JSON
const fullResponseJson = JSON.parse(response);
settings.addLog('response', fullResponseJson);
const content = fullResponseJson.choices?.[0]?.message?.content || response;
const jsonStr = content.replace(/```json\s?|\s?```/g, '').trim();
evaluationResult.value = JSON.parse(jsonStr);
} catch (parseErr) {
console.error('Failed to parse evaluation result:', response);
const response = await executeTranslationRequest({
apiAddress: modelConfig.apiBaseUrl,
apiKey: modelConfig.apiKey,
payload: requestBody,
logger: settings,
logType: 'evaluation',
});
const parsedEvaluation = tryParseEvaluationResult(extractAssistantContent(response));
if (!parsedEvaluation.ok) {
console.error(parsedEvaluation.error, response);
settings.addLog('error', `Evaluation parsing error: ${response}`);
}
} catch (err: any) {
settings.addLog('error', `Evaluation error: ${String(err)}`);
evaluationResult.value = parsedEvaluation.result;
} catch {
} finally {
isEvaluating.value = false;
}
@@ -184,18 +168,11 @@ const refineTranslation = async () => {
const originalTranslation = targetText.value;
targetText.value = '';
let apiBaseUrl = settings.apiBaseUrl;
let apiKey = settings.apiKey;
let modelName = settings.modelName;
if (settings.evaluationProfileId) {
const profile = settings.profiles.find(p => p.id === settings.evaluationProfileId);
if (profile) {
apiBaseUrl = profile.apiBaseUrl;
apiKey = profile.apiKey;
modelName = profile.modelName;
}
}
const modelConfig = resolveModelConfig({
apiBaseUrl: settings.apiBaseUrl,
apiKey: settings.apiKey,
modelName: settings.modelName,
}, settings.profiles, settings.evaluationProfileId);
const refinementSystemPrompt = settings.refinementPromptTemplate
.replace(/{SOURCE_LANG}/g, sourceLang.value.englishName)
@@ -207,26 +184,25 @@ const refineTranslation = async () => {
const formattedSuggestions = selectedTexts.map((text, i) => `${i + 1}. ${text}`).join('\n');
const refinementUserPrompt = `[Source Text]\n${sourceText.value}\n\n[Current Translation]\n${originalTranslation}\n\n[User Feedback]\n${formattedSuggestions}`;
const requestBody = {
model: modelName,
const requestBody: TranslationPayload = {
model: modelConfig.modelName,
messages: [ { role: "system", content: refinementSystemPrompt }, { role: "user", content: refinementUserPrompt } ],
stream: settings.enableStreaming
};
const requestId = crypto.randomUUID();
settings.addLog('request', { type: 'refinement', ...requestBody }, generateCurl(apiBaseUrl, apiKey, requestBody));
try {
if (settings.enableStreaming) activeStreamRequestId.value = requestId;
const response = await invoke<string>('translate', { apiAddress: apiBaseUrl, apiKey: apiKey, payload: requestBody, requestId });
const response = await executeTranslationRequest({
apiAddress: modelConfig.apiBaseUrl,
apiKey: modelConfig.apiKey,
payload: requestBody,
logger: settings,
logType: 'refinement',
onStreamStart: (requestId) => {
activeStreamRequestId.value = requestId;
},
});
if (settings.enableStreaming) {
settings.addLog('response', targetText.value);
} else {
const fullResponseJson = JSON.parse(response);
settings.addLog('response', fullResponseJson);
targetText.value = fullResponseJson.choices?.[0]?.message?.content || response;
}
if (!settings.enableStreaming) targetText.value = extractAssistantContent(response);
if (evaluationResult.value?.suggestions) {
appliedSuggestionIds.value.push(...selectedSuggestionIds.value);
@@ -239,9 +215,7 @@ const refineTranslation = async () => {
});
}
} catch (err: any) {
const errorMsg = String(err);
settings.addLog('error', errorMsg);
targetText.value = `Error: ${errorMsg}`;
targetText.value = `Error: ${String(err)}`;
} finally {
isRefining.value = false;
activeStreamRequestId.value = null;
@@ -268,27 +242,28 @@ const translate = async () => {
? `[Context]\n${context.value}\n\n[Text to Translate]\n${sourceText.value}`
: `[Text to Translate]\n${sourceText.value}`;
const requestBody = {
const requestBody: TranslationPayload = {
model: settings.modelName,
messages: [ { role: "system", content: systemMessage }, { role: "user", content: userMessage } ],
stream: settings.enableStreaming
};
const requestId = crypto.randomUUID();
settings.addLog('request', requestBody, generateCurl(settings.apiBaseUrl, settings.apiKey, requestBody));
try {
if (settings.enableStreaming) activeStreamRequestId.value = requestId;
const response = await invoke<string>('translate', { apiAddress: settings.apiBaseUrl, apiKey: settings.apiKey, payload: requestBody, requestId });
const response = await executeTranslationRequest({
apiAddress: settings.apiBaseUrl,
apiKey: settings.apiKey,
payload: requestBody,
logger: settings,
onStreamStart: (requestId) => {
activeStreamRequestId.value = requestId;
},
});
let finalTargetText = '';
if (settings.enableStreaming) {
finalTargetText = targetText.value;
settings.addLog('response', response);
} else {
const fullResponseJson = JSON.parse(response);
settings.addLog('response', fullResponseJson);
finalTargetText = fullResponseJson.choices?.[0]?.message?.content || response;
finalTargetText = extractAssistantContent(response);
targetText.value = finalTargetText;
}
@@ -303,9 +278,7 @@ const translate = async () => {
modelName: settings.modelName
});
} catch (err: any) {
const errorMsg = String(err);
settings.addLog('error', errorMsg);
targetText.value = `Error: ${errorMsg}`;
targetText.value = `Error: ${String(err)}`;
} finally {
isTranslating.value = false;
activeStreamRequestId.value = null;

View File

@@ -0,0 +1,164 @@
import { invoke } from '@tauri-apps/api/core';
import type { ApiProfile } from '../stores/settings';
interface Message {
role: string;
content: string;
}
export interface TranslationPayload {
model: string;
messages: Message[];
stream: boolean;
}
export interface TranslationChunkEvent {
request_id: string;
chunk: string;
}
export interface EvaluationSuggestion {
id: number;
text: string;
importance: number;
}
export interface EvaluationResult {
score: number;
analysis: string;
suggestions: EvaluationSuggestion[];
}
export interface ModelConfig {
apiBaseUrl: string;
apiKey: string;
modelName: string;
}
type LogType = 'request' | 'response' | 'error';
interface Logger {
addLog: (type: LogType, content: any, curl?: string) => void;
}
interface ExecuteTranslationRequestOptions {
apiAddress: string;
apiKey: string;
payload: TranslationPayload;
logger: Logger;
logType?: string;
onStreamStart?: (requestId: string) => void;
}
const FALLBACK_EVALUATION_RESULT: EvaluationResult = {
score: 0,
analysis: '无法解析审计结果,请查看日志',
suggestions: [],
};
export function resolveModelConfig(
primary: ModelConfig,
profiles: ApiProfile[],
profileId: string | null,
): ModelConfig {
if (!profileId) return primary;
const profile = profiles.find((item) => item.id === profileId);
if (!profile) return primary;
return {
apiBaseUrl: profile.apiBaseUrl,
apiKey: profile.apiKey,
modelName: profile.modelName,
};
}
export function generateCurl(apiBaseUrl: string, apiKey: string, body: TranslationPayload) {
const fullUrl = `${apiBaseUrl.replace(/\/$/, '')}/chat/completions`;
const maskedKey = apiKey ? `${apiKey.slice(0, 6)}...${apiKey.slice(-4)}` : 'YOUR_API_KEY';
return `curl "${fullUrl}" \\
-H "Content-Type: application/json" \\
-H "Authorization: Bearer ${maskedKey}" \\
-d '${JSON.stringify(body, null, 2)}'`;
}
export async function executeTranslationRequest({
apiAddress,
apiKey,
payload,
logger,
logType,
onStreamStart,
}: ExecuteTranslationRequestOptions) {
const requestId = crypto.randomUUID();
const requestLog = logType ? { type: logType, ...payload } : payload;
logger.addLog('request', requestLog, generateCurl(apiAddress, apiKey, payload));
try {
if (payload.stream) onStreamStart?.(requestId);
const response = await invoke<string>('translate', {
apiAddress,
apiKey,
payload,
requestId,
});
logger.addLog('response', payload.stream ? response : safeParseJson(response) ?? response);
return response;
} catch (error) {
logger.addLog('error', String(error));
throw error;
}
}
export function safeParseJson(value: string) {
try {
return JSON.parse(value);
} catch {
return null;
}
}
export function extractAssistantContent(response: string) {
const parsed = safeParseJson(response);
if (parsed && typeof parsed === 'object') {
return (parsed as any).choices?.[0]?.message?.content || response;
}
return response;
}
export function tryParseEvaluationResult(raw?: string) {
if (!raw) {
return {
ok: false as const,
error: 'Evaluation result is empty',
result: FALLBACK_EVALUATION_RESULT,
};
}
try {
let cleanStr = raw.trim();
if (cleanStr.startsWith('```')) {
cleanStr = cleanStr.replace(/^```[a-zA-Z]*\n?/, '').replace(/\n?```$/, '').trim();
}
const parsed = JSON.parse(cleanStr);
return {
ok: true as const,
result: {
score: Number(parsed.score) || 0,
analysis: typeof parsed.analysis === 'string' ? parsed.analysis : '',
suggestions: Array.isArray(parsed.suggestions) ? parsed.suggestions : [],
} satisfies EvaluationResult,
};
} catch (error) {
return {
ok: false as const,
error: `Failed to parse evaluation JSON: ${String(error)}`,
result: FALLBACK_EVALUATION_RESULT,
};
}
}