Add interactive CLI client with /add, /search, /quit commands

This commit is contained in:
2026-05-12 11:56:41 +00:00
parent 71f1ac67da
commit b8a26b0829
+57
View File
@@ -0,0 +1,57 @@
#!/usr/bin/env python3
"""Interactive test CLI client for RAG agent with /add, /search, /quit commands."""
from rag_tools import search_knowledge_base, add_to_knowledge_base
PROMPT = "> "
def main():
print("=== RAG Knowledge Base CLI ===")
print("Commands: /add <title> | /search <query> | /quit")
print("-" * 40)
while True:
try:
user_input = input(PROMPT).strip()
except
SÑQ\œÜÙ^XØ\[\œ\
Nˆš[
žYHHŠBˆœXZˆYˆÝ\Ù\Ú[œ]ˆÛÛ[YBˆYˆ\Ù\Ú[œ] œÝ\ÝÚ]
ÜÙX\˜ÚŠNˆ]Y\žHH\Ù\Ú[œ]ÎÜÙX\˜ÚŽÒ .strip()
if not query:
print("Use: /search <your_query>")
continue
print("\nSearching...")
result = search_knowledge_base(query)
print(f"\n{result}\n")
elif user_input.startswith("/add "):
title_and_content = user_input[5:].strip()
if "|" in title_and_content:
parts = title_and_content.split("|", 1)
title = parts[0].strip()
content = parts[1].strip()
else:
title = title_and_content
content = input("Enter content: ").strip()
if not content:
print("Error: content cannot be empty.")
continue
result = add_to_knowledge_base(content, title)
print(f\n{result}\n")
elif user_input == "/quit":
print("Bye!")
break
else:
print(f"Unknown command: {user_input}")
print("Available: /add <title>|<content>, /search <query>, /quit")
if __name__ == "__main__":
main()