vent_agent_2.0/extensions/result_platform/extension.py

from contextlib import asynccontextmanager

import config
from extensions.result_platform.client import ResultPlatformClient
from extensions.result_platform.settings import ResultPlatformSettings


def enabled():
    return config.get_bool("RESULT_PLATFORM_ENABLED", False)


def validate_config():
    ResultPlatformSettings.from_config()


@asynccontextmanager
async def result_service(dependencies):
    client = ResultPlatformClient(ResultPlatformSettings.from_config())
    try:
        yield client
    finally:
        await client.close()


def setup(registry):
    registry.register_service("result_platform", result_service)