change /knowledge commands in shell

src/aios_kernel/__init__.py

@@ -23,5 +23,6 @@ from .local_stability_node import Local_Stability_ComputeNode
 from .stability_node import Stability_ComputeNode
 from .local_st_compute_node import LocalSentenceTransformer_Text_ComputeNode,LocalSentenceTransformer_Image_ComputeNode
 from .compute_node_config import ComputeNodeConfig
+from .ai_function import SimpleAIFunction
 AIOS_Version = "0.5.1, build 2023-9-28"
 

src/aios_kernel/agent.py

@@ -122,7 +122,8 @@ class AIAgent:
             self.contact_prompt_str = config["contact_prompt"]
 
         if config.get("owner_env") is not None:
-            self.owner_env = Environment.get_env_by_id(config["owner_env"])
+            self.owner_env = config.get("owner_env")
+       
 
         if config.get("powerby") is not None:
             self.powerby = config["powerby"]