github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-24 15:43:54 +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'. traversal_source (str): The traversal source to use for queries. Defaults to 'g'.
message_serializer (Optional[Any]): The message serializer to use for requests. message_serializer (Optional[Any]): The message serializer to use for requests.
Defaults to serializer.GraphSONSerializersV2d0() 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 *Security note*: Make sure that the database connection uses credentials
that are narrowly-scoped to only include necessary permissions. that are narrowly-scoped to only include necessary permissions.
Failure to do so may result in data corruption or loss, since the calling 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, password: Optional[str] = None,
traversal_source: str = "g", traversal_source: str = "g",
message_serializer: Optional[Any] = None, message_serializer: Optional[Any] = None,
include_edge_properties: bool = False,
) -> None: ) -> None:
"""Create a new Gremlin graph wrapper instance.""" """Create a new Gremlin graph wrapper instance."""
try: try:
@ -70,6 +73,7 @@ class GremlinGraph(GraphStore):
else serializer.GraphSONSerializersV2d0(), else serializer.GraphSONSerializersV2d0(),
) )
self.schema: str = "" self.schema: str = ""
self.include_edge_properties = include_edge_properties
@property @property
def get_schema(self) -> str: def get_schema(self) -> str:
@ -102,11 +106,27 @@ class GremlinGraph(GraphStore):
[ [
"Vertex labels are the following:", "Vertex labels are the following:",
",".join(vertex_schema), ",".join(vertex_schema),
"Edge labes are the following:", "Edge labels are the following:",
",".join(edge_schema), ",".join(edge_schema),
f"Vertices have following properties:\n{vertex_properties}", 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]]: def query(self, query: str, params: dict = {}) -> List[Dict[str, Any]]:
q = self.client.submit(query) q = self.client.submit(query)