feat: Add python client for remote registry server (#3941)

* add remote registry Signed-off-by: tokoko <[email protected]> * format and lint remote registry code Signed-off-by: tokoko <[email protected]> * add read-only registry exception Signed-off-by: tokoko <[email protected]> --------- Signed-off-by: tokoko <[email protected]>
feast-dev · Mar 7, 2024 · 42a7b81 · 42a7b81
1 parent fa8492d
commit 42a7b81
Show file tree

Hide file tree

Showing 7 changed files with 454 additions and 1 deletion.
diff --git a/sdk/python/feast/errors.py b/sdk/python/feast/errors.py
@@ -415,3 +415,8 @@ def __init__(self):
 class PushSourceNotFoundException(Exception):
     def __init__(self, push_source_name: str):
         super().__init__(f"Unable to find push source '{push_source_name}'.")
+
+
+class ReadOnlyRegistryException(Exception):
+    def __init__(self):
+        super().__init__("Registry implementation is read-only.")
diff --git a/sdk/python/feast/feature_store.py b/sdk/python/feast/feature_store.py
@@ -164,6 +164,10 @@ def __init__(
             self._registry = SnowflakeRegistry(
                 registry_config, self.config.project, None
             )
+        elif registry_config and registry_config.registry_type == "remote":
+            from feast.infra.registry.remote import RemoteRegistry
+
+            self._registry = RemoteRegistry(registry_config, self.config.project, None)
         else:
             r = Registry(self.config.project, registry_config, repo_path=self.repo_path)
             r._initialize_registry(self.config.project)

diff --git a/sdk/python/feast/infra/registry/base_registry.py b/sdk/python/feast/infra/registry/base_registry.py
@@ -246,7 +246,7 @@ def delete_feature_view(self, name: str, project: str, commit: bool = True):
     @abstractmethod
     def get_stream_feature_view(
         self, name: str, project: str, allow_cache: bool = False
-    ):
+    ) -> StreamFeatureView:
         """
         Retrieves a stream feature view.
 

diff --git a/sdk/python/feast/infra/registry/registry.py b/sdk/python/feast/infra/registry/registry.py
@@ -178,6 +178,10 @@ def __new__(
             from feast.infra.registry.snowflake import SnowflakeRegistry
 
             return SnowflakeRegistry(registry_config, project, repo_path)
+        elif registry_config and registry_config.registry_type == "remote":
+            from feast.infra.registry.remote import RemoteRegistry
+
+            return RemoteRegistry(registry_config, project, repo_path)
         else:
             return super(Registry, cls).__new__(cls)