benchmark/embedding/benchmark.py [68:85]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    ):
        """Initialize the benchmark with the given parameters.

        Args:
            server_address: Address of the gRPC server to benchmark
            batch_sizes: List of batch sizes to test
            num_requests: Number of requests to send for each batch size
            concurrency: Maximum number of concurrent requests
            prompt_length: Maximum length of generated prompts
            model_name: Name of the model to use for encoding
            distribution: Distribution type for prompt lengths and batch sizes
        """
        self.server_address = server_address
        self.batch_sizes = batch_sizes
        self.num_requests = num_requests
        self.concurrency = concurrency
        self.prompt_length = prompt_length
        self.model_name = model_name
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



benchmark/embedding/benchmark_http.py [29:35]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    ):
        self.server_address = server_address
        self.batch_sizes = batch_sizes
        self.num_requests = num_requests
        self.concurrency = concurrency
        self.prompt_length = prompt_length
        self.model_name = model_name
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -