fix 2
This commit is contained in:
@@ -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;
|
||||
|
||||
@@ -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;
|
||||
|
||||
164
src/lib/translation-service.ts
Normal file
164
src/lib/translation-service.ts
Normal 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,
|
||||
};
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user