Default to gemini-1.5-flash-latest in Google Generative AI (#118367)

Default to flash
This commit is contained in:
tronikos 2024-05-29 05:04:47 -07:00 committed by GitHub
parent 4056c4c2cc
commit aeee222df4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 8 additions and 8 deletions

View file

@ -8,7 +8,7 @@ CONF_PROMPT = "prompt"
CONF_RECOMMENDED = "recommended"
CONF_CHAT_MODEL = "chat_model"
RECOMMENDED_CHAT_MODEL = "models/gemini-1.5-pro-latest"
RECOMMENDED_CHAT_MODEL = "models/gemini-1.5-flash-latest"
CONF_TEMPERATURE = "temperature"
RECOMMENDED_TEMPERATURE = 1.0
CONF_TOP_P = "top_p"

View file

@ -12,7 +12,7 @@
'top_k': 64,
'top_p': 0.95,
}),
'model_name': 'models/gemini-1.5-pro-latest',
'model_name': 'models/gemini-1.5-flash-latest',
'safety_settings': dict({
'DANGEROUS': 'BLOCK_MEDIUM_AND_ABOVE',
'HARASSMENT': 'BLOCK_MEDIUM_AND_ABOVE',
@ -64,7 +64,7 @@
'top_k': 64,
'top_p': 0.95,
}),
'model_name': 'models/gemini-1.5-pro-latest',
'model_name': 'models/gemini-1.5-flash-latest',
'safety_settings': dict({
'DANGEROUS': 'BLOCK_MEDIUM_AND_ABOVE',
'HARASSMENT': 'BLOCK_MEDIUM_AND_ABOVE',
@ -128,7 +128,7 @@
'top_k': 64,
'top_p': 0.95,
}),
'model_name': 'models/gemini-1.5-pro-latest',
'model_name': 'models/gemini-1.5-flash-latest',
'safety_settings': dict({
'DANGEROUS': 'BLOCK_MEDIUM_AND_ABOVE',
'HARASSMENT': 'BLOCK_MEDIUM_AND_ABOVE',
@ -184,7 +184,7 @@
'top_k': 64,
'top_p': 0.95,
}),
'model_name': 'models/gemini-1.5-pro-latest',
'model_name': 'models/gemini-1.5-flash-latest',
'safety_settings': dict({
'DANGEROUS': 'BLOCK_MEDIUM_AND_ABOVE',
'HARASSMENT': 'BLOCK_MEDIUM_AND_ABOVE',
@ -240,7 +240,7 @@
'top_k': 64,
'top_p': 0.95,
}),
'model_name': 'models/gemini-1.5-pro-latest',
'model_name': 'models/gemini-1.5-flash-latest',
'safety_settings': dict({
'DANGEROUS': 'BLOCK_MEDIUM_AND_ABOVE',
'HARASSMENT': 'BLOCK_MEDIUM_AND_ABOVE',
@ -296,7 +296,7 @@
'top_k': 64,
'top_p': 0.95,
}),
'model_name': 'models/gemini-1.5-pro-latest',
'model_name': 'models/gemini-1.5-flash-latest',
'safety_settings': dict({
'DANGEROUS': 'BLOCK_MEDIUM_AND_ABOVE',
'HARASSMENT': 'BLOCK_MEDIUM_AND_ABOVE',

View file

@ -5,7 +5,7 @@
'api_key': '**REDACTED**',
}),
'options': dict({
'chat_model': 'models/gemini-1.5-pro-latest',
'chat_model': 'models/gemini-1.5-flash-latest',
'dangerous_block_threshold': 'BLOCK_MEDIUM_AND_ABOVE',
'harassment_block_threshold': 'BLOCK_MEDIUM_AND_ABOVE',
'hate_block_threshold': 'BLOCK_MEDIUM_AND_ABOVE',