apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-24 12:41:02 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Fixing collection management translation

Browse source
This commit is contained in:
Cyber MacGeddon 2025-09-19 14:32:21 +01:00
parent 0a3711d660
commit 34c2ae5ca6
3 changed files with 12 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
trustgraph-base/trustgraph/messaging/translators/collection.py
View file

@ -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
print("RESULT IS", result, flush=True)
return result

4
trustgraph-flow/trustgraph/gateway/dispatch/collection_management.py
View file

@ -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)
print("RESPONSE", message, flush=True)
return self.response_translator.from_response_with_completion(message)

3
trustgraph-flow/trustgraph/librarian/collection_manager.py
View file

@ -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()
)