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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: appkit-assistant
-Version: 0.17.0
+Version: 0.17.2
 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.2}/pyproject.toml

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

{appkit_assistant-0.17.0 → appkit_assistant-0.17.2}/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.2}/src/appkit_assistant/backend/processor.py

@@ -3,6 +3,7 @@ Base processor interface for AI processing services.
 """
 
 import abc
+import asyncio
 import logging
 from collections.abc import AsyncGenerator
 
@@ -40,6 +41,7 @@ class Processor(abc.ABC):
         model_id: str,
         files: list[str] | None = None,
         mcp_servers: list[MCPServer] | None = None,
+        cancellation_token: asyncio.Event | None = None,
     ) -> AsyncGenerator[Chunk, None]:
         """
         Process the thread using an AI model.
@@ -49,6 +51,7 @@ class Processor(abc.ABC):
             model_id: The ID of the model to use.
             files: Optional list of file paths that were uploaded.
             mcp_servers: Optional list of MCP servers to use as tools.
+            cancellation_token: Optional event to signal cancellation.
 
         Returns:
             An async generator that yields Chunk objects containing different content

{appkit_assistant-0.17.0 → appkit_assistant-0.17.2}/src/appkit_assistant/backend/processors/claude_responses_processor.py

@@ -5,6 +5,7 @@ Supports MCP tools, file uploads (images and documents), extended thinking,
 and automatic citation extraction.
 """
 
+import asyncio
 import base64
 import json
 import logging
@@ -69,6 +70,7 @@ class ClaudeResponsesProcessor(BaseClaudeProcessor):
         mcp_servers: list[MCPServer] | None = None,
         payload: dict[str, Any] | None = None,
         user_id: int | None = None,
+        cancellation_token: asyncio.Event | None = None,
     ) -> AsyncGenerator[Chunk, None]:
         """Process messages using Claude Messages API with streaming."""
         if not self.client:
@@ -103,6 +105,9 @@ class ClaudeResponsesProcessor(BaseClaudeProcessor):
                 # Process streaming events
                 async with stream as response:
                     async for event in response:
+                        if cancellation_token and cancellation_token.is_set():
+                            logger.info("Processing cancelled by user")
+                            break
                         chunk = self._handle_event(event)
                         if chunk:
                             yield chunk

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

@@ -2,6 +2,7 @@
 Gemini responses processor for generating AI responses using Google's GenAI API.
 """
 
+import asyncio
 import copy
 import json
 import logging
@@ -72,6 +73,7 @@ class GeminiResponsesProcessor(BaseGeminiProcessor):
         mcp_servers: list[MCPServer] | None = None,
         payload: dict[str, Any] | None = None,
         user_id: int | None = None,
+        cancellation_token: asyncio.Event | None = None,
     ) -> AsyncGenerator[Chunk, None]:
         """Process messages using Google GenAI API with native MCP support."""
         if not self.client:
@@ -120,7 +122,7 @@ class GeminiResponsesProcessor(BaseGeminiProcessor):
         try:
             # Generate content with MCP tools
             async for chunk in self._stream_with_mcp(
-                model.model, contents, config, mcp_sessions
+                model.model, contents, config, mcp_sessions, cancellation_token
             ):
                 yield chunk
 
@@ -193,11 +195,14 @@ class GeminiResponsesProcessor(BaseGeminiProcessor):
         contents: list[types.Content],
         config: types.GenerateContentConfig,
         mcp_sessions: list[Any],
+        cancellation_token: asyncio.Event | None = None,
     ) -> AsyncGenerator[Chunk, None]:
         """Stream responses with MCP tool support."""
         if not mcp_sessions:
             # No MCP sessions, direct streaming
-            async for chunk in self._stream_generation(model_name, contents, config):
+            async for chunk in self._stream_generation(
+                model_name, contents, config, cancellation_token
+            ):
                 yield chunk
             return
 
@@ -225,7 +230,7 @@ class GeminiResponsesProcessor(BaseGeminiProcessor):
 
             # Stream with automatic function calling loop
             async for chunk in self._stream_with_tool_loop(
-                model_name, contents, config, tool_contexts
+                model_name, contents, config, tool_contexts, cancellation_token
             ):
                 yield chunk
 
@@ -235,12 +240,17 @@ class GeminiResponsesProcessor(BaseGeminiProcessor):
         contents: list[types.Content],
         config: types.GenerateContentConfig,
         tool_contexts: list[MCPToolContext],
+        cancellation_token: asyncio.Event | None = None,
     ) -> AsyncGenerator[Chunk, None]:
         """Stream generation with tool call handling loop."""
         max_tool_rounds = 10
         current_contents = list(contents)
 
         for _round_num in range(max_tool_rounds):
+            if cancellation_token and cancellation_token.is_set():
+                logger.info("Processing cancelled by user")
+                break
+
             response = await self.client.aio.models.generate_content(
                 model=model_name, contents=current_contents, config=config
             )
@@ -560,6 +570,7 @@ class GeminiResponsesProcessor(BaseGeminiProcessor):
         model_name: str,
         contents: list[types.Content],
         config: types.GenerateContentConfig,
+        cancellation_token: asyncio.Event | None = None,
     ) -> AsyncGenerator[Chunk, None]:
         """Stream generation from Gemini model."""
         # generate_content_stream returns an awaitable that yields an async generator
@@ -567,6 +578,9 @@ class GeminiResponsesProcessor(BaseGeminiProcessor):
             model=model_name, contents=contents, config=config
         )
         async for chunk in stream:
+            if cancellation_token and cancellation_token.is_set():
+                logger.info("Processing cancelled by user")
+                break
             processed = self._handle_chunk(chunk)
             if processed:
                 yield processed
@@ -643,6 +657,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.2}/src/appkit_assistant/backend/processors/lorem_ipsum_processor.py

@@ -6,6 +6,7 @@ import asyncio
 import logging
 import random
 from collections.abc import AsyncGenerator
+from typing import Any
 
 from appkit_assistant.backend.models import (
     AIModel,
@@ -58,6 +59,8 @@ class LoremIpsumProcessor(Processor):
         model_id: str,
         files: list[str] | None = None,  # noqa: ARG002
         mcp_servers: list[MCPServer] | None = None,  # noqa: ARG002
+        cancellation_token: asyncio.Event | None = None,
+        **kwargs: Any,  # noqa: ARG002
     ) -> AsyncGenerator[Chunk, None]:
         """
         Generate a Lorem Ipsum response of varying lengths based on the model_id.
@@ -67,6 +70,8 @@ class LoremIpsumProcessor(Processor):
             model_id: The model ID (determines response length).
             files: Optional list of files (ignored for this processor).
             mcp_servers: Optional list of MCP servers (ignored for this processor).
+            cancellation_token: Optional event to signal cancellation.
+            **kwargs: Additional arguments.
 
         Returns:
             An async generator that yields Chunk objects with text content.
@@ -84,9 +89,13 @@ class LoremIpsumProcessor(Processor):
 
         num_paragraphs = random.randint(4, 8)  # noqa: S311
         for i in range(num_paragraphs):
+            if cancellation_token and cancellation_token.is_set():
+                break
             paragraph = random.choice(LOREM_PARAGRAPHS)  # noqa: S311
             words = paragraph.split()
             for word in words:
+                if cancellation_token and cancellation_token.is_set():
+                    break
                 content = word + " "
                 await asyncio.sleep(0.01)
                 yield Chunk(

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

@@ -1,3 +1,4 @@
+import asyncio
 import logging
 from collections.abc import AsyncGenerator
 from typing import Any
@@ -27,6 +28,8 @@ class OpenAIChatCompletionsProcessor(BaseOpenAIProcessor):
         files: list[str] | None = None,  # noqa: ARG002
         mcp_servers: list[MCPServer] | None = None,
         payload: dict[str, Any] | None = None,
+        cancellation_token: asyncio.Event | None = None,
+        **kwargs: Any,  # noqa: ARG002
     ) -> AsyncGenerator[Chunk, None]:
         """Process messages using the Chat Completions API.
 
@@ -36,6 +39,8 @@ class OpenAIChatCompletionsProcessor(BaseOpenAIProcessor):
             files: File attachments (not used in chat completions)
             mcp_servers: MCP servers (will log warning if provided)
             payload: Additional payload parameters
+            cancellation_token: Optional event to signal cancellation
+            **kwargs: Additional arguments
         """
         if not self.client:
             raise ValueError("OpenAI Client not initialized.")
@@ -63,26 +68,46 @@ class OpenAIChatCompletionsProcessor(BaseOpenAIProcessor):
 
             if isinstance(session, AsyncStream):
                 async for event in session:
+                    if cancellation_token and cancellation_token.is_set():
+                        logger.info("Processing cancelled by user")
+                        break
                     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.2}/src/appkit_assistant/backend/processors/openai_responses_processor.py

@@ -1,3 +1,4 @@
+import asyncio
 import json
 import logging
 from collections.abc import AsyncGenerator
@@ -52,6 +53,7 @@ class OpenAIResponsesProcessor(BaseOpenAIProcessor):
         mcp_servers: list[MCPServer] | None = None,
         payload: dict[str, Any] | None = None,
         user_id: int | None = None,
+        cancellation_token: asyncio.Event | None = None,
     ) -> AsyncGenerator[Chunk, None]:
         """Process messages using simplified content accumulator pattern."""
         if not self.client:
@@ -73,6 +75,9 @@ class OpenAIResponsesProcessor(BaseOpenAIProcessor):
             try:
                 if hasattr(session, "__aiter__"):  # Streaming
                     async for event in session:
+                        if cancellation_token and cancellation_token.is_set():
+                            logger.info("Processing cancelled by user")
+                            break
                         chunk = self._handle_event(event)
                         if chunk:
                             yield chunk

{appkit_assistant-0.17.0 → appkit_assistant-0.17.2}/src/appkit_assistant/backend/processors/perplexity_processor.py

@@ -1,3 +1,4 @@
+import asyncio
 import enum
 import logging
 import os
@@ -90,6 +91,8 @@ class PerplexityProcessor(OpenAIChatCompletionsProcessor):
         files: list[str] | None = None,
         mcp_servers: list[MCPServer] | None = None,  # noqa: ARG002
         payload: dict[str, Any] | None = None,
+        cancellation_token: asyncio.Event | None = None,
+        **kwargs: Any,
     ) -> AsyncGenerator[Chunk, None]:
         if model_id not in self.models:
             logger.error("Model %s not supported by Perplexity processor", model_id)
@@ -117,5 +120,7 @@ class PerplexityProcessor(OpenAIChatCompletionsProcessor):
             files=files,
             mcp_servers=None,
             payload=perplexity_payload,
+            cancellation_token=cancellation_token,
+            **kwargs,
         ):
             yield response

{appkit_assistant-0.17.0 → appkit_assistant-0.17.2}/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.2}/src/appkit_assistant/components/composer.py

@@ -53,13 +53,25 @@ def composer_input(placeholder: str = "Frage etwas...") -> rx.Component:
 
 
 def submit() -> rx.Component:
-    return rx.fragment(
+    return rx.cond(
+        ThreadState.processing,
+        rx.tooltip(
+            rx.button(
+                rx.icon("square", size=14, fill="currentColor"),
+                on_click=ThreadState.request_cancellation,
+                color_scheme="red",
+                variant="solid",
+                type="button",
+                cursor="pointer",
+                loading=ThreadState.cancellation_requested,
+            ),
+            content="Stoppen",
+        ),
         rx.button(
             rx.icon("arrow-right", size=18),
             id="composer-submit",
             name="composer_submit",
             type="submit",
-            loading=ThreadState.processing,
         ),
     )
 

{appkit_assistant-0.17.0 → appkit_assistant-0.17.2}/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.2}/src/appkit_assistant/state/thread_state.py

@@ -10,10 +10,12 @@ This module contains ThreadState which manages the current active thread:
 See thread_list_state.py for ThreadListState which manages the thread list sidebar.
 """
 
+import asyncio
 import json
 import logging
 import uuid
 from collections.abc import AsyncGenerator
+from datetime import UTC, datetime
 from typing import Any
 
 import reflex as rx
@@ -60,6 +62,7 @@ class ThreadState(rx.State):
     ai_models: list[AIModel] = []
     selected_model: str = ""
     processing: bool = False
+    cancellation_requested: bool = False
     messages: list[Message] = []
     prompt: str = ""
     suggestions: list[Suggestion] = []
@@ -74,6 +77,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:
@@ -103,6 +110,8 @@ class ThreadState(rx.State):
     _current_user_id: str = ""
     _skip_user_message: bool = False  # Skip adding user message (for OAuth resend)
     _pending_file_cleanup: list[str] = []  # Files to delete after processing
+    # Internal cancellation event
+    _cancel_event: asyncio.Event | None = None
 
     # -------------------------------------------------------------------------
     # Computed properties
@@ -510,6 +519,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 +594,84 @@ 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()
+
+    @rx.event
+    def request_cancellation(self) -> None:
+        """Signal that the current processing should be cancelled."""
+        self.cancellation_requested = True
+        if self._cancel_event:
+            self._cancel_event.set()
+            logger.info("Cancellation requested by user")
+
     async def _process_message(self) -> None:
         """Process the current message and stream the response."""
         logger.debug("Processing message: %s", self.prompt)
@@ -552,6 +700,9 @@ class ThreadState(rx.State):
             self.uploaded_files = []
             self._pending_file_cleanup = file_paths
 
+            # Initialize cancellation event
+            self._cancel_event = asyncio.Event()
+
         first_response_received = False
         try:
             async for chunk in processor.process(
@@ -560,6 +711,7 @@ class ThreadState(rx.State):
                 files=file_paths or None,
                 mcp_servers=mcp_servers,
                 user_id=user_id,
+                cancellation_token=self._cancel_event,
             ):
                 first_response_received = await self._handle_stream_chunk(
                     chunk=chunk,
@@ -741,7 +893,9 @@ class ThreadState(rx.State):
             if self.messages:
                 self.messages[-1].done = True
             self.processing = False
+            self.cancellation_requested = False
             self.current_activity = ""
+            self._cancel_event = None
 
             # Clean up uploaded files from disk
             if self._pending_file_cleanup: