From ac0711defbae9dacf92aede347552ebb85c8cbc2 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Mon, 9 Jun 2025 15:54:07 +0200
Subject: [PATCH] Adds description to context in graph completion and its
 subclasses (#934)

<!-- .github/pull_request_template.md -->

## Description
Adds description to context in graph completion and its subclasses

## DCO Affirmation
I affirm that all code in every commit of this pull request conforms to
the terms of the Topoteretes Developer Certificate of Origin.
---
 cognee/modules/retrieval/graph_completion_retriever.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cognee/modules/retrieval/graph_completion_retriever.py b/cognee/modules/retrieval/graph_completion_retriever.py
index 3c5e85089..5b78e5b47 100644
--- a/cognee/modules/retrieval/graph_completion_retriever.py
+++ b/cognee/modules/retrieval/graph_completion_retriever.py
@@ -50,7 +50,7 @@ class GraphCompletionRetriever(BaseRetriever):
                         content = text
                     else:
                         name = node.attributes.get("name", "Unnamed Node")
-                        content = name
+                        content = node.attributes.get("description", name)
                     nodes[node.id] = {"node": node, "name": name, "content": content}
         return nodes