File size: 5,582 Bytes
31da410 cda8be4 31da410 cda8be4 31da410 cda8be4 31da410 cda8be4 31da410 cda8be4 31da410 cda8be4 31da410 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 | // Gemini 格式转换工具
import config from '../../config/config.js';
import { generateRequestId } from '../idGenerator.js';
import { convertGeminiToolsToAntigravity } from '../toolConverter.js';
import { getSignatureContext, createThoughtPart, modelMapping, isEnableThinking } from './common.js';
import { normalizeGeminiParameters, toGenerationConfig } from '../parameterNormalizer.js';
/**
* 为 functionCall 生成唯一 ID
*/
function generateFunctionCallId() {
return `call_${Date.now()}_${Math.random().toString(36).substr(2, 9)}`;
}
/**
* 处理 functionCall 和 functionResponse 的 ID 匹配
*/
function processFunctionCallIds(contents) {
const functionCallIds = [];
// 收集所有 functionCall 的 ID
contents.forEach(content => {
if (content.role === 'model' && content.parts && Array.isArray(content.parts)) {
content.parts.forEach(part => {
if (part.functionCall) {
if (!part.functionCall.id) {
part.functionCall.id = generateFunctionCallId();
}
functionCallIds.push(part.functionCall.id);
}
});
}
});
// 为 functionResponse 分配对应的 ID
let responseIndex = 0;
contents.forEach(content => {
if (content.role === 'user' && content.parts && Array.isArray(content.parts)) {
content.parts.forEach(part => {
if (part.functionResponse) {
if (!part.functionResponse.id && responseIndex < functionCallIds.length) {
part.functionResponse.id = functionCallIds[responseIndex];
responseIndex++;
}
}
});
}
});
}
/**
* 处理 model 消息中的 thought 和签名
*/
function processModelThoughts(content, reasoningSignature, toolSignature) {
const parts = content.parts;
// 查找 thought 和独立 thoughtSignature 的位置
let thoughtIndex = -1;
let signatureIndex = -1;
let signatureValue = null;
for (let i = 0; i < parts.length; i++) {
const part = parts[i];
if (part.thought === true && !part.thoughtSignature) {
thoughtIndex = i;
}
if (part.thoughtSignature && !part.thought) {
signatureIndex = i;
signatureValue = part.thoughtSignature;
}
}
// 合并或添加 thought 和签名
if (thoughtIndex !== -1 && signatureIndex !== -1) {
parts[thoughtIndex].thoughtSignature = signatureValue;
parts.splice(signatureIndex, 1);
} else if (thoughtIndex !== -1 && signatureIndex === -1) {
parts[thoughtIndex].thoughtSignature = reasoningSignature;
} else if (thoughtIndex === -1) {
parts.unshift(createThoughtPart(' ', reasoningSignature));
}
// 收集独立的签名 parts（用于 functionCall）
const standaloneSignatures = [];
for (let i = parts.length - 1; i >= 0; i--) {
const part = parts[i];
if (part.thoughtSignature && !part.thought && !part.functionCall && !part.text) {
standaloneSignatures.unshift({ index: i, signature: part.thoughtSignature });
}
}
// 为 functionCall 分配签名
let sigIndex = 0;
for (let i = 0; i < parts.length; i++) {
const part = parts[i];
if (part.functionCall && !part.thoughtSignature) {
if (sigIndex < standaloneSignatures.length) {
part.thoughtSignature = standaloneSignatures[sigIndex].signature;
sigIndex++;
} else {
part.thoughtSignature = toolSignature;
}
}
}
// 移除已使用的独立签名 parts
for (let i = standaloneSignatures.length - 1; i >= 0; i--) {
if (i < sigIndex) {
parts.splice(standaloneSignatures[i].index, 1);
}
}
}
export function generateGeminiRequestBody(geminiBody, modelName, token) {
const enableThinking = isEnableThinking(modelName);
const actualModelName = modelMapping(modelName);
const request = JSON.parse(JSON.stringify(geminiBody));
if (request.contents && Array.isArray(request.contents)) {
processFunctionCallIds(request.contents);
if (enableThinking) {
const { reasoningSignature, toolSignature } = getSignatureContext(token.sessionId, actualModelName);
request.contents.forEach(content => {
if (content.role === 'model' && content.parts && Array.isArray(content.parts)) {
processModelThoughts(content, reasoningSignature, toolSignature);
}
});
}
}
// 使用统一参数规范化模块处理 Gemini 格式参数
const normalizedParams = normalizeGeminiParameters(request.generationConfig || {});
// 转换为 generationConfig 格式
request.generationConfig = toGenerationConfig(normalizedParams, enableThinking, actualModelName);
request.sessionId = token.sessionId;
delete request.safetySettings;
// 转换工具定义
if (request.tools && Array.isArray(request.tools)) {
request.tools = convertGeminiToolsToAntigravity(request.tools, token.sessionId, actualModelName);
}
// 添加工具配置
if (request.tools && request.tools.length > 0 && !request.toolConfig) {
request.toolConfig = { functionCallingConfig: { mode: 'VALIDATED' } };
}
const existingText = request.systemInstruction?.parts?.[0]?.text || '';
const mergedText = existingText ? `${config.systemInstruction}\n\n${existingText}` : config.systemInstruction ?? "";
request.systemInstruction = {
role: 'user',
parts: [{ text: mergedText }]
};
//console.log(JSON.stringify(request, null, 2))
const requestBody = {
project: token.projectId,
requestId: generateRequestId(),
request: request,
model: actualModelName,
userAgent: 'antigravity'
};
return requestBody;
}
|