diff --git a/main.py b/main.py index c550b80..78d317c 100644 --- a/main.py +++ b/main.py @@ -1,8 +1,19 @@ -""" -Entry point for the RAG agent. +"""Entry point for the RAG agent. + +This script simply imports and runs the chat loop defined in ``agent.py``. """ -from agent import main +from agent import agent_executor if __name__ == "__main__": - main() + print("RAG Agent ready. Type 'exit' to quit.") + while True: + user_input = input("Запрос: ") + if user_input.lower() in {"exit", "quit", "q"}: + print("Bye!") + break + try: + result = agent_executor.invoke({"input": user_input}) + print(result["output"]) + except Exception as e: + print(f"Error: {e}")