appkit-assistant 0.7.1__py3-none-any.whl

appkit_assistant/backend/repositories.py

@@ -0,0 +1,96 @@
+"""Repository for MCP server data access operations."""
+
+import logging
+
+import reflex as rx
+
+from appkit_assistant.backend.models import MCPServer
+
+logger = logging.getLogger(__name__)
+
+
+class MCPServerRepository:
+    """Repository class for MCP server database operations."""
+
+    @staticmethod
+    async def get_all() -> list[MCPServer]:
+        """Retrieve all MCP servers ordered by name."""
+        async with rx.asession() as session:
+            result = await session.exec(MCPServer.select().order_by(MCPServer.name))
+            return result.all()
+
+    @staticmethod
+    async def get_by_id(server_id: int) -> MCPServer | None:
+        """Retrieve an MCP server by ID."""
+        async with rx.asession() as session:
+            result = await session.exec(
+                MCPServer.select().where(MCPServer.id == server_id)
+            )
+            return result.first()
+
+    @staticmethod
+    async def create(
+        name: str,
+        url: str,
+        headers: str,
+        description: str | None = None,
+        prompt: str | None = None,
+    ) -> MCPServer:
+        """Create a new MCP server."""
+        async with rx.asession() as session:
+            server = MCPServer(
+                name=name,
+                url=url,
+                headers=headers,
+                description=description,
+                prompt=prompt,
+            )
+            session.add(server)
+            await session.commit()
+            await session.refresh(server)
+            logger.debug("Created MCP server: %s", name)
+            return server
+
+    @staticmethod
+    async def update(
+        server_id: int,
+        name: str,
+        url: str,
+        headers: str,
+        description: str | None = None,
+        prompt: str | None = None,
+    ) -> MCPServer | None:
+        """Update an existing MCP server."""
+        async with rx.asession() as session:
+            result = await session.exec(
+                MCPServer.select().where(MCPServer.id == server_id)
+            )
+            server = result.first()
+            if server:
+                server.name = name
+                server.url = url
+                server.headers = headers
+                server.description = description
+                server.prompt = prompt
+                await session.commit()
+                await session.refresh(server)
+                logger.debug("Updated MCP server: %s", name)
+                return server
+            logger.warning("MCP server with ID %s not found for update", server_id)
+            return None
+
+    @staticmethod
+    async def delete(server_id: int) -> bool:
+        """Delete an MCP server by ID."""
+        async with rx.asession() as session:
+            result = await session.exec(
+                MCPServer.select().where(MCPServer.id == server_id)
+            )
+            server = result.first()
+            if server:
+                await session.delete(server)
+                await session.commit()
+                logger.debug("Deleted MCP server: %s", server.name)
+                return True
+            logger.warning("MCP server with ID %s not found for deletion", server_id)
+            return False

appkit_assistant/backend/system_prompt.py

@@ -0,0 +1,56 @@
+from typing import Final
+
+SYSTEM_PROMPT: Final[str] = """
+# System Prompt: Kontextbewusster, Tool-orientierter Chat-Client
+
+## 1) Auftrag
+Interpretiere Benutzereingaben semantisch, berücksichtige Kontext (Verlauf, Metadaten, Projekte) und führe die geeignetsten Tools aus, um präzise, nachvollziehbare Ergebnisse zu liefern.
+
+## 2) Prioritäten
+1. Korrektheit und Prägnanz vor Länge.
+2. Tool-Einsatz, wenn verfügbar und sinnvoll; ansonsten fundierte Eigenableitung.
+3. Ergebnisse stets direkt anzeigen (kein Warten/Platzhalter).
+
+## 3) Ausgabeformate
+- **Code:** In Markdown-Codeblöcken mit korrektem Sprach-Tag.
+- **Diagramme:** Immer in Mermaid-Syntax als korrekter Markdown Source.
+- **Analysen/Vergleiche:** Datengetrieben; Tabellen verwenden.
+- **Bilder (wichtig):** Mit Bilderzeugungs-Tools generieren und **immer inline** im Chat anzeigen. Bei realen Personen nur nach vorgängiger Zustimmung.
+
+## 4) Tool-Nutzung
+- Wähle genau **ein** primäres Tool pro Aufgabe (falls mehrere möglich, wähle das mit größtem Nutzen).
+- Nutze Capability-Deskriptoren, führe Tool(s) deterministisch aus, zeige Output unmittelbar.
+- Exploratives Vorgehen ist erlaubt, sofern Ziel und Kontext klar sind.
+- Falls kein Tool passt: direkt antworten (strukturierte Begründung implizit, nicht ausgeben).
+
+{mcp_prompts}
+
+## 5) Kontext
+- Berücksichtige durchgehend Gesprächsverlauf, Nutzerrolle, Organisation und laufende Projekte.
+- Halte Kohärenz über mehrere Turns; vermeide Wiederholungen.
+
+## 6) Fehler- & Ausnahmebehandlung
+- **Toolfehler/Unverfügbarkeit:** Kurz informieren und sofort eine belastbare Alternative liefern (z. B. lokale Schätzung/Analyse).
+- **Mehrdeutigkeit:** Triff eine nachvollziehbare Annahme und liefere ein vollständiges Ergebnis.
+- **Kein geeignetes Tool:** Antwort mit eigener Inferenz, klar strukturiert.
+
+## 7) Qualitätskriterien
+- Präzise, testbare Aussagen; wenn sinnvoll, mit Zahlen/Tabellen.
+- Klare Struktur (Überschriften, Listen, Tabellen, Codeblöcke, Diagramme).
+- Konsistente Terminologie; keine redundanten Passagen.
+
+## 8) Beispiele (Format)
+```python
+def hello_world():
+    print("Hello, world!")
+```
+```mermaid
+flowchart TD
+    A["LLM/Chat-Client"] --> B["MCP Client"]
+    B --> C{{"Transport"}}
+    C -->|stdio| D["FastMCP Server (lokal)"]
+    C -->|http| E["FastMCP Server (remote)"]
+    D --> F["@mcp.tool web_search()"]
+    E --> F
+```
+"""

appkit_assistant/components/__init__.py

@@ -0,0 +1,38 @@
+from appkit_assistant.backend.models import Suggestion
+from appkit_assistant.components.composer import composer
+from appkit_assistant.components.thread import Assistant
+from appkit_assistant.components.message import MessageComponent
+from appkit_assistant.backend.models import (
+    AIModel,
+    Chunk,
+    ChunkType,
+    MCPServer,
+    Message,
+    MessageType,
+    ThreadModel,
+    ThreadStatus,
+)
+from appkit_assistant.state.thread_state import (
+    ThreadState,
+    ThreadListState,
+)
+from appkit_assistant.components.mcp_server_table import mcp_servers_table
+
+__all__ = [
+    "AIModel",
+    "Assistant",
+    "Chunk",
+    "ChunkType",
+    "MCPServer",
+    "Message",
+    "MessageComponent",
+    "MessageType",
+    "Suggestion",
+    "ThreadList",
+    "ThreadListState",
+    "ThreadModel",
+    "ThreadState",
+    "ThreadStatus",
+    "composer",
+    "mcp_servers_table",
+]

appkit_assistant/components/composer.py

@@ -0,0 +1,154 @@
+from collections.abc import Callable
+
+import reflex as rx
+
+import appkit_mantine as mn
+from appkit_assistant.components.tools_modal import tools_popover
+from appkit_assistant.state.thread_state import ThreadState
+
+
+def render_model_option(model: dict) -> rx.Component:
+    return rx.hstack(
+        rx.cond(
+            model.icon,
+            rx.image(
+                src=rx.color_mode_cond(
+                    light=f"/icons/{model.icon}.svg",
+                    dark=f"/icons/{model.icon}_dark.svg",
+                ),
+                width="13px",
+                margin_right="8px",
+            ),
+            None,
+        ),
+        rx.text(model.text),
+        align="center",
+        spacing="0",
+    )
+
+
+def composer_input(placeholder: str = "Frage etwas...") -> rx.Component:
+    return rx.text_area(
+        id="composer-area",
+        name="composer_prompt",
+        placeholder=placeholder,
+        value=ThreadState.prompt,
+        auto_height=True,
+        enter_key_submit=True,
+        # stil
+        border="0",
+        outline="none",
+        variant="soft",
+        background_color=rx.color("white", 1, alpha=False),
+        padding="9px 3px",
+        size="3",
+        min_height="24px",
+        max_height="244px",
+        resize="none",
+        rows="1",
+        width="100%",
+        on_change=ThreadState.set_prompt,
+    )
+
+
+def submit() -> rx.Component:
+    return rx.fragment(
+        rx.button(
+            rx.icon("arrow-right", size=18),
+            id="composer-submit",
+            name="composer_submit",
+            type="submit",
+            loading=ThreadState.processing,
+        ),
+    )
+
+
+def add_attachment(show: bool = False) -> rx.Component | None:
+    if not show:
+        return None
+
+    return rx.tooltip(
+        rx.button(
+            rx.icon("paperclip", size=18),
+            rx.text("2 files", size="1", color="gray.2"),
+            id="composer-attachment",
+            variant="ghost",
+            padding="8px",
+            access_key="s",
+        ),
+        content="Manage Attachments…",
+    )
+
+
+def choose_model(show: bool = False) -> rx.Component | None:
+    if not show:
+        return None
+
+    return rx.cond(
+        ThreadState.ai_models,
+        mn.rich_select(
+            mn.rich_select.map(
+                ThreadState.ai_models,
+                renderer=render_model_option,
+                value=lambda model: model.id,
+            ),
+            placeholder="Wähle ein Modell",
+            value=ThreadState.selected_model,
+            on_change=ThreadState.set_selected_model,
+            name="model-select",
+            width="252px",
+            position="top",
+        ),
+        None,
+    )
+
+
+def tools(show: bool = False) -> rx.Component:
+    """Render tools button with conditional visibility."""
+    return rx.cond(
+        show,
+        rx.hstack(
+            tools_popover(),
+            spacing="1",
+            align="center",
+        ),
+        rx.fragment(),  # Empty fragment when hidden
+    )
+
+
+def clear(show: bool = True) -> rx.Component | None:
+    if not show:
+        return None
+
+    return rx.tooltip(
+        rx.button(
+            rx.icon("paintbrush", size=17),
+            variant="ghost",
+            padding="8px",
+            on_click=ThreadState.clear,
+        ),
+        content="Chatverlauf löschen",
+    )
+
+
+def composer(*children, on_submit: Callable, **kwargs) -> rx.Component:
+    return rx.vstack(
+        rx.form.root(
+            *children,
+            on_submit=on_submit,
+        ),
+        **kwargs,
+    )
+
+
+class ComposerComponent(rx.ComponentNamespace):
+    __call__ = staticmethod(composer)
+    add_attachment = staticmethod(add_attachment)
+    choose_model = staticmethod(choose_model)
+    clear = staticmethod(clear)
+    input = staticmethod(composer_input)
+    submit = staticmethod(submit)
+    tools = staticmethod(tools)
+
+
+composer = ComposerComponent()

appkit_assistant/components/composer_key_handler.py

@@ -0,0 +1,38 @@
+import reflex as rx
+
+
+class KeyboardShortcuts(rx.Component):
+    library = None  # No external library needed
+    tag = None
+
+    def add_imports(self) -> dict:
+        return {
+            "react": [rx.ImportVar(tag="useEffect")],
+        }
+
+    def add_hooks(self) -> list[str]:
+        return [
+            """
+useEffect(() => {
+    const textarea = document.getElementById('composer-area');
+    const submitBtn = document.getElementById('composer-submit');
+
+    if (textarea && submitBtn) {
+        const handleKeydown = (e) => {
+            if (e.key === 'Enter' && !e.shiftKey) {
+                e.preventDefault();
+                if (textarea.value.trim() && !submitBtn.disabled) {
+                    submitBtn.click();
+                }
+            }
+        };
+        textarea.addEventListener('keydown', handleKeydown);
+        return () => textarea.removeEventListener('keydown', handleKeydown);
+    }
+}, []);
+            """
+        ]
+
+
+# Create an instance function
+keyboard_shortcuts = KeyboardShortcuts.create