allow Exception
This commit is contained in:
@@ -2,16 +2,17 @@ from pydantic import BaseModel
|
||||
from enum import Enum
|
||||
from typing import Optional
|
||||
|
||||
FunctionStatus = Enum("FunctionStatus", ["SUCCESS", "FAILED"])
|
||||
|
||||
|
||||
|
||||
class FunctionResponse(BaseModel):
|
||||
name: str
|
||||
status: FunctionStatus
|
||||
success: bool
|
||||
message: str = ""
|
||||
args: Optional[dict] = None
|
||||
|
||||
def __init__(self, name: str, status: FunctionStatus, message: str = "", args: Optional[dict] = None):
|
||||
super().__init__(name=name, status=status, message=message, args=args)
|
||||
def __init__(self, name: str, success: bool, message: str = "", args: Optional[dict] = None):
|
||||
super().__init__(name=name, success=success, message=message, args=args)
|
||||
|
||||
def to_dict(self):
|
||||
return self.model_dump()
|
||||
|
||||
Reference in New Issue
Block a user