feat(py): Added the indexer and the retriever for the resolve_action plugin

Author: hendrixmar

py/plugins/dev-local-vectorstore/src/genkit/plugins/dev_local_vectorstore/plugin_api.py

@@ -67,9 +67,18 @@ def resolve_action(
         kind: ActionKind,
         name: str,
     ) -> None:
-        ...
 
-    def _configure_dev_local_retriever(self, ai: GenkitRegistry) -> Action:
+        if kind is ActionKind.RETRIEVER:
+            self._configure_dev_local_retriever(ai, name)
+            return None
+
+        if kind is ActionKind.INDEXER:
+            self._configure_dev_local_indexer(ai, name)
+
+
+        return None
+
+    def _configure_dev_local_retriever(self, ai: GenkitRegistry, name: str | None = None) -> Action:
         """Registers Local Vector Store retriever for provided parameters.
 
         Args:
@@ -81,7 +90,7 @@ def _configure_dev_local_retriever(self, ai: GenkitRegistry) -> Action:
         """
         retriever = DevLocalVectorStoreRetriever(
             ai=ai,
-            index_name=self.index_name,
+            index_name=name or self.index_name,
             embedder=self.embedder,
             embedder_options=self.embedder_options,
         )
@@ -92,7 +101,7 @@ def _configure_dev_local_retriever(self, ai: GenkitRegistry) -> Action:
             fn=retriever.retrieve,
         )
 
-    def _configure_dev_local_indexer(self, ai: GenkitRegistry) -> Action:
+    def _configure_dev_local_indexer(self, ai: GenkitRegistry, name: str | None = None) -> Action:
         """Registers Local Vector Store indexer for provided parameters.
 
         Args:
@@ -104,13 +113,16 @@ def _configure_dev_local_indexer(self, ai: GenkitRegistry) -> Action:
         """
         indexer = DevLocalVectorStoreIndexer(
             ai=ai,
-            index_name=self.index_name,
+            index_name= name or self.index_name,
             embedder=self.embedder,
             embedder_options=self.embedder_options,
         )
 
         DevLocalVectorStore._indexers[self.index_name] = indexer
 
+
+
+
     @classmethod
     async def index(cls, index_name: str, documents: Docs) -> None:
         """Lookups the Local Vector Store indexer for provided index name.

py/plugins/dev-local-vectorstore/tests/test_plugin_api.py

@@ -27,14 +27,30 @@
 @pytest.mark.parametrize(
     'kind, name',
     [
-        (ActionKind.MODEL, 'test_model'),
-        (ActionKind.EMBEDDER, 'test_embedder'),
+        (ActionKind.INDEXER, 'test_indexer'),
+        (ActionKind.RETRIEVER, 'test_retriever'),
     ],
 )
 def test_action_resolve(kind, name, vectorstore_plugin_instance):
     """Test initialize method of Vectorstore plugin."""
     ai_mock = MagicMock(spec=Genkit)
     assert hasattr(vectorstore_plugin_instance, "resolve_action")
-    assert  vectorstore_plugin_instance.resolve_action(ai_mock, kind, name) is None
 
+    if kind == ActionKind.RETRIEVER:
+        vectorstore_plugin_instance._configure_dev_local_retriever = MagicMock()
 
+        assert vectorstore_plugin_instance.resolve_action(ai_mock, kind, name) is None
+
+        vectorstore_plugin_instance._configure_dev_local_retriever.assert_called_once_with(
+            ai_mock,
+            name
+        )
+    else:
+        vectorstore_plugin_instance._configure_dev_local_indexer = MagicMock()
+
+        assert vectorstore_plugin_instance.resolve_action(ai_mock, kind, name) is None
+
+        vectorstore_plugin_instance._configure_dev_local_indexer.assert_called_once_with(
+            ai_mock,
+            name
+        )