diff --git a/src/aios_kernel/knowledge_pipeline.py b/src/aios_kernel/knowledge_pipeline.py index 9b6b111..759793a 100644 --- a/src/aios_kernel/knowledge_pipeline.py +++ b/src/aios_kernel/knowledge_pipeline.py @@ -267,6 +267,7 @@ class KnowledgeEmailSource: class KnowledgeDirSource: def __init__(self, config): self.config = config + config["path"] = os.path.abspath(config["path"]) self.config["type"] = "dir" @classmethod @@ -342,6 +343,11 @@ class KnowledgePipline: self.add_email_source(KnowledgeEmailSource(source_config)) if source_config['type'] == 'dir': self.add_dir_source(KnowledgeDirSource(source_config)) + user_data_dir = AIStorage.get_instance().get_myai_dir() + default_dir = os.path.abspath(f"{user_data_dir}/data") + if not os.path.exists(default_dir): + os.makedirs(default_dir) + self.add_dir_source(KnowledgeDirSource({"path": default_dir})) return True diff --git a/src/service/aios_shell/aios_shell.py b/src/service/aios_shell/aios_shell.py index a3aef85..2f64d1c 100644 --- a/src/service/aios_shell/aios_shell.py +++ b/src/service/aios_shell/aios_shell.py @@ -319,7 +319,7 @@ class AIOS_Shell: async def handle_knowledge_commands(self, args): show_text = FormattedText([("class:title", "sub command not support!\n" - "/knowledge add email | dir\n" + "/knowledge add dir\n" "/knowledge journal [$topn]\n")]) if len(args) < 1: return show_text @@ -585,7 +585,7 @@ def print_welcome_screen(): \033[1;94m\tGive your Agent a Telegram account :\033[0m /connect $agent_name \033[1;94m\tAdd personal files to the AI Knowledge Base. \033[0m \t\t1) Copy your file to ~/myai/data -\t\t2) /knowlege add dir +\t\t2) /knowledge add dir \033[1;94m\tSearch your knowledge base :\033[0m /open Mia \033[1;94m\tCheck the progress of AI reading personal data :\033[0m /knowledge journal \033[1;94m\tOpen AI Bash (For Developer Only):\033[0m /open ai_bash @@ -665,7 +665,7 @@ async def main(): '/history $num $offset', '/connect $target', '/contact $name', - '/knowledge add email | dir', + '/knowledge add dir', '/knowledge journal [$topn]', '/set_config $key', '/enable $feature',