From f6ad23eb6613cfc57ebd8883c3815ce4dc0fb1bf Mon Sep 17 00:00:00 2001
From: Rana Banerjee <rana1224@gmail.com>
Date: Sat, 24 Feb 2024 00:56:22 +0530
Subject: [PATCH] Return the object along with the subject when extracting the
keywords (#11326)
---
.../llama_index/core/indices/knowledge_graph/retrievers.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/llama-index-core/llama_index/core/indices/knowledge_graph/retrievers.py b/llama-index-core/llama_index/core/indices/knowledge_graph/retrievers.py
index 91bd967192..4256eeb3d3 100644
--- a/llama-index-core/llama_index/core/indices/knowledge_graph/retrievers.py
+++ b/llama-index-core/llama_index/core/indices/knowledge_graph/retrievers.py
@@ -173,6 +173,10 @@ class KGTableRetriever(BaseRetriever):
keyword = rel_text.split(",")[0]
if keyword:
keywords.append(keyword.strip("(\"'"))
+ # Return the Object as well
+ keyword = rel_text.split(",")[2]
+ if keyword:
+ keywords.append(keyword.strip(" ()\"'"))
return keywords
def _retrieve(
--
GitLab