Python version compat

Author: stellasia

src/neo4j_genai/components/kg_builder.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import asyncio
 from typing import Any
 

src/neo4j_genai/components/rag.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import asyncio
 from typing import Any
 
@@ -36,7 +38,6 @@ async def process(self, prompt: str) -> dict[str, Any]:
     pipe.add_component("generate", LLM())
     pipe.connect("retrieve", "augment", {"context": "retrieve.context"})
     pipe.connect("augment", "generate", {"prompt": "augment.prompt"})
-    pipe.connect("generate", "retrieve", {"prompt": "augment.prompt"})
 
     query = "my question"
     print(

src/neo4j_genai/core/graph.py

@@ -2,6 +2,8 @@
 Basic graph structure for Pipeline
 """
 
+from __future__ import annotations
+
 from typing import Any, Optional
 
 

src/neo4j_genai/core/pipeline.py

@@ -14,6 +14,8 @@
 e.g. SchemaBuilder + Chunker => ERExtractor
 """
 
+from __future__ import annotations
+
 import asyncio
 import enum
 import logging

src/neo4j_genai/core/stores.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import abc
 from typing import Any