fix(graph): improve LLM tool UX for GetSubgraphTool and ReadGraphTool

This commit is contained in:
Riz Ashraf committed 2026-10-10 12:50:07 +01:00
1 parent eed9a9165c
commit 39528411f4
2 files changed
+5 -2

No files matched your search

+3 -1
View File
@@ -100,6 +100,8 @@ pub struct DeleteRelationsTool {
pub struct ReadGraphTool {
/// Optional namespace to restrict the read to.
pub namespace: Option<String>,
/// Optional entity_type to filter entities by (e.g. 'DatabaseTable').
pub entity_type: Option<String>,
/// Optional token budget cap. Truncates graph outputs to fit within the specified token budget.
pub max_tokens: Option<usize>,
/// Optional maximum number of entities to return (pagination).
@@ -213,7 +215,7 @@ pub struct GetSubgraphTool {
pub root_entity: Option<String>,
/// Maximum search depth (hops). Defaults to 2.
pub depth: Option<u32>,
/// Output format: 'json' (raw entities and relations) or 'markdown_tree' (compact topology tree). Defaults to 'json'.
/// Output format: 'json' (raw entities and relations) or 'markdown_tree' (compact topology tree). Defaults to 'markdown_tree'.
pub format: Option<SubgraphFormat>,
/// Optional namespace filter.
pub namespace: Option<String>,