AutoGen笔记：02.工具使用

poetry add python-dotenv
poetry add autogen-agentchat  
poetry add "autogen-ext[openai]"

def vacation_destination(city: str) -> tuple[str, str]:
    """
    Checks if a specific vacation destination is available

    Args:
        city (str): Name of the city to check

    Returns:
        tuple: Contains city name and availability status ('Available' or 'Unavailable')
    """
    destinations = {
        "Barcelona": "Available",
        "Tokyo": "Unavailable",
        "Cape Town": "Available",
        "Vancouver": "Available",
        "Dubai": "Unavailable",
    }

    if city in destinations:
        return city, destinations[city]
    else:
        return city, "City not found"

from autogen_core.tools import FunctionTool

get_vacations = FunctionTool(
    vacation_destination,
    description="Search for vacation destinations and if they are availabe or not.",
)

from autogen_agentchat.agents import AssistantAgent

agent = AssistantAgent(
    name="assistant",
    model_client=client,
    tools=[get_vacations],
    system_message="You are a travel agent that helps users find vacation destinations.",
    reflect_on_tool_use=True,
)

import asyncio
from dotenv import load_dotenv
from autogen_ext.models.openai import OpenAIChatCompletionClient
from autogen_core.tools import FunctionTool
from autogen_agentchat.agents import AssistantAgent
from autogen_agentchat.messages import TextMessage
from autogen_core import CancellationToken

load_dotenv()
client = OpenAIChatCompletionClient(model="gpt-4o")


def vacation_destination(city: str) -> tuple[str, str]:
    """
    Checks if a specific vacation destination is available

    Args:
        city (str): Name of the city to check

    Returns:
        tuple: Contains city name and availability status ('Available' or 'Unavailable')
    """
    destinations = {
        "Barcelona": "Available",
        "Tokyo": "Unavailable",
        "Cape Town": "Available",
        "Vancouver": "Available",
        "Dubai": "Unavailable",
    }

    if city in destinations:
        return city, destinations[city]
    else:
        return city, "City not found"


get_vacations = FunctionTool(
    vacation_destination,
    description="Search for vacation destinations and if they are availabe or not.",
)

agent = AssistantAgent(
    name="assistant",
    model_client=client,
    tools=[get_vacations],
    system_message="You are a travel agent that helps users find vacation destinations.",
    reflect_on_tool_use=True,
)


async def main():
    response = await agent.on_messages(
        [TextMessage(content="I would like to take a trip to Tokyo", source="user")],
        cancellation_token=CancellationToken(),
    )
    print(response.inner_messages)
    print(response.chat_message)


asyncio.run(main())

[ToolCallRequestEvent(source='assistant', models_usage=RequestUsage(prompt_tokens=78, completion_tokens=16), content=[FunctionCall(id='call_lpPgiefSjGHIF3BmduZzQcYh', arguments='{"city":"Tokyo"}', name='vacation_destination')], type='ToolCallRequestEvent'), ToolCallExecutionEvent(source='assistant', models_usage=None, content=[FunctionExecutionResult(content="('Tokyo', 'Unavailable')", call_id='call_lpPgiefSjGHIF3BmduZzQcYh', is_error=False)], type='ToolCallExecutionEvent')]
source='assistant' models_usage=RequestUsage(prompt_tokens=66, completion_tokens=62) content='It looks like detailed recommendations for Tokyo are currently unavailable. However, I’d be happy to share general advice! Tokyo is a vibrant city with a mix of ancient traditions and cutting-edge modernity. Is there something specific you’re looking for in your trip—like food, culture, shopping, or something else?' type='TextMessage'