Pārlūkot izejas kodu

Config: Update repository and API to use COLLECTION_NAME from .env

Dr-Swopt 3 dienas atpakaļ
vecāks
revīzija
2e5a32e038
2 mainītis faili ar 4 papildinājumiem un 3 dzēšanām
  1. 2 1
      src/api/main.py
  2. 2 2
      src/infrastructure/repository.py

+ 2 - 1
src/api/main.py

@@ -25,7 +25,8 @@ vision_service = VertexVisionService(
 )
 repo = MongoPalmOilRepository(
     uri=os.getenv("MONGO_URI"),
-    db_name=os.getenv("DB_NAME", "palm_oil_db")
+    db_name=os.getenv("DB_NAME", "palm_oil_db"),
+    collection_name=os.getenv("COLLECTION_NAME", "ffb_records")
 )
 analyze_use_case = AnalyzeBunchUseCase(vision_service, repo)
 

+ 2 - 2
src/infrastructure/repository.py

@@ -2,9 +2,9 @@ from pymongo import MongoClient
 from src.domain.models import PalmOilBunch
 
 class MongoPalmOilRepository:
-    def __init__(self, uri: str, db_name: str):
+    def __init__(self, uri: str, db_name: str, collection_name: str):
         self.client = MongoClient(uri)
-        self.collection = self.client[db_name]["ffb_records"]
+        self.collection = self.client[db_name][collection_name]
 
     def save(self, bunch: PalmOilBunch):
         # Convert dataclass to dict for MongoDB