Skip to content

Commit c99b9ac

Browse files
authored
fix: gemini vertex 2.0 flash models don't support profiles (#254)
Add `supports_profiles=False` parameter to Gemini 2.0 Flash models in the model catalog. Context: gemini flash 2.0 models do not need Profiles, and this was not consistent for the vertex/developer models, so fixing for the vertex models.
1 parent d44974b commit c99b9ac

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/fenic/core/_inference/model_catalog.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -766,6 +766,7 @@ def _initialize_google_vertex_models(self):
766766
context_window_length=1_048_576,
767767
max_output_tokens=8_192,
768768
max_temperature=2.0,
769+
supports_profiles=False,
769770
supports_pdf_parsing=True,
770771
),
771772
snapshots=["gemini-2.0-flash-lite-001"],
@@ -782,6 +783,7 @@ def _initialize_google_vertex_models(self):
782783
context_window_length=1_048_576,
783784
max_output_tokens=8_192,
784785
max_temperature=2.0,
786+
supports_profiles=False,
785787
supports_pdf_parsing=True,
786788
),
787789
snapshots=["gemini-2.0-flash-001", "gemini-2.0-flash-exp"],

0 commit comments

Comments
 (0)