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.
This commit is contained in:
parent
4eb71ccaf4
commit
ac0711defb
1 changed files with 1 additions and 1 deletions
|
|
@ -50,7 +50,7 @@ class GraphCompletionRetriever(BaseRetriever):
|
||||||
content = text
|
content = text
|
||||||
else:
|
else:
|
||||||
name = node.attributes.get("name", "Unnamed Node")
|
name = node.attributes.get("name", "Unnamed Node")
|
||||||
content = name
|
content = node.attributes.get("description", name)
|
||||||
nodes[node.id] = {"node": node, "name": name, "content": content}
|
nodes[node.id] = {"node": node, "name": name, "content": content}
|
||||||
return nodes
|
return nodes
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue