Rename process->run for Components for consistency with Pipeline

Author: stellasia

src/neo4j_genai/components/kg_builder.py

@@ -7,12 +7,12 @@
 
 
 class DocumentChunker(Component):
-    async def process(self, text: str) -> dict[str, Any]:
+    async def run(self, text: str) -> dict[str, Any]:
         return {"chunks": [t.strip() for t in text.split(".") if t.strip()]}
 
 
 class SchemaBuilder(Component):
-    async def process(self, schema: dict[str, Any]) -> dict[str, Any]:
+    async def run(self, schema: dict[str, Any]) -> dict[str, Any]:
         return {"schema": schema}
 
 
@@ -25,7 +25,7 @@ async def _process_chunk(self, chunk: str, schema: str) -> dict[str, Any]:
             }
         }
 
-    async def process(self, chunks: list[str], schema: str) -> dict[str, Any]:
+    async def run(self, chunks: list[str], schema: str) -> dict[str, Any]:
         tasks = [self._process_chunk(chunk, schema) for chunk in chunks]
         result = await asyncio.gather(*tasks)
         merged_result: dict[str, Any] = {"data": {"entities": [], "relations": []}}
@@ -36,7 +36,7 @@ async def process(self, chunks: list[str], schema: str) -> dict[str, Any]:
 
 
 class Writer(Component):
-    async def process(
+    async def run(
         self, entities: dict[str, Any], relations: dict[str, Any]
     ) -> dict[str, Any]:
         return {

src/neo4j_genai/components/rag.py

@@ -16,22 +16,25 @@ def search(self, *args: Any, **kwargs: Any) -> RetrieverResult:
             ]
         )
 
-    async def process(self, query: str) -> dict[str, Any]:
+    async def run(self, query: str) -> dict[str, Any]:
         res = self.search(query)
         return {"context": "\n".join(c.content for c in res.items)}
 
 
 class PromptTemplate(Component):
-    async def process(self, query: str, context: list[str]) -> dict[str, Any]:
+    async def run(self, query: str, context: list[str]) -> dict[str, Any]:
         return {"prompt": f"my prompt using '{context}', query '{query}'"}
 
 
 class LLM(Component):
-    async def process(self, prompt: str) -> dict[str, Any]:
+    async def run(self, prompt: str) -> dict[str, Any]:
         return {"answer": f"some text based on '{prompt}'"}
 
 
 if __name__ == "__main__":
+    # retriever = Retriever()
+    # print(asyncio.run(retriever.run("my context item 1")))
+
     pipe = Pipeline()
     pipe.add_component("retrieve", Retriever())
     pipe.add_component("augment", PromptTemplate())

src/neo4j_genai/core/pipeline.py

@@ -49,7 +49,7 @@ class RunResult(BaseModel):
 
 
 class Component:
-    async def process(self, *args: Any, **kwargs: Any) -> dict[str, Any]:
+    async def run(self, *args: Any, **kwargs: Any) -> dict[str, Any]:
         return {}
 
 
@@ -63,7 +63,7 @@ def __init__(self, name: str, component: Component, pipeline: "Pipeline"):
     async def execute(self, **kwargs: Any) -> RunResult:
         logger.debug(f"Running component {self.name} with {kwargs}")
         self.status = RunStatus.RUNNING
-        res = await self.component.process(**kwargs)
+        res = await self.component.run(**kwargs)
         self.status = RunStatus.DONE
         return RunResult(
             status=self.status,