Reformat the error to match OpenAI's

This commit is contained in:
Brendan Dolan-Gavitt 2022-11-01 12:46:55 -04:00
commit b7b85461af
2 changed files with 33 additions and 4 deletions

View file

@ -0,0 +1,19 @@
from typing import *
class FauxPilotException(Exception):
def __init__(self, message: str, type: Optional[str] = None, param: Optional[str] = None, code: Optional[int] = None):
super().__init__(message)
self.message = message
self.type = type
self.param = param
self.code = code
def json(self):
return {
'error': {
'message': self.message,
'type': self.type,
'param': self.param,
'code': self.code
}
}