Update the cloud function to use gemini flash (#42)

Co-authored-by: Luka Fontanilla <[email protected]>
looker-open-source · Jun 4, 2024 · 92103b7 · 92103b7
1 parent ee71c80
commit 92103b7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/explore-assistant-cloud-function/main.py b/explore-assistant-cloud-function/main.py
@@ -63,7 +63,7 @@ def has_valid_signature(request):
 
     return hmac.compare_digest(signature, expected_signature)
 
-def generate_looker_query(contents, parameters=None):
+def generate_looker_query(contents, parameters=None, model_name="gemini-1.5-flash"):
 
    # Define default parameters
     default_parameters = {