feat: add document store

HaoXuAI · HaoXuAI · commit d2e0a5998340 · 2024-03-31T16:46:09.000-07:00
diff --git a/sdk/python/feast/infra/online_stores/contrib/postgres.py b/sdk/python/feast/infra/online_stores/contrib/postgres.py
@@ -340,11 +340,7 @@ def online_search(
 
         return result
 
-    def create_index(
-            self,
-            config: RepoConfig,
-            table: str
-    ):
+    def create_index(self, config: RepoConfig, table: str):
         document_store_config = config.document_store_config
         with self._get_conn(config) as conn, conn.cursor() as cur:
             cur.execute(
diff --git a/sdk/python/feast/infra/online_stores/document_store.py b/sdk/python/feast/infra/online_stores/document_store.py
@@ -5,9 +5,9 @@
 import numpy as np
 
 from feast.feature_view import FeatureView
+from feast.infra.online_stores.online_store import OnlineStore
 from feast.protos.feast.types.Value_pb2 import Value as ValueProto
 from feast.repo_config import FeastConfigBaseModel, RepoConfig
-from feast.infra.online_stores.online_store import OnlineStore
 
 
 class DocumentStoreIndexConfig(FeastConfigBaseModel):
@@ -16,14 +16,13 @@ class DocumentStoreIndexConfig(FeastConfigBaseModel):
 
 
 class DocumentStore(OnlineStore):
-
     @abstractmethod
     def online_search(
-            self,
-            config: RepoConfig,
-            table: FeatureView,
-            requested_feature: str,
-            embeddings: np.ndarray,
-            top_k: int,
+        self,
+        config: RepoConfig,
+        table: FeatureView,
+        requested_feature: str,
+        embeddings: np.ndarray,
+        top_k: int,
     ) -> List[Tuple[Optional[datetime], Optional[Dict[str, ValueProto]]]]:
-        raise NotImplementedError("You have to implement this!")
+        raise NotImplementedError("You have to implement this!")
