Improve chunk relate code and add read test case

This commit is contained in:
liyaxing
2023-09-26 02:30:59 +08:00
committed by tsukasa
parent d25f2956f9
commit 8b862c0075
6 changed files with 55 additions and 16 deletions
+17
View File
@@ -24,6 +24,7 @@ from knowledge import (
ObjectRelationStore,
KnowledgeStore,
EmailObject,
ImageObject,
)
import asyncio
import unittest
@@ -57,6 +58,22 @@ class TestVectorSTorage(unittest.TestCase):
ret2 = obj.encode()
self.assertEqual(ret, ret2)
images = email_object.get_rich_text().get_images()
image_keys = list(images.keys())
print("got image list: ", image_keys)
image_id = images[image_keys[1]]
print(f"got image object: {image_keys[1]} {image_id.to_base58()}")
buf = KnowledgeStore().get_object_store().get_object(image_id)
image_obj= ImageObject.decode(buf)
file_size = image_obj.get_file_size()
print(f"got image object: {image_id.to_base58()}, size: {file_size}")
image_data = KnowledgeStore().get_chunk_reader().read_chunk_list_to_single_bytes(image_obj.get_chunk_list())
self.assertEqual(file_size, len(image_data))
def test_relation(self):
obj1 = ObjectID.hash_data("12345".encode("utf-8"))