Query fix

backend_apis/app/main.py

@@ -110,7 +110,7 @@
 # Text models
 llm_latest = bison_latest.from_pretrained(model_name="text-bison")
 llm_ga = bison_ga.from_pretrained(model_name="text-bison@002")
-model_version_name = config["global"]["model_version_name"]
+model_version_name = config["models"]["code_model_name"]
 gemini_llm = GenerativeModel(model_version_name)
 
 TEXT_MODEL_NAME = config["models"]["text_model_name"]

frontend/src/app/audiences/audiences.component.html

@@ -135,8 +135,8 @@
             value="What are the customer emails ordered by the sum of transactions value by customers in New York City?">
             What are the customer emails ordered by the sum of transactions value by customers in New York City?
         </option>
-        <option value="What are the top 50 customer emails ordered by the loyalty score?">
-            What are the top 50 customer emails ordered by the loyalty score?
+        <option value="What are the customer emails ordered by the total transactions in app by the customer in the state of Georgia?">
+            What are the customer emails ordered by the total transactions in app by the customer in the state of Georgia?
         </option>
         <option value="What are the top 50 customer emails ordered by loyalty score?">What are the top 50 customer
             emails ordered by loyalty score?</option>

infra/templates/config.toml.tftpl

@@ -17,11 +17,6 @@ project_id = "${project_id}"
 location = "${region}"
 
 
-# Gemini Model Version name
-# https://ai.google.dev/gemini-api/docs/models/gemini
-model_version_name = "gemini-1.5-pro"
-
-
 # Audiences
 # Dataset ID of the BigQuery dataset. If you change the dataset_id during the deployment 
 # of this solution, change it here as well.
@@ -208,7 +203,7 @@ Product description: """
 # documentation for more information.
 text_model_name = "text-bison"
 image_model_name = "imagegeneration"
-code_model_name = "code-bison"
+code_model_name = "gemini-1.5-pro"
 
 [data_sample]