From 7edc3f72fb1ebf029bd518a8aa84fd5ac2dc8ff0 Mon Sep 17 00:00:00 2001
From: Cyber MacGeddon <cybermaggedon@gmail.com>
Date: Wed, 4 Mar 2026 16:09:03 +0000
Subject: [PATCH] Minor fixes

---
 .../trustgraph/librarian/blob_store.py        |  4 +-
 .../trustgraph/librarian/librarian.py         | 52 ++-----------------
 2 files changed, 7 insertions(+), 49 deletions(-)

diff --git a/trustgraph-flow/trustgraph/librarian/blob_store.py b/trustgraph-flow/trustgraph/librarian/blob_store.py
index 05d21b3a..9b138c40 100644
--- a/trustgraph-flow/trustgraph/librarian/blob_store.py
+++ b/trustgraph-flow/trustgraph/librarian/blob_store.py
@@ -159,10 +159,10 @@ class BlobStore:
         etag = self.client._upload_part(
             bucket_name=self.bucket_name,
             object_name=object_name,
+            data=data,
+            headers={"Content-Length": str(len(data))},
             upload_id=upload_id,
             part_number=part_number,
-            data=io.BytesIO(data),
-            headers={"Content-Length": str(len(data))},
         )
 
         logger.debug(f"Uploaded part {part_number} for {object_id}, etag={etag}")
diff --git a/trustgraph-flow/trustgraph/librarian/librarian.py b/trustgraph-flow/trustgraph/librarian/librarian.py
index 831e015a..a0bea9e6 100644
--- a/trustgraph-flow/trustgraph/librarian/librarian.py
+++ b/trustgraph-flow/trustgraph/librarian/librarian.py
@@ -73,13 +73,7 @@ class Librarian:
 
         logger.debug("Add complete")
 
-        return LibrarianResponse(
-            error = None,
-            document_metadata = None,
-            content = None,
-            document_metadatas = None,
-            processing_metadatas = None,
-        )
+        return LibrarianResponse()
 
     async def remove_document(self, request):
 
@@ -122,13 +116,7 @@ class Librarian:
 
         logger.debug("Remove complete")
 
-        return LibrarianResponse(
-            error = None,
-            document_metadata = None,
-            content = None,
-            document_metadatas = None,
-            processing_metadatas = None,
-        )
+        return LibrarianResponse()
 
     async def update_document(self, request):
 
@@ -146,13 +134,7 @@ class Librarian:
 
         logger.debug("Update complete")
 
-        return LibrarianResponse(
-            error = None,
-            document_metadata = None,
-            content = None,
-            document_metadatas = None,
-            processing_metadatas = None,
-        )
+        return LibrarianResponse()
 
     async def get_document_metadata(self, request):
 
@@ -169,8 +151,6 @@ class Librarian:
             error = None,
             document_metadata = doc,
             content = None,
-            document_metadatas = None,
-            processing_metadatas = None,
         )
 
     async def get_document_content(self, request):
@@ -192,8 +172,6 @@ class Librarian:
             error = None,
             document_metadata = None,
             content = base64.b64encode(content),
-            document_metadatas = None,
-            processing_metadatas = None,
         )
 
     async def add_processing(self, request):
@@ -239,13 +217,7 @@ class Librarian:
 
         logger.debug("Add complete")
 
-        return LibrarianResponse(
-            error = None,
-            document_metadata = None,
-            content = None,
-            document_metadatas = None,
-            processing_metadatas = None,
-        )
+        return LibrarianResponse()
 
     async def remove_processing(self, request):
 
@@ -265,13 +237,7 @@ class Librarian:
 
         logger.debug("Remove complete")
 
-        return LibrarianResponse(
-            error = None,
-            document_metadata = None,
-            content = None,
-            document_metadatas = None,
-            processing_metadatas = None,
-        )
+        return LibrarianResponse()
 
     async def list_documents(self, request):
 
@@ -286,11 +252,7 @@ class Librarian:
             ]
 
         return LibrarianResponse(
-            error = None,
-            document_metadata = None,
-            content = None,
             document_metadatas = docs,
-            processing_metadatas = None,
         )
 
     async def list_processing(self, request):
@@ -298,10 +260,6 @@ class Librarian:
         procs = await self.table_store.list_processing(request.user)
 
         return LibrarianResponse(
-            error = None,
-            document_metadata = None,
-            content = None,
-            document_metadatas = None,
             processing_metadatas = procs,
         )