Fix Vector database index name

.gitignore

@@ -1,4 +1,4 @@
+venv
 .venv
 .env
 __pycache__/
-```

api/ingest.py

@@ -6,6 +6,7 @@
 
 from models.ingest import RequestPayload
 from service.embedding import EmbeddingService, get_encoder
+from utils.summarise import SUMMARY_SUFFIX
 
 router = APIRouter()
 
@@ -37,7 +38,7 @@ async def ingest(payload: RequestPayload) -> Dict:
         embedding_service.generate_embeddings(
             nodes=summary_chunks,
             encoder=encoder,
-            index_name=f"{payload.index_name}-summary",
+            index_name=f"{payload.index_name}-{SUMMARY_SUFFIX}",
         ),
     )
 

service/router.py

@@ -8,6 +8,7 @@
 from models.query import RequestPayload
 from service.embedding import get_encoder
 from service.vector_database import VectorService, get_vector_service
+from utils.summarise import SUMMARY_SUFFIX
 
 
 def create_route_layer() -> RouteLayer:
@@ -48,7 +49,7 @@ async def query(payload: RequestPayload) -> List:
 
     if decision == "summarize":
         vector_service: VectorService = get_vector_service(
-            index_name=f"{payload.index_name}summary",
+            index_name=f"{payload.index_name}-{SUMMARY_SUFFIX}",
             credentials=payload.vector_database,
             encoder=encoder,
         )

utils/summarise.py

@@ -18,6 +18,9 @@ def _generate_content(document: Document) -> str:
 Your summary:"""
 
 
+SUMMARY_SUFFIX = "summary"
+
+
 async def completion(document: Document):
     content = _generate_content(document)
     completion = await client.chat.completions.create(