34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
|
import uuid
|
||
|
|
from typing import Dict, Optional
|
||
|
|
|
||
|
|
from dataclasses_json import DataClassJsonMixin
|
||
|
|
from pydantic import Field
|
||
|
|
from pydantic.dataclasses import dataclass
|
||
|
|
|
||
|
|
from chainlit.context import context
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class Action(DataClassJsonMixin):
|
||
|
|
# Name of the action, this should be used in the action_callback
|
||
|
|
name: str
|
||
|
|
# The parameters to call this action with.
|
||
|
|
payload: Dict
|
||
|
|
# The label of the action. This is what the user will see.
|
||
|
|
label: str = ""
|
||
|
|
# The tooltip of the action button. This is what the user will see when they hover the action.
|
||
|
|
tooltip: str = ""
|
||
|
|
# The lucid icon name for this action.
|
||
|
|
icon: Optional[str] = None
|
||
|
|
# This should not be set manually, only used internally.
|
||
|
|
forId: Optional[str] = None
|
||
|
|
# The ID of the action
|
||
|
|
id: str = Field(default_factory=lambda: str(uuid.uuid4()))
|
||
|
|
|
||
|
|
async def send(self, for_id: str):
|
||
|
|
self.forId = for_id
|
||
|
|
await context.emitter.emit("action", self.to_dict())
|
||
|
|
|
||
|
|
async def remove(self):
|
||
|
|
await context.emitter.emit("remove_action", self.to_dict())
|