mirror of
https://github.com/tiennm99/litellm.git
synced 2026-06-17 16:48:54 +00:00
Xiaohan Fu
07fe9e8604
* implement failopen option default to True * introduce a config to set the timeout limit (default to 30)
86 lines
2.6 KiB
Python
86 lines
2.6 KiB
Python
"""
|
|
Minimal mock GraySwan monitor server that intentionally responds slowly.
|
|
|
|
Usage:
|
|
python scripts/mock_grayswan_timeout_server.py --port 8787 --delay 35
|
|
|
|
Point GRAYSWAN_API_BASE at http://127.0.0.1:8787 so the guardrail hits this
|
|
endpoint and times out (the guardrail client has a 30s timeout).
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import json
|
|
import logging
|
|
import time
|
|
from http.server import BaseHTTPRequestHandler, HTTPServer
|
|
from typing import Optional
|
|
|
|
LOG = logging.getLogger("mock_grayswan")
|
|
logging.basicConfig(level=logging.INFO, format="%(asctime)s %(levelname)s %(message)s")
|
|
|
|
|
|
class SlowHandler(BaseHTTPRequestHandler):
|
|
delay_seconds: float = 35.0
|
|
|
|
def log_message(self, fmt: str, *args) -> None: # noqa: D401
|
|
"""Route handler logs through the logging module."""
|
|
LOG.info("%s - %s", self.address_string(), fmt % args)
|
|
|
|
def _read_body(self) -> Optional[bytes]:
|
|
content_length = self.headers.get("content-length")
|
|
if content_length is None:
|
|
return None
|
|
try:
|
|
length = int(content_length)
|
|
except ValueError:
|
|
return None
|
|
return self.rfile.read(length)
|
|
|
|
def do_POST(self) -> None: # noqa: N802
|
|
if self.path != "/cygnal/monitor":
|
|
self.send_error(404, "Not Found")
|
|
return
|
|
|
|
body = self._read_body()
|
|
LOG.info("Received POST %s body=%s", self.path, body)
|
|
|
|
LOG.info("Sleeping for %.1fs to trigger client timeout", self.delay_seconds)
|
|
time.sleep(self.delay_seconds)
|
|
|
|
response = {"status": "ok", "delayed": self.delay_seconds}
|
|
response_bytes = json.dumps(response).encode("utf-8")
|
|
|
|
self.send_response(200)
|
|
self.send_header("Content-Type", "application/json")
|
|
self.send_header("Content-Length", str(len(response_bytes)))
|
|
self.end_headers()
|
|
self.wfile.write(response_bytes)
|
|
|
|
|
|
def main() -> None:
|
|
parser = argparse.ArgumentParser(description="Mock GraySwan monitor server")
|
|
parser.add_argument("--port", type=int, default=8787, help="Port to listen on")
|
|
parser.add_argument(
|
|
"--delay",
|
|
type=float,
|
|
default=35.0,
|
|
help="Seconds to delay before responding (must exceed guardrail timeout)",
|
|
)
|
|
args = parser.parse_args()
|
|
|
|
SlowHandler.delay_seconds = args.delay
|
|
server = HTTPServer(("0.0.0.0", args.port), SlowHandler)
|
|
LOG.info("Starting mock server on port %d with delay %.1fs", args.port, args.delay)
|
|
try:
|
|
server.serve_forever()
|
|
except KeyboardInterrupt:
|
|
LOG.info("Shutting down mock server")
|
|
finally:
|
|
server.server_close()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|