CalvinBackup/llmsecops-research
1
0
Fork 0
mirror of https://github.com/lightbroker/llmsecops-research.git synced 2026-02-12 14:42:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

error handling

Browse Source
This commit is contained in:
Adam Wilson
2025-04-24 11:39:04 -06:00
parent c3cf6a5b54
commit 3563053ac3
1 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
tests/api/controller.py
View File

@@ -26,10 +26,6 @@ class ApiController:
request_body = env['wsgi.input'].read(request_body_size)
request_json = json.loads(request_body.decode('utf-8'))
prompt = request_json.get('prompt')
# for now, just reading request and echoing back in response
# data = json.loads(prompt)
# response_body = json.dumps(data).encode('utf-8')
data = self.get_service_response(prompt)
response_body = json.dumps(data).encode('utf-8')
@@ -51,7 +47,13 @@ class ApiController:
try:
handler = self.routes.get((method,path), self.__http_200_ok)
return handler(env, start_response)
except json.JSONDecodeError:
start_response('400 Bad Request', self.response_headers)
return [json.dumps({'error': 'Invalid JSON'}).encode('utf-8')]
except json.JSONDecodeError as e:
response_body = e.msg.encode('utf-8')
response_headers = [('Content-Type', 'text/plain'), ('Content-Length', str(len(response_body)))]
start_response('400 Bad Request', response_headers)
return [response_body]
except Exception as e:
response_body = e.msg.encode('utf-8')
response_headers = [('Content-Type', 'text/plain'), ('Content-Length', str(len(response_body)))]
start_response('500 Internal Server Error', response_headers)
return [response_body]