Paul assistant

backend/chainlit/__init__.py

@@ -24,6 +24,10 @@
 
 import chainlit.input_widget as input_widget
 from chainlit.action import Action
+
+# import Assistant
+from chainlit.assistant import Assistant
+from chainlit.assistant_settings import AssistantSettings
 from chainlit.cache import cache
 from chainlit.chat_context import chat_context
 from chainlit.chat_settings import ChatSettings
@@ -66,6 +70,27 @@
     logger.info("Loaded .env file")
 
 
+# assistant-related callbacks setters
+# ---------------------------------
+@trace
+def on_create_assistant(
+    func: Callable[[Optional[User], AssistantSettings], Any]
+) -> Callable[[Optional[User], AssistantSettings], Any]:
+    config.code.on_create_assistant = wrap_user_function(func)
+    return func
+
+
+@trace
+def on_list_assistants(
+    func: Callable[[Optional[User]], List[Assistant]]
+) -> Callable[[Optional[User]], List[Assistant]]:
+    config.code.on_list_assistants = wrap_user_function(func)
+    return func
+
+
+# ---------------------------------
+
+
 @trace
 def password_auth_callback(func: Callable[[str, str], Optional[User]]) -> Callable:
     """
@@ -402,6 +427,9 @@ def acall(self):
     "TaskStatus",
     "Video",
     "ChatSettings",
+    "AssistantSettings",
+    # assistant
+    "Assistant",
     "input_widget",
     "Message",
     "ErrorMessage",
@@ -421,6 +449,10 @@ def acall(self):
     "action_callback",
     "author_rename",
     "on_settings_update",
+    # assistant-related callbacks setters
+    "on_create_assistant",
+    "on_list_assistants",
+    # end of assistant-related callbacks setters
     "password_auth_callback",
     "header_auth_callback",
     "sleep",

backend/chainlit/assistant.py

@@ -0,0 +1,16 @@
+from typing import List, Dict
+from chainlit.input_widget import InputWidget
+
+class Assistant:
+    input_widgets: List[InputWidget] = []
+    settings_values: Dict = {}
+
+    def __init__(self, input_widgets: List[InputWidget], settings_values: Dict):
+        self.input_widgets = input_widgets
+        self.settings_values = settings_values
+
+    def to_dict(self):
+        return {
+            "input_widgets": [widget.__repr__() for widget in self.input_widgets],
+            "settings_values": self.settings_values
+        }

backend/chainlit/assistant_settings.py

@@ -0,0 +1,35 @@
+import logging
+from typing import List
+
+from chainlit.context import context
+from chainlit.input_widget import InputWidget
+from pydantic.dataclasses import Field, dataclass
+
+
+@dataclass
+class AssistantSettings:
+    """Useful to create chat settings that the user can change."""
+
+    inputs: List[InputWidget] = Field(default_factory=list, exclude=True)
+
+    def __init__(
+        self,
+        inputs: List[InputWidget],
+    ) -> None:
+        self.inputs = inputs
+
+    def settings(self):
+        return dict(
+            [(input_widget.id, input_widget.initial) for input_widget in self.inputs]
+        )
+
+    async def send(self):
+        settings = self.settings()
+        context.emitter.set_assistant_settings(settings)
+
+        inputs_content = [input_widget.to_dict() for input_widget in self.inputs]
+        # logging.info(f"Sending assistant settings: {inputs_content}")
+        await context.emitter.emit("assistant_settings", inputs_content)
+
+        # logging.info(f"Assistant settings sent: {settings}")
+        return settings

backend/chainlit/config.py

@@ -21,7 +21,7 @@
     from chainlit.types import AudioChunk, ChatProfile, Starter, ThreadDict
     from chainlit.user import User
     from fastapi import Request, Response
-
+    from chainlit.assistant import Assistant
 
 BACKEND_ROOT = os.path.dirname(__file__)
 PACKAGE_ROOT = os.path.dirname(os.path.dirname(BACKEND_ROOT))
@@ -291,6 +291,12 @@ class CodeSettings:
     )
     set_starters: Optional[Callable[[Optional["User"]], List["Starter"]]] = None
 
+    # assistant-related callback function
+    on_create_assistant: Optional[Callable[[Optional["User"], Any], Any]] = None
+    on_list_assistants: Optional[
+        Callable[[Optional["User"]], List["Assistant"]]
+    ] = None
+
 
 @dataclass()
 class ProjectSettings(DataClassJsonMixin):

backend/chainlit/emitter.py

@@ -114,6 +114,10 @@ async def set_chat_settings(self, settings: dict):
         """Stub method to set chat settings."""
         pass
 
+    async def set_assistant_settings(self, settings: dict):
+        """Stub method to send assistant settings to the UI."""
+        pass
+
     async def send_action_response(
         self, id: str, status: bool, response: Optional[str] = None
     ):
@@ -361,6 +365,9 @@ def send_token(self, id: str, token: str, is_sequence=False, is_input=False):
     def set_chat_settings(self, settings: Dict[str, Any]):
         self.session.chat_settings = settings
 
+    def set_assistant_settings(self, settings: Dict[str, Any]):
+        self.session.assistant_settings = settings
+
     def send_action_response(
         self, id: str, status: bool, response: Optional[str] = None
     ):

backend/chainlit/input_widget.py

@@ -161,6 +161,28 @@ def to_dict(self) -> Dict[str, Any]:
             "description": self.description,
         }
 
+@dataclass
+class FileUploadInput(InputWidget):
+    """Useful to create a file upload input."""
+
+    type: InputWidgetType = "fileupload"
+    initial: Optional[str] = None
+    placeholder: Optional[str] = None
+    accept: List[str] = Field(default_factory=lambda: [])
+    max_size_mb: Optional[int] = None
+    max_files: Optional[int] = None
+
+    def to_dict(self) -> Dict[str, Any]:
+        return {
+            "type": self.type,
+            "id": self.id,
+            "label": self.label,
+            "initial": self.initial,
+            "placeholder": self.placeholder,
+            "tooltip": self.tooltip,
+            "description": self.description,
+        }
+
 
 @dataclass
 class Tags(InputWidget):

backend/chainlit/server.py

@@ -5,6 +5,8 @@
 import shutil
 import urllib.parse
 from typing import Any, Optional, Union
+import uuid
+import os
 
 from chainlit.oauth_providers import get_oauth_provider
 from chainlit.secret import random_secret
@@ -194,11 +196,11 @@ def get_build_dir(local_target: str, packaged_target: str):
 
 router = APIRouter(prefix=ROOT_PATH)
 
-app.mount(
-    f"{ROOT_PATH}/public",
-    StaticFiles(directory="public", check_dir=False),
-    name="public",
-)
+# app.mount(
+#     f"{ROOT_PATH}/static",
+#     StaticFiles(directory="static", check_dir=False),
+#     name="static",
+# )
 
 app.mount(
     f"{ROOT_PATH}/assets",
@@ -218,7 +220,6 @@ def get_build_dir(local_target: str, packaged_target: str):
     name="copilot",
 )
 
-
 # -------------------------------------------------------------------------------
 #                               SLACK HANDLER
 # -------------------------------------------------------------------------------
@@ -921,7 +922,6 @@ async def get_avatar(avatar_id: str):
     avatar_id = avatar_id.strip().lower().replace(" ", "_")
 
     avatar_path = os.path.join(APP_ROOT, "public", "avatars", f"{avatar_id}.*")
-
     files = glob.glob(avatar_path)
 
     if files:
@@ -931,6 +931,34 @@ async def get_avatar(avatar_id: str):
     else:
         return await get_favicon()
 
+# post avatar/{avatar_id} (only for authenticated users)
+@router.post("/avatars/{avatar_id}")
+async def upload_avatar(
+    avatar_id: str,
+    file: UploadFile,
+    current_user: Annotated[
+        Union[None, User, PersistedUser], Depends(get_current_user)
+    ],
+):
+    # if not current_user:
+    #     raise HTTPException(status_code=401, detail="Not authenticated")
+
+    # Save the file to the avatars directory
+    try:
+        # file_extension = os.path.splitext(file.filename)[1]
+        # avatar_filename = f"{avatar_id}{file_extension}"
+        avatar_path = os.path.join(APP_ROOT, "public", "avatars", avatar_id)
+
+        # Ensure the avatars directory exists
+        os.makedirs(os.path.dirname(avatar_path), exist_ok=True)
+
+        with open(avatar_path, "wb") as f:
+            f.write(await file.read())
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+    return {"id": avatar_id}
+
 
 @router.head("/")
 def status_check():

backend/chainlit/session.py

@@ -16,6 +16,7 @@
 )
 
 import aiofiles
+from chainlit.assistant import Assistant
 from chainlit.logger import logger
 
 if TYPE_CHECKING:
@@ -72,8 +73,12 @@ def __init__(
         user_env: Optional[Dict[str, str]],
         # Chat profile selected before the session was created
         chat_profile: Optional[str] = None,
+        # Selected assistant
+        selected_assistant: Optional[Assistant] = None,
         # Origin of the request
         http_referer: Optional[str] = None,
+        # assistant settings
+        assistant_settings: Optional[Dict[str, Any]] = None,
     ):
         if thread_id:
             self.thread_id_to_resume = thread_id
@@ -90,7 +95,9 @@ def __init__(
 
         self.id = id
 
+        self.assistant_settings = assistant_settings
         self.chat_settings: Dict[str, Any] = {}
+        self.selected_assistant = selected_assistant
 
     @property
     def files_dir(self):
@@ -153,6 +160,7 @@ def to_persistable(self) -> Dict:
         user_session = user_sessions.get(self.id) or {}  # type: Dict
         user_session["chat_settings"] = self.chat_settings
         user_session["chat_profile"] = self.chat_profile
+        user_session["selected_assistant"] = self.selected_assistant
         user_session["http_referer"] = self.http_referer
         user_session["client_type"] = self.client_type
         metadata = clean_metadata(user_session)
@@ -176,6 +184,10 @@ def __init__(
         user_env: Optional[Dict[str, str]] = None,
         # Origin of the request
         http_referer: Optional[str] = None,
+        # assistant settings
+        assistant_settings: Optional[Dict[str, Any]] = None,
+        # selected assistant
+        selected_assistant: Optional[Assistant] = None,
     ):
         super().__init__(
             id=id,
@@ -185,6 +197,8 @@ def __init__(
             client_type=client_type,
             user_env=user_env,
             http_referer=http_referer,
+            assistant_settings=assistant_settings,
+            selected_assistant=selected_assistant,
         )
 
     def delete(self):
@@ -228,10 +242,14 @@ def __init__(
         token: Optional[str] = None,
         # Chat profile selected before the session was created
         chat_profile: Optional[str] = None,
+        # Selected assistant
+        selected_assistant: Optional[Assistant] = None,
         # Languages of the user's browser
         languages: Optional[str] = None,
         # Origin of the request
         http_referer: Optional[str] = None,
+        # chat settings
+        assistant_settings: Optional[Dict[str, Any]] = None,
     ):
         super().__init__(
             id=id,
@@ -241,7 +259,9 @@ def __init__(
             user_env=user_env,
             client_type=client_type,
             chat_profile=chat_profile,
+            selected_assistant=selected_assistant,
             http_referer=http_referer,
+            assistant_settings=assistant_settings,
         )
 
         self.socket_id = socket_id