appkit-assistant 0.8.0__py3-none-any.whl → 0.10.0__py3-none-any.whl

appkit_assistant/components/system_prompt_editor.py

@@ -0,0 +1,78 @@
+import reflex as rx
+
+import appkit_mantine as mn
+from appkit_assistant.state.system_prompt_state import SystemPromptState
+from appkit_ui.components.dialogs import delete_dialog
+
+
+def system_prompt_editor() -> rx.Component:
+    """Admin-UI für das System Prompt Versioning mit appkit-mantine & appkit-ui.
+
+    Uses a hybrid approach for the textarea:
+    - default_value + on_change prevents cursor jumping during typing
+    - key prop forces re-render when selecting a different version
+    - This gives us both smooth editing AND the ability to update from select
+    """
+    return rx.vstack(
+        mn.markdown_preview(
+            source=(
+                """
+Der System-Prompt legt fest, wie sich der Assistent verhält. Bitte stellen Sie sicher,
+dass der Platzhalter `{mcp_prompts}` immer im Text enthalten ist. Dieser Platzhalter ist
+notwendig, damit das System im Hintergrund die benötigten Funktionen und Werkzeuge
+automatisch einfügen kann."""
+            ),
+            width="100%",
+        ),
+        mn.textarea(
+            placeholder="System-Prompt hier eingeben (max. 10.000 Zeichen)...",
+            description=f"{SystemPromptState.char_count} / 10.000 Zeichen",
+            default_value=SystemPromptState.current_prompt,
+            on_change=SystemPromptState.set_current_prompt,
+            error=SystemPromptState.error_message,
+            rows=21,
+            variant="filled",
+            width="100%",
+            key=SystemPromptState.textarea_key,
+        ),
+        rx.hstack(
+            mn.select(
+                placeholder="Aktuell",
+                data=SystemPromptState.versions,
+                value=SystemPromptState.selected_version_str,
+                on_change=SystemPromptState.set_selected_version,
+                clearable=False,
+                searchable=False,
+                width="280px",
+            ),
+            delete_dialog(
+                title="Version endgültig löschen?",
+                content="die ausgewählte Version",
+                on_click=SystemPromptState.delete_version,
+                icon_button=True,
+                class_name="dialog",
+                variant="outline",
+                color_scheme="red",
+                disabled=SystemPromptState.is_loading
+                | (SystemPromptState.selected_version_id == 0),
+            ),
+            rx.spacer(),
+            rx.button(
+                "Neue Version speichern",
+                on_click=SystemPromptState.save_current,
+                disabled=SystemPromptState.is_loading
+                | (
+                    SystemPromptState.current_prompt
+                    == SystemPromptState.last_saved_prompt
+                ),
+                loading=SystemPromptState.is_loading,
+            ),
+            align="center",
+            width="100%",
+            spacing="4",
+        ),
+        width="100%",
+        max_width="960px",
+        padding="6",
+        spacing="5",
+    )

appkit_assistant/components/thread.py

@@ -4,15 +4,11 @@ from collections.abc import Callable
 import reflex as rx
 
 import appkit_mantine as mn
+from appkit_assistant.backend.models import Message, MessageType
 from appkit_assistant.components import composer
 from appkit_assistant.components.message import MessageComponent
 from appkit_assistant.components.threadlist import ThreadList
-from appkit_assistant.state.thread_state import (
-    Message,
-    MessageType,
-    ThreadListState,
-    ThreadState,
-)
+from appkit_assistant.state.thread_state import ThreadState
 
 logger = logging.getLogger(__name__)
 
@@ -55,7 +51,7 @@ class Assistant:
                         lambda suggestion: Assistant.suggestion(
                             prompt=suggestion.prompt,
                             icon=suggestion.icon,
-                            update_prompt=ThreadState.update_prompt,
+                            update_prompt=ThreadState.set_prompt,
                         ),
                     ),
                     spacing="4",
@@ -122,7 +118,7 @@ class Assistant:
                 ),
                 rx.hstack(
                     composer.tools(
-                        show=with_tools and ThreadState.current_model_supports_tools
+                        show=with_tools and ThreadState.selected_model_supports_tools
                     ),
                     composer.add_attachment(show=with_attachments),
                     composer.clear(show=with_clear),
@@ -156,9 +152,6 @@ class Assistant:
         # if suggestions is not None:
         #     ThreadState.set_suggestions(suggestions)
 
-        if with_thread_list:
-            ThreadState.with_thread_list = with_thread_list
-
         return rx.flex(
             rx.cond(
                 ThreadState.messages,
@@ -207,7 +200,10 @@ class Assistant:
                 spacing="0",
                 flex_shrink=0,
                 z_index=1000,
-                on_mount=ThreadState.load_available_mcp_servers,
+                on_mount=[
+                    ThreadState.set_with_thread_list(with_thread_list),
+                    ThreadState.load_mcp_servers,
+                ],
             ),
             **props,
         )
@@ -216,12 +212,8 @@ class Assistant:
     def thread_list(
         *items,
         with_footer: bool = False,
-        default_model: str | None = None,
         **props,
     ) -> rx.Component:
-        if default_model:
-            ThreadListState.default_model = default_model
-
         return rx.flex(
             rx.flex(
                 ThreadList.header(

appkit_assistant/components/threadlist.py

@@ -1,6 +1,8 @@
 import reflex as rx
 
-from appkit_assistant.state.thread_state import ThreadListState, ThreadModel
+from appkit_assistant.backend.models import ThreadModel
+from appkit_assistant.state.thread_list_state import ThreadListState
+from appkit_assistant.state.thread_state import ThreadState
 
 
 class ThreadList:
@@ -13,7 +15,7 @@ class ThreadList:
                     rx.text(title),
                     size="2",
                     margin_right="28px",
-                    on_click=ThreadListState.create_thread(),
+                    on_click=ThreadState.new_thread(),
                     width="95%",
                 ),
                 content="Neuen Chat starten",
@@ -46,24 +48,28 @@ class ThreadList:
                 min_width="0",
                 title=thread.title,
             ),
-            rx.tooltip(
-                rx.button(
-                    rx.icon(
-                        "trash",
-                        size=13,
-                        stroke_width=1.5,
+            rx.cond(
+                ThreadListState.loading_thread_id == thread.thread_id,
+                rx.spinner(size="1", margin_left="6px", margin_right="6px"),
+                rx.tooltip(
+                    rx.button(
+                        rx.icon(
+                            "trash",
+                            size=13,
+                            stroke_width=1.5,
+                        ),
+                        variant="ghost",
+                        size="1",
+                        margin_left="0px",
+                        margin_right="0px",
+                        color_scheme="gray",
+                        on_click=ThreadListState.delete_thread(thread.thread_id),
                     ),
-                    variant="ghost",
-                    size="1",
-                    margin_left="0px",
-                    margin_right="0px",
-                    color_scheme="gray",
-                    on_click=ThreadListState.delete_thread(thread.thread_id),
+                    content="Chat löschen",
+                    flex_shrink=0,
                 ),
-                content="Chat löschen",
-                flex_shrink=0,
             ),
-            on_click=ThreadListState.select_thread(thread.thread_id),
+            on_click=ThreadState.load_thread(thread.thread_id),
             flex_direction=["row"],
             margin_right="10px",
             margin_bottom="8px",
@@ -113,18 +119,25 @@ class ThreadList:
                     ThreadListState.threads,
                     ThreadList.thread_list_item,
                 ),
-                rx.text(
-                    "Keine Chats vorhanden.",
-                    size="2",
-                    white_space="nowrap",
-                    overflow="hidden",
-                    text_overflow="ellipsis",
-                    flex_grow="1",
-                    min_width="0",
-                    margin_right="10px",
-                    margin_bottom="8px",
-                    padding="6px",
-                    align="center",
+                rx.cond(
+                    ThreadListState.loading,
+                    rx.vstack(
+                        rx.skeleton(height="34px", width="210px"),
+                        rx.skeleton(height="34px", width="210px"),
+                        rx.skeleton(height="34px", width="210px"),
+                        spacing="2",
+                    ),
+                    rx.text(
+                        "Noch keine Chats vorhanden. "
+                        "Klicke auf 'Neuer Chat', um zu beginnen.",
+                        size="2",
+                        flex_grow="1",
+                        min_width="0",
+                        margin_right="10px",
+                        margin_bottom="8px",
+                        padding="6px",
+                        align="center",
+                    ),
                 ),
             ),
             scrollbars="vertical",

appkit_assistant/components/tools_modal.py

@@ -92,6 +92,6 @@ def tools_popover() -> rx.Component:
             side="top",
         ),
         open=ThreadState.show_tools_modal,
-        on_open_change=ThreadState.set_show_tools_modal,
+        on_open_change=ThreadState.toogle_tools_modal,
         placement="bottom-start",
     )

appkit_assistant/configuration.py

@@ -8,3 +8,4 @@ class AssistantConfig(BaseConfig):
     openai_base_url: str | None = None
     openai_api_key: SecretStr | None = None
     google_api_key: SecretStr | None = None
+    azure_ai_projects_endpoint: str | None = None

appkit_assistant/state/system_prompt_state.py

@@ -0,0 +1,179 @@
+import logging
+from collections.abc import AsyncGenerator
+from typing import Any, Final
+
+import reflex as rx
+from reflex.state import State
+
+from appkit_assistant.backend.repositories import SystemPromptRepository
+from appkit_assistant.backend.system_prompt_cache import invalidate_prompt_cache
+from appkit_user.authentication.states import UserSession
+
+logger = logging.getLogger(__name__)
+
+MAX_PROMPT_LENGTH: Final[int] = 10000
+
+
+class SystemPromptState(State):
+    """State für System Prompt Editing und Versionierung."""
+
+    current_prompt: str = ""
+    last_saved_prompt: str = ""
+    versions: list[dict[str, str | int]] = []
+    prompt_map: dict[str, str] = {}
+    selected_version_id: int = 0
+    is_loading: bool = False
+    error_message: str = ""
+    char_count: int = 0
+    textarea_key: int = 0
+
+    async def load_versions(self) -> None:
+        """Alle System Prompt Versionen laden."""
+        self.is_loading = True
+        self.error_message = ""
+        try:
+            prompts = await SystemPromptRepository.get_all()
+            self.versions = [
+                {
+                    "value": str(p.version),
+                    "label": (
+                        f"Version {p.version} - "
+                        f"{p.created_at.strftime('%d.%m.%Y %H:%M')}"
+                    ),
+                }
+                for p in prompts
+            ]
+
+            self.prompt_map = {str(p.version): p.prompt for p in prompts}
+
+            if prompts:
+                latest = prompts[0]
+                self.selected_version_id = latest.version
+
+                if not self.current_prompt:
+                    self.current_prompt = latest.prompt
+                    self.last_saved_prompt = latest.prompt
+                else:
+                    self.last_saved_prompt = latest.prompt
+            else:
+                self.selected_version_id = 0
+                if not self.current_prompt:
+                    self.current_prompt = ""
+                self.last_saved_prompt = self.current_prompt
+
+            self.char_count = len(self.current_prompt)
+            # Force textarea to re-render with loaded content
+            self.textarea_key += 1
+
+            logger.info("Loaded %s system prompt versions", len(self.versions))
+        except Exception as exc:
+            self.error_message = f"Fehler beim Laden: {exc!s}"
+            logger.exception("Failed to load system prompt versions")
+        finally:
+            self.is_loading = False
+
+    async def save_current(self) -> AsyncGenerator[Any, Any]:
+        if self.current_prompt == self.last_saved_prompt:
+            yield rx.toast.info("Es wurden keine Änderungen erkannt.")
+            return
+
+        if not self.current_prompt.strip():
+            self.error_message = "Prompt darf nicht leer sein."
+            yield rx.toast.error("Prompt darf nicht leer sein.")
+            return
+
+        if len(self.current_prompt) > MAX_PROMPT_LENGTH:
+            self.error_message = "Prompt darf maximal 20.000 Zeichen enthalten."
+            yield rx.toast.error("Prompt ist zu lang (max. 20.000 Zeichen).")
+            return
+
+        self.is_loading = True
+        self.error_message = ""
+        try:
+            user_session: UserSession = await self.get_state(UserSession)
+            user_id = user_session.user_id
+
+            await SystemPromptRepository.create(
+                prompt=self.current_prompt,
+                user_id=user_id,
+            )
+
+            self.last_saved_prompt = self.current_prompt
+
+            # Invalidate cache to force reload of new prompt version
+            await invalidate_prompt_cache()
+            logger.info("System prompt cache invalidated after save")
+
+            await self.load_versions()
+
+            yield rx.toast.success("Neue Version erfolgreich gespeichert.")
+            logger.info("Saved new system prompt version by user %s", user_id)
+        except Exception as exc:
+            self.error_message = f"Fehler beim Speichern: {exc!s}"
+            logger.exception("Failed to save system prompt")
+            yield rx.toast.error(f"Fehler: {exc!s}")
+        finally:
+            self.is_loading = False
+
+    async def delete_version(self) -> AsyncGenerator[Any, Any]:
+        if not self.selected_version_id:
+            self.error_message = "Keine Version ausgewählt."
+            yield rx.toast.error("Bitte zuerst eine Version auswählen.")
+            return
+
+        self.is_loading = True
+        self.error_message = ""
+        try:
+            success = await SystemPromptRepository.delete(self.selected_version_id)
+            if success:
+                self.selected_version_id = 0
+
+                # Invalidate cache since latest version might have changed
+                await invalidate_prompt_cache()
+                logger.info("System prompt cache invalidated after deletion")
+
+                await self.load_versions()
+                yield rx.toast.success("Version erfolgreich gelöscht.")
+            else:
+                self.error_message = "Version nicht gefunden."
+                yield rx.toast.error("Version nicht gefunden.")
+        except Exception as exc:
+            self.error_message = f"Fehler beim Löschen: {exc!s}"
+            logger.exception("Failed to delete version")
+            yield rx.toast.error(f"Fehler: {exc!s}")
+        finally:
+            self.is_loading = False
+
+    def set_current_prompt(self, value: str) -> None:
+        """Update current prompt text and char count.
+
+        This is called on every keystroke but doesn't cause cursor jumping
+        because we use default_value in the textarea component.
+        """
+        self.current_prompt = value
+        self.char_count = len(value)
+
+    def set_selected_version(self, value: str | None) -> None:
+        """Handle version selection and load corresponding prompt.
+
+        When a version is selected, we update the prompt content and force
+        the textarea to re-render by changing its key.
+        """
+        if value is None or value == "":
+            return
+
+        self.selected_version_id = int(value)
+
+        # Load the prompt for the selected version
+        if value in self.prompt_map:
+            self.current_prompt = self.prompt_map[value]
+            self.char_count = len(self.current_prompt)
+            # Force textarea to re-render with new content
+            self.textarea_key += 1
+
+    @rx.var
+    def selected_version_str(self) -> str:
+        """Return selected version as string for the select component."""
+        if self.selected_version_id == 0:
+            return ""
+        return str(self.selected_version_id)