mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-07-22 11:41:02 +02:00
Get operations
This commit is contained in:
parent
fc4654d54d
commit
4efb23e5da
5 changed files with 135 additions and 3 deletions
|
|
@ -75,7 +75,7 @@ class LibrarianRequestor(ServiceRequestor):
|
|||
)
|
||||
|
||||
if message.content:
|
||||
response["content"] = message.content
|
||||
response["content"] = message.content.decode("utf-8")
|
||||
|
||||
if message.document_metadatas != None:
|
||||
response["document-metadatas"] = [
|
||||
|
|
|
|||
|
|
@ -60,3 +60,14 @@ class BlobStore:
|
|||
|
||||
print("Remove blob complete", flush=True)
|
||||
|
||||
|
||||
async def get(self, object_id):
|
||||
|
||||
# FIXME: Loop retry
|
||||
resp = self.minio.get_object(
|
||||
bucket_name = self.bucket_name,
|
||||
object_name = "doc/" + str(object_id),
|
||||
)
|
||||
|
||||
return resp.read()
|
||||
|
||||
|
|
|
|||
|
|
@ -134,10 +134,48 @@ class Librarian:
|
|||
)
|
||||
|
||||
async def get_document_metadata(self, request):
|
||||
raise RuntimeError("Not implemented")
|
||||
|
||||
print("GET DOC...")
|
||||
|
||||
doc = await self.table_store.get_document(
|
||||
request.user,
|
||||
request.document_id
|
||||
)
|
||||
|
||||
print("Get complete", flush=True)
|
||||
|
||||
return LibrarianResponse(
|
||||
error = None,
|
||||
document_metadata = doc,
|
||||
content = None,
|
||||
document_metadatas = None,
|
||||
processing_metadatas = None,
|
||||
)
|
||||
|
||||
async def get_document_content(self, request):
|
||||
raise RuntimeError("Not implemented")
|
||||
|
||||
print("GET DOC CONTENT...")
|
||||
|
||||
object_id = await self.table_store.get_document_object_id(
|
||||
request.user,
|
||||
request.document_id
|
||||
)
|
||||
|
||||
print("Now", object_id)
|
||||
|
||||
content = await self.blob_store.get(
|
||||
object_id
|
||||
)
|
||||
|
||||
print("Get complete", flush=True)
|
||||
|
||||
return LibrarianResponse(
|
||||
error = None,
|
||||
document_metadata = None,
|
||||
content = base64.b64encode(content),
|
||||
document_metadatas = None,
|
||||
processing_metadatas = None,
|
||||
)
|
||||
|
||||
async def add_processing(self, request):
|
||||
raise RuntimeError("Not implemented")
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue