OpenAI「Agents SDK」⑦ハンドオフ

from agents import Agent, Runner
import asyncio

billing_agent = Agent(
    name="Billing Support",
    instructions="支払いに関する問い合わせを処理する。",
)

refund_agent = Agent(
    name="Refund Support",
    instructions="返品に関する問い合わせを処理する。",
)

cs_agent = Agent(
    name="Customer Service",
    instructions="あなたは、カスタマーサポートの受付エージェントです。お客様の問い合わせに応じて専門のエージェントにタスクを委譲します。",
    # ハンドオフを設定
    handoffs=[
        billing_agent,
        refund_agent,
    ]
)

async def main():
    # 例1: 支払いに関する問い合わせ
    input = "私の注文の支払いはどうなっていますか？"
    result = await Runner.run(cs_agent, input)
    print(f"ユーザ: {input}")
    print(f"{result.last_agent.name}: {result.final_output}")

    print("-" * 20)

    # 例2: 返品に関する問い合わせ
    input = "注文を返品したいのですが・・・。"
    result = await Runner.run(cs_agent, input)
    print(f"ユーザ: {input}")
    print(f"{result.last_agent.name}: {result.final_output}")

if __name__ == "__main__":
    asyncio.run(main())

ユーザ: 私の注文の支払いはどうなっていますか？
Billing Support: 支払いの詳細についてお調べいたしますので、少々お待ちください。お手数ですが、注文番号を教えていただけますか？
--------------------
ユーザ: 注文を返品したいのですが・・・。
Refund Support: 返品をご希望ですね。手続きを進めるために、注文番号や購入日などの詳細を教えていただけますか？

from agents import Agent, Runner, RunContextWrapper, handoff
import asyncio

# ハンドオフ時に呼び出されるコールバック
def on_handoff(ctx: RunContextWrapper[None]):
    print("ハンドオフが行われました")

billing_agent = Agent(
    name="Billing Support",
    instructions="支払いに関する問い合わせを処理する。",
)

refund_agent = Agent(
    name="Refund Support",
    instructions="返品に関する問い合わせを処理する。",
)

billing_handoff_obj = handoff(
    agent=billing_agent,
    on_handoff=on_handoff,
    tool_name_override="billing_operation_handoff",
    tool_description_override=(
        "支払いに関しての全ての問い合わせを処理する、支払処理専門のエージェント。"
    ),
)

refund_handoff_obj = handoff(
    agent=refund_agent,
    on_handoff=on_handoff,
    tool_name_override="refund_operation_handoff",
    tool_description_override=(
        "返品に関しての全ての問い合わせを処理する、返品処理専門のエージェント。"
    ),
)

cs_agent = Agent(
    name="Customer Service",
    instructions="あなたは、カスタマーサポートの受付エージェントです。お客様の問い合わせに応じて専門のエージェントにタスクを委譲します。",
    handoffs=[
        billing_handoff_obj,
        refund_handoff_obj,
    ]
)

async def main():
    # 例1: 支払いに関する問い合わせ
    input = "私の注文の支払いはどうなっていますか？"
    result = await Runner.run(cs_agent, input)
    print(f"ユーザ: {input}")
    print(f"{result.last_agent.name}: {result.final_output}")

    print("-" * 20)

    # 例2: 返品に関する問い合わせ
    input = "注文を返品したいのですが・・・。"
    result = await Runner.run(cs_agent, input)
    print(f"ユーザ: {input}")
    print(f"{result.last_agent.name}: {result.final_output}")

if __name__ == "__main__":
    asyncio.run(main())

ハンドオフが行われました
ユーザ: 私の注文の支払いはどうなっていますか？
Billing Support: 支払い状況について詳しく調べるために、注文番号や関連情報を教えていただけますか？
--------------------
ハンドオフが行われました
ユーザ: 注文を返品したいのですが・・・。
Refund Support: 返品をご希望ですね。返品に関する詳細をお手伝いいたしますので、以下の情報を教えていただけますか？

1. 注文番号
2. 商品名
3. 商品が未使用・未開封か
4. 返品理由

これらの情報をもとに、返品手続きを進めさせていただきます。

from pydantic import BaseModel

from agents import Agent, Runner, RunContextWrapper, handoff
import asyncio

refund_agent = Agent(
    name="Refund Support",
    instructions="返品に関する問い合わせを処理する。",
)

# ハンドオフで入力として渡すデータのモデル
class EscalationData(BaseModel):
    reason: str

# ハンドオフ時に呼び出されるコールバックで、入力データ受け取る
async def on_handoff(ctx: RunContextWrapper[None], input_data: EscalationData):
    print(f"refund agentにハンドオフされました。理由: {input_data.reason}")

# ハンドオフの設定
refund_handoff = handoff(
    agent=refund_agent,
    on_handoff=on_handoff,
    input_type=EscalationData,
)

cs_agent = Agent(
    name="Customer Service",
    instructions="あなたは、カスタマーサポートの受付エージェントです。お客様の問い合わせに応じて専門のエージェントにタスクを委譲します。",
    handoffs=[refund_handoff],
)

async def main():
    # 例1: 支払いに関する問い合わせ
    input = "私の注文の支払いはどうなっていますか？"
    result = await Runner.run(cs_agent, input)
    print(f"ユーザ: {input}")
    print(f"{result.last_agent.name}: {result.final_output}")

    print("-" * 20)

    # 例2: 返品に関する問い合わせ
    input = "注文を返品したいのですが・・・。"
    result = await Runner.run(cs_agent, input)
    print(f"ユーザ: {input}")
    print(f"{result.last_agent.name}: {result.final_output}")

if __name__ == "__main__":
    asyncio.run(main())

ユーザ: 私の注文の支払いはどうなっていますか？
Customer Service: ご注文の詳細を確認し、支払い状況をお調べします。注文番号を教えていただけますか？
--------------------
返品エージェントにハンドオフされました。理由: お客様が注文を返品したいとおっしゃっています。
ユーザ: 注文を返品したいのですが・・・。
Refund Support: 返品の件でお手伝いしますね。注文番号を教えていただけますか？また、返品の理由もお伺いしたいです。

from agents import Agent, Runner, handoff, function_tool, SQLiteSession
from agents.extensions import handoff_filters
import asyncio

# 注文番号から注文情報を参照するダミーの関数
@function_tool
def get_info_from_order_id(order_id: str) -> str:
    """注文番号から注文情報を取得する"""
    return (
        "注文番号: " + order_id + "\n"
        "注文日: 2025-01-01\n"
        "注文内容: 商品A 1本\n"
        "注文金額: 1000円\n"
        "発送状況: 入荷待ち\n"
    )

# 注文番号から注文情報を参照するダミーの関数
@function_tool
def cancel_order_from_order_id(order_id: str) -> str:
    """注文番号から注文をキャンセルする"""
    return "注文をキャンセルしました。"

cancel_agent = Agent(
    name="Cancel Agent",
    instructions="キャンセルに関する問い合わせを処理する。",
    tools=[cancel_order_from_order_id],
)

# 入力フィルタを設定したCancel Agentへのハンドオフを定義
handoff_cancel = handoff(
    agent=cancel_agent,
    input_filter=handoff_filters.remove_all_tools, 
)

cs_agent = Agent(
    name="Customer Service",
    instructions="あなたは、カスタマーサポートの受付エージェントです。お客様の問い合わせに応じて専門のエージェントにタスクを委譲します。",
    handoffs=[handoff_cancel],
    tools=[get_info_from_order_id],
)

async def main():
    session = SQLiteSession("user_123")
    input = "私の注文はどうなっていますか？"
    result = await Runner.run(cs_agent, input, session=session)
    print(f"ユーザ: {input}")
    print(f"{result.last_agent.name}: {result.final_output}")

    print("-" * 20)

    input = "注文番号は001です。"
    result = await Runner.run(cs_agent, input, session=session)
    print(f"ユーザ: {input}")
    print(f"{result.last_agent.name}: {result.final_output}")

    print("-" * 20)

    # 例2: 返品に関する問い合わせ
    input = "入荷がわからないのであれば、注文をキャンセルしたいのですが・・・。"
    result = await Runner.run(cs_agent, input, session=session)
    print(f"ユーザ: {input}")
    print(f"{result.last_agent.name}: {result.final_output}")

if __name__ == "__main__":
    asyncio.run(main())

ユーザ: 私の注文はどうなっていますか？
Customer Service: 注文番号を教えていただけますか？確認いたします。
--------------------
ユーザ: 注文番号は001です。
Customer Service: 注文番号001について確認しました。

- **注文日**: 2025-01-01
- **注文内容**: 商品A 1本
- **注文金額**: 1000円
- **発送状況**: 入荷待ち

その他のご質問があれば教えてください。
--------------------
ユーザ: 入荷がわからないのであれば、注文をキャンセルしたいのですが・・・。
Cancel Agent: 注文番号001のキャンセルが完了しました。その他にお手伝いできることがあれば教えてください。

from agents import Agent
from agents.extensions.handoff_prompt import RECOMMENDED_PROMPT_PREFIX

billing_agent = Agent(
    name="Billing agent",
    instructions=f"""{RECOMMENDED_PROMPT_PREFIX}
    お客様の請求に関する問い合わせを処理する。""",
)

print(billing_agent.instructions)

from agents import Agent
from agents.extensions.handoff_prompt import prompt_with_handoff_instructions

billing_agent = Agent(
    name="Billing agent",
    instructions=prompt_with_handoff_instructions(
        "お客様の請求に関する問い合わせを処理する。"
    ),
)

print(billing_agent.instructions)

# System context
You are part of a multi-agent system called the Agents SDK, designed to make agent coordination and execution easy. Agents uses two primary abstraction: **Agents** and **Handoffs**. An agent encompasses instructions and tools and can hand off a conversation to another agent when appropriate. Handoffs are achieved by calling a handoff function, generally named `transfer_to_<agent_name>`. Transfers between agents are handled seamlessly in the background; do not mention or draw attention to these transfers in your conversation with the user.

お客様の請求に関する問い合わせを処理する。