From d18922df2ca953e5a0d151db2a4afeb917e53030 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=D0=A0=D0=B8=D0=BD=D0=B0=D1=80=20=D0=9C=D0=B8=D1=80=D0=B7?= =?UTF-8?q?=D0=B0=D0=B3=D0=B8=D1=82=D0=BE=D0=B2?= Date: Wed, 17 Jun 2026 13:10:25 +0000 Subject: [PATCH] =?UTF-8?q?=D0=9E=D0=B1=D0=BD=D0=BE=D0=B2=D0=BB=D0=B5?= =?UTF-8?q?=D0=BD=20vectorstore.py:=20remove=20main.py?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- main.py | 51 --------------------------------------------------- 1 file changed, 51 deletions(-) delete mode 100644 main.py diff --git a/main.py b/main.py deleted file mode 100644 index bb8c6e6..0000000 --- a/main.py +++ /dev/null @@ -1,51 +0,0 @@ -import argparse - -from dotenv import load_dotenv - -from agent import agent -from vectorstore import create_vectorstore, load_documents - - -def ingest_documents(directory: str) -> None: - vectorstore = create_vectorstore() - chunk_count = load_documents(directory, vectorstore) - print(f"Loaded {chunk_count} chunks from {directory} into ChromaDB.") - - -def run_chat() -> None: - print("RAG agent ready. Type 'exit' to quit.") - while True: - user_input = input("You: ").strip() - if user_input.lower() == "exit": - print("Goodbye!") - return - if not user_input: - continue - - result = agent.invoke( - {"messages": [{"role": "user", "content": user_input}]} - ) - final_message = result["messages"][-1] - print(f"Assistant: {final_message.content}") - - -def main() -> None: - load_dotenv() - - parser = argparse.ArgumentParser() - parser.add_argument( - "--ingest", - metavar="DIRECTORY", - help="Load .txt and .md files from DIRECTORY into ChromaDB before the demo.", - ) - args = parser.parse_args() - - if args.ingest: - ingest_documents(args.ingest) - return - - run_chat() - - -if __name__ == "__main__": - main()