agent.py

# Tencent is pleased to support the open source community by making tRPC-Agent-Python available.
#
# Copyright (C) 2026 Tencent. All rights reserved.
#
# tRPC-Agent-Python is licensed under Apache-2.0.
""" Agent module"""

from trpc_agent_sdk.agents import LlmAgent
from trpc_agent_sdk.models import LLMModel
from trpc_agent_sdk.models import OpenAIModel
from trpc_agent_sdk.tools import LongRunningFunctionTool

from .config import get_model_config
from .prompts import MAIN_AGENT_INSTRUCTION
from .prompts import SUB_AGENT_INSTRUCTION
from .tools import check_system_critical_operation
from .tools import human_approval_required


def _create_model() -> LLMModel:
    """ Create a model"""
    api_key, url, model_name = get_model_config()
    model = OpenAIModel(model_name=model_name, api_key=api_key, base_url=url)
    return model


def create_agent() -> LlmAgent:
    """ Create an agent with long-running function tools and sub-agents"""
    model = _create_model()

    approval_tool = LongRunningFunctionTool(human_approval_required)
    critical_operation_tool = LongRunningFunctionTool(check_system_critical_operation)

    system_operations_agent = LlmAgent(
        name="system_operations_agent",
        model=model,
        description="System operations specialist that handles critical operations requiring human approval",
        instruction=SUB_AGENT_INSTRUCTION,
        tools=[critical_operation_tool],
        disallow_transfer_to_parent=True,
        output_key="system_ops_result",
    )

    agent = LlmAgent(
        name="human_in_loop_agent",
        description="Agent demonstrating long-running tools with human-in-the-loop and sub-agents",
        model=model,
        instruction=MAIN_AGENT_INSTRUCTION,
        tools=[approval_tool],
        sub_agents=[system_operations_agent],
    )
    return agent


root_agent = create_agent()