appkit-assistant 0.17.3__tar.gz → 1.0.0__tar.gz

{appkit_assistant-0.17.3 → appkit_assistant-1.0.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: appkit-assistant
-Version: 0.17.3
+Version: 1.0.0
 Summary: Add your description here
 Project-URL: Homepage, https://github.com/jenreh/appkit
 Project-URL: Documentation, https://github.com/jenreh/appkit/tree/main/docs
@@ -16,14 +16,16 @@ Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Classifier: Topic :: Software Development :: User Interfaces
 Requires-Python: >=3.13
-Requires-Dist: anthropic>=0.40.0
+Requires-Dist: anthropic>=0.77.0
 Requires-Dist: appkit-commons
 Requires-Dist: appkit-mantine
 Requires-Dist: appkit-ui
-Requires-Dist: google-genai>=1.52.0
-Requires-Dist: mcp>=1.0.0
-Requires-Dist: openai>=2.3.0
-Requires-Dist: reflex>=0.8.22
+Requires-Dist: apscheduler>=3.11.2
+Requires-Dist: google-genai>=1.60.0
+Requires-Dist: mcp>=1.26.0
+Requires-Dist: openai>=2.16.0
+Requires-Dist: python-multipart>=0.0.22
+Requires-Dist: reflex>=0.8.26
 Description-Content-Type: text/markdown
 
 # appkit-assistant

{appkit_assistant-0.17.3 → appkit_assistant-1.0.0}/pyproject.toml

@@ -1,16 +1,18 @@
 [project]
 dependencies = [
-    "anthropic>=0.40.0",
+    "anthropic>=0.77.0",
+    "apscheduler>=3.11.2",
     "appkit-commons",
     "appkit-mantine",
     "appkit-ui",
-    "google-genai>=1.52.0",
-    "mcp>=1.0.0",
-    "openai>=2.3.0",
-    "reflex>=0.8.22",
+    "google-genai>=1.60.0",
+    "mcp>=1.26.0",
+    "openai>=2.16.0",
+    "reflex>=0.8.26",
+    "python-multipart>=0.0.22",
 ]
 name = "appkit-assistant"
-version = "0.17.3"
+version = "1.0.0"
 description = "Add your description here"
 readme = "README.md"
 authors = [{ name = "Jens Rehpöhler" }]

{appkit_assistant-0.17.3/src/appkit_assistant/backend → appkit_assistant-1.0.0/src/appkit_assistant/backend/database}/models.py

@@ -1,14 +1,12 @@
 import json
-import uuid
 from datetime import UTC, datetime
-from enum import StrEnum
 from typing import Any
 
 import reflex as rx
-from pydantic import BaseModel
 from sqlalchemy.sql import func
 from sqlmodel import Column, DateTime, Field
 
+from appkit_assistant.backend.schemas import MCPAuthType, ThreadStatus
 from appkit_commons.database.configuration import DatabaseConfig
 from appkit_commons.database.entities import EncryptedString
 from appkit_commons.registry import service_registry
@@ -32,135 +30,6 @@ class EncryptedJSON(EncryptedString):
         return value
 
 
-class ChunkType(StrEnum):
-    """Enum for chunk types."""
-
-    TEXT = "text"  # default
-    ANNOTATION = "annotation"  # for text annotations
-    IMAGE = "image"
-    IMAGE_PARTIAL = "image_partial"  # for streaming image generation
-    THINKING = "thinking"  # when the model is "thinking" / reasoning
-    THINKING_RESULT = "thinking_result"  # when the "thinking" is done
-    ACTION = "action"  # when the user needs to take action
-    TOOL_RESULT = "tool_result"  # result from a tool
-    TOOL_CALL = "tool_call"  # calling a tool
-    COMPLETION = "completion"  # when response generation is complete
-    AUTH_REQUIRED = "auth_required"  # user needs to authenticate (MCP)
-    ERROR = "error"  # when an error occurs
-    LIFECYCLE = "lifecycle"
-
-
-class Chunk(BaseModel):
-    """Model for text chunks."""
-
-    type: ChunkType
-    text: str
-    chunk_metadata: dict[str, str] = {}
-
-
-class ThreadStatus(StrEnum):
-    """Enum for thread status."""
-
-    NEW = "new"
-    ACTIVE = "active"
-    IDLE = "idle"
-    WAITING = "waiting"
-    ERROR = "error"
-    DELETED = "deleted"
-    ARCHIVED = "archived"
-
-
-class MessageType(StrEnum):
-    """Enum for message types."""
-
-    HUMAN = "human"
-    SYSTEM = "system"
-    ASSISTANT = "assistant"
-    TOOL_USE = "tool_use"
-    ERROR = "error"
-    INFO = "info"
-    WARNING = "warning"
-
-
-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
-    attachments: list[str] = []  # List of filenames for display
-
-
-class ThinkingType(StrEnum):
-    REASONING = "reasoning"
-    TOOL_CALL = "tool_call"
-
-
-class ThinkingStatus(StrEnum):
-    IN_PROGRESS = "in_progress"
-    COMPLETED = "completed"
-    ERROR = "error"
-
-
-class Thinking(BaseModel):
-    type: ThinkingType
-    id: str  # reasoning_session_id or tool_id
-    text: str
-    status: ThinkingStatus = ThinkingStatus.IN_PROGRESS
-    tool_name: str | None = None
-    parameters: str | None = None
-    result: str | None = None
-    error: str | None = None
-
-
-class AIModel(BaseModel):
-    id: str
-    text: str
-    icon: str = "codesandbox"
-    stream: bool = False
-    tenant_key: str = ""
-    project_id: int = 0
-    model: str = "default"
-    temperature: float = 0.05
-    supports_tools: bool = False
-    supports_attachments: bool = False
-    keywords: list[str] = []
-    disabled: bool = False
-    requires_role: str | None = None
-
-
-class Suggestion(BaseModel):
-    prompt: str
-    icon: str = ""
-
-
-class UploadedFile(BaseModel):
-    """Model for tracking uploaded files in the composer."""
-
-    filename: str
-    file_path: str
-    size: int = 0
-
-
-class ThreadModel(BaseModel):
-    thread_id: str
-    title: str = ""
-    active: bool = False
-    state: ThreadStatus = ThreadStatus.NEW
-    prompt: str | None = ""
-    messages: list[Message] = []
-    ai_model: str = ""
-
-
-class MCPAuthType(StrEnum):
-    """Enum for MCP server authentication types."""
-
-    NONE = "none"
-    API_KEY = "api_key"
-    OAUTH_DISCOVERY = "oauth_discovery"
-
-
 class MCPServer(rx.Model, table=True):
     """Model for MCP (Model Context Protocol) server configuration."""
 
@@ -197,6 +66,7 @@ class MCPServer(rx.Model, table=True):
     oauth_discovered_at: datetime | None = Field(
         default=None, sa_column=Column(DateTime(timezone=True), nullable=True)
     )
+    active: bool = Field(default=True, nullable=False)
 
 
 class SystemPrompt(rx.Model, table=True):
@@ -228,6 +98,7 @@ class AssistantThread(rx.Model, table=True):
     ai_model: str = Field(default="", nullable=False)
     active: bool = Field(default=False, nullable=False)
     messages: list[dict[str, Any]] = Field(default=[], sa_column=Column(EncryptedJSON))
+    vector_store_id: str | None = Field(default=None, nullable=True)
     created_at: datetime = Field(
         default_factory=lambda: datetime.now(UTC),
         sa_column=Column(DateTime(timezone=True)),
@@ -271,3 +142,32 @@ class AssistantMCPUserToken(rx.Model, table=True):
         default_factory=lambda: datetime.now(UTC),
         sa_column=Column(DateTime(timezone=True), onupdate=func.now()),
     )
+
+
+class AssistantFileUpload(rx.Model, table=True):
+    """Model for tracking files uploaded to OpenAI for vector search.
+
+    Each file is associated with a thread and vector store.
+    """
+
+    __tablename__ = "assistant_file_uploads"
+
+    id: int | None = Field(default=None, primary_key=True)
+    filename: str = Field(max_length=255, nullable=False)
+    openai_file_id: str = Field(max_length=255, nullable=False, index=True)
+    vector_store_id: str = Field(max_length=255, nullable=False, index=True)
+    vector_store_name: str = Field(max_length=255, default="", nullable=False)
+    thread_id: int = Field(
+        index=True, nullable=False, foreign_key="assistant_thread.id"
+    )
+    user_id: int = Field(index=True, nullable=False)
+    file_size: int = Field(default=0, nullable=False)
+
+    created_at: datetime = Field(
+        default_factory=lambda: datetime.now(UTC),
+        sa_column=Column(DateTime(timezone=True)),
+    )
+    updated_at: datetime = Field(
+        default_factory=lambda: datetime.now(UTC),
+        sa_column=Column(DateTime(timezone=True), onupdate=func.now()),
+    )

{appkit_assistant-0.17.3/src/appkit_assistant/backend → appkit_assistant-1.0.0/src/appkit_assistant/backend/database}/repositories.py

@@ -7,7 +7,8 @@ from sqlalchemy import select
 from sqlalchemy.ext.asyncio import AsyncSession
 from sqlalchemy.orm import defer
 
-from appkit_assistant.backend.models import (
+from appkit_assistant.backend.database.models import (
+    AssistantFileUpload,
     AssistantThread,
     MCPServer,
     SystemPrompt,
@@ -30,6 +31,18 @@ class MCPServerRepository(BaseRepository[MCPServer, AsyncSession]):
         result = await session.execute(stmt)
         return list(result.scalars().all())
 
+    async def find_all_active_ordered_by_name(
+        self, session: AsyncSession
+    ) -> list[MCPServer]:
+        """Retrieve all active MCP servers ordered by name."""
+        stmt = (
+            select(MCPServer)
+            .where(MCPServer.active == True)  # noqa: E712
+            .order_by(MCPServer.name)
+        )
+        result = await session.execute(stmt)
+        return list(result.scalars().all())
+
 
 class SystemPromptRepository(BaseRepository[SystemPrompt, AsyncSession]):
     """Repository class for system prompt database operations.
@@ -156,7 +169,86 @@ class ThreadRepository(BaseRepository[AssistantThread, AsyncSession]):
         return list(result.scalars().all())
 
 
+class FileUploadRepository(BaseRepository[AssistantFileUpload, AsyncSession]):
+    """Repository class for file upload database operations."""
+
+    @property
+    def model_class(self) -> type[AssistantFileUpload]:
+        return AssistantFileUpload
+
+    async def find_unique_vector_stores(
+        self, session: AsyncSession
+    ) -> list[tuple[str, str]]:
+        """Get unique vector store IDs with names from all file uploads.
+
+        Returns:
+            List of tuples (vector_store_id, vector_store_name).
+        """
+        stmt = (
+            select(
+                AssistantFileUpload.vector_store_id,
+                AssistantFileUpload.vector_store_name,
+            )
+            .distinct()
+            .order_by(AssistantFileUpload.vector_store_id)
+        )
+        result = await session.execute(stmt)
+        return [(row[0], row[1] or "") for row in result.all()]
+
+    async def find_by_vector_store(
+        self, session: AsyncSession, vector_store_id: str
+    ) -> list[AssistantFileUpload]:
+        """Get all files for a specific vector store."""
+        stmt = (
+            select(AssistantFileUpload)
+            .where(AssistantFileUpload.vector_store_id == vector_store_id)
+            .order_by(AssistantFileUpload.created_at.desc())
+        )
+        result = await session.execute(stmt)
+        return list(result.scalars().all())
+
+    async def find_by_thread(
+        self, session: AsyncSession, thread_id: int
+    ) -> list[AssistantFileUpload]:
+        """Get all files for a specific thread."""
+        stmt = (
+            select(AssistantFileUpload)
+            .where(AssistantFileUpload.thread_id == thread_id)
+            .order_by(AssistantFileUpload.created_at.desc())
+        )
+        result = await session.execute(stmt)
+        return list(result.scalars().all())
+
+    async def delete_file(
+        self, session: AsyncSession, file_id: int
+    ) -> AssistantFileUpload | None:
+        """Delete a file upload by ID and return the deleted record."""
+        stmt = select(AssistantFileUpload).where(AssistantFileUpload.id == file_id)
+        result = await session.execute(stmt)
+        file_upload = result.scalars().first()
+        if file_upload:
+            await session.delete(file_upload)
+            await session.flush()
+            return file_upload
+        return None
+
+    async def delete_by_vector_store(
+        self, session: AsyncSession, vector_store_id: str
+    ) -> list[AssistantFileUpload]:
+        """Delete all files for a vector store and return the deleted records."""
+        stmt = select(AssistantFileUpload).where(
+            AssistantFileUpload.vector_store_id == vector_store_id
+        )
+        result = await session.execute(stmt)
+        files = list(result.scalars().all())
+        for file_upload in files:
+            await session.delete(file_upload)
+        await session.flush()
+        return files
+
+
 # Export instances
 mcp_server_repo = MCPServerRepository()
 system_prompt_repo = SystemPromptRepository()
 thread_repo = ThreadRepository()
+file_upload_repo = FileUploadRepository()

{appkit_assistant-0.17.3 → appkit_assistant-1.0.0}/src/appkit_assistant/backend/model_manager.py

@@ -4,8 +4,8 @@ import logging
 import threading
 from typing import Self
 
-from appkit_assistant.backend.models import AIModel
-from appkit_assistant.backend.processor import Processor
+from appkit_assistant.backend.processors.processor_base import ProcessorBase
+from appkit_assistant.backend.schemas import AIModel
 
 logger = logging.getLogger(__name__)
 
@@ -29,13 +29,13 @@ class ModelManager:
     def __init__(self):
         """Initialize the service manager if not already initialized."""
         if not hasattr(self, "_initialized"):
-            self._processors: dict[str, Processor] = {}
+            self._processors: dict[str, ProcessorBase] = {}
             self._models: dict[str, AIModel] = {}
             self._model_to_processor: dict[str, str] = {}
             self._initialized = True
             logger.debug("ModelManager initialized")
 
-    def register_processor(self, processor_name: str, processor: Processor) -> None:
+    def register_processor(self, processor_name: str, processor: ProcessorBase) -> None:
         """
         Register a processor with the service manager.
 
@@ -59,7 +59,7 @@ class ModelManager:
 
         logger.debug("Registered processor: %s", processor_name)
 
-    def get_processor_for_model(self, model_id: str) -> Processor | None:
+    def get_processor_for_model(self, model_id: str) -> ProcessorBase | None:
         """
         Get the processor that supports the specified model.
 

appkit_assistant-1.0.0/src/appkit_assistant/backend/models/__init__.py

@@ -0,0 +1,28 @@
+from appkit_assistant.backend.models.anthropic import (
+    CLAUDE_HAIKU_4_5,
+    CLAUDE_SONNET_4_5,
+)
+from appkit_assistant.backend.models.google import GEMINI_3_FLASH, GEMINI_3_PRO
+from appkit_assistant.backend.models.openai import GPT_5_1, GPT_5_MINI, GPT_5_2
+from appkit_assistant.backend.models.perplexity import (
+    SONAR,
+    SONAR_DEEP_RESEARCH,
+    SONAR_PRO,
+    SONAR_REASONING,
+)
+from appkit_assistant.backend.schemas import AIModel
+
+__all__ = [
+    "CLAUDE_HAIKU_4_5",
+    "CLAUDE_SONNET_4_5",
+    "GEMINI_3_FLASH",
+    "GEMINI_3_PRO",
+    "GPT_5_1",
+    "GPT_5_2",
+    "GPT_5_MINI",
+    "SONAR",
+    "SONAR_DEEP_RESEARCH",
+    "SONAR_PRO",
+    "SONAR_REASONING",
+    "AIModel",
+]

appkit_assistant-1.0.0/src/appkit_assistant/backend/models/anthropic.py

@@ -0,0 +1,31 @@
+"""
+Claude model definitions for the Anthropic's Claude API.
+"""
+
+from typing import Final
+
+from appkit_assistant.backend.schemas import (
+    AIModel,
+)
+
+CLAUDE_HAIKU_4_5: Final = AIModel(
+    id="claude-haiku-4.5",
+    text="Claude 4.5 Haiku",
+    icon="anthropic",
+    model="claude-haiku-4-5",
+    stream=True,
+    supports_attachments=False,
+    supports_tools=True,
+    temperature=1.0,
+)
+
+CLAUDE_SONNET_4_5: Final = AIModel(
+    id="claude-sonnet-4.5",
+    text="Claude 4.5 Sonnet",
+    icon="anthropic",
+    model="claude-sonnet-4-5",
+    stream=True,
+    supports_attachments=False,
+    supports_tools=True,
+    temperature=1.0,
+)

appkit_assistant-1.0.0/src/appkit_assistant/backend/models/google.py

@@ -0,0 +1,27 @@
+"""
+Gemini model definitions for Google's GenAI API.
+"""
+
+from typing import Final
+
+from appkit_assistant.backend.schemas import AIModel
+
+GEMINI_3_PRO: Final = AIModel(
+    id="gemini-3-pro-preview",
+    text="Gemini 3 Pro",
+    icon="googlegemini",
+    model="gemini-3-pro-preview",
+    stream=True,
+    supports_attachments=False,
+    supports_tools=True,
+)
+
+GEMINI_3_FLASH: Final = AIModel(
+    id="gemini-3-flash-preview",
+    text="Gemini 3 Flash",
+    icon="googlegemini",
+    model="gemini-3-flash-preview",
+    stream=True,
+    supports_attachments=False,
+    supports_tools=True,
+)

appkit_assistant-1.0.0/src/appkit_assistant/backend/models/openai.py

@@ -0,0 +1,50 @@
+from typing import Final
+
+from appkit_assistant.backend.schemas import AIModel
+
+O3: Final = AIModel(
+    id="o3",
+    text="o3 Reasoning",
+    icon="openai",
+    model="o3",
+    temperature=1,
+    stream=True,
+    supports_attachments=False,
+    supports_tools=True,
+)
+
+GPT_5_MINI: Final = AIModel(
+    id="gpt-5-mini",
+    text="GPT 5 Mini",
+    icon="openai",
+    model="gpt-5-mini",
+    stream=True,
+    supports_attachments=True,
+    supports_tools=True,
+    supports_search=True,
+    temperature=1,
+)
+
+GPT_5_1: Final = AIModel(
+    id="gpt-5.1",
+    text="GPT 5.1",
+    icon="openai",
+    model="gpt-5.1",
+    stream=True,
+    supports_attachments=True,
+    supports_tools=True,
+    supports_search=True,
+    temperature=1,
+)
+
+GPT_5_2: Final = AIModel(
+    id="gpt-5.2",
+    text="GPT 5.2",
+    icon="openai",
+    model="gpt-5.2",
+    stream=True,
+    supports_attachments=True,
+    supports_tools=True,
+    supports_search=True,
+    temperature=1,
+)

appkit_assistant-1.0.0/src/appkit_assistant/backend/models/perplexity.py

@@ -0,0 +1,56 @@
+import enum
+
+from appkit_assistant.backend.schemas import AIModel
+
+
+class ContextSize(enum.StrEnum):
+    """Enum for context size options."""
+
+    LOW = "low"
+    MEDIUM = "medium"
+    HIGH = "high"
+
+
+class PerplexityAIModel(AIModel):
+    """AI model for Perplexity API."""
+
+    search_context_size: ContextSize = ContextSize.MEDIUM
+    search_domain_filter: list[str] = []
+
+
+SONAR = PerplexityAIModel(
+    id="sonar",
+    text="Perplexity Sonar",
+    icon="perplexity",
+    model="sonar",
+    stream=True,
+)
+
+SONAR_PRO = PerplexityAIModel(
+    id="sonar-pro",
+    text="Perplexity Sonar Pro",
+    icon="perplexity",
+    model="sonar-pro",
+    stream=True,
+    keywords=["sonar", "perplexity"],
+)
+
+SONAR_DEEP_RESEARCH = PerplexityAIModel(
+    id="sonar-deep-research",
+    text="Perplexity Deep Research",
+    icon="perplexity",
+    model="sonar-deep-research",
+    search_context_size=ContextSize.HIGH,
+    stream=True,
+    keywords=["reasoning", "deep", "research", "perplexity"],
+)
+
+SONAR_REASONING = PerplexityAIModel(
+    id="sonar-reasoning",
+    text="Perplexity Reasoning",
+    icon="perplexity",
+    model="sonar-reasoning",
+    search_context_size=ContextSize.HIGH,
+    stream=True,
+    keywords=["reasoning", "perplexity"],
+)

appkit_assistant-1.0.0/src/appkit_assistant/backend/processors/__init__.py

@@ -0,0 +1,29 @@
+from appkit_assistant.backend.processors.claude_responses_processor import (
+    ClaudeResponsesProcessor,
+)
+from appkit_assistant.backend.processors.processor_base import ProcessorBase
+from appkit_assistant.backend.processors.perplexity_processor import PerplexityProcessor
+from appkit_assistant.backend.processors.streaming_base import StreamingProcessorBase
+from appkit_assistant.backend.processors.openai_chat_completion_processor import (
+    OpenAIChatCompletionsProcessor,
+)
+from appkit_assistant.backend.processors.openai_responses_processor import (
+    OpenAIResponsesProcessor,
+)
+from appkit_assistant.backend.processors.lorem_ipsum_processor import (
+    LoremIpsumProcessor,
+)
+from appkit_assistant.backend.processors.gemini_responses_processor import (
+    GeminiResponsesProcessor,
+)
+
+__all__ = [
+    "ClaudeResponsesProcessor",
+    "GeminiResponsesProcessor",
+    "LoremIpsumProcessor",
+    "OpenAIChatCompletionsProcessor",
+    "OpenAIResponsesProcessor",
+    "PerplexityProcessor",
+    "ProcessorBase",
+    "StreamingProcessorBase",
+]