Spaces:
Runtime error
Runtime error
Fix minor bugs with poorly qualified symbols
Browse files
src/ctp_slack_bot/db/repositories/mongo_db_vectorized_chunk_repository.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
from dependency_injector.resources import AsyncResource
|
2 |
from loguru import logger
|
3 |
-
from pymongo import ReturnDocument
|
4 |
from typing import Any, Collection, Dict, Iterable, Mapping, Optional, Self, Sequence, Set
|
5 |
|
6 |
from ctp_slack_bot.core import Settings
|
@@ -149,7 +149,7 @@ class MongoVectorizedChunkRepository(VectorRepositoryBase, VectorizedChunkReposi
|
|
149 |
async def ensure_indices_exist(self: Self) -> None:
|
150 |
await super().ensure_indices_exist()
|
151 |
index_name = "parent_chunk_unique"
|
152 |
-
existing_indices = await collection.index_information()
|
153 |
if index_name not in existing_indices:
|
154 |
await self.collection.create_index([("parent_id", ASCENDING), ("chunk_id", ASCENDING)], unique=True, name=index_name)
|
155 |
|
@@ -157,5 +157,5 @@ class MongoVectorizedChunkRepositoryResource(AsyncResource):
|
|
157 |
async def init(self: Self, settings: Settings, mongo_db: MongoDB) -> MongoVectorizedChunkRepository:
|
158 |
vectorized_chunk_collection = await mongo_db.get_collection("vectorized_chunks")
|
159 |
vectorized_chunk_repository = MongoVectorizedChunkRepository(settings=settings, collection=vectorized_chunk_collection)
|
160 |
-
await vectorized_chunk_repository.
|
161 |
return vectorized_chunk_repository
|
|
|
1 |
from dependency_injector.resources import AsyncResource
|
2 |
from loguru import logger
|
3 |
+
from pymongo import ASCENDING, ReturnDocument
|
4 |
from typing import Any, Collection, Dict, Iterable, Mapping, Optional, Self, Sequence, Set
|
5 |
|
6 |
from ctp_slack_bot.core import Settings
|
|
|
149 |
async def ensure_indices_exist(self: Self) -> None:
|
150 |
await super().ensure_indices_exist()
|
151 |
index_name = "parent_chunk_unique"
|
152 |
+
existing_indices = await self.collection.index_information()
|
153 |
if index_name not in existing_indices:
|
154 |
await self.collection.create_index([("parent_id", ASCENDING), ("chunk_id", ASCENDING)], unique=True, name=index_name)
|
155 |
|
|
|
157 |
async def init(self: Self, settings: Settings, mongo_db: MongoDB) -> MongoVectorizedChunkRepository:
|
158 |
vectorized_chunk_collection = await mongo_db.get_collection("vectorized_chunks")
|
159 |
vectorized_chunk_repository = MongoVectorizedChunkRepository(settings=settings, collection=vectorized_chunk_collection)
|
160 |
+
await vectorized_chunk_repository.ensure_indices_exist()
|
161 |
return vectorized_chunk_repository
|
src/ctp_slack_bot/db/repositories/vector_repository_base.py
CHANGED
@@ -19,7 +19,7 @@ class VectorRepositoryBase(ABC, BaseModel):
|
|
19 |
|
20 |
async def ensure_indices_exist(self: Self) -> None:
|
21 |
"""Ensure that indices exist."""
|
22 |
-
self.ensure_search_index_exists()
|
23 |
|
24 |
async def ensure_search_index_exists(self: Self) -> None:
|
25 |
"""
|
|
|
19 |
|
20 |
async def ensure_indices_exist(self: Self) -> None:
|
21 |
"""Ensure that indices exist."""
|
22 |
+
await self.ensure_search_index_exists()
|
23 |
|
24 |
async def ensure_search_index_exists(self: Self) -> None:
|
25 |
"""
|