Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
# SPDX-FileCopyrightText: 2023-present deepset GmbH <info@deepset.ai>
#
# SPDX-License-Identifier: Apache-2.0
from typing import Any
from typing import Any, ClassVar

from haystack import component, default_to_dict
from haystack.components.embedders import OpenAIDocumentEmbedder
Expand Down Expand Up @@ -30,6 +30,14 @@ class STACKITDocumentEmbedder(OpenAIDocumentEmbedder):
```
"""

SUPPORTED_MODELS: ClassVar[list[str]] = [
"intfloat/e5-mistral-7b-instruct",
"Qwen/Qwen3-VL-Embedding-8B",
]
"""A non-exhaustive list of embedding models supported by this component.
See https://docs.stackit.cloud/products/data-and-ai/ai-model-serving/basics/available-shared-models
for the full list."""

def __init__(
self,
model: str,
Expand Down
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
# SPDX-FileCopyrightText: 2023-present deepset GmbH <info@deepset.ai>
#
# SPDX-License-Identifier: Apache-2.0
from typing import Any
from typing import Any, ClassVar

from haystack import component, default_to_dict
from haystack.components.embedders import OpenAITextEmbedder
Expand All @@ -23,6 +23,14 @@ class STACKITTextEmbedder(OpenAITextEmbedder):
```
"""

SUPPORTED_MODELS: ClassVar[list[str]] = [
"intfloat/e5-mistral-7b-instruct",
"Qwen/Qwen3-VL-Embedding-8B",
]
"""A non-exhaustive list of embedding models supported by this component.
See https://docs.stackit.cloud/products/data-and-ai/ai-model-serving/basics/available-shared-models
for the full list."""

def __init__(
self,
model: str,
Expand Down
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
# SPDX-FileCopyrightText: 2025-present deepset GmbH <info@deepset.ai>
#
# SPDX-License-Identifier: Apache-2.0
from typing import Any
from typing import Any, ClassVar

from haystack import component, default_to_dict
from haystack.components.generators.chat import OpenAIChatGenerator
Expand Down Expand Up @@ -38,6 +38,19 @@ class STACKITChatGenerator(OpenAIChatGenerator):
```
"""

SUPPORTED_MODELS: ClassVar[list[str]] = [
"Qwen/Qwen3-VL-235B-A22B-Instruct-FP8",
"cortecs/Llama-3.3-70B-Instruct-FP8-Dynamic",
"openai/gpt-oss-120b",
"google/gemma-3-27b-it",
"openai/gpt-oss-20b",
"neuralmagic/Mistral-Nemo-Instruct-2407-FP8",
"neuralmagic/Meta-Llama-3.1-8B-Instruct-FP8",
]
"""A non-exhaustive list of chat models supported by this component.
See https://docs.stackit.cloud/products/data-and-ai/ai-model-serving/basics/available-shared-models
for the full list."""

def __init__(
self,
model: str,
Expand Down
7 changes: 7 additions & 0 deletions integrations/stackit/tests/test_stackit_chat_generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -63,6 +63,13 @@ def mock_chat_completion():


class TestSTACKITChatGenerator:
def test_supported_models(self):
"""SUPPORTED_MODELS is a non-empty list of strings."""
models = STACKITChatGenerator.SUPPORTED_MODELS
assert isinstance(models, list)
assert len(models) > 0
assert all(isinstance(m, str) for m in models)

def test_init_default(self, monkeypatch):
monkeypatch.setenv("STACKIT_API_KEY", "test-api-key")
component = STACKITChatGenerator(model="neuralmagic/Meta-Llama-3.1-8B-Instruct-FP8")
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,13 @@


class TestSTACKITDocumentEmbedder:
def test_supported_models(self):
"""SUPPORTED_MODELS is a non-empty list of strings."""
models = STACKITDocumentEmbedder.SUPPORTED_MODELS
assert isinstance(models, list)
assert len(models) > 0
assert all(isinstance(m, str) for m in models)

def test_init_default(self, monkeypatch):
monkeypatch.setenv("STACKIT_API_KEY", "test-api-key")

Expand Down
7 changes: 7 additions & 0 deletions integrations/stackit/tests/test_stackit_text_embedder.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,13 @@


class TestSTACKITTextEmbedder:
def test_supported_models(self):
"""SUPPORTED_MODELS is a non-empty list of strings."""
models = STACKITTextEmbedder.SUPPORTED_MODELS
assert isinstance(models, list)
assert len(models) > 0
assert all(isinstance(m, str) for m in models)

def test_init_default(self, monkeypatch):
monkeypatch.setenv("STACKIT_API_KEY", "test-api-key")

Expand Down