[Bug]: Infinite credit spend in grounding_with_vais example

[xangelix]

File Name

gemini/grounding/grounding_with_vais.ipynb

What happened?

Infinite credit spend in one of two examples for the Vertex Search feature. Not a fun surprise.

This code is missing a break condition and encourages the end user to let it run unattended, which will result in an infinite loop of queries that incur costs.

generative-ai/gemini/grounding/grounding_with_vais.ipynb

Lines 420 to 434 in 312646e

	"search_client = vais.SearchServiceClient(client_options=client_options)\n",
	"\n",
	"while True:\n",
	" try:\n",
	" search_client.search(\n",
	" vais.SearchRequest(\n",
	" serving_config=(\n",
	" f\"projects/{PROJECT_ID}/locations/{VAIS_LOCATION}/collections/default_collection/engines/{engine_id}/servingConfigs/default_search\"\n",
	" ),\n",
	" query=\"Who is the CEO?\",\n",
	" page_size=10,\n",
	" )\n",
	" )\n",
	" except Exception:\n",
	" time.sleep(10)"

Relevant log output

Code of Conduct

• I agree to follow this project's Code of Conduct