Oliver/hm_ai_qc_report_tool
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update default Google model to gemini-2.5-flash

Browse source 
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
nickviljoen 2026-03-21 18:59:00 +02:00
parent 5fd5f0fc4f
commit b4e94ad4eb
4 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
core/models/usage_log.py
View file

@ -10,7 +10,7 @@ COST_PER_MILLION_TOKENS = {
'gpt-4-turbo': {'input': 10.00, 'output': 30.00, 'blended': 15.00},
},
'google': {
'gemini-2.0-flash': {'input': 0.10, 'output': 0.40, 'blended': 0.20},
'gemini-2.5-flash': {'input': 0.10, 'output': 0.40, 'blended': 0.20},
'gemini-2.5-pro': {'input': 1.25, 'output': 10.00, 'blended': 3.50},
'gemini-1.5-pro': {'input': 1.25, 'output': 5.00, 'blended': 2.50},
},

2
core/services/llm_config.py
View file

@ -34,7 +34,7 @@ class LLMConfig:
},
'google': {
'api_key_env': 'GOOGLE_API_KEY',
'models': ['gemini-2.0-flash', 'gemini-2.5-pro', 'gemini-1.5-pro']
'models': ['gemini-2.5-flash', 'gemini-2.5-pro', 'gemini-1.5-pro']
},
'anthropic': {
'api_key_env': 'ANTHROPIC_API_KEY',

4
modules/hm_qc/routes.py
View file

@ -217,7 +217,7 @@ def execute():
# Override LLM provider if user selected one
if llm_provider:
provider_models = {'openai': 'gpt-4o', 'google': 'gemini-2.0-flash'}
provider_models = {'openai': 'gpt-4o', 'google': 'gemini-2.5-flash'}
for check in profile.get('checks', []):
if check.get('llm_provider'):
check['llm_provider'] = llm_provider
@ -300,7 +300,7 @@ def execute_batch():
# Override LLM provider if user selected one
if llm_provider:
provider_models = {'openai': 'gpt-4o', 'google': 'gemini-2.0-flash'}
provider_models = {'openai': 'gpt-4o', 'google': 'gemini-2.5-flash'}
for check in profile.get('checks', []):
if check.get('llm_provider'):
check['llm_provider'] = llm_provider

2
modules/video_qc/routes.py
View file

@ -120,7 +120,7 @@ def execute():
if not session_id:
return jsonify({'error': 'Missing session_id'}), 400
provider_models = {'openai': 'gpt-4o', 'google': 'gemini-2.0-flash'}
provider_models = {'openai': 'gpt-4o', 'google': 'gemini-2.5-flash'}
llm_model = provider_models.get(llm_provider, 'gpt-4o')
upload_path = os.path.join(