in ember/benchmark.py [0:0]
def send_request(self) -> Dict:
"""Send a single request with batch_size messages."""
messages = self.generate_batch_messages()
payload = {
"model": self.model,
"documents": [{"content": message} for message in messages]
}
start_time = time.time()
try:
response = requests.post(self.url, json=payload, timeout=30)
response.raise_for_status()
duration = time.time() - start_time
return {
"success": True,
"duration": duration,
"status_code": response.status_code,
"num_messages": len(messages)
}
except requests.exceptions.RequestException as e:
return {
"success": False,
"duration": time.time() - start_time,
"error": str(e),
"num_messages": len(messages)
}