def main()

in ember/benchmark.py [0:0]

• 22 lines of code
• 1 McCabe index (conditional complexity)

def main():
    parser = argparse.ArgumentParser(description='Embedding Service Benchmark')
    parser.add_argument('--requests', type=int, default=50,
                       help='Total number of requests to make')
    parser.add_argument('--batch-size', type=int, default=64,
                       help='Maximum number of documents per request')
    parser.add_argument('--concurrent', type=int, default=5,
                       help='Number of concurrent requests')
    parser.add_argument('--warm-up', type=int, default=2,
                       help='Number of warm-up iterations')
    args = parser.parse_args()

    config = BenchmarkConfig(
        total_requests=args.requests,
        max_batch_size=args.batch_size,
        concurrent_requests=args.concurrent,
        warm_up_iterations=args.warm_up
    )

    benchmark = EmbeddingBenchmark(config)
    
    # Perform warm-up
    benchmark.warm_up()
    
    # Run batch benchmark
    results = benchmark.run_batch_benchmark()
    analysis = benchmark.analyze_results(results)
    
    # Print results
    benchmark.print_results(analysis)