Fixed exports for NeptuneOpenCypherQAChain (#8439)

## Description
The imports for `NeptuneOpenCypherQAChain` are failing. This PR adds the
chain class to the `__init__.py` file to fix this issue.

## Maintainers
@dev2049 
@krlawrence
This commit is contained in:
Piyush Jain
2023-07-30 20:36:22 -07:00
committed by GitHub
parent e98e2b2b81
commit b2f8a5bae9
4 changed files with 37 additions and 6 deletions

View File

@@ -34,6 +34,7 @@ from langchain.chains.graph_qa.cypher import GraphCypherQAChain
from langchain.chains.graph_qa.hugegraph import HugeGraphQAChain
from langchain.chains.graph_qa.kuzu import KuzuQAChain
from langchain.chains.graph_qa.nebulagraph import NebulaGraphQAChain
from langchain.chains.graph_qa.neptune_cypher import NeptuneOpenCypherQAChain
from langchain.chains.graph_qa.sparql import GraphSparqlQAChain
from langchain.chains.hyde.base import HypotheticalDocumentEmbedder
from langchain.chains.llm import LLMChain
@@ -101,6 +102,7 @@ __all__ = [
"MultiRouteChain",
"NatBotChain",
"NebulaGraphQAChain",
"NeptuneOpenCypherQAChain",
"OpenAIModerationChain",
"OpenAPIEndpointChain",
"QAGenerationChain",

View File

@@ -0,0 +1,2 @@
def test_import() -> None:
from langchain.chains import NeptuneOpenCypherQAChain # noqa: F401

View File

@@ -0,0 +1,2 @@
def test_import() -> None:
from langchain.graphs import NeptuneGraph # noqa: F401