github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-23 15:19:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

community: Add edge properties to the gremlin graph schema (#30449)

Browse Source 
Description: Extend the gremlin graph schema to include the edge
properties, grouped by its triples; i.e: `inVLabel` and `outVLabel`.
This should give more context when crafting queries to run against a
gremlin graph db
This commit is contained in:
Ian Muge 2025-03-25 00:03:01 +01:00 committed by GitHub
parent b60e6f6efa
commit a8ce63903d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 21 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
libs/community/langchain_community/graphs/gremlin_graph.py
View File

@ -20,6 +20,8 @@ class GremlinGraph(GraphStore):
traversal_source (str): The traversal source to use for queries. Defaults to 'g'.
message_serializer (Optional[Any]): The message serializer to use for requests.
Defaults to serializer.GraphSONSerializersV2d0()
include_edge_properties (bool): Whether to include edge properties in
the gremlin graph schema. Defaults to False
*Security note*: Make sure that the database connection uses credentials
that are narrowly-scoped to only include necessary permissions.
Failure to do so may result in data corruption or loss, since the calling
@ -46,6 +48,7 @@ class GremlinGraph(GraphStore):
password: Optional[str] = None,
traversal_source: str = "g",
message_serializer: Optional[Any] = None,
include_edge_properties: bool = False,
) -> None:
"""Create a new Gremlin graph wrapper instance."""
try:
@ -70,6 +73,7 @@ class GremlinGraph(GraphStore):
else serializer.GraphSONSerializersV2d0(),
)
self.schema: str = ""
self.include_edge_properties = include_edge_properties
@property
def get_schema(self) -> str:
@ -102,11 +106,27 @@ class GremlinGraph(GraphStore):
[
"Vertex labels are the following:",
",".join(vertex_schema),
"Edge labes are the following:",
"Edge labels are the following:",
",".join(edge_schema),
f"Vertices have following properties:\n{vertex_properties}",
]
)
if self.include_edge_properties:
edge_properties = (
self.client.submit(
"g.E().group().by(label)"
".by(project('inVLabel', 'outVLabel','properties')"
".by(inV().label()).by(outV().label()).by(properties().key().dedup()"
".fold()).dedup().fold())"
)
.all()
.result()[0]
)
self.structured_schema["edge_props"] = edge_properties
self.schema += (
f"\nEdges have the following properties, grouped by label and"
f" the distinct inV and outV labels:\n {edge_properties}"
)
def query(self, query: str, params: dict = {}) -> List[Dict[str, Any]]:
q = self.client.submit(query)