code_runner_client.py 960 B

12345678910111213141516171819202122232425
  1. import httpx
  2. from core_domain import CodeExecutionRequestContract, CodeExecutionResponseContract
  3. class CodeRunnerClientError(Exception):
  4. pass
  5. class CodeRunnerClient:
  6. def __init__(self, base_url: str, timeout_seconds: float = 60.0) -> None:
  7. self.base_url = base_url.rstrip("/")
  8. self.timeout_seconds = timeout_seconds
  9. def execute_code(
  10. self,
  11. payload: CodeExecutionRequestContract) -> CodeExecutionResponseContract:
  12. try:
  13. with httpx.Client(timeout=self.timeout_seconds) as client:
  14. response = client.post(
  15. f"{self.base_url}/code/execute",
  16. json=payload.model_dump(mode="json"))
  17. response.raise_for_status()
  18. return CodeExecutionResponseContract.model_validate(response.json())
  19. except httpx.HTTPError as exc:
  20. raise CodeRunnerClientError(f"code-runner-service request failed: {exc}") from exc