Spaces:
Running
Running
Update neo4j_connectors.py
Browse files- neo4j_connectors.py +5 -5
neo4j_connectors.py
CHANGED
|
@@ -83,11 +83,11 @@ class GraphDB:
|
|
| 83 |
return [record["skill"] for record in result]
|
| 84 |
|
| 85 |
def get_user_skills(self, user_id):
|
| 86 |
-
|
| 87 |
-
|
| 88 |
-
|
| 89 |
-
|
| 90 |
-
|
| 91 |
result = session.run(query, user_id=user_id)
|
| 92 |
return [(record["skill"], record["confidence"]) for record in result]
|
| 93 |
|
|
|
|
| 83 |
return [record["skill"] for record in result]
|
| 84 |
|
| 85 |
def get_user_skills(self, user_id):
|
| 86 |
+
query = """
|
| 87 |
+
MATCH (u:User {user_id: $user_id})-[r:HAS_SKILL]->(s:Skill)
|
| 88 |
+
RETURN s.name AS skill, r.confidence AS confidence
|
| 89 |
+
"""
|
| 90 |
+
with self.driver.session() as session:
|
| 91 |
result = session.run(query, user_id=user_id)
|
| 92 |
return [(record["skill"], record["confidence"]) for record in result]
|
| 93 |
|