24 lines
572 B
Python
24 lines
572 B
Python
from pydantic import BaseModel
|
|
from enum import Enum
|
|
from typing import Optional
|
|
|
|
|
|
|
|
|
|
class FunctionResponse(BaseModel):
|
|
name: str
|
|
success: bool
|
|
message: str = ""
|
|
args: Optional[dict] = None
|
|
|
|
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()
|
|
|
|
def to_json(self):
|
|
return self.model_dump_json()
|
|
|
|
def to_string(self):
|
|
return self.model_dump_json() |