18 lines
756 B
Python
18 lines
756 B
Python
# Copyright (c) 2025 Patrick Motsch
|
|
# All rights reserved.
|
|
|
|
"""Shared helpers for AI workflow actions."""
|
|
|
|
|
|
def applyCommonAiParams(parameters: dict, request) -> None:
|
|
"""Apply common AI parameters (requireNeutralization, allowedModels) from node to request."""
|
|
requireNeutralization = parameters.get("requireNeutralization")
|
|
if requireNeutralization is not None:
|
|
request.requireNeutralization = bool(requireNeutralization)
|
|
|
|
allowedModels = parameters.get("allowedModels")
|
|
if allowedModels and isinstance(allowedModels, list):
|
|
if not request.options:
|
|
from modules.datamodels.datamodelAi import AiCallOptions
|
|
request.options = AiCallOptions()
|
|
request.options.allowedModels = allowedModels
|