From 4b11725bfe8d172c5c25cdd9608b03ea97382c60 Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Tue, 21 Apr 2026 11:57:57 +0100 Subject: [PATCH] Fix cassandra fix --- .../trustgraph/query/triples/cassandra/service.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/trustgraph-flow/trustgraph/query/triples/cassandra/service.py b/trustgraph-flow/trustgraph/query/triples/cassandra/service.py index 27e8efb9..efce5968 100755 --- a/trustgraph-flow/trustgraph/query/triples/cassandra/service.py +++ b/trustgraph-flow/trustgraph/query/triples/cassandra/service.py @@ -395,7 +395,7 @@ class Processor(TriplesQueryService): else: # For specific patterns, fall back to non-streaming # (these typically return small result sets anyway) - async for batch, is_final in self._fallback_stream(query, batch_size): + async for batch, is_final in self._fallback_stream(workspace, query, batch_size): yield batch, is_final return @@ -452,9 +452,9 @@ class Processor(TriplesQueryService): logger.error(f"Exception in streaming query: {e}", exc_info=True) raise e - async def _fallback_stream(self, query, batch_size): + async def _fallback_stream(self, workspace, query, batch_size): """Fallback to non-streaming query with post-hoc batching.""" - triples = await self.query_triples(query) + triples = await self.query_triples(workspace, query) for i in range(0, len(triples), batch_size): batch = triples[i:i + batch_size]