admin-valentin/AIClient-2-API
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor(antigravity-core): 优化工具配置逻辑并清理代码

Browse source 
- 重构工具配置检查逻辑，避免创建空对象
- 移除已注释的maxOutputTokens处理代码
- 统一代码缩进格式
- 清理JSON Schema属性处理逻辑
This commit is contained in:
hex2077 2026-01-06 22:37:23 +08:00
parent ff8b5a01d2
commit 717acbb2f4
1 changed files with 17 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
src/gemini/antigravity-core.js
View file

@ -258,7 +258,7 @@ function normalizeAntigravityThinking(modelName, payload, isClaudeModel) {
function geminiToAntigravity(modelName, payload, projectId) { function geminiToAntigravity(modelName, payload, projectId) {
// 深拷贝请求体,避免修改原始对象 // 深拷贝请求体,避免修改原始对象
let template = JSON.parse(JSON.stringify(payload)); let template = JSON.parse(JSON.stringify(payload));
const isClaudeModel = isClaude(modelName); const isClaudeModel = isClaude(modelName);
// 设置基本字段 // 设置基本字段
@ -281,13 +281,12 @@ function geminiToAntigravity(modelName, payload, projectId) {
} }
// 设置工具配置 // 设置工具配置
if (!template.request.toolConfig) { if (template.request.toolConfig) {
template.request.toolConfig = {}; if (!template.request.toolConfig.functionCallingConfig) {
template.request.toolConfig.functionCallingConfig = {};
}
template.request.toolConfig.functionCallingConfig.mode = 'VALIDATED';
} }
if (!template.request.toolConfig.functionCallingConfig) {
template.request.toolConfig.functionCallingConfig = {};
}
template.request.toolConfig.functionCallingConfig.mode = 'VALIDATED';
// 当模型是 Claude 时，禁止使用 tools // 当模型是 Claude 时，禁止使用 tools
if (isClaudeModel) { if (isClaudeModel) {
@ -301,11 +300,11 @@ function geminiToAntigravity(modelName, payload, projectId) {
// 对于非 Claude 模型，删除 maxOutputTokens // 对于非 Claude 模型，删除 maxOutputTokens
// Claude 模型需要保留 maxOutputTokens // Claude 模型需要保留 maxOutputTokens
if (!isClaudeModel) { // if (!isClaudeModel) { 注释了cc用不了
if (template.request.generationConfig && template.request.generationConfig.maxOutputTokens) { if (template.request.generationConfig && template.request.generationConfig.maxOutputTokens) {
delete template.request.generationConfig.maxOutputTokens; delete template.request.generationConfig.maxOutputTokens;
} }
} // }
// 处理 Thinking 配置 // 处理 Thinking 配置
// 对于非 gemini-3-* 模型，将 thinkingLevel 转换为 thinkingBudget // 对于非 gemini-3-* 模型，将 thinkingLevel 转换为 thinkingBudget
@ -319,22 +318,22 @@ function geminiToAntigravity(modelName, payload, projectId) {
} }
// 清理所有工具声明中的 JSON Schema 属性（移除 Google API 不支持的属性如 exclusiveMinimum 等） // 清理所有工具声明中的 JSON Schema 属性（移除 Google API 不支持的属性如 exclusiveMinimum 等）
if (template.request.tools && Array.isArray(template.request.tools)) { if (template.request.tools && Array.isArray(template.request.tools)) {
template.request.tools.forEach((tool) => { template.request.tools.forEach((tool) => {
if (tool.functionDeclarations && Array.isArray(tool.functionDeclarations)) { if (tool.functionDeclarations && Array.isArray(tool.functionDeclarations)) {
tool.functionDeclarations.forEach((funcDecl) => { tool.functionDeclarations.forEach((funcDecl) => {
// 对于 Claude 模型，处理 parametersJsonSchema // 对于 Claude 模型，处理 parametersJsonSchema
if (isClaudeModel && funcDecl.parametersJsonSchema) { if (isClaudeModel && funcDecl.parametersJsonSchema) {
funcDecl.parameters = cleanJsonSchemaProperties(funcDecl.parametersJsonSchema); funcDecl.parameters = cleanJsonSchemaProperties(funcDecl.parametersJsonSchema);
delete funcDecl.parameters.$schema; delete funcDecl.parameters.$schema;
delete funcDecl.parametersJsonSchema; delete funcDecl.parametersJsonSchema;
} else if (funcDecl.parameters) { } else if (funcDecl.parameters) {
funcDecl.parameters = cleanJsonSchemaProperties(funcDecl.parameters); funcDecl.parameters = cleanJsonSchemaProperties(funcDecl.parameters);
} }
}); });
} }
}); });
} }
// 如果是图像模型，增加参数 "generationConfig.imageConfig.imageSize": "4K" // 如果是图像模型，增加参数 "generationConfig.imageConfig.imageSize": "4K"
if (isImageModel(modelName)) { if (isImageModel(modelName)) {