Skip to content

Commit 1e04d7d

Browse files
committed
fix
1 parent f5dcdec commit 1e04d7d

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

samples/snippets/multimodal_test.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -190,7 +190,7 @@ def test_multimodal_example(gcs_bucket_snippets: str) -> None:
190190
# [END bigquery_dataframes_multimodal_create_embedding]
191191

192192
# [START bigquery_dataframes_multimodal_create_df_products_mm]
193-
df_images = bpd.read_gbq("cymbal_pets.product_images")
193+
df_images = bpd.read_gbq("SELECT * FROM cymbal_pets.product_images")
194194
df_products = bpd.read_gbq("cymbal_pets.products")
195195

196196
df_products_mm = df_images.merge(df_products, on="uri").drop(columns="uri")
@@ -406,7 +406,7 @@ def chunk_pdf(src_ref: str, chunk_size: int, overlap_size: int) -> list[str]:
406406
# [END bigquery_dataframes_multimodal_define_chunk_pdf]
407407

408408
# [START bigquery_dataframes_multimodal_apply_chunk_pdf]
409-
df_manuals = bpd.read_gbq("cymbal_pets.product_manuals")
409+
df_manuals = bpd.read_gbq("SELECT * FROM cymbal_pets.product_manuals")
410410
df_manuals["url"] = bbq.to_json_string(
411411
bbq.obj.get_access_url(df_manuals["ref"], "R")
412412
)
@@ -476,7 +476,9 @@ def chunk_pdf(src_ref: str, chunk_size: int, overlap_size: int) -> list[str]:
476476
# [END bigquery_dataframes_create_external_table_all]
477477

478478
# [START bigquery_dataframes_create_manual_to_chunks]
479-
df1 = bpd.read_gbq("cymbal_pets.product_manuals_all").sort_values("uri")
479+
df1 = bpd.read_gbq("SELECT * FROM cymbal_pets.product_manuals_all").sort_values(
480+
"uri"
481+
)
480482
df2 = df1.copy()
481483
df1["name"] = df1["uri"].str.extract(r".*/([^.]*).[^/]+")
482484
df2["name"] = df2["uri"].str.extract(r".*/([^.]*)_page[0-9]+.[^/]+")

0 commit comments

Comments
 (0)