Operate Factory Robots with Intent

# agent_service/factory_agent.py

from google.adk.agents import Agent
from google.adk.tools import tool

@tool
def set_operation_mode(mode: str) -> dict:
    """Set the workcell operation mode.
    modes: can_priority | fruit_priority | balanced | high_capacity | low_power
    """
    return {"action": "set_operation_mode", "mode": mode}

@tool
def set_production_target(item_type: str, count: int) -> dict:
    """Set a production target for cans or fruits."""
    return {"action": "set_production_target", "item_type": item_type, "count": count}

@tool
def disable_robot(robot: str) -> dict:
    """Take a robot offline. robots: UR3e | UR5e | UR10e | ScaraT6"""
    return {"action": "disable_robot", "robot": robot}

@tool
def enable_robot(robot: str) -> dict:
    """Bring a robot back online."""
    return {"action": "enable_robot", "robot": robot}

@tool
def get_workcell_status() -> dict:
    """Return a production report for the current workcell state."""
    return {"action": "get_workcell_status"}

factory_agent = Agent(
    model="gemini-2.0-flash",
    tools=[set_operation_mode, set_production_target,
           disable_robot, enable_robot, get_workcell_status],
)