diff --git a/cli.py b/cli.py new file mode 100644 index 0000000..9632ace --- /dev/null +++ b/cli.py @@ -0,0 +1,29 @@ +import sys +from rag_tools import add_to_knowledge_base, search_knowledge_base + +def run_cli(): + print("RAG CLI. Commands: /add <content>, /search <query>, /quit") + while True: + try: + line = input(">>> ") + except EOFError: + break + if not line: + continue + if line.startswith("/quit"): + break + if line.startswith("/add"): + parts = line.split(maxsplit=2) + if len(parts) < 3: + print("Usage: /add <title> <content>") + continue + title, content = parts[1], parts[2] + print(add_to_knowledge_base(content, title)) + elif line.startswith("/search"): + query = line[len("/search"):].strip() + print(search_knowledge_base(query)) + else: + print("Unknown command") + +if __name__ == "__main__": + run_cli()