From 92103b7e90d4977c0769f2c26c9036f230335eed Mon Sep 17 00:00:00 2001 From: Jawad Laraqui Date: Tue, 4 Jun 2024 14:32:47 -0400 Subject: [PATCH] Update the cloud function to use gemini flash (#42) Co-authored-by: Luka Fontanilla <51471651+LukaFontanilla@users.noreply.github.com> --- explore-assistant-cloud-function/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/explore-assistant-cloud-function/main.py b/explore-assistant-cloud-function/main.py index c393ea41..5bcdd17f 100644 --- 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 = {