fix(server): add embedding capability to rits provider (#1480)

Signed-off-by: Radek Ježek <[email protected]>

Author: jezekra1

apps/agentstack-server/src/agentstack_server/domain/models/model_provider.py

@@ -220,6 +220,6 @@ class ModelWithScore(BaseModel):
     ModelProviderType.PERPLEXITY: {ModelCapability.LLM},
     ModelProviderType.TOGETHER: {ModelCapability.LLM},
     ModelProviderType.VOYAGE: {ModelCapability.EMBEDDING},
-    ModelProviderType.RITS: {ModelCapability.LLM},
+    ModelProviderType.RITS: {ModelCapability.LLM, ModelCapability.EMBEDDING},
     ModelProviderType.OTHER: {ModelCapability.LLM, ModelCapability.EMBEDDING},  # Other can support both
 }

apps/agentstack-server/src/agentstack_server/exceptions.py

@@ -86,8 +86,10 @@ def __init__(self, entity: str, max_size: int, status_code: int = status.HTTP_41
 
 class ModelLoadFailedError(PlatformError):
     def __init__(self, provider: ModelProvider, exception: HTTPError, status_code: int = status.HTTP_400_BAD_REQUEST):
+        from agentstack_server.application import extract_messages
+
         super().__init__(
-            f"Failed to load models from {provider.type} provider ({provider.base_url}): {exception}",
+            f"Failed to load models from {provider.type} provider ({provider.base_url}): {extract_messages(exception)}",
             status_code=status_code,
         )
 

apps/agentstack-server/tests/e2e/conftest.py

@@ -1,10 +1,10 @@
 # Copyright 2025 © BeeAI a Series of LF Projects, LLC
 # SPDX-License-Identifier: Apache-2.0
-
+import json
 import logging
 import socket
 from collections.abc import AsyncIterator, Awaitable, Callable
-from contextlib import asynccontextmanager, closing
+from contextlib import asynccontextmanager, closing, suppress
 from typing import Any
 
 import httpx
@@ -59,10 +59,15 @@ def free_port() -> int:
 
 @pytest.fixture()
 async def setup_real_llm(test_configuration, setup_platform_client):
-    await ModelProvider.create(
-        name="test_config",
-        type=test_configuration.llm_provider_type,
-        base_url=test_configuration.llm_api_base.get_secret_value(),
-        api_key=test_configuration.llm_api_key.get_secret_value(),
-    )
+    try:
+        await ModelProvider.create(
+            name="test_config",
+            type=test_configuration.llm_provider_type,
+            base_url=test_configuration.llm_api_base.get_secret_value(),
+            api_key=test_configuration.llm_api_key.get_secret_value(),
+        )
+    except httpx.HTTPStatusError as ex:
+        with suppress(Exception):
+            ex = Exception(str(f"Failed to setup LLM - {ex}\n{json.dumps(ex.response.text, indent=2)}"))
+        raise ex
     await SystemConfiguration.update(default_llm_model=test_configuration.llm_model)