26 lines
604 B
Python
26 lines
604 B
Python
"""
|
|
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() |