mirror of
https://github.com/lightbroker/llmsecops-research.git
synced 2026-03-16 15:26:32 +00:00
22 lines
742 B
Python
22 lines
742 B
Python
from src.text_generation.entrypoints.http_api_controller import HttpApiController
|
|
from src.text_generation.services.logging.file_logging_service import FileLoggingService
|
|
from wsgiref.simple_server import make_server
|
|
|
|
|
|
class RestApiServer:
|
|
def __init__(self):
|
|
self.logger = FileLoggingService(filename='text_generation.server.log')
|
|
|
|
def listen(self):
|
|
try:
|
|
port = 9999
|
|
controller = HttpApiController()
|
|
with make_server('', port, controller) as wsgi_srv:
|
|
print(f'listening on port {port}...')
|
|
wsgi_srv.serve_forever()
|
|
except Exception as e:
|
|
self.logger.debug(e)
|
|
|
|
if __name__ == '__main__':
|
|
srv = RestApiServer()
|
|
srv.listen() |