All files / api/src/ai-providers/google model-capabilities.ts

100% Statements 14/14
100% Branches 0/0
100% Functions 0/0
100% Lines 14/14

Press n or j to go to the next uncovered block, b, p or k for the previous block.

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              1x 1x   1x 1x   1x   1x 1x 1x   1x 1x 1x 1x 1x 1x  
/**
 * Google AI model capabilities configuration.
 *
 * Defines which parameters are supported by different Google (Gemini) models.
 */
 
import type { ProviderModelCapabilities } from '@shared/types/ai-providers/model-capabilities';
import { FunctionName } from '@shared/types/api/request';
import { AIProvider } from '@shared/types/constants';
 
export const googleModelCapabilities: ProviderModelCapabilities = {
  provider: AIProvider.GOOGLE,
 
  models: [
    // Gemini 2.0 Flash Thinking models (support thinking parameter)
    {
      modelPattern: /^gemini-2\.0-flash-thinking-exp(-\d+)?$/,
      endpointConfigs: {
        // These models support thinking parameter - no restrictions
        [FunctionName.CHAT_COMPLETE]: {},
        [FunctionName.STREAM_CHAT_COMPLETE]: {},
      },
    },
  ],
};