import sys
from rag_tools import add_to_knowledge_base, search_knowledge_base

def run_cli():
    print("RAG CLI. Commands: /add <title> <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()