naas-abi-core 1.2.1__tar.gz → 1.4.0__tar.gz

{naas_abi_core-1.2.1 → naas_abi_core-1.4.0}/=0.9.13

@@ -1,6 +1,6 @@
 Collecting uv
   Downloading uv-0.9.21-py3-none-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (11 kB)
 Downloading uv-0.9.21-py3-none-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (22.2 MB)
-   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 22.2/22.2 MB 352.7 MB/s  0:00:00
+   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 22.2/22.2 MB 290.7 MB/s  0:00:00
 Installing collected packages: uv
 Successfully installed uv-0.9.21

{naas_abi_core-1.2.1 → naas_abi_core-1.4.0}/CHANGELOG.md

@@ -2,6 +2,60 @@
 
 <!-- version list -->
 
+## v1.4.0 (2026-01-02)
+
+### Bug Fixes
+
+- Add default logo api
+  ([`0036796`](https://github.com/jupyter-naas/abi/commit/0036796586ac904879f4dd1fa4a40a400bd0e4ef))
+
+- Error aws neptune
+  ([`76ffc39`](https://github.com/jupyter-naas/abi/commit/76ffc39a0d9408a68d39bf704172c4ff5f5d98c3))
+
+- Remove connections from api doc
+  ([`4f629bf`](https://github.com/jupyter-naas/abi/commit/4f629bf0df5e3f3df30453f2531879301b05252d))
+
+### Features
+
+- Add boto3-stubs
+  ([`47dccd1`](https://github.com/jupyter-naas/abi/commit/47dccd144ab4811c1c4e481287b39026d07bb7ec))
+
+### Refactoring
+
+- Move workflows ontology yaml to naas module
+  ([`b4e8e9f`](https://github.com/jupyter-naas/abi/commit/b4e8e9fd738a338c462c9df2525e043cfdf5095d))
+
+
+## v1.3.0 (2026-01-02)
+
+### Bug Fixes
+
+- Get env value from .env if not in env
+  ([`b984252`](https://github.com/jupyter-naas/abi/commit/b98425297e596142ee39ece11f6542ed713e4d4e))
+
+- Improve config first pass loading for secret service
+  ([`f598e79`](https://github.com/jupyter-naas/abi/commit/f598e7995e7984a6bbe51e1a7e8a4a34de121968))
+
+- Missing f string + display config file name in debug
+  ([`17e4bac`](https://github.com/jupyter-naas/abi/commit/17e4bac2541642aa97fe5f59522c6cbc578c60fa))
+
+- Remove duplicated logger
+  ([`7e9ffb8`](https://github.com/jupyter-naas/abi/commit/7e9ffb81023126fe56c3150aba12e885260956a8))
+
+- Update package management & add types-pyyaml
+  ([`5a66600`](https://github.com/jupyter-naas/abi/commit/5a666002264744f571015ac45eef3e583b458f40))
+
+### Documentation
+
+- Add docstring
+  ([`76766b9`](https://github.com/jupyter-naas/abi/commit/76766b9acafee1d0dc5b2ee32057b0c3e5a0ba75))
+
+### Features
+
+- Test adaptors dotenv & naas
+  ([`d3b7204`](https://github.com/jupyter-naas/abi/commit/d3b7204f663f4ed969446db8475d97197227336b))
+
+
 ## v1.2.1 (2025-12-30)
 
 

{naas_abi_core-1.2.1 → naas_abi_core-1.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: naas-abi-core
-Version: 1.2.1
+Version: 1.4.0
 Summary: Abi framework allowing you to build your AI system.
 Author-email: Maxime Jublou <maxime@naas.ai>, Florent Ravenel <florent@naas.ai>, Jeremy Ravenel <jeremy@naas.ai>
 Requires-Python: <4,>=3.10
@@ -28,14 +28,18 @@ Requires-Dist: spacy>=3.8.7
 Requires-Dist: sparqlwrapper>=2.0.0
 Requires-Dist: sse-starlette<3,>=2.1.3
 Requires-Dist: starlette>=0.46.2
+Requires-Dist: types-paramiko>=4.0.0.20250822
+Requires-Dist: types-pyyaml>=6.0.12.20250915
 Requires-Dist: types-tqdm>=4.67.0.20250809
 Provides-Extra: all
+Requires-Dist: boto3-stubs>=1.42.19; extra == 'all'
 Requires-Dist: boto3<2,>=1.38.19; extra == 'all'
 Requires-Dist: langchain-openai>=0.3.35; extra == 'all'
 Requires-Dist: paramiko<4.0.0,>=3.5.1; extra == 'all'
 Requires-Dist: qdrant-client>=1.14.3; extra == 'all'
 Requires-Dist: sshtunnel>=0.4.0; extra == 'all'
 Provides-Extra: aws
+Requires-Dist: boto3-stubs>=1.42.19; extra == 'aws'
 Requires-Dist: boto3<2,>=1.38.19; extra == 'aws'
 Provides-Extra: openrouter
 Requires-Dist: langchain-openai>=0.3.35; extra == 'openrouter'

naas_abi_core-1.4.0/naas_abi_core/apps/api/openapi_doc.py

@@ -0,0 +1,144 @@
+TAGS_METADATA = [
+    {
+        "name": "Overview",
+        "description": """
+### Project Overview
+The **ABI** (Artificial Business Intelligence) project is a Python-based backend framework designed to serve as the core infrastructure for building an Organizational AI System. 
+This system empowers businesses to integrate, manage, and scale AI-driven operations with a focus on ontology, assistant-driven workflows, and analytics.\n
+Designed for flexibility and scalability, ABI provides a customizable framework suitable for organizations aiming to create intelligent, automated systems tailored to their needs.
+
+### API Overview
+The ABI API allows users and applications to interact with ABI's capabilities for business process automation and intelligence.\n
+This document describes the current version of the ABI API, which provides access to agents, pipelines, workflows, integrations, ontology management and analytics features.
+        """,
+    },
+    {
+        "name": "Authentication",
+        "description": """
+Authentication uses a Bearer token that can be provided either in the Authorization header (e.g. 'Authorization: Bearer `<token>`') or as a query parameter (e.g. '?token=`<token>`'). 
+The token must match the `ABI_API_KEY` environment variable.
+Contact your administrator to get the token.
+
+*Authentication with Authorization header:*
+
+```python
+import requests
+
+url = "https://<your-registry-name>.default.space.naas.ai/agents/abi/completion"
+
+headers = {
+    "Authorization": f"Bearer {token}"
+}
+
+response = requests.post(url, headers=headers)
+print(response.json())
+```
+
+*Authentication with query parameter:*
+
+```python
+import requests
+
+url = "https://<your-registry-name>.default.space.naas.ai/agents/abi/completion?token=<token>"
+
+response = requests.post(url)
+print(response.json())
+```
+        """,
+    },
+    {
+        "name": "Agents",
+        "description": """
+API endpoints for interacting with ABI's agents.
+
+### Core Agents:
+- Abi: Manages and coordinates other agents
+- Ontology: Manages and coordinates other agents
+- Naas: Manages and coordinates other agents
+- Support: Provides help and guidance for using ABI
+
+### Marketplace Agents:
+- Custom agents with deep expertise in specific domains
+- Can be configured and trained for specialized tasks
+- Extensible through custom tools and knowledge bases
+
+Each agent can be accessed through dedicated endpoints that allow:
+- Completion requests for generating responses
+- Chat interactions for ongoing conversations
+- Tool execution for specific tasks
+- Configuration updates for customizing behavior
+
+Agents leverage various tools including integrations, pipelines and workflows to accomplish tasks. They can be extended with custom tools and knowledge to enhance their capabilities.
+
+        """,
+    },
+    {
+        "name": "Pipelines",
+        "description": """
+API endpoints for interacting with ABI's pipelines.
+        """,
+    },
+    {
+        "name": "Workflows",
+        "description": """
+API endpoints for interacting with ABI's workflows.
+        """,
+    },
+]
+
+API_LANDING_HTML = """
+<!DOCTYPE html>
+<html>
+    <head>
+        <title>[TITLE]</title>
+        <link rel="icon" type="image/x-icon" href="/static/favicon.ico">
+        <style>
+            body {
+                font-family: Arial, sans-serif;
+                display: flex;
+                flex-direction: column;
+                align-items: center;
+                justify-content: center;
+                height: 100vh;
+                margin: 0;
+                background-color: #000000;
+                color: white;
+            }
+            .logo {
+                width: 200px;
+                margin-bottom: 20px;
+            }
+            h1 {
+                font-size: 48px;
+                margin-bottom: 40px;
+            }
+            .buttons {
+                display: flex;
+                gap: 20px;
+            }
+            a {
+                padding: 12px 24px;
+                font-size: 18px;
+                border: none;
+                border-radius: 4px;
+                cursor: pointer;
+                text-decoration: none;
+                color: white;
+                background-color: #007bff;
+                transition: background-color 0.2s;
+            }
+            a:hover {
+                background-color: #0056b3;
+            }
+        </style>
+    </head>
+    <body>
+        <img src="/static/[LOGO_NAME]" alt="Logo" class="logo">
+        <h1>Welcome to [TITLE]!</h1>
+        <p>[TITLE] is a tool that allows you to interact with ABI's capabilities for business process automation and intelligence.</p>
+        <div class="buttons">
+            <a href="/redoc">Go to Documentation</a>
+        </div>
+    </body>
+</html>
+"""

{naas_abi_core-1.2.1 → naas_abi_core-1.4.0}/naas_abi_core/engine/engine_configuration/EngineConfiguration.py

@@ -129,7 +129,22 @@ class EngineConfiguration(BaseModel):
 
             def __getattr__(self, name):
                 if self.secret_service is None:
-                    return 0
+                    # This rule is used only when doing the first pass configuration to load the secret service.
+
+                    # First priority is to check the environment variables.
+                    if name in os.environ:
+                        return os.environ.get(name)
+                    else:
+                        # If the environment variable is not found, we check the .env file. ONLY FOR THE FIRST PASS CONFIGURATION.
+                        from dotenv import dotenv_values
+
+                        secrets = dotenv_values()
+                        if name in secrets:
+                            return secrets.get(name)
+                        else:
+                            return f"Secret '{name}' not found while loading the secret service. Please provide it via the environment variables or .env file."
+                elif name in os.environ:
+                    return os.environ.get(name)
                 secret = self.secret_service.get(name)
                 if secret is None:
                     if not sys.stdin.isatty():
@@ -144,9 +159,6 @@ class EngineConfiguration(BaseModel):
                     return value
                 return secret
 
-            def get(self, key, default=None):
-                return 0
-
         first_pass_data = yaml.safe_load(
             StringIO(Template(yaml_content).render(secret=SecretServiceWrapper()))
         )
@@ -157,13 +169,11 @@ class EngineConfiguration(BaseModel):
         # Here we can now template the yaml by using `yaml_content` and the secret service.
         # Using Jinja2 template engine.
 
-        logger.debug("Yaml content: {yaml_content}")
+        logger.debug(f"Yaml content: {yaml_content}")
 
         template = Template(yaml_content)
         templated_yaml = template.render(secret=SecretServiceWrapper(secret_service))
 
-        logger.debug(f"Templated yaml: {templated_yaml}")
-
         data = yaml.safe_load(StringIO(templated_yaml))
 
         logger.debug(f"Data: {data}")
@@ -178,17 +188,28 @@ class EngineConfiguration(BaseModel):
         if configuration_yaml is not None:
             return cls.from_yaml_content(configuration_yaml)
 
-        logger.debug(f"Loading configuration from {os.getenv('ENV')}")
+        # Get ENV value
+        from dotenv import dotenv_values
+
+        env = os.getenv("ENV")
+        if not env:
+            env = dotenv_values().get("ENV")
 
-        if os.path.exists(f"config.{os.getenv('ENV')}.yaml"):
-            return cls.from_yaml(f"config.{os.getenv('ENV')}.yaml")
+        # First we check the environment variable.
+        if os.path.exists(f"config.{env}.yaml"):
+            config_file = f"config.{env}.yaml"
+        # If the config.{env}.yaml file is not found, we check the config.yaml file.
         elif os.path.exists("config.yaml"):
-            return cls.from_yaml("config.yaml")
+            config_file = "config.yaml"
         else:
             raise FileNotFoundError(
                 "Configuration file not found. Please create a config.yaml file or config.{env}.yaml file."
             )
 
+        logger.debug(f"Loading configuration from {config_file}")
+
+        return cls.from_yaml(config_file)
+
 
 if __name__ == "__main__":
     config = EngineConfiguration.load_configuration()

{naas_abi_core-1.2.1 → naas_abi_core-1.4.0}/naas_abi_core/engine/engine_configuration/EngineConfiguration_ObjectStorageService.py

@@ -17,18 +17,46 @@ from typing_extensions import Self
 
 
 class ObjectStorageAdapterFSConfiguration(BaseModel):
+    """Object storage adapter filesystem configuration.
+
+    object_storage_adapter:
+      adapter: "fs"
+      config:
+        base_path: "storage/datastore"
+    """
     base_path: str
 
 
 class ObjectStorageAdapterS3Configuration(BaseModel):
+    """Object storage adapter S3 configuration.
+
+    object_storage_adapter:
+      adapter: "s3"
+      config:
+        bucket_name: "my-bucket"
+        base_prefix: "my-prefix"
+        access_key_id: "{{ secret.AWS_ACCESS_KEY_ID }}"
+        secret_access_key: "{{ secret.AWS_SECRET_ACCESS_KEY }}"
+        session_token: "{{ secret.AWS_SESSION_TOKEN }}"
+    """
     bucket_name: str
+    base_prefix: str
     access_key_id: str
     secret_access_key: str
-    base_prefix: str
     session_token: str | None = None
 
 
 class ObjectStorageAdapterNaasConfiguration(BaseModel):
+    """Object storage adapter Naas configuration.
+
+    object_storage_adapter:
+      adapter: "naas"
+      config:
+        naas_api_key: "{{ secret.NAAS_API_KEY }}"
+        workspace_id: "{{ secret.WORKSPACE_ID }}"
+        storage_name: "{{ secret.STORAGE_NAME }}"
+        base_prefix: "my-prefix"
+    """
     naas_api_key: str
     workspace_id: str
     storage_name: str

{naas_abi_core-1.2.1 → naas_abi_core-1.4.0}/naas_abi_core/engine/engine_configuration/EngineConfiguration_SecretService.py

@@ -18,7 +18,38 @@ if TYPE_CHECKING:
     )
 
 
+class DotenvSecretConfiguration(BaseModel):
+    """Dotenv secret configuration.
+
+    secret_adapters:
+      - adapter: "dotenv"
+        config: {}
+    """
+    pass
+
+
+class NaasSecretConfiguration(BaseModel):
+    """Naas secret configuration.
+
+    secret_adapters:
+      - adapter: "naas"
+        config:
+          naas_api_key: "{{ secret.NAAS_API_KEY }}"
+          naas_api_url: "https://api.naas.ai"
+    """
+    naas_api_key: str
+    naas_api_url: str
+
+
 class Base64SecretConfiguration(BaseModel):
+    """Base64 secret configuration.
+
+    secret_adapters:
+      - adapter: "base64"
+        config:
+          secret_adapter: *secret_adapter
+          base64_secret_key: "{{ secret.BASE64_SECRET_KEY }}"
+    """
     secret_adapter: "SecretAdapterConfiguration"
     base64_secret_key: str
 
@@ -30,15 +61,6 @@ class Base64SecretConfiguration(BaseModel):
         return Base64Secret(self.secret_adapter.load(), self.base64_secret_key)
 
 
-class NaasSecretConfiguration(BaseModel):
-    naas_api_key: str
-    naas_api_url: str
-
-
-class DotenvSecretConfiguration(BaseModel):
-    pass
-
-
 class SecretAdapterConfiguration(GenericLoader):
     adapter: Literal["dotenv", "naas", "base64", "custom"]
     config: (

naas_abi_core-1.4.0/naas_abi_core/engine/engine_configuration/EngineConfiguration_SecretService_test.py

@@ -0,0 +1,74 @@
+from naas_abi_core.engine.engine_configuration.EngineConfiguration_SecretService import (
+    DotenvSecretConfiguration,
+    NaasSecretConfiguration,
+    SecretAdapterConfiguration,
+    SecretServiceConfiguration,
+)
+from naas_abi_core.services.secret.Secret import Secret
+from naas_abi_core.services.secret.SecretPorts import ISecretAdapter
+
+
+def test_secret_service_configuration():
+    from naas_abi_core.services.secret.adaptors.secondary.dotenv_secret_secondaryadaptor import (
+        DotenvSecretSecondaryAdaptor,
+    )
+
+    configuration = SecretServiceConfiguration(
+        secret_adapters=[
+            SecretAdapterConfiguration(
+                adapter="dotenv", config=DotenvSecretConfiguration()
+            )
+        ]
+    )
+    assert configuration.secret_adapters is not None
+    assert len(configuration.secret_adapters) == 1
+
+    secret_adapter = configuration.secret_adapters[0].load()
+
+    assert secret_adapter is not None
+    assert isinstance(secret_adapter, ISecretAdapter)
+    assert isinstance(secret_adapter, DotenvSecretSecondaryAdaptor)
+
+    secret_service = configuration.load()
+
+    assert secret_service is not None
+    assert isinstance(secret_service, Secret)
+
+
+def test_secret_service_configuration_naas():
+    import os
+
+    from dotenv import load_dotenv
+    from naas_abi_core.services.secret.adaptors.secondary.NaasSecret import (
+        NaasSecret,
+    )
+
+    load_dotenv()
+
+    naas_api_key = os.getenv("NAAS_API_KEY")
+    if naas_api_key is None:
+        raise ValueError("NAAS_API_KEY is not set")
+
+    naas_api_url = os.getenv("NAAS_API_URL")
+    if naas_api_url is None:
+        raise ValueError("NAAS_API_URL is not set")
+
+    configuration = SecretServiceConfiguration(
+        secret_adapters=[
+            SecretAdapterConfiguration(
+                adapter="naas",
+                config=NaasSecretConfiguration(
+                    naas_api_key=naas_api_key,
+                    naas_api_url=naas_api_url,
+                ),
+            )
+        ]
+    )
+    assert configuration.secret_adapters is not None
+    assert len(configuration.secret_adapters) == 1
+
+    secret_adapter = configuration.secret_adapters[0].load()
+
+    assert secret_adapter is not None
+    assert isinstance(secret_adapter, ISecretAdapter)
+    assert isinstance(secret_adapter, NaasSecret)

{naas_abi_core-1.2.1 → naas_abi_core-1.4.0}/naas_abi_core/engine/engine_configuration/EngineConfiguration_TripleStoreService.py

@@ -28,6 +28,7 @@ class OxigraphAdapterConfiguration(BaseModel):
         oxigraph_url: "http://localhost:7878"
         timeout: 60
     """
+
     oxigraph_url: str = "http://localhost:7878"
     timeout: int = 60
 
@@ -38,11 +39,12 @@ class AWSNeptuneAdapterConfiguration(BaseModel):
     triple_store_adapter:
       adapter: "aws_neptune"
       config:
-        aws_region_name: "eu-west-3"
+        aws_region_name: "{{ secret.AWS_REGION }}"
         aws_access_key_id: "{{ secret.AWS_ACCESS_KEY_ID }}"
         aws_secret_access_key: "{{ secret.AWS_SECRET_ACCESS_KEY }}"
-        db_instance_identifier: "{{ secret.DB_INSTANCE_IDENTIFIER }}"
+        db_instance_identifier: "{{ secret.AWS_NEPTUNE_DB_INSTANCE_IDENTIFIER }}"
     """
+
     aws_region_name: str
     aws_access_key_id: str
     aws_secret_access_key: str
@@ -55,15 +57,16 @@ class AWSNeptuneSSHTunnelAdapterConfiguration(AWSNeptuneAdapterConfiguration):
     triple_store_adapter:
       adapter: "aws_neptune_sshtunnel"
       config:
-        aws_region_name: "eu-west-3"
+        aws_region_name: "{{ secret.AWS_REGION }}"
         aws_access_key_id: "{{ secret.AWS_ACCESS_KEY_ID }}"
         aws_secret_access_key: "{{ secret.AWS_SECRET_ACCESS_KEY }}"
-        db_instance_identifier: "{{ secret.DB_INSTANCE_IDENTIFIER }}"   
-        bastion_host: "bastion.example.com"
-        bastion_port: 22
-        bastion_user: "ubuntu"
-        bastion_private_key: "{{ secret.BASTION_PRIVATE_KEY }}"
+        db_instance_identifier: "{{ secret.AWS_NEPTUNE_DB_INSTANCE_IDENTIFIER }}"
+        bastion_host: "{{ secret.AWS_BASTION_HOST }}"
+        bastion_port: "{{ secret.AWS_BASTION_PORT }}"
+        bastion_user: "{{ secret.AWS_BASTION_USER }}"
+        bastion_private_key: "{{ secret.AWS_BASTION_PRIVATE_KEY }}"
     """
+
     bastion_host: str
     bastion_port: int
     bastion_user: str
@@ -79,6 +82,7 @@ class TripleStoreAdapterFilesystemConfiguration(BaseModel):
         store_path: "storage/triplestore"
         triples_path: "triples"
     """
+
     store_path: str
     triples_path: str = "triples"
 
@@ -92,6 +96,7 @@ class TripleStoreAdapterObjectStorageConfiguration(BaseModel):
         object_storage_service: *object_storage_service
         triples_prefix: "triples"
     """
+
     object_storage_service: ObjectStorageServiceConfiguration
     triples_prefix: str = "triples"
 

naas_abi_core-1.4.0/naas_abi_core/engine/engine_configuration/EngineConfiguration_TripleStoreService_test.py

@@ -0,0 +1,109 @@
+from naas_abi_core.engine.engine_configuration.EngineConfiguration_TripleStoreService import (
+    AWSNeptuneSSHTunnelAdapterConfiguration,
+    OxigraphAdapterConfiguration,
+    TripleStoreAdapterConfiguration,
+    TripleStoreAdapterFilesystemConfiguration,
+    TripleStoreServiceConfiguration,
+)
+from naas_abi_core.services.triple_store.TripleStorePorts import ITripleStorePort
+from naas_abi_core.services.triple_store.TripleStoreService import TripleStoreService
+
+
+def test_triple_store_service_configuration():
+    from naas_abi_core.services.triple_store.adaptors.secondary.TripleStoreService__SecondaryAdaptor__Filesystem import (
+        TripleStoreService__SecondaryAdaptor__Filesystem,
+    )
+
+    configuration = TripleStoreServiceConfiguration(
+        triple_store_adapter=TripleStoreAdapterConfiguration(
+            adapter="fs",
+            config=TripleStoreAdapterFilesystemConfiguration(
+                store_path="storage/triplestore/test", triples_path="triples"
+            ),
+        )
+    )
+    assert configuration.triple_store_adapter is not None
+
+    triple_store_adapter = configuration.triple_store_adapter.load()
+
+    assert triple_store_adapter is not None
+    assert isinstance(triple_store_adapter, ITripleStorePort)
+    assert isinstance(
+        triple_store_adapter, TripleStoreService__SecondaryAdaptor__Filesystem
+    )
+
+    triple_store_service = configuration.load()
+
+    assert triple_store_service is not None
+    assert isinstance(triple_store_service, TripleStoreService)
+
+
+def test_triple_store_service_configuration_naas():
+    from naas_abi_core.services.triple_store.adaptors.secondary.Oxigraph import (
+        Oxigraph,
+    )
+
+    configuration = TripleStoreServiceConfiguration(
+        triple_store_adapter=TripleStoreAdapterConfiguration(
+            adapter="oxigraph",
+            config=OxigraphAdapterConfiguration(
+                oxigraph_url="http://localhost:7878",
+                timeout=120,
+            ),
+        )
+    )
+    assert configuration.triple_store_adapter is not None
+
+    triple_store_adapter = configuration.triple_store_adapter.load()
+
+    assert triple_store_adapter is not None
+    assert isinstance(triple_store_adapter, ITripleStorePort)
+    assert isinstance(triple_store_adapter, Oxigraph)
+
+    triple_store_service = configuration.load()
+
+    assert triple_store_service is not None
+    assert isinstance(triple_store_service, TripleStoreService)
+
+
+def test_triple_store_service_configuration_aws_neptune_sshtunnel():
+    import os
+
+    from dotenv import load_dotenv
+
+    load_dotenv()
+
+    from naas_abi_core.services.triple_store.adaptors.secondary.AWSNeptune import (
+        AWSNeptuneSSHTunnel,
+    )
+
+    configuration = TripleStoreServiceConfiguration(
+        triple_store_adapter=TripleStoreAdapterConfiguration(
+            adapter="aws_neptune_sshtunnel",
+            config=AWSNeptuneSSHTunnelAdapterConfiguration(
+                aws_region_name=os.environ.get("AWS_REGION"),
+                aws_access_key_id=os.environ.get("AWS_ACCESS_KEY_ID"),
+                aws_secret_access_key=os.environ.get("AWS_SECRET_ACCESS_KEY"),
+                db_instance_identifier=os.environ.get(
+                    "AWS_NEPTUNE_DB_INSTANCE_IDENTIFIER"
+                ),
+                bastion_host=os.environ.get("AWS_BASTION_HOST"),
+                bastion_port=int(os.environ.get("AWS_BASTION_PORT")),
+                bastion_user=os.environ.get("AWS_BASTION_USER"),
+                bastion_private_key=os.environ.get("AWS_BASTION_PRIVATE_KEY"),
+            ),
+        ),
+    )
+
+    assert configuration.triple_store_adapter is not None
+
+    triple_store_adapter = configuration.triple_store_adapter.load()
+
+    assert triple_store_adapter is not None
+    assert isinstance(triple_store_adapter, ITripleStorePort)
+    assert isinstance(triple_store_adapter, AWSNeptuneSSHTunnel)
+
+    triple_store_service = configuration.load()
+
+    assert triple_store_service is not None
+    assert isinstance(triple_store_service, TripleStoreService)