Function main

03-ai-agent-for-devops/code/07/src/main.py:44–101

Main interactive loop

()

def main():
    """Main interactive loop"""
    try:
        # Validate configuration
        Config.validate()
        
        # Print banner
        print_banner()
        
        # Initialize agent
        print("Initializing AI agent...")
        agent = LogAnalyzerAgent()
        print("Ready!\n")
        
        # Main loop
        while True:
            try:
                # Get user input
                user_input = input("You: ").strip()
                
                # Handle empty input
                if not user_input:
                    continue
                
                # Handle commands
                if user_input.lower() in ['quit', 'exit', 'q']:
                    print("\nGoodbye!")
                    break
                
                if user_input.lower() == 'clear':
                    agent.clear_history()
                    print("\nConversation history cleared.\n")
                    continue
                
                if user_input.lower() == 'help':
                    print_help()
                    continue
                
                # Process query
                response = agent.process_query(user_input)
                print(f"\nAgent: {response}\n")
            
            except KeyboardInterrupt:
                print("\n\nInterrupted. Type 'quit' to exit.\n")
                continue
            except EOFError:
                print("\n\nGoodbye!")
                break
    
    except ValueError as e:
        print(f"\nConfiguration Error: {e}")
        print("Please check your .env file and ensure GEMINI_API_KEY is set.")
        sys.exit(1)
    except Exception as e:
        print(f"\nUnexpected Error: {e}")
        import traceback
        traceback.print_exc()
        sys.exit(1)