test prompt from knowledge object

test/test_knowledge_base.py

@@ -18,39 +18,32 @@ root.addHandler(handler)
 
 
 from knowledge import ObjectID, HashValue, EmailObjectBuilder
-from aios_kernel import KnowledgeBase, AgentPrompt
+from aios_kernel import KnowledgeBase, AgentPrompt, OpenAI_ComputeNode, ComputeKernel
 import asyncio
 import unittest
 
-async def test_embedding_email():
-    data = HashValue.hash_data("1233".encode("utf-8"));
-    print(data.to_base58())
-    print(data.to_base36())
-    
-    data2 = HashValue.from_base58(data.to_base58())
-    self.assertEqual(data.to_base36(), data2.to_base36())
-    
-    data2 = HashValue.from_base36(data.to_base36())
-    self.assertEqual(data.to_base58(), data2.to_base58())
+async def test_embedding_email(test):
+    open_ai_node = OpenAI_ComputeNode()
+    open_ai_node.start()
+    ComputeKernel().add_compute_node(open_ai_node)
+      
+    email_folder = os.path.join(dir_path, "../rootfs/data/email/")
+    print("explore emails in folder ", email_folder)
+    for root, dirs, files in os.walk(email_folder):
+        for dir in dirs:    
+            email_object = EmailObjectBuilder({}, os.path.join(root, dir)).build()
+            await KnowledgeBase().insert_object(email_object)
 
-    email_folder = "F:\\system\\Downloads\\8081ffdb80925f5bff9c6ab9c4756c7d"
-    email_object = EmailObjectBuilder({}, email_folder).build()
-
-    await KnowledgeBase().do_embedding(email_object)
-
-
-async def test_query_email():
     msg_prompt = AgentPrompt()
     msg_prompt.messages = [{"role":"user","content":"abcdef"}]
        
-    KnowledgeBase().query(msg_prompt)
+    await KnowledgeBase().query_prompt(msg_prompt)
+
+   
 
 class TestKnowledgeBase(unittest.TestCase):
     def test_embedding(self):
-        asyncio.run(test_embedding_email())
-
-    def test_query(self):
-        asyncio.run(test_query_email())
+        asyncio.run(test_embedding_email(self))
         
         
 if __name__ == "__main__":