From 34c2ae5ca64e73b6622673d32319b10c75592564 Mon Sep 17 00:00:00 2001
From: Cyber MacGeddon <cybermaggedon@gmail.com>
Date: Fri, 19 Sep 2025 14:32:21 +0100
Subject: [PATCH] Fixing collection management translation

---
 .../trustgraph/messaging/translators/collection.py         | 7 ++++++-
 .../trustgraph/gateway/dispatch/collection_management.py   | 4 +++-
 trustgraph-flow/trustgraph/librarian/collection_manager.py | 3 +++
 3 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/trustgraph-base/trustgraph/messaging/translators/collection.py b/trustgraph-base/trustgraph/messaging/translators/collection.py
index 5ad31f26..38ac813b 100644
--- a/trustgraph-base/trustgraph/messaging/translators/collection.py
+++ b/trustgraph-base/trustgraph/messaging/translators/collection.py
@@ -54,6 +54,7 @@ class CollectionManagementResponseTranslator(MessageTranslator):
     """Translator for CollectionManagementResponse schema objects"""
 
     def to_pulsar(self, data: Dict[str, Any]) -> CollectionManagementResponse:
+
         # Handle error
         error = None
         if "error" in data and data["error"]:
@@ -86,6 +87,8 @@ class CollectionManagementResponseTranslator(MessageTranslator):
     def from_pulsar(self, obj: CollectionManagementResponse) -> Dict[str, Any]:
         result = {}
 
+        print("COLLECTIONMGMT", obj, flush=True)
+
         if obj.error is not None:
             result["error"] = {
                 "type": obj.error.type,
@@ -106,4 +109,6 @@ class CollectionManagementResponseTranslator(MessageTranslator):
                     "updated_at": coll.updated_at
                 })
 
-        return result
\ No newline at end of file
+        print("RESULT IS", result, flush=True)
+
+        return result
diff --git a/trustgraph-flow/trustgraph/gateway/dispatch/collection_management.py b/trustgraph-flow/trustgraph/gateway/dispatch/collection_management.py
index 6e78db48..f2755ae8 100644
--- a/trustgraph-flow/trustgraph/gateway/dispatch/collection_management.py
+++ b/trustgraph-flow/trustgraph/gateway/dispatch/collection_management.py
@@ -22,7 +22,9 @@ class CollectionManagementRequestor(ServiceRequestor):
         self.response_translator = TranslatorRegistry.get_response_translator("collection-management")
 
     def to_request(self, body):
+        print("REQUEST", body, flush=True)
         return self.request_translator.to_pulsar(body)
 
     def from_response(self, message):
-        return self.response_translator.from_response_with_completion(message)
\ No newline at end of file
+        print("RESPONSE", message, flush=True)
+        return self.response_translator.from_response_with_completion(message)
diff --git a/trustgraph-flow/trustgraph/librarian/collection_manager.py b/trustgraph-flow/trustgraph/librarian/collection_manager.py
index 03868095..f830db28 100644
--- a/trustgraph-flow/trustgraph/librarian/collection_manager.py
+++ b/trustgraph-flow/trustgraph/librarian/collection_manager.py
@@ -116,6 +116,7 @@ class CollectionManager:
             ]
 
             return CollectionManagementResponse(
+                error=None,
                 collections=collection_metadata,
                 timestamp=datetime.now().isoformat()
             )
@@ -186,6 +187,7 @@ class CollectionManager:
                 )
 
             return CollectionManagementResponse(
+                error=None,
                 collections=[collection_metadata],
                 timestamp=datetime.now().isoformat()
             )
@@ -269,6 +271,7 @@ class CollectionManager:
             del self.pending_deletions[deletion_key]
 
             return CollectionManagementResponse(
+                error=None,
                 timestamp=datetime.now().isoformat()
             )