github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-21 18:39:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

community[minor]: add neptune analytics graph (#20047)

Browse Source 
Replacement for PR
[#19772](https://github.com/langchain-ai/langchain/pull/19772).

---------

Co-authored-by: Dave Bechberger <dbechbe@amazon.com>
Co-authored-by: bechbd <bechbd@users.noreply.github.com>
This commit is contained in:
Piyush Jain
2024-04-09 07:20:59 -07:00
committed by GitHub
parent ad9750403b
commit cd7abc495a
5 changed files with 302 additions and 117 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
libs/langchain/langchain/chains/graph_qa/neptune_cypher.py
View File

@@ -3,7 +3,7 @@ from __future__ import annotations
import re
from typing import Any, Dict, List, Optional
from langchain_community.graphs import NeptuneGraph
from langchain_community.graphs import BaseNeptuneGraph
from langchain_core.callbacks import CallbackManagerForChainRun
from langchain_core.language_models import BaseLanguageModel
from langchain_core.prompts.base import BasePromptTemplate
@@ -107,7 +107,7 @@ class NeptuneOpenCypherQAChain(Chain):
response = chain.run(query)
"""
graph: NeptuneGraph = Field(exclude=True)
graph: BaseNeptuneGraph = Field(exclude=True)
cypher_generation_chain: LLMChain
qa_chain: LLMChain
input_key: str = "query" #: :meta private: