reset embedding effective

src/api/services/postgresql_service.py

@@ -0,0 +1,38 @@
+import logging
+from typing import List
+from src.api.database import Database
+
+logger = logging.getLogger(__name__)
+
+class PostgresqlService:
+    def __init__(self, db: Database):
+        self.db = db
+
+
+    async def get_db_rows_from_dataset_name(self, dataset) -> List:
+        query_mapping = {
+            "re-mind/product_type_embedding": {
+                "column": "type",
+                "table": "product_producttype",
+            },
+            "re-mind/marketplace_name_embedding": {
+                "column": "name",
+                "table": "invoice_marketplace",
+            },
+            "re-mind/manufacturer_name_embedding": {
+                "column": "name",
+                "table": "product_manufacturer",
+            },
+            "re-mind/seller_name_embedding": {
+                "column": "name",
+                "table": "product_seller",
+            }
+        }
+
+        db_table = query_mapping[dataset]["table"]
+        column = query_mapping[dataset]["column"]
+        query = f"SELECT {column} FROM {db_table}"
+
+        results = await self.db.fetch(query)
+
+        return results