Skip to content

Commit b9fa5fa

Browse files
Pass namespace correctly form ObjectAPI embeddings ingestion to vector ingestion (#352)
Pass namespace correctly form ObjectAPI embeddings ingestion to vector ingestion
1 parent d9bc472 commit b9fa5fa

File tree

2 files changed

+8
-0
lines changed

2 files changed

+8
-0
lines changed

apis/python/src/tiledb/vector_search/object_api/embeddings_ingestion.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -297,12 +297,15 @@ def create_dag(
297297
limit=1,
298298
retry_policy="Always",
299299
),
300+
namespace=namespace,
300301
)
301302
else:
302303
d = dag.DAG(
303304
name="embedding-generation",
304305
mode=Mode.REALTIME,
305306
max_workers=workers,
307+
# TODO: `default` is not an actual namespace. This is a temp fix to
308+
# be able to run DAGs locally.
306309
namespace="default",
307310
)
308311

@@ -451,6 +454,7 @@ def create_dag(
451454
index_timestamp=index_timestamp,
452455
storage_version=obj_index.index.storage_version,
453456
config=config,
457+
namespace=namespace,
454458
mode=vector_indexing_mode,
455459
**kwargs,
456460
)
@@ -468,12 +472,15 @@ def submit_local(d, func, *args, **kwargs):
468472
mode=Mode.BATCH,
469473
name="ingest-embeddings-driver",
470474
max_workers=1,
475+
namespace=namespace,
471476
)
472477
else:
473478
d = dag.DAG(
474479
mode=Mode.REALTIME,
475480
name="ingest-embeddings-driver",
476481
max_workers=1,
482+
# TODO: `default` is not an actual namespace. This is a temp fix to
483+
# be able to run DAGs locally.
477484
namespace="default",
478485
)
479486
submit = partial(submit_local, d)

apis/python/src/tiledb/vector_search/object_api/object_index.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -336,6 +336,7 @@ def update_index(
336336
embeddings_generation_mode=embeddings_generation_mode,
337337
vector_indexing_mode=vector_indexing_mode,
338338
config=config,
339+
namespace=namespace,
339340
environment_variables=environment_variables,
340341
**kwargs,
341342
)

0 commit comments

Comments
 (0)