deploy-vllm2 / test_server.py
Damien Benveniste
modified
e89ed1a
raw
history blame contribute delete
982 Bytes
from http.server import HTTPServer, BaseHTTPRequestHandler
import json
class SimpleHandler(BaseHTTPRequestHandler):
def do_GET(self):
self.send_response(200)
self.send_header('Content-type', 'application/json')
self.end_headers()
response = json.dumps({"message": "Server is running"})
self.wfile.write(response.encode())
def do_POST(self):
content_length = int(self.headers['Content-Length'])
post_data = self.rfile.read(content_length)
self.send_response(200)
self.send_header('Content-type', 'application/json')
self.end_headers()
response = json.dumps({"message": "Received POST request", "data": post_data.decode()})
self.wfile.write(response.encode())
def run(port=8000):
server_address = ('', port)
httpd = HTTPServer(server_address, SimpleHandler)
print(f'Starting server on port {port}...')
httpd.serve_forever()
if __name__ == '__main__':
run()