#  Copyright 2026 zhaoxi826
#
#  Licensed under the Apache License, Version 2.0 (the "License");
#  you may not use this file except in compliance with the License.
#  You may obtain a copy of the License at
#
#       http://www.apache.org/licenses/LICENSE-2.0
#
#  Unless required by applicable law or agreed to in writing, software
#  distributed under the License is distributed on an "AS IS" BASIS,
#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#  See the License for the specific language governing permissions and
#  limitations under the License.


from pydantic import Field
from kilostar.core.work.workflow.workflow import WorkflowStep
from kilostar.utils.agent_model import ResponseModel, RequestModel, DepsModel


class ControlNodeResponse(ResponseModel):
    """控制节点回复的基类。"""

    pass


class ControlNodeInput(RequestModel):
    """控制节点输入的基类，承载一次调度所需的入参。"""

    pass


class ControlNodeDeps(DepsModel):
    """控制节点运行期依赖，注入到 pydantic-ai Agent 的 RunContext。"""

    workflow_step: WorkflowStep
    # In the future, this can be dynamically populated with tools specific to the current task execution


class ForWorkflow(ControlNodeResponse):
    """控制节点执行单个工作流步骤的输出模型。"""

    output: str = Field(
        ..., description="控制节点执行特定工作流步骤的结果。包含执行细节和输出数据。"
    )


class ForWorkflowInput(ControlNodeInput):
    """控制节点针对工作流步骤的输入模型。"""

    workflow_step: WorkflowStep