github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-04 20:46:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix buggy output from GraphQAChain (#7372)

Browse Source 
fixes https://github.com/hwchase17/langchain/issues/7289
A simple fix of the buggy output of `graph_qa`. If we have several
entities with triplets then the last entry of `triplets` for a given
entity merges with the first entry of the `triplets` of the next entity.
This commit is contained in:
Bearnardd
2023-07-07 23:19:53 +02:00
committed by GitHub
parent 7cdf97ba9b
commit 184ede4e48
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
langchain/chains/graph_qa/base.py
View File

@@ -75,9 +75,10 @@ class GraphQAChain(Chain):
) )
entities = get_entities(entity_string) entities = get_entities(entity_string)
context = "" context = ""
all_triplets = []
for entity in entities: for entity in entities:
triplets = self.graph.get_entity_knowledge(entity) all_triplets.extend(self.graph.get_entity_knowledge(entity))
context += "\n".join(triplets) context = "\n".join(all_triplets)
_run_manager.on_text("Full Context:", end="\n", verbose=self.verbose) _run_manager.on_text("Full Context:", end="\n", verbose=self.verbose)
_run_manager.on_text(context, color="green", end="\n", verbose=self.verbose) _run_manager.on_text(context, color="green", end="\n", verbose=self.verbose)
result = self.qa_chain( result = self.qa_chain(