fix model
This commit is contained in:
@@ -257,8 +257,22 @@ const evaluateMessage = async (messageId: string) => {
|
|||||||
|
|
||||||
const userPrompt = `[Source Text]\n${msg.original}\n\n[Current Translation]\n${msg.translated}`;
|
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;
|
||||||
|
|
||||||
|
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 = {
|
const requestBody = {
|
||||||
model: settings.modelName,
|
model: evalModelName,
|
||||||
messages: [
|
messages: [
|
||||||
{ role: "system", content: systemPrompt },
|
{ role: "system", content: systemPrompt },
|
||||||
{ role: "user", content: userPrompt }
|
{ role: "user", content: userPrompt }
|
||||||
@@ -266,12 +280,12 @@ const evaluateMessage = async (messageId: string) => {
|
|||||||
stream: false
|
stream: false
|
||||||
};
|
};
|
||||||
|
|
||||||
settings.addLog('request', { type: 'conversation-eval', ...requestBody }, generateCurl(settings.apiBaseUrl, settings.apiKey, requestBody));
|
settings.addLog('request', { type: 'conversation-eval', ...requestBody }, generateCurl(evalApiBaseUrl, evalApiKey, requestBody));
|
||||||
|
|
||||||
try {
|
try {
|
||||||
const response = await invoke<string>('translate', {
|
const response = await invoke<string>('translate', {
|
||||||
apiAddress: settings.apiBaseUrl,
|
apiAddress: evalApiBaseUrl,
|
||||||
apiKey: settings.apiKey,
|
apiKey: evalApiKey,
|
||||||
payload: requestBody
|
payload: requestBody
|
||||||
});
|
});
|
||||||
const fullResponseJson = JSON.parse(response);
|
const fullResponseJson = JSON.parse(response);
|
||||||
|
|||||||
Reference in New Issue
Block a user