Add main.py

This commit is contained in:
2026-06-01 17:40:35 +00:00
parent c8e28697ff
commit 3d0e1c3713
+26
View File
@@ -0,0 +1,26 @@
"""
Simple CLI to chat with the RAG agent.
"""
import os
import sys
from agent import agent_executor
def main():
print("Welcome to the RAG agent. Type 'exit' to quit.")
while True:
try:
user_input = input("\nQuery: ")
except (KeyboardInterrupt, EOFError):
print("\nExiting.")
break
if user_input.strip().lower() in {"exit", "quit", "q"}:
print("Goodbye!")
break
# Run agent
response = agent_executor.run(user_input)
print("\nAnswer:\n", response)
if __name__ == "__main__":
main()