diff --git a/amplifier_module_provider_gemini/__init__.py b/amplifier_module_provider_gemini/__init__.py index 479285c..2e149cc 100644 --- a/amplifier_module_provider_gemini/__init__.py +++ b/amplifier_module_provider_gemini/__init__.py @@ -200,6 +200,10 @@ def get_info(self) -> ProviderInfo: "max_tokens": 8192, "temperature": 0.7, "timeout": 600.0, + "context_window": 1_048_576, + "max_output_tokens": 65_536, + # Gemini charges 2x input / 1.5x output above 200k tokens + "context_budget_cap": 200_000, }, config_fields=[ ConfigField(