apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-06-13 08:45:13 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix cassandra fix

Browse source
This commit is contained in:
Cyber MacGeddon 2026-04-21 11:57:57 +01:00
parent 0bc97b9374
commit 4b11725bfe
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
trustgraph-flow/trustgraph/query/triples/cassandra/service.py
View file

@ -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]