Delete obsolete cli.py

2026-06-05 13:07:12 +00:00
parent abfa2ae700
commit 8e9477e19f
1 changed files with 0 additions and 90 deletions
@@ -1,90 +0,0 @@
 """Interactive CLI for the RAG agent.
 The CLI supports the following commands:
 * ``/add <dir>`` – Load all .txt/.md files from *dir* into Qdrant.
 * ``/search <query>`` – Run the agent on *query* and print the answer.
 * ``/quit`` – Exit the program.
 The vector store is created on first use and persisted in
 ``./qdrant_db``.
 """
 import os
 from pathlib import Path
 # Load environment variables (e.g., TAVILY_API_KEY)
 from dotenv import load_dotenv
 load_dotenv()
 # Local modules
 from .vector_store import create_vectorstore, load_documents
 from .agent import create_agent
 from . import tools
 # ---------------------------------------------------------------------------
 # Global state
 # ---------------------------------------------------------------------------
 VECTORSTORE = None
 AGENT = None
 # ---------------------------------------------------------------------------
 # Helper functions
 # ---------------------------------------------------------------------------
 def init_vectorstore() -> None:
    global VECTORSTORE
    if VECTORSTORE is None:
        VECTORSTORE = create_vectorstore()
        # Expose the store to the tools module so that the decorated
        # functions can access it via the global name ``vectorstore``.
        tools.vectorstore = VECTORSTORE
 def init_agent() -> None:
    global AGENT
    if AGENT is None:
        AGENT = create_agent()
 def add_documents(directory: str) -> None:
    init_vectorstore()
    load_documents(directory, VECTORSTORE)
 def run_query(query: str) -> None:
    init_agent()
    # The agent expects a dictionary with the key ``input``.
    result = AGENT.invoke({"input": query})
    print("\nAnswer:\n", result)
 # ---------------------------------------------------------------------------
 # CLI loop
 # ---------------------------------------------------------------------------
 def main() -> None:
    print("RAG Agent CLI.  Type '/quit' to exit.")
    while True:
        try:
            line = input("> ")
        except (EOFError, KeyboardInterrupt):
            print("\nBye!")
            break
        if not line:
            continue
        if line.strip() == "/quit":
            print("Bye!")
            break
        if line.startswith("/add "):
            _, dir_path = line.split(" ", 1)
            dir_path = dir_path.strip()
            if not os.path.isdir(dir_path):
                print(f"{dir_path} is not a directory.")
                continue
            add_documents(dir_path)
            print("Documents added.")
        elif line.startswith("/search "):
            _, query = line.split(" ", 1)
            run_query(query.strip())
        else:
            print("Unknown command. Use /add, /search, or /quit.")
 if __name__ == "__main__":
    main()