appkit-assistant 0.17.0__tar.gz → 0.17.1__tar.gz

{appkit_assistant-0.17.0 → appkit_assistant-0.17.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: appkit-assistant
-Version: 0.17.0
+Version: 0.17.1
 Summary: Add your description here
 Project-URL: Homepage, https://github.com/jenreh/appkit
 Project-URL: Documentation, https://github.com/jenreh/appkit/tree/main/docs

{appkit_assistant-0.17.0 → appkit_assistant-0.17.1}/pyproject.toml

@@ -10,7 +10,7 @@ dependencies = [
     "reflex>=0.8.22",
 ]
 name = "appkit-assistant"
-version = "0.17.0"
+version = "0.17.1"
 description = "Add your description here"
 readme = "README.md"
 authors = [{ name = "Jens Rehpöhler" }]

{appkit_assistant-0.17.0 → appkit_assistant-0.17.1}/src/appkit_assistant/backend/models.py

@@ -1,4 +1,5 @@
 import json
+import uuid
 from datetime import UTC, datetime
 from enum import StrEnum
 from typing import Any
@@ -82,7 +83,9 @@ class MessageType(StrEnum):
 
 
 class Message(BaseModel):
+    id: str = Field(default_factory=lambda: str(uuid.uuid4()))
     text: str
+    original_text: str | None = None  # To store original text if edited
     editable: bool = False
     type: MessageType
     done: bool = False

{appkit_assistant-0.17.0 → appkit_assistant-0.17.1}/src/appkit_assistant/backend/processors/gemini_responses_processor.py

@@ -643,6 +643,9 @@ class GeminiResponsesProcessor(BaseGeminiProcessor):
         content = candidate.content
 
         # List comprehension for text parts
+        if not content.parts:
+            return None
+
         text_parts = [part.text for part in content.parts if part.text]
 
         if text_parts:

{appkit_assistant-0.17.0 → appkit_assistant-0.17.1}/src/appkit_assistant/backend/processors/openai_chat_completion_processor.py

@@ -66,23 +66,40 @@ class OpenAIChatCompletionsProcessor(BaseOpenAIProcessor):
                     if event.choices and event.choices[0].delta:
                         content = event.choices[0].delta.content
                         if content:
-                            yield self._create_chunk(content, model.model, stream=True)
+                            yield self._create_chunk(
+                                content,
+                                model.model,
+                                stream=True,
+                                message_id=event.id,
+                            )
             else:
                 content = session.choices[0].message.content
                 if content:
-                    yield self._create_chunk(content, model.model)
+                    yield self._create_chunk(
+                        content, model.model, message_id=session.id
+                    )
         except Exception as e:
             raise e
 
-    def _create_chunk(self, content: str, model: str, stream: bool = False) -> Chunk:
+    def _create_chunk(
+        self,
+        content: str,
+        model: str,
+        stream: bool = False,
+        message_id: str | None = None,
+    ) -> Chunk:
+        metadata = {
+            "source": "chat_completions",
+            "streaming": str(stream),
+            "model": model,
+        }
+        if message_id:
+            metadata["message_id"] = message_id
+
         return Chunk(
             type=ChunkType.TEXT,
             text=content,
-            chunk_metadata={
-                "source": "chat_completions",
-                "streaming": str(stream),
-                "model": model,
-            },
+            chunk_metadata=metadata,
         )
 
     def _convert_messages_to_openai_format(

{appkit_assistant-0.17.0 → appkit_assistant-0.17.1}/src/appkit_assistant/backend/response_accumulator.py

@@ -52,6 +52,14 @@ class ResponseAccumulator:
 
     def process_chunk(self, chunk: Chunk) -> None:
         """Process a single chunk and update internal state."""
+        # Update message ID if provided in metadata
+        if (
+            self.messages
+            and self.messages[-1].type == MessageType.ASSISTANT
+            and "message_id" in chunk.chunk_metadata
+        ):
+            self.messages[-1].id = chunk.chunk_metadata["message_id"]
+
         if chunk.type == ChunkType.TEXT:
             if self.messages and self.messages[-1].type == MessageType.ASSISTANT:
                 self.messages[-1].text += chunk.text

{appkit_assistant-0.17.0 → appkit_assistant-0.17.1}/src/appkit_assistant/components/message.py

@@ -12,6 +12,7 @@ from appkit_assistant.state.thread_state import (
     ThreadState,
 )
 from appkit_ui.components.collabsible import collabsible
+from appkit_ui.components.dialogs import delete_dialog
 
 message_styles = {
     "spacing": "4",
@@ -85,6 +86,70 @@ class AuthCardComponent:
         )
 
 
+class MessageActionsBar:
+    """Component for message action buttons (copy, download, retry)."""
+
+    @staticmethod
+    def render(message: Message) -> rx.Component:
+        return rx.hstack(
+            rx.tooltip(
+                rx.icon_button(
+                    rx.icon("copy", size=14),
+                    on_click=ThreadState.copy_message(message.text),
+                    variant="ghost",
+                    size="1",
+                    color_scheme="gray",
+                ),
+                content="Kopieren",
+            ),
+            rx.tooltip(
+                rx.icon_button(
+                    rx.icon("download", size=14),
+                    on_click=ThreadState.download_message(message.text, message.id),
+                    variant="ghost",
+                    size="1",
+                    color_scheme="gray",
+                ),
+                content="Herunterladen",
+            ),
+            rx.tooltip(
+                delete_dialog(
+                    title="Nachricht löschen",
+                    content="diese Nachricht",
+                    on_click=ThreadState.delete_message(message.id),
+                    icon_button=True,
+                    variant="ghost",
+                    size="1",
+                    color_scheme="gray",
+                ),
+                content="Löschen",
+            ),
+            rx.cond(
+                (message.type == MessageType.ASSISTANT)
+                | (message.type == MessageType.ERROR),
+                rx.tooltip(
+                    rx.icon_button(
+                        rx.cond(
+                            ThreadState.processing,
+                            rx.spinner(size="1"),
+                            rx.icon("refresh-cw", size=14),
+                        ),
+                        on_click=ThreadState.retry_message(message.id),
+                        variant="ghost",
+                        size="1",
+                        color_scheme="gray",
+                        disabled=ThreadState.processing,
+                    ),
+                    content="Erneut generieren (folgende Nachrichten werden entfernt)",
+                ),
+                rx.fragment(),
+            ),
+            spacing="3",
+            margin_top="-9px",
+            margin_left="9px",
+        )
+
+
 class MessageComponent:
     @staticmethod
     def _file_badge(filename: str) -> rx.Component:
@@ -116,31 +181,101 @@ class MessageComponent:
 
     @staticmethod
     def human_message(message: Message) -> rx.Component:
-        return rx.hstack(
-            rx.spacer(),
+        return rx.cond(
+            ThreadState.editing_message_id == message.id,
+            # Edit Mode
             rx.vstack(
-                rx.box(
-                    rx.text(
-                        message.text,
-                        padding="0.5em",
-                        border_radius="10px",
-                        white_space="pre-line",
+                rx.text_area(
+                    value=ThreadState.edited_message_content,
+                    on_change=ThreadState.set_edited_message_content,
+                    height="112px",
+                    width="824px",
+                    auto_focus=True,
+                    bg=rx.color("gray", 3),
+                    variant="soft",
+                ),
+                rx.hstack(
+                    rx.button(
+                        "Abbrechen",
+                        on_click=ThreadState.cancel_edit,
+                        variant="soft",
+                        color_scheme="gray",
                     ),
-                    padding="4px",
-                    max_width="100%",
-                    background_color=rx.color_mode_cond(
-                        light=rx.color("accent", 3),
-                        dark=rx.color("accent", 3),
+                    rx.button("Senden", on_click=ThreadState.submit_edited_message),
+                    justify="end",
+                    width="100%",
+                    spacing="2",
+                ),
+                style=message_styles,
+                align="end",
+            ),
+            rx.vstack(
+                rx.hstack(
+                    rx.spacer(),
+                    rx.vstack(
+                        rx.box(
+                            rx.text(
+                                message.text,
+                                padding="0.5em",
+                                border_radius="10px",
+                                white_space="pre-line",
+                            ),
+                            padding="4px",
+                            max_width="800px",
+                            background_color=rx.color_mode_cond(
+                                light=rx.color("accent", 3),
+                                dark=rx.color("accent", 3),
+                            ),
+                            border_radius="9px",
+                        ),
+                        MessageComponent._attachments_row(message.attachments),
+                        align="end",
+                        spacing="1",
                     ),
-                    border_radius="9px",
                 ),
-                MessageComponent._attachments_row(message.attachments),
+                rx.hstack(
+                    rx.spacer(),
+                    rx.tooltip(
+                        rx.icon_button(
+                            rx.icon("pencil", size=14),
+                            on_click=ThreadState.set_editing_mode(
+                                message.id, message.text
+                            ),
+                            variant="ghost",
+                            size="1",
+                            color_scheme="gray",
+                        ),
+                        content="Bearbeiten",
+                    ),
+                    rx.tooltip(
+                        delete_dialog(
+                            title="Nachricht löschen",
+                            content="diese Nachricht",
+                            on_click=ThreadState.delete_message(message.id),
+                            icon_button=True,
+                            variant="ghost",
+                            size="1",
+                            color_scheme="gray",
+                        ),
+                        content="Löschen",
+                    ),
+                    rx.tooltip(
+                        rx.icon_button(
+                            rx.icon("copy", size=14),
+                            on_click=ThreadState.copy_message(message.text),
+                            variant="ghost",
+                            size="1",
+                            color_scheme="gray",
+                        ),
+                        content="Kopieren",
+                    ),
+                    spacing="3",
+                    justify="end",
+                    margin_right="9px",
+                ),
                 align="end",
-                spacing="1",
+                style=message_styles,
             ),
-            max_width="80%",
-            margin_top="24px",
-            style=message_styles,
         )
 
     @staticmethod
@@ -190,12 +325,6 @@ class MessageComponent:
                         color=rx.color("gray", 8),
                         margin_right="9px",
                     ),
-                    rx.hstack(
-                        rx.el.span(""),
-                        rx.el.span(""),
-                        rx.el.span(""),
-                        rx.el.span(""),
-                    ),
                     class_name="loading",
                     height="40px",
                     color=rx.color("gray", 8),
@@ -206,23 +335,24 @@ class MessageComponent:
                     padding_right="18px",
                 ),
                 # Actual message content
-                mn.markdown_preview(
-                    source=message.text,
-                    enable_mermaid=message.done,
-                    enable_katex=message.done,
-                    security_level="standard",
+                rx.box(
+                    mn.markdown_preview(
+                        source=message.text,
+                        enable_mermaid=message.done,
+                        enable_katex=message.done,
+                        security_level="standard",
+                        class_name="markdown",
+                    ),
                     padding="0.5em",
-                    border_radius="9px",
+                    margin_top="18px",
                     max_width="90%",
-                    class_name="markdown",
                 ),
-                # rx.markdown(
-                #     message.text,
-                #     padding="0.5em",
-                #     border_radius="9px",
-                #     max_width="90%",
-                #     class_name="markdown",
-                # ),
+            ),
+            # Actions bar
+            rx.cond(
+                message.done,
+                MessageActionsBar.render(message),
+                rx.fragment(),
             ),
             spacing="3",
             width="100%",
@@ -263,7 +393,7 @@ class MessageComponent:
         )
 
     @staticmethod
-    def error_message(message: str) -> rx.Component:
+    def error_message(message: Message) -> rx.Component:
         return rx.hstack(
             rx.avatar(
                 fallback="!",
@@ -273,15 +403,20 @@ class MessageComponent:
                 margin_top="16px",
                 color_scheme="red",
             ),
-            rx.callout(
-                message,
-                icon="triangle-alert",
-                color_scheme="red",
-                max_width="90%",
-                size="1",
-                padding="0.5em",
-                border_radius="9px",
-                margin_top="18px",
+            rx.vstack(
+                rx.callout(
+                    message.text,
+                    icon="triangle-alert",
+                    color_scheme="red",
+                    max_width="100%",
+                    size="1",
+                    padding="0.5em",
+                    border_radius="9px",
+                    margin_top="18px",
+                ),
+                MessageActionsBar.render(message),
+                width="90%",
+                spacing="2",
             ),
             style=message_styles,
         )
@@ -328,7 +463,7 @@ class MessageComponent:
                 ),
                 (
                     MessageType.ERROR,
-                    MessageComponent.error_message(message.text),
+                    MessageComponent.error_message(message),
                 ),
                 (
                     MessageType.SYSTEM,

{appkit_assistant-0.17.0 → appkit_assistant-0.17.1}/src/appkit_assistant/state/thread_state.py

@@ -14,6 +14,7 @@ import json
 import logging
 import uuid
 from collections.abc import AsyncGenerator
+from datetime import UTC, datetime
 from typing import Any
 
 import reflex as rx
@@ -74,6 +75,10 @@ class ThreadState(rx.State):
     # File upload state
     uploaded_files: list[UploadedFile] = []
 
+    # Editing state
+    editing_message_id: str | None = None
+    edited_message_content: str = ""
+
     # Internal logic helper (not reactive)
     @property
     def _thread_service(self) -> ThreadService:
@@ -510,6 +515,67 @@ class ThreadState(rx.State):
     # Message processing
     # -------------------------------------------------------------------------
 
+    @rx.event
+    def set_editing_mode(self, message_id: str, content: str) -> None:
+        """Enable editing mode for a message."""
+        self.editing_message_id = message_id
+        self.edited_message_content = content
+
+    @rx.event
+    def set_edited_message_content(self, content: str) -> None:
+        """Set the content of the message currently being edited."""
+        self.edited_message_content = content
+
+    @rx.event
+    def cancel_edit(self) -> None:
+        """Cancel editing mode."""
+        self.editing_message_id = None
+        self.edited_message_content = ""
+
+    @rx.event(background=True)
+    async def submit_edited_message(self) -> AsyncGenerator[Any, Any]:
+        """Submit edited message."""
+        async with self:
+            content = self.edited_message_content.strip()
+            if len(content) < 1:
+                yield rx.toast.error(
+                    "Nachricht darf nicht leer sein", position="top-right"
+                )
+                return
+
+            # Find message index
+            msg_index = -1
+            for i, m in enumerate(self.messages):
+                if m.id == self.editing_message_id:
+                    msg_index = i
+                    break
+
+            if msg_index == -1:
+                self.cancel_edit()
+                return
+
+            target_message = self.messages[msg_index]
+
+            # Update message
+            target_message.original_text = (
+                target_message.original_text or target_message.text
+            )
+            target_message.text = content
+
+            # Remove all messages AFTER this one
+            self.messages = self.messages[: msg_index + 1]
+
+            # Set prompt to bypass empty check in _begin_message_processing
+            self.prompt = content
+            self._skip_user_message = True
+
+            # Clear edit state
+            self.editing_message_id = None
+            self.edited_message_content = ""
+
+        # Trigger processing
+        await self._process_message()
+
     @rx.event(background=True)
     async def submit_message(self) -> AsyncGenerator[Any, Any]:
         """Submit a message and process the response."""
@@ -524,6 +590,76 @@ class ThreadState(rx.State):
             }
         """)
 
+    @rx.event(background=True)
+    async def delete_message(self, message_id: str) -> None:
+        """Delete a message from the conversation."""
+        async with self:
+            self.messages = [m for m in self.messages if m.id != message_id]
+            self._thread.messages = self.messages
+
+            if self._thread.state != ThreadStatus.NEW:
+                await self._thread_service.save_thread(
+                    self._thread, self.current_user_id
+                )
+
+    @rx.event
+    def copy_message(self, text: str) -> list[Any]:
+        """Copy message text to clipboard."""
+        return [
+            rx.set_clipboard(text),
+            rx.toast.success("Nachricht kopiert"),
+        ]
+
+    @rx.event
+    def download_message(self, text: str, message_id: str) -> Any:
+        """Download message as markdown file."""
+        timestamp = datetime.now(UTC).strftime("%Y%m%d_%H%M%S")
+        filename = (
+            f"message_{message_id}_{timestamp}.md"
+            if message_id
+            else f"message_{timestamp}.md"
+        )
+
+        # Use JavaScript to trigger download
+        return rx.call_script(f"""
+            const blob = new Blob([{json.dumps(text)}], {{type: 'text/markdown'}});
+            const url = URL.createObjectURL(blob);
+            const a = document.createElement('a');
+            a.href = url;
+            a.download = '{filename}';
+            document.body.appendChild(a);
+            a.click();
+            document.body.removeChild(a);
+            URL.revokeObjectURL(url);
+        """)
+
+    @rx.event(background=True)
+    async def retry_message(self, message_id: str) -> None:
+        """Retry generating a message."""
+        async with self:
+            # Find message index
+            index = -1
+            for i, msg in enumerate(self.messages):
+                if msg.id == message_id:
+                    index = i
+                    break
+
+            if index == -1:
+                return
+
+            # Keep context up to this message
+            # effectively removing this message and everything after
+            self.messages = self.messages[:index]
+
+            # Set prompt to bypass check (content checks)
+            self.prompt = "Regenerate"
+
+            # Flag to skip adding a new user message
+            self._skip_user_message = True
+
+        # Trigger processing directly
+        await self._process_message()
+
     async def _process_message(self) -> None:
         """Process the current message and stream the response."""
         logger.debug("Processing message: %s", self.prompt)