fix node to object mapping in FGARetriever (#62)

deepu105 · web-flow · commit 31ee4d858500 · 2026-04-07T13:22:09.000-04:00
diff --git a/packages/auth0-ai-llamaindex/auth0_ai_llamaindex/FGARetriever.py b/packages/auth0-ai-llamaindex/auth0_ai_llamaindex/FGARetriever.py
@@ -116,7 +116,7 @@ async def _async_filter_FGA(
                 }.values()
             )
             node_to_obj = {
-                nodeWithScore: check.object
+                nodeWithScore.id_: check.object
                 for check, nodeWithScore in zip(all_checks, nodes)
             }
 
@@ -132,8 +132,8 @@ async def _async_filter_FGA(
             return [
                 node
                 for node in nodes
-                if node_to_obj[node] in permissions_map
-                and permissions_map[node_to_obj[node]]
+                if node_to_obj[node.id_] in permissions_map
+                and permissions_map[node_to_obj[node.id_]]
             ]
 
     async def _aretrieve(self, query_bundle: QueryBundle) -> List[NodeWithScore]:

Original file line number	Diff line number	Diff line change
`@@ -116,7 +116,7 @@ async def _async_filter_FGA(`
`116`	`116`	`}.values()`
`117`	`117`	`)`
`118`	`118`	`node_to_obj = {`
`119`		`- nodeWithScore: check.object`
	`119`	`+ nodeWithScore.id_: check.object`
`120`	`120`	`for check, nodeWithScore in zip(all_checks, nodes)`
`121`	`121`	`}`
`122`	`122`
`@@ -132,8 +132,8 @@ async def _async_filter_FGA(`
`132`	`132`	`return [`
`133`	`133`	`node`
`134`	`134`	`for node in nodes`
`135`		`- if node_to_obj[node] in permissions_map`
`136`		`- and permissions_map[node_to_obj[node]]`
	`135`	`+ if node_to_obj[node.id_] in permissions_map`
	`136`	`+ and permissions_map[node_to_obj[node.id_]]`
`137`	`137`	`]`
`138`	`138`
`139`	`139`	`async def _aretrieve(self, query_bundle: QueryBundle) -> List[NodeWithScore]:`