appkit-assistant 0.17.0__py3-none-any.whl → 0.17.1__py3-none-any.whl

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/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/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/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/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/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)

{appkit_assistant-0.17.0.dist-info → appkit_assistant-0.17.1.dist-info}/METADATA

@@ -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.dist-info → appkit_assistant-0.17.1.dist-info}/RECORD

@@ -3,18 +3,18 @@ appkit_assistant/pages.py,sha256=gDvBweUO2WjrhP1RE5AAkjL1_S-givWr3CkkGZKws_E,471
 appkit_assistant/backend/file_manager.py,sha256=54SYphu6FsxbEYuMx8ohQiSAeY2gGDV1q3S6RZuNku0,3153
 appkit_assistant/backend/mcp_auth_service.py,sha256=lYQEe4yOZ48ear6dvcuOXsaOc6RClIBMsOOkV7SG5Aw,27768
 appkit_assistant/backend/model_manager.py,sha256=fmv3yP63LxDnne4vjT7IzETTI2aSxViC2FSUfHQajlk,4382
-appkit_assistant/backend/models.py,sha256=vymgJ7npF-L-4WGXWwVUdHqLo3MAypPPaaD_SyDcAIw,8198
+appkit_assistant/backend/models.py,sha256=GCFWUUPsVspcFEv9naQ8n2bMU8FzMNrmqm3VgW1Fqfw,8346
 appkit_assistant/backend/processor.py,sha256=LtkkC6v0idNtbuMm3Sw5J9zIjYqtNhtxYxjYefPSS9A,2135
 appkit_assistant/backend/repositories.py,sha256=R-7kYdxg4RWQrTEOU4tbcOEhJA_FlesWrt65UpItRSU,5547
-appkit_assistant/backend/response_accumulator.py,sha256=dFJZiXcDn4V2POSk7Us6gQLsrUIRBMLdoOP61LwutdI,9766
+appkit_assistant/backend/response_accumulator.py,sha256=BCK-Ut_Wmo7rhEqOb7QlqXO8TrtANjL80FbX_AvMl1Q,10056
 appkit_assistant/backend/system_prompt_cache.py,sha256=83OIyixeTb3HKOy3XIzPyTAE-G2JyqrfcG8xVeTS2Ls,5514
 appkit_assistant/backend/processors/claude_base.py,sha256=j0DhBn8EVAjW_bfCghXaEHyORO1raUNdQeemVWCKJlA,5376
 appkit_assistant/backend/processors/claude_responses_processor.py,sha256=_quuN5FRsPSpQjaNIceN-Z66Fb3RnhXtcsS4gE0NrPA,32980
 appkit_assistant/backend/processors/gemini_base.py,sha256=ijCa8-_xdddD6ms_pkUjCj5kHZOoQ1wfw-7AhhXu6vo,2286
-appkit_assistant/backend/processors/gemini_responses_processor.py,sha256=pzbczuw7MsoABdkuJGNxlebV8POA2D72XMzLoMhj9k4,26429
+appkit_assistant/backend/processors/gemini_responses_processor.py,sha256=V5jZ83a3vi60aFTtRnbwP2q1PyEGw1iB6dYne-4CgI4,26484
 appkit_assistant/backend/processors/lorem_ipsum_processor.py,sha256=WR4s1RyOi5xgWVpre04xoxoMLIAINkpdvRCkSfNuoMc,4699
 appkit_assistant/backend/processors/openai_base.py,sha256=TBQCZW7RxaPMag_DX_tpLyMpZ_PZ_KiO8_sFN49OCpk,4424
-appkit_assistant/backend/processors/openai_chat_completion_processor.py,sha256=nTxouoXDU6VcQr8UhA2KiMNt60KvIwM8cH9Z8lo4dXY,4218
+appkit_assistant/backend/processors/openai_chat_completion_processor.py,sha256=XFp_V9Yp4KcTmicbCe3fl_3G2tP5Ffm-xLVqW50HyQw,4629
 appkit_assistant/backend/processors/openai_responses_processor.py,sha256=hysK7mlsG2t8SfltQpCvKA6MxpeIomRMGX-tSMaUshc,29492
 appkit_assistant/backend/processors/perplexity_processor.py,sha256=weHukv78MSCF_uSCKGSMpNYHsET9OB8IhpvUiMfPQ8A,3355
 appkit_assistant/backend/services/thread_service.py,sha256=LpM8ZZHt1o4MYEKzH_XPURSi3qS6p3pAQA53tOE53MU,4663
@@ -24,7 +24,7 @@ appkit_assistant/components/composer_key_handler.py,sha256=KyZYyhxzFR8DH_7F_DrvT
 appkit_assistant/components/mcp_oauth.py,sha256=puLwxAhmF25BjnZMdJbKIfC6bFXK2D8LybOX0kD7Ri4,1737
 appkit_assistant/components/mcp_server_dialogs.py,sha256=afIImmhfrNyLmxDZBpCxHxvD8HKpDanIloLEC8dJgro,23444
 appkit_assistant/components/mcp_server_table.py,sha256=1dziN7hDDvE8Y3XcdIs0wUPv1H64kP9gRAEjgH9Yvzo,2323
-appkit_assistant/components/message.py,sha256=xFMcrSMErHbbgZGl0rujNwhL7nRGwIp-S-a47cYwaJ4,14605
+appkit_assistant/components/message.py,sha256=SJcwWfSw5XxiSHtKirJUun9jkx3-ATpZ0AHLv82KwXc,19824
 appkit_assistant/components/system_prompt_editor.py,sha256=REl33zFmcpYRe9kxvFrBRYg40dV4L4FtVC_3ibLsmrU,2940
 appkit_assistant/components/thread.py,sha256=-KK1vcEmITR_oDGLbQhIa2WxRCVxEqel12m4haS7y9w,8461
 appkit_assistant/components/threadlist.py,sha256=1xVakSTQYi5-wgED3fTJVggeIjL_fkthehce0wKUYtM,4896
@@ -33,7 +33,7 @@ appkit_assistant/state/mcp_oauth_state.py,sha256=6MofExrbOOEl_YUcUOqcSTN3h7KAaER
 appkit_assistant/state/mcp_server_state.py,sha256=3AFvy53xx_eLTxw-LfJklPTgq4Ohqu4xs1QlLs-kU4U,11387
 appkit_assistant/state/system_prompt_state.py,sha256=zdnYrTnl7EszALRiodu6pcuQUd2tmtPG1eJ10j_OotI,7705
 appkit_assistant/state/thread_list_state.py,sha256=DEOR5Nklj1qfYaxSRMXCZdZRv2iq2Jb37JSg739_wL4,10250
-appkit_assistant/state/thread_state.py,sha256=xLydHA5YDFj3J1fGBQoAZsWg3zlcKOh2-9SzvlGvyU0,34742
-appkit_assistant-0.17.0.dist-info/METADATA,sha256=jzEwclPQN3VXn5APgK9EQMLiGKnug-wI-JRxPu44Kao,9498
-appkit_assistant-0.17.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-appkit_assistant-0.17.0.dist-info/RECORD,,
+appkit_assistant/state/thread_state.py,sha256=Hz7vRsBJ2cTV_xENuMoZBVSo2FGbR4uovdwiXF1zixk,39292
+appkit_assistant-0.17.1.dist-info/METADATA,sha256=t0hdCrMbCpvbETpblS3bmLtJ0ltfMXhGI9MTOtnrzII,9498
+appkit_assistant-0.17.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+appkit_assistant-0.17.1.dist-info/RECORD,,