torbjorn/my-pal-mcp-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'origin/main'

Browse Source
This commit is contained in:
Fahad
2025-06-16 18:20:17 +04:00
parent 357452b7ba c3d44525b6
commit 93399b6d10
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
providers/gemini.py
View File

@@ -73,8 +73,8 @@ class GeminiModelProvider(ModelProvider):
from utils.model_restrictions import get_restriction_service from utils.model_restrictions import get_restriction_service
restriction_service = get_restriction_service() restriction_service = get_restriction_service()
if not restriction_service.is_allowed(ProviderType.GOOGLE, resolved_name, model_name): if not restriction_service.is_allowed(ProviderType.GOOGLE, model_name, resolved_name):
raise ValueError(f"Gemini model '{model_name}' is not allowed by restriction policy.") raise ValueError(f"Gemini model '{resolved_name}' is not allowed by restriction policy.")
config = self.SUPPORTED_MODELS[resolved_name] config = self.SUPPORTED_MODELS[resolved_name]
@@ -109,7 +109,7 @@ class GeminiModelProvider(ModelProvider):
"""Generate content using Gemini model.""" """Generate content using Gemini model."""
# Validate parameters # Validate parameters
resolved_name = self._resolve_model_name(model_name) resolved_name = self._resolve_model_name(model_name)
self.validate_parameters(resolved_name, temperature) self.validate_parameters(model_name, temperature)
# Prepare content parts (text and potentially images) # Prepare content parts (text and potentially images)
parts = [] parts = []
@@ -150,7 +150,7 @@ class GeminiModelProvider(ModelProvider):
generation_config.max_output_tokens = max_output_tokens generation_config.max_output_tokens = max_output_tokens
# Add thinking configuration for models that support it # Add thinking configuration for models that support it
capabilities = self.get_capabilities(resolved_name) capabilities = self.get_capabilities(model_name)
if capabilities.supports_extended_thinking and thinking_mode in self.THINKING_BUDGETS: if capabilities.supports_extended_thinking and thinking_mode in self.THINKING_BUDGETS:
# Get model's max thinking tokens and calculate actual budget # Get model's max thinking tokens and calculate actual budget
model_config = self.SUPPORTED_MODELS.get(resolved_name) model_config = self.SUPPORTED_MODELS.get(resolved_name)
@@ -255,7 +255,7 @@ class GeminiModelProvider(ModelProvider):
from utils.model_restrictions import get_restriction_service from utils.model_restrictions import get_restriction_service
restriction_service = get_restriction_service() restriction_service = get_restriction_service()
if not restriction_service.is_allowed(ProviderType.GOOGLE, resolved_name, model_name): if not restriction_service.is_allowed(ProviderType.GOOGLE, model_name, resolved_name):
logger.debug(f"Gemini model '{model_name}' -> '{resolved_name}' blocked by restrictions") logger.debug(f"Gemini model '{model_name}' -> '{resolved_name}' blocked by restrictions")
return False return False