Merge pull request #15 from 3coins/add-graphs

Added neptune graphs.
langchain-ai · Apr 15, 2024 · 688a3a5 · 688a3a5
2 parents 9dd050b + 5a3347f
commit 688a3a5
Show file tree

Hide file tree

Showing 6 changed files with 761 additions and 3 deletions.
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"]