fix merge bugs.

This commit is contained in:
Liu Zhicong
2023-11-30 21:14:24 -08:00
parent 50e22817c1
commit 83149b5590
3 changed files with 31 additions and 4 deletions
+2 -1
View File
@@ -1,3 +1,4 @@
from .open_ai_node import * from .open_ai_node import *
from .openai_tts_node import * from .openai_tts_node import *
from .whisper_node import * from .whisper_node import *
from .dall_e_compute_node import *
@@ -9,9 +9,7 @@ import base64
from PIL import Image from PIL import Image
from .compute_task import ComputeTask, ComputeTaskResult, ComputeTaskState, ComputeTaskType, ComputeTaskResultCode from aios import ComputeTask, ComputeTaskResult, ComputeTaskState, ComputeTaskType, ComputeTaskResultCode,ComputeNode, AIStorage, UserConfig
from .compute_node import ComputeNode
from .storage import AIStorage, UserConfig
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
+28
View File
@@ -0,0 +1,28 @@
import sys
import os
import logging
import asyncio
import time
import unittest
dir_path = os.path.dirname(os.path.realpath(__file__))
sys.path.append("{}/../src/".format(dir_path))
from aios_kernel import WorkspaceEnvironment
async def test_workspace():
test_env = WorkspaceEnvironment("test")
test_env._add_document_dir(f"{dir_path}/../rootfs/test_doc")
test_env._start_scan_document()
catalogs = await test_env.get_knowledege_catalog()
print(catalogs)
asyncio.sleep(60*60)
if __name__ == "__main__":
asyncio.run(test_workspace())
print("OK!")
time.sleep(60*60)