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 91bd96719272beb7e23fe4b702fa4d44914d8576..4256eeb3d3832e853b22627b3007e9d1593c9303 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(