From 39b8646a8ab65d242ec6220674f8775052411bf5 Mon Sep 17 00:00:00 2001 From: Vitaly Terentyev Date: Thu, 19 Feb 2026 16:56:48 +0400 Subject: [PATCH] Fix renaming of embeddable_to_dict_fn --- .../beam-ml/bigquery_vector_ingestion_and_search.ipynb | 2 +- sdks/python/apache_beam/ml/rag/ingestion/bigquery_it_test.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/notebooks/beam-ml/bigquery_vector_ingestion_and_search.ipynb b/examples/notebooks/beam-ml/bigquery_vector_ingestion_and_search.ipynb index b1becd294ff0..318d1329f3dd 100644 --- a/examples/notebooks/beam-ml/bigquery_vector_ingestion_and_search.ipynb +++ b/examples/notebooks/beam-ml/bigquery_vector_ingestion_and_search.ipynb @@ -1275,7 +1275,7 @@ " },\n", " schema_config=SchemaConfig(\n", " schema=SCHEMA,\n", - " chunk_to_dict_fn=chunk_to_dict_custom\n", + " embeddable_to_dict_fn=chunk_to_dict_custom\n", " )\n", ")" ], diff --git a/sdks/python/apache_beam/ml/rag/ingestion/bigquery_it_test.py b/sdks/python/apache_beam/ml/rag/ingestion/bigquery_it_test.py index b21da0443467..7ab2472341d3 100644 --- a/sdks/python/apache_beam/ml/rag/ingestion/bigquery_it_test.py +++ b/sdks/python/apache_beam/ml/rag/ingestion/bigquery_it_test.py @@ -138,7 +138,7 @@ def test_custom_schema(self): 'name': 'source', 'type': 'STRING' }] }, - chunk_to_dict_fn=lambda chunk: { + embeddable_to_dict_fn=lambda chunk: { 'id': chunk.id, 'embedding': chunk.embedding.dense_embedding, 'source': chunk.metadata.get('source') })