From 6b1de83fa4209b900844e71f4499a0ee7917d406 Mon Sep 17 00:00:00 2001 From: Shreyas Pimpalgaonkar Date: Tue, 22 Oct 2024 10:43:01 -0700 Subject: [PATCH] rename to raw_chunks --- py/core/main/api/ingestion_router.py | 2 +- py/core/providers/database/vector.py | 4 ++-- py/shared/abstractions/vector.py | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/py/core/main/api/ingestion_router.py b/py/core/main/api/ingestion_router.py index 30f4e4046..084176553 100644 --- a/py/core/main/api/ingestion_router.py +++ b/py/core/main/api/ingestion_router.py @@ -359,7 +359,7 @@ async def ingest_chunks_app( @self.base_endpoint async def create_vector_index_app( table_name: Optional[VectorTableName] = Body( - default=VectorTableName.TEXT_CHUNKS, + default=VectorTableName.RAW_CHUNKS, description=create_vector_descriptions.get("table_name"), ), index_method: IndexMethod = Body( diff --git a/py/core/providers/database/vector.py b/py/core/providers/database/vector.py index f64877f5c..16a3939ce 100644 --- a/py/core/providers/database/vector.py +++ b/py/core/providers/database/vector.py @@ -531,8 +531,8 @@ async def create_index( ArgError: If an invalid index method is used, or if *replace* is False and an index already exists. """ - if table_name == VectorTableName.TEXT_CHUNKS: - table_name_str = f"{self.project_name}.{VectorTableName.TEXT_CHUNKS}" + if table_name == VectorTableName.RAW_CHUNKS: + table_name_str = f"{self.project_name}.{VectorTableName.RAW_CHUNKS}" col_name = "vec" elif table_name == VectorTableName.ENTITIES_DOCUMENT: table_name_str = ( diff --git a/py/shared/abstractions/vector.py b/py/shared/abstractions/vector.py index d51d761bf..e994ab5bf 100644 --- a/py/shared/abstractions/vector.py +++ b/py/shared/abstractions/vector.py @@ -105,7 +105,7 @@ class VectorTableName(str, Enum): """ - TEXT_CHUNKS = "vector" + RAW_CHUNKS = "vector" ENTITIES_DOCUMENT = "entity_embedding" ENTITIES_COLLECTION = "entity_collection" # TODO: Add support for triples