MCP Run Python

🚀 PydanticAIを使用した銀行サポートエージェントの構築

このドキュメントでは、PydanticAIを使用して銀行サポートエージェントを構築する方法を紹介します。依存性注入、動的システムプロンプト、ツール関数の定義などの機能を含んでいます。

🚀 クイックスタート

PydanticAIの使用方法については、以下のリンクを参照してください。

• 示例文档
• ドキュメント
• API参考文档

💻 使用例

基本的な使用法

1. 依存関係の定義

from dataclasses import dataclass

from pydantic import BaseModel, Field
from pydantic_ai import Agent, RunContext

from bank_database import DatabaseConn


# 支持代理所需的依赖项，包括客户ID和数据库连接。
@dataclass
class SupportDependencies:
    customer_id: int
    db: DatabaseConn

2. 出力モデルの定義

# 输出模型用于定义支持代理返回的结果结构。
class SupportOutput(BaseModel):
    support_advice: str = Field(description='建议提供给客户的信息')
    block_card: bool = Field(description="是否需要阻止客户的卡片")
    risk: int = Field(description='查询的风险等级', ge=0, le=10)

3. サポートエージェントの定義

# 支持代理将作为银行的一线支持，处理客户查询。
support_agent = Agent(
    'openai:gpt-4o',
    deps_type=SupportDependencies,
    output_type=SupportOutput,
    system_prompt=(
        '你是一位银行支持代理，请根据客户需求提供支持，并评估查询的风险等级。'
    ),
)

4. 動的システムプロンプトの定義

# 动态系统提示可以根据依赖项中的客户名称自定义响应。
@support_agent.system_prompt
async def add_customer_name(ctx: RunContext[SupportDependencies]) -> str:
    customer_name = await ctx.deps.db.customer_name(id=ctx.deps.customer_id)
    return f"客户的姓名是 {customer_name!r}"

5. ツール関数の定義

# 工具函数允许代理调用外部功能，如查询客户余额。
@support_agent.tool
async def customer_balance(
        ctx: RunContext[SupportDependencies], include_pending: bool
) -> float:
    """返回客户的当前账户余额。"""
    balance = await ctx.deps.db.customer_balance(
        id=ctx.deps.customer_id,
        include_pending=include_pending,
    )
    return balance

6. エージェントを使用したクエリの処理

async def main():
    deps = SupportDependencies(customer_id=123, db=DatabaseConn())
    result = await support_agent.run('What is my balance?', deps=deps)
    print(result.output)
    # 示例输出：
    # support_advice='您好，John先生，您当前的账户余额为 $123.45。' block_card=False risk=1

    result = await support_agent.run('I just lost my card!', deps=deps)
    print(result.output)
    # 示例输出：
    # support_advice="非常抱歉听到这个消息，John先生。为了防止未经授权的交易，我们暂时阻止了您的卡片使用。" block_card=True risk=8

📚 まとめ

PydanticAIを使用することで、依存性注入、動的システムプロンプト、ツール関数を備えたサポートエージェントを簡単に構築できます。これらのサンプルが皆様のお役に立てば幸いです。