wehub-resource-sync
b7f52be4c9
CI / Run CI (push) Has been cancelled
CI / check-backend (push) Has been cancelled
CI / check-frontend (push) Has been cancelled
CI / tests (push) Has been cancelled
CI / e2e-tests (push) Has been cancelled
Copilot Setup Steps / copilot-setup-steps (push) Has been cancelled
50 lines
1.5 KiB
Python
50 lines
1.5 KiB
Python
from typing import Any, List
|
|
|
|
from pydantic import Field
|
|
from pydantic.dataclasses import dataclass
|
|
|
|
from chainlit.context import context
|
|
from chainlit.input_widget import InputWidget, Tab
|
|
|
|
|
|
@dataclass
|
|
class ChatSettings:
|
|
"""Useful to create chat settings that the user can change."""
|
|
|
|
inputs: List[InputWidget] | List[Tab] = Field(default_factory=list, exclude=True)
|
|
|
|
def __init__(
|
|
self,
|
|
inputs: List[InputWidget] | List[Tab],
|
|
) -> None:
|
|
self.inputs = inputs
|
|
|
|
def settings(self):
|
|
def collect_settings(
|
|
values: dict[str, Any], inputs: List[InputWidget] | List[Tab]
|
|
) -> None:
|
|
for input in inputs:
|
|
if isinstance(input, Tab):
|
|
collect_settings(values, input.inputs)
|
|
else:
|
|
values[input.id] = input.initial
|
|
|
|
settings: dict[str, Any] = {}
|
|
collect_settings(settings, self.inputs)
|
|
return settings
|
|
|
|
def _inputs_as_dicts(self) -> List[dict[str, Any]]:
|
|
return [input_widget.to_dict() for input_widget in self.inputs]
|
|
|
|
async def refresh(self):
|
|
"""Push settings widgets to the UI without updating session chat_settings."""
|
|
settings = self.settings()
|
|
await context.emitter.emit("chat_settings", self._inputs_as_dicts())
|
|
return settings
|
|
|
|
async def send(self):
|
|
settings = self.settings()
|
|
context.emitter.set_chat_settings(settings)
|
|
await context.emitter.emit("chat_settings", self._inputs_as_dicts())
|
|
return settings
|