From f46b8bb00803f962b268235c5ef389b476812fe2 Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Mon, 29 Sep 2025 23:19:43 +0100 Subject: [PATCH] Fixing collection management param error --- .../trustgraph/storage/doc_embeddings/milvus/write.py | 2 +- .../trustgraph/storage/doc_embeddings/pinecone/write.py | 2 +- .../trustgraph/storage/doc_embeddings/qdrant/write.py | 2 +- .../trustgraph/storage/graph_embeddings/milvus/write.py | 2 +- .../trustgraph/storage/graph_embeddings/pinecone/write.py | 2 +- .../trustgraph/storage/graph_embeddings/qdrant/write.py | 2 +- trustgraph-flow/trustgraph/storage/triples/cassandra/write.py | 2 +- trustgraph-flow/trustgraph/storage/triples/falkordb/write.py | 2 +- trustgraph-flow/trustgraph/storage/triples/memgraph/write.py | 2 +- trustgraph-flow/trustgraph/storage/triples/neo4j/write.py | 2 +- 10 files changed, 10 insertions(+), 10 deletions(-) diff --git a/trustgraph-flow/trustgraph/storage/doc_embeddings/milvus/write.py b/trustgraph-flow/trustgraph/storage/doc_embeddings/milvus/write.py index 2827b9b5..f723d5e1 100755 --- a/trustgraph-flow/trustgraph/storage/doc_embeddings/milvus/write.py +++ b/trustgraph-flow/trustgraph/storage/doc_embeddings/milvus/write.py @@ -93,7 +93,7 @@ class Processor(DocumentEmbeddingsStoreService): help=f'Milvus store URI (default: {default_store_uri})' ) - async def on_storage_management(self, message): + async def on_storage_management(self, message, consumer, flow): """Handle storage management requests""" logger.info(f"Storage management request: {message.operation} for {message.user}/{message.collection}") diff --git a/trustgraph-flow/trustgraph/storage/doc_embeddings/pinecone/write.py b/trustgraph-flow/trustgraph/storage/doc_embeddings/pinecone/write.py index e3b32ea3..29092748 100644 --- a/trustgraph-flow/trustgraph/storage/doc_embeddings/pinecone/write.py +++ b/trustgraph-flow/trustgraph/storage/doc_embeddings/pinecone/write.py @@ -198,7 +198,7 @@ class Processor(DocumentEmbeddingsStoreService): help=f'Pinecone region, (default: {default_region}' ) - async def on_storage_management(self, message): + async def on_storage_management(self, message, consumer, flow): """Handle storage management requests""" logger.info(f"Storage management request: {message.operation} for {message.user}/{message.collection}") diff --git a/trustgraph-flow/trustgraph/storage/doc_embeddings/qdrant/write.py b/trustgraph-flow/trustgraph/storage/doc_embeddings/qdrant/write.py index 2a13dc32..f53de0c5 100644 --- a/trustgraph-flow/trustgraph/storage/doc_embeddings/qdrant/write.py +++ b/trustgraph-flow/trustgraph/storage/doc_embeddings/qdrant/write.py @@ -141,7 +141,7 @@ class Processor(DocumentEmbeddingsStoreService): help=f'Qdrant API key (default: None)' ) - async def on_storage_management(self, message): + async def on_storage_management(self, message, consumer, flow): """Handle storage management requests""" logger.info(f"Storage management request: {message.operation} for {message.user}/{message.collection}") diff --git a/trustgraph-flow/trustgraph/storage/graph_embeddings/milvus/write.py b/trustgraph-flow/trustgraph/storage/graph_embeddings/milvus/write.py index ce7287fb..a93bc9a9 100755 --- a/trustgraph-flow/trustgraph/storage/graph_embeddings/milvus/write.py +++ b/trustgraph-flow/trustgraph/storage/graph_embeddings/milvus/write.py @@ -89,7 +89,7 @@ class Processor(GraphEmbeddingsStoreService): help=f'Milvus store URI (default: {default_store_uri})' ) - async def on_storage_management(self, message): + async def on_storage_management(self, message, consumer, flow): """Handle storage management requests""" logger.info(f"Storage management request: {message.operation} for {message.user}/{message.collection}") diff --git a/trustgraph-flow/trustgraph/storage/graph_embeddings/pinecone/write.py b/trustgraph-flow/trustgraph/storage/graph_embeddings/pinecone/write.py index d2dd0223..cc845737 100755 --- a/trustgraph-flow/trustgraph/storage/graph_embeddings/pinecone/write.py +++ b/trustgraph-flow/trustgraph/storage/graph_embeddings/pinecone/write.py @@ -197,7 +197,7 @@ class Processor(GraphEmbeddingsStoreService): help=f'Pinecone region, (default: {default_region}' ) - async def on_storage_management(self, message): + async def on_storage_management(self, message, consumer, flow): """Handle storage management requests""" logger.info(f"Storage management request: {message.operation} for {message.user}/{message.collection}") diff --git a/trustgraph-flow/trustgraph/storage/graph_embeddings/qdrant/write.py b/trustgraph-flow/trustgraph/storage/graph_embeddings/qdrant/write.py index f013a839..e3eb0362 100755 --- a/trustgraph-flow/trustgraph/storage/graph_embeddings/qdrant/write.py +++ b/trustgraph-flow/trustgraph/storage/graph_embeddings/qdrant/write.py @@ -148,7 +148,7 @@ class Processor(GraphEmbeddingsStoreService): help=f'Qdrant API key' ) - async def on_storage_management(self, message): + async def on_storage_management(self, message, consumer, flow): """Handle storage management requests""" logger.info(f"Storage management request: {message.operation} for {message.user}/{message.collection}") diff --git a/trustgraph-flow/trustgraph/storage/triples/cassandra/write.py b/trustgraph-flow/trustgraph/storage/triples/cassandra/write.py index f057aed2..86cc1dc7 100755 --- a/trustgraph-flow/trustgraph/storage/triples/cassandra/write.py +++ b/trustgraph-flow/trustgraph/storage/triples/cassandra/write.py @@ -123,7 +123,7 @@ class Processor(TriplesStoreService): await self.storage_request_consumer.start() await self.storage_response_producer.start() - async def on_storage_management(self, message): + async def on_storage_management(self, message, consumer, flow): """Handle storage management requests""" logger.info(f"Storage management request: {message.operation} for {message.user}/{message.collection}") diff --git a/trustgraph-flow/trustgraph/storage/triples/falkordb/write.py b/trustgraph-flow/trustgraph/storage/triples/falkordb/write.py index c9b371c2..fe410da8 100755 --- a/trustgraph-flow/trustgraph/storage/triples/falkordb/write.py +++ b/trustgraph-flow/trustgraph/storage/triples/falkordb/write.py @@ -191,7 +191,7 @@ class Processor(TriplesStoreService): await self.storage_request_consumer.start() await self.storage_response_producer.start() - async def on_storage_management(self, message): + async def on_storage_management(self, message, consumer, flow): """Handle storage management requests""" logger.info(f"Storage management request: {message.operation} for {message.user}/{message.collection}") diff --git a/trustgraph-flow/trustgraph/storage/triples/memgraph/write.py b/trustgraph-flow/trustgraph/storage/triples/memgraph/write.py index 61b2f5e5..40aaa6e7 100755 --- a/trustgraph-flow/trustgraph/storage/triples/memgraph/write.py +++ b/trustgraph-flow/trustgraph/storage/triples/memgraph/write.py @@ -323,7 +323,7 @@ class Processor(TriplesStoreService): await self.storage_request_consumer.start() await self.storage_response_producer.start() - async def on_storage_management(self, message): + async def on_storage_management(self, message, consumer, flow): """Handle storage management requests""" logger.info(f"Storage management request: {message.operation} for {message.user}/{message.collection}") diff --git a/trustgraph-flow/trustgraph/storage/triples/neo4j/write.py b/trustgraph-flow/trustgraph/storage/triples/neo4j/write.py index 9f8ce7e2..23e601ef 100755 --- a/trustgraph-flow/trustgraph/storage/triples/neo4j/write.py +++ b/trustgraph-flow/trustgraph/storage/triples/neo4j/write.py @@ -274,7 +274,7 @@ class Processor(TriplesStoreService): await self.storage_request_consumer.start() await self.storage_response_producer.start() - async def on_storage_management(self, message): + async def on_storage_management(self, message, consumer, flow): """Handle storage management requests""" logger.info(f"Storage management request: {message.operation} for {message.user}/{message.collection}")