Skip to content

Commit

Permalink
rename to raw_chunks
Browse files Browse the repository at this point in the history
  • Loading branch information
shreyaspimpalgaonkar committed Oct 22, 2024
1 parent f606d1c commit 6b1de83
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 4 deletions.
2 changes: 1 addition & 1 deletion py/core/main/api/ingestion_router.py
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand Down
4 changes: 2 additions & 2 deletions py/core/providers/database/vector.py
Original file line number Diff line number Diff line change
Expand Up @@ -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 = (
Expand Down
2 changes: 1 addition & 1 deletion py/shared/abstractions/vector.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 6b1de83

Please sign in to comment.