Added neptune graphs.

langchain-ai · Apr 10, 2024 · f0d4b94 · f0d4b94
1 parent e0317f2
commit f0d4b94
Show file tree

Hide file tree

Showing 6 changed files with 760 additions and 1 deletion.
diff --git a/libs/aws/langchain_aws/__init__.py b/libs/aws/langchain_aws/__init__.py
@@ -1,4 +1,5 @@
 from langchain_aws.chat_models import BedrockChat, ChatBedrock
+from langchain_aws.graphs import NeptuneAnalyticsGraph, NeptuneGraph
 from langchain_aws.llms import Bedrock, BedrockLLM, SagemakerEndpoint
 from langchain_aws.retrievers import (
     AmazonKendraRetriever,
@@ -13,4 +14,6 @@
     "SagemakerEndpoint",
     "AmazonKendraRetriever",
     "AmazonKnowledgeBasesRetriever",
+    "NeptuneAnalyticsGraph",
+    "NeptuneGraph",
 ]
diff --git a/libs/aws/langchain_aws/graphs/__init__.py b/libs/aws/langchain_aws/graphs/__init__.py
@@ -0,0 +1,7 @@
+from langchain_aws.graphs.neptune_graph import (
+    BaseNeptuneGraph,
+    NeptuneAnalyticsGraph,
+    NeptuneGraph,
+)
+
+__all__ = ["BaseNeptuneGraph", "NeptuneAnalyticsGraph", "NeptuneGraph"]