github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-17 18:23:59 +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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats 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(