Function get_toolkit_versions

python/composio/utils/toolkit_version.py:34–78

Gets toolkit versions configuration by merging environment variables, user-provided defaults, and fallbacks. Priority order: 1. If default_versions is a string, use it as a global version for all toolkits 2. User-provided toolkit version mappings (default_versions dict) 3. Envi

(
    default_versions: t.Optional[ToolkitVersionParam] = None,
)

def get_toolkit_versions(
    default_versions: t.Optional[ToolkitVersionParam] = None,
) -> ToolkitVersionParam:
    """
    Gets toolkit versions configuration by merging environment variables, user-provided defaults, and fallbacks.

    Priority order:
    1. If default_versions is a string, use it as a global version for all toolkits
    2. User-provided toolkit version mappings (default_versions dict)
    3. Environment variables (COMPOSIO_TOOLKIT_VERSION_<TOOLKIT_NAME>)
    4. Fallback to 'latest' if no versions are configured

    :param default_versions: Optional default versions configuration (string for global version or dict mapping toolkit names to versions)
    :return: Toolkit versions configuration - either a string for global version or dict mapping toolkit names to versions
    """
    # If already set by user as a string, use it as global version for all toolkits
    if isinstance(default_versions, str):
        return default_versions

    # Check if there are envs similar to COMPOSIO_TOOLKIT_VERSION_GITHUB then extract the toolkit name
    toolkit_versions_from_env: ToolkitVersions = {}
    for key, value in os.environ.items():
        if key.startswith("COMPOSIO_TOOLKIT_VERSION_"):
            toolkit_name = key.replace("COMPOSIO_TOOLKIT_VERSION_", "")
            toolkit_versions_from_env[toolkit_name.lower()] = value

    # If the provided default versions is a dict, normalize the keys to be lower case
    # Use user provided values as overrides
    user_provided_toolkit_versions: ToolkitVersions = {}
    if default_versions and isinstance(default_versions, dict):
        user_provided_toolkit_versions = {
            key.lower(): value for key, value in default_versions.items()
        }

    # Final toolkit versions
    toolkit_versions = {
        **toolkit_versions_from_env,
        **user_provided_toolkit_versions,
    }

    # If the toolkit_versions are empty, use 'latest'
    if len(toolkit_versions) == 0:
        return "latest"

    return toolkit_versions