Fixing collection management param error

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}")
 

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}")
 

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}")
 

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}")
 

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}")
 

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}")
 

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}")
 

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}")
 

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}")
 

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}")