Fix renaming of embeddable_to_dict_fn (#37646)

Amar3tto · web-flow · commit 120ceab62a70 · 2026-02-19T10:29:17.000-05:00
diff --git 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
@@ -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')
         })
