allow Exception

This commit is contained in:
CakeCN
2025-09-14 14:12:35 +08:00
parent d6c4f8cfc0
commit d5b16eec52
2 changed files with 40 additions and 39 deletions

View File

@@ -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()