Add experimental Neo4J adapter

Author: s-heppner

.gitignore

@@ -21,3 +21,6 @@
 
 # customized config files
 /test/test_config.ini
+
+# Neo4J Data
+compose_with_neo4j/data

compose_with_neo4j/compose.yaml

@@ -0,0 +1,25 @@
+services:
+  semantic_matching_service:
+    build: ..
+    container_name: semantic_matching_service
+    restart: unless-stopped
+    ports:
+      - "8000:8000"
+    working_dir: /app/semantic_matcher
+    command: ["python", "service.py"]
+
+  neo4j:
+    image: neo4j:latest
+    container_name: neo4j
+    volumes:
+      - ./data/neo4j/logs:/logs
+      - ./data/neo4j/config:/config
+      - ./data/neo4j/data:/data
+      - ./data/neo4j/plugins:/plugins
+    environment:
+      - NEO4J_AUTH=neo4j/your_password
+      - NEO4J_dbms_usage__report_enabled=false
+    ports:
+      - "7474:7474"
+      - "7687:7687"
+    restart: always

pyproject.toml

@@ -17,6 +17,7 @@ dependencies = [
     "uvicorn>=0.21.1",
     "requests>=2.31.0",
     "networkx>=3.4.2",
+
 ]
 
 [project.optional-dependencies]
@@ -27,6 +28,9 @@ dev = [
     "types-networkx",  # Type hints for networkx
     "types-requests",   # Type hints for requests
 ]
+neo4j = [
+    "neo4j>=5.28.1",
+]
 
 [tool.setuptools]
 packages = { find = { include = ["semantic_matcher"], exclude = ["test*"] } }

semantic_matcher/algorithm.py

@@ -4,6 +4,7 @@
 
 import networkx as nx
 from pydantic import BaseModel
+from neo4j import GraphDatabase
 
 
 class SemanticMatchGraph(nx.DiGraph):
@@ -53,6 +54,34 @@ def from_file(cls, filename: str) -> "SemanticMatchGraph":
             )
         return graph
 
+    def to_neo4j(self, url: str, user: str, password: str):
+        driver = GraphDatabase.driver(url, auth=(user, password))
+        with driver.session() as session:
+            # Optional: clear previous graph
+            session.run("MATCH (n) DETACH DELETE n")
+
+            for u, v, data in self.edges(data=True):
+                session.run("""
+                    MERGE (a:Semantic {id: $u})
+                    MERGE (b:Semantic {id: $v})
+                    MERGE (a)-[:MATCH {score: $score}]->(b)
+                """, u=u, v=v, score=data["weight"])
+        driver.close()
+
+    @classmethod
+    def from_neo4j(cls, url: str, user: str, password: str) -> "SemanticMatchGraph":
+        graph = cls()
+        driver = GraphDatabase.driver(url, auth=(user, password))
+        with driver.session() as session:
+            result = session.run("""
+                MATCH (a:Semantic)-[r:MATCH]->(b:Semantic)
+                RETURN a.id AS u, b.id AS v, r.score AS score
+            """)
+            for record in result:
+                graph.add_semantic_match(record["u"], record["v"], record["score"])
+        driver.close()
+        return graph
+
 
 class SemanticMatch(BaseModel):
     base_semantic_id: str
@@ -145,3 +174,17 @@ def find_semantic_matches(
                 heapq.heappush(pq, (-new_score, neighbor, path + [node]))  # Push updated path
 
     return results
+
+
+if __name__ == '__main__':
+    graph_complex = SemanticMatchGraph()
+    graph_complex.add_edge("A", "B", weight=0.9)
+    graph_complex.add_edge("A", "C", weight=0.8)
+    graph_complex.add_edge("B", "D", weight=0.7)
+    graph_complex.add_edge("C", "D", weight=0.6)
+    graph_complex.add_edge("D", "E", weight=0.5)
+    graph_complex.to_neo4j(
+        url="bolt://localhost:7687",
+        user="neo4j",
+        password="your_password"
+    )