12: エージェントのためのツールの構築

from langchain.tools import tool
 
@tool
def get_weather(city: str) -> str:
    """指定された都市の現在の天気を取得します。"""
    return f"It's always sunny in {city}!"

print(get_weather.name)
# 出力: get_weather
 
print(get_weather.description)
# 出力: 指定された都市の現在の天気を取得します。
 
print(get_weather.args)
# 出力: {'city': {'title': 'City', 'type': 'string'}}

result = get_weather.invoke({"city": "Paris"})
print(result)
# 出力: It's always sunny in Paris!

@tool("web_search")
def search(query: str) -> str:
    """ウェブで情報を検索します。"""
    return f"Results for: {query}"
 
print(search.name)
# 出力: web_search

@tool("calculator", description="算術演算を実行します。あらゆる数学の問題にこれを使ってください。")
def calc(expression: str) -> str:
    """数式の文字列を評価します。"""
    return str(eval(expression))  # 警告: eval() は安全ではありません。本番環境では絶対に使わないでください。

from pydantic import BaseModel, Field
from langchain.tools import tool
 
class WeatherInput(BaseModel):
    """天気クエリのための入力です。"""
    location: str = Field(description="都市名 (例: ソウル、東京)")
    units: str = Field(default="celsius", description="温度の単位 (celsius または fahrenheit)")
 
@tool(args_schema=WeatherInput)
def get_weather_detailed(location: str, units: str = "celsius") -> str:
    """選択した温度の単位で現在の天気を取得します。"""
    temp = 22 if units == "celsius" else 72
    return f"Current weather in {location}: {temp} degrees {units[0].upper()}"

# product_service.py
 
PRODUCTS = {
    1: {"name": "Wireless Mouse", "price": 29.99, "stock": 120},
    2: {"name": "Mechanical Keyboard", "price": 89.99, "stock": 0},
    3: {"name": "USB-C Hub", "price": 45.50, "stock": 35},
    4: {"name": "Laptop Stand", "price": 39.00, "stock": 8},
}
 
def fetch_product(product_id: int) -> dict:
    """IDで商品情報を検索します。"""
    product = PRODUCTS.get(product_id)
    if product is None:
        raise ValueError(f"Product with ID {product_id} not found.")
    return product
 
def fetch_stock(product_id: int) -> int:
    """商品の在庫数量を返します。"""
    product = PRODUCTS.get(product_id)
    if product is None:
        raise ValueError(f"Product with ID {product_id} not found.")
    return product["stock"]

from langchain.tools import tool
from product_service import fetch_product
 
@tool
def get_product(product_id: int) -> str:
    """IDで商品を検索します。その名前、価格、在庫レベルを返します。"""
    product = fetch_product(product_id)
    return f"Product {product_id}: {product['name']} — ${product['price']:.2f}, {product['stock']} in stock."

print(get_product.invoke({"product_id": 1}))
# 出力: Product 1: Wireless Mouse — $29.99, 120 in stock.

print(get_product.invoke({"product_id": 99}))
# ValueError: Product with ID 99 not found.

from langchain.tools import tool
from product_service import fetch_product
 
@tool
def get_product(product_id: int) -> str:
    """IDで商品を検索します。その名前、価格、在庫レベルを返します。"""
    try:
        product = fetch_product(product_id)
        return f"Product {product_id}: {product['name']} — ${product['price']:.2f}, {product['stock']} in stock."
    except ValueError as e:
        return f"Error: {e}"
    except Exception as e:
        return f"Unexpected error looking up product {product_id}: {e}"

print(get_product.invoke({"product_id": 1}))
# 出力: Product 1: Wireless Mouse — $29.99, 120 in stock.
 
print(get_product.invoke({"product_id": 99}))
# 出力: Error: Product with ID 99 not found.

from product_service import fetch_stock
 
@tool
def check_stock(product_id: int) -> str:
    """商品が現在在庫にあるかどうかを確認します。"""
    try:
        stock = fetch_stock(product_id)
        if stock > 0:
            return f"{stock} units available."
        return "Out of stock."
    except ValueError as e:
        return f"Error: {e}"
    except Exception as e:
        return f"Unexpected error checking stock for product {product_id}: {e}"