Community version: UI adjustments

.gitignore

@@ -212,7 +212,6 @@ examples/
 *.webp
 
 # Portable Network Graphics
-*.png
 
 # Animated Portable Network Graphics
 *.apng
@@ -225,7 +224,6 @@ examples/
 *.tif
 
 # Scalable Vector Graphics
-*.svg
 *.svgz
 
 # Portable Document Format

backend/migrations/versions/8403bb364491_unique_mcp_a2a_server_url_per_user_.py

@@ -0,0 +1,43 @@
+"""Unique mcp/a2a server url per user constraint
+
+Revision ID: 8403bb364491
+Revises: 046a06a0b0a4
+Create Date: 2025-07-07 13:05:36.680696
+
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "8403bb364491"
+down_revision: Union[str, None] = "046a06a0b0a4"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_constraint("a2acards_server_url_key", "a2acards", type_="unique")
+    op.create_unique_constraint(
+        "uq_a2a_card_server_url", "a2acards", ["creator_id", "server_url"]
+    )
+    op.drop_constraint("mcpservers_server_url_key", "mcpservers", type_="unique")
+    op.create_unique_constraint(
+        "uq_mcp_server_url", "mcpservers", ["creator_id", "server_url"]
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_constraint("uq_mcp_server_url", "mcpservers", type_="unique")
+    op.create_unique_constraint(
+        "mcpservers_server_url_key", "mcpservers", ["server_url"]
+    )
+    op.drop_constraint("uq_a2a_card_server_url", "a2acards", type_="unique")
+    op.create_unique_constraint("a2acards_server_url_key", "a2acards", ["server_url"])
+    # ### end Alembic commands ###

backend/src/models/__init__.py

@@ -291,7 +291,7 @@ class MCPServer(Base):
     name: Mapped[str] = mapped_column(nullable=True)
     description: Mapped[str] = mapped_column(nullable=True)
 
-    server_url: Mapped[str] = mapped_column(unique=True, nullable=False)
+    server_url: Mapped[str] = mapped_column(nullable=False)
 
     mcp_tools: Mapped[List["MCPTool"]] = relationship(back_populates="mcp_server")
 
@@ -306,6 +306,10 @@ class MCPServer(Base):
 
     is_active: Mapped[bool]
 
+    __table_args__ = (
+        UniqueConstraint("creator_id", "server_url", name="uq_mcp_server_url"),
+    )
+
     def __repr__(self) -> str:
         return f"<MCPServer(host={self.server_url!r}>"
 
@@ -336,7 +340,7 @@ class A2ACard(Base):
     alias: Mapped[str] = mapped_column(nullable=True)
     description: Mapped[str] = mapped_column(nullable=True)
 
-    server_url: Mapped[str] = mapped_column(unique=True, nullable=False)
+    server_url: Mapped[str] = mapped_column(nullable=False)
     card_content: Mapped[not_null_json_column]
 
     is_active: Mapped[bool]
@@ -349,6 +353,9 @@ class A2ACard(Base):
     creator_id: Mapped[uuid.UUID] = mapped_column(
         ForeignKey("users.id", ondelete="CASCADE"), nullable=True, index=True
     )
+    __table_args__ = (
+        UniqueConstraint("creator_id", "server_url", name="uq_a2a_card_server_url"),
+    )
 
 
 class ChatMessage(Base):

backend/src/repositories/a2a.py

@@ -1,4 +1,5 @@
 import logging
+import traceback
 from datetime import datetime
 from typing import Optional
 from uuid import UUID
@@ -112,6 +113,9 @@ async def add_url(
             )
 
         except IntegrityError:
+            print("=" * 30)
+            print(traceback.format_exc())
+            print("=" * 30)
             await db.rollback()
             return await self.update_server_is_active_state(
                 db=db, user_id=user_id, agent_dto=a2a_card_dto

backend/src/repositories/model_config.py

@@ -5,6 +5,7 @@
 from sqlalchemy import and_, select
 from sqlalchemy.ext.asyncio import AsyncSession
 from sqlalchemy.orm import selectinload
+from src.auth.encrypt import decrypt_secret
 from src.models import ModelConfig, ModelProvider, User
 from src.repositories.base import CRUDBase
 from src.schemas.api.model_config.dto import ModelConfigDTO, ModelProviderDTO
@@ -289,6 +290,18 @@ async def update_provider(
             upd_in.api_key = None
             return await self.update(db=db, db_obj=provider_obj, obj_in=upd_in.dump())
 
+        try:
+            new_api_key = decrypt_secret(upd_in.api_key)
+            prev_api_key = decrypt_secret(provider_obj.api_key)
+            if prev_api_key == new_api_key:
+                return await self.update(
+                    db=db, db_obj=provider_obj, obj_in=upd_in.dump()
+                )
+        except ValueError:
+            # result of 'prev_api_key' decryption is a value that differs from the previously
+            # encrypted key -> cryptography library raises ValueError -> apply encryption to the new value
+            pass
+
         api_key = validate_and_encrypt_provider_api_key(api_key=upd_in.api_key)
         upd_in.api_key = api_key
         return await self.update(db=db, db_obj=provider_obj, obj_in=upd_in.dump())

backend/src/routes/websocket.py

@@ -29,6 +29,7 @@
 
 settings = get_settings()
 logger = logging.getLogger(__name__)
+
 ws_router = APIRouter()
 
 
@@ -151,6 +152,9 @@ async def handle_frontend_ws(
                     f"Message validation failed. Details: {validation_exception_handler(exc=e)}"  # noqa: E501
                 )
             chat_title = message_obj.message[:20]
+            if not chat_title:
+                chat_title = "New Chat"
+
             chat = await chat_repo.get_chat_by_session_id(
                 db=db, session_id=session_id, user_model=user_model
             )

frontend/components.json

@@ -0,0 +1,21 @@
+{
+  "$schema": "https://ui.shadcn.com/schema.json",
+  "style": "new-york",
+  "rsc": false,
+  "tsx": true,
+  "tailwind": {
+    "config": "./tailwind.config.js",
+    "css": "src/index.css",
+    "baseColor": "neutral",
+    "cssVariables": true,
+    "prefix": ""
+  },
+  "aliases": {
+    "components": "@/components",
+    "utils": "@/utils/utils",
+    "ui": "@/components/ui",
+    "lib": "@/utils",
+    "hooks": "@/hooks"
+  },
+  "iconLibrary": "lucide"
+}