Added support for Gemini models

This commit is contained in:
Badri Narayanan S
2025-12-27 14:09:20 +05:30
parent 9b7dcf3a6c
commit c1e1dbb0ef
13 changed files with 641 additions and 176 deletions

View File

@@ -84,6 +84,40 @@ export const MAX_WAIT_BEFORE_ERROR_MS = 120000; // 2 minutes - throw error if wa
// Thinking model constants
export const MIN_SIGNATURE_LENGTH = 50; // Minimum valid thinking signature length
// Gemini-specific limits
export const GEMINI_MAX_OUTPUT_TOKENS = 16384;
/**
* Get the model family from model name (dynamic detection, no hardcoded list).
* @param {string} modelName - The model name from the request
* @returns {'claude' | 'gemini' | 'unknown'} The model family
*/
export function getModelFamily(modelName) {
const lower = (modelName || '').toLowerCase();
if (lower.includes('claude')) return 'claude';
if (lower.includes('gemini')) return 'gemini';
return 'unknown';
}
/**
* Check if a model supports thinking/reasoning output.
* @param {string} modelName - The model name from the request
* @returns {boolean} True if the model supports thinking blocks
*/
export function isThinkingModel(modelName) {
const lower = (modelName || '').toLowerCase();
// Claude thinking models have "thinking" in the name
if (lower.includes('claude') && lower.includes('thinking')) return true;
// Gemini thinking models: explicit "thinking" in name, OR gemini version 3+
if (lower.includes('gemini')) {
if (lower.includes('thinking')) return true;
// Check for gemini-3 or higher (e.g., gemini-3, gemini-3.5, gemini-4, etc.)
const versionMatch = lower.match(/gemini-(\d+)/);
if (versionMatch && parseInt(versionMatch[1], 10) >= 3) return true;
}
return false;
}
// Google OAuth configuration (from opencode-antigravity-auth)
export const OAUTH_CONFIG = {
clientId: '1071006060591-tmhssin2h21lcre235vtolojh4g403ep.apps.googleusercontent.com',
@@ -117,6 +151,9 @@ export default {
MAX_ACCOUNTS,
MAX_WAIT_BEFORE_ERROR_MS,
MIN_SIGNATURE_LENGTH,
GEMINI_MAX_OUTPUT_TOKENS,
getModelFamily,
isThinkingModel,
OAUTH_CONFIG,
OAUTH_REDIRECT_URI
};