Refactor custom.py for better error handling
- Modifies the `CustomComponent` class to handle errors using `HTTPException` from FastAPI - Removes the unused `HTTPExceptionWithTraceback` class - Updates the error response format with `err.msg` and `traceback.format_exc()`
This commit is contained in:
parent
a21582589c
commit
ae5bc1e203
1 changed files with 2 additions and 13 deletions
|
|
@ -12,12 +12,6 @@ from langchain.agents.tools import Tool
|
||||||
from fastapi import HTTPException
|
from fastapi import HTTPException
|
||||||
|
|
||||||
|
|
||||||
class HTTPExceptionWithTraceback(HTTPException):
|
|
||||||
def __init__(self, status_code, detail=None, traceback=None):
|
|
||||||
super().__init__(status_code, detail)
|
|
||||||
self.traceback = traceback
|
|
||||||
|
|
||||||
|
|
||||||
class Function(BaseModel):
|
class Function(BaseModel):
|
||||||
code: str
|
code: str
|
||||||
function: Optional[Callable] = None
|
function: Optional[Callable] = None
|
||||||
|
|
@ -170,13 +164,8 @@ class CustomComponent(BaseModel):
|
||||||
detail={
|
detail={
|
||||||
'error': err.msg,
|
'error': err.msg,
|
||||||
'traceback': traceback.format_exc()
|
'traceback': traceback.format_exc()
|
||||||
}
|
},
|
||||||
)
|
) from err
|
||||||
# raise HTTPExceptionWithTraceback(
|
|
||||||
# status_code=400,
|
|
||||||
# detail=err.msg,
|
|
||||||
# traceback=traceback.format_exc()
|
|
||||||
# ) from err
|
|
||||||
|
|
||||||
for node in module.body:
|
for node in module.body:
|
||||||
if isinstance(node, (ast.Import, ast.ImportFrom)):
|
if isinstance(node, (ast.Import, ast.ImportFrom)):
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue