AutoGen笔记：02.工具使用

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

 1def vacation_destination(city: str) -> tuple[str, str]:
 2    """
 3    Checks if a specific vacation destination is available
 4
 5    Args:
 6        city (str): Name of the city to check
 7
 8    Returns:
 9        tuple: Contains city name and availability status ('Available' or 'Unavailable')
10    """
11    destinations = {
12        "Barcelona": "Available",
13        "Tokyo": "Unavailable",
14        "Cape Town": "Available",
15        "Vancouver": "Available",
16        "Dubai": "Unavailable",
17    }
18
19    if city in destinations:
20        return city, destinations[city]
21    else:
22        return city, "City not found"

1from autogen_core.tools import FunctionTool
2
3get_vacations = FunctionTool(
4    vacation_destination,
5    description="Search for vacation destinations and if they are availabe or not.",
6)

1from autogen_agentchat.agents import AssistantAgent
2
3agent = AssistantAgent(
4    name="assistant",
5    model_client=client,
6    tools=[get_vacations],
7    system_message="You are a travel agent that helps users find vacation destinations.",
8    reflect_on_tool_use=True,
9)

 1import asyncio
 2from dotenv import load_dotenv
 3from autogen_ext.models.openai import OpenAIChatCompletionClient
 4from autogen_core.tools import FunctionTool
 5from autogen_agentchat.agents import AssistantAgent
 6from autogen_agentchat.messages import TextMessage
 7from autogen_core import CancellationToken
 8
 9load_dotenv()
10client = OpenAIChatCompletionClient(model="gpt-4o")
11
12
13def vacation_destination(city: str) -> tuple[str, str]:
14    """
15    Checks if a specific vacation destination is available
16
17    Args:
18        city (str): Name of the city to check
19
20    Returns:
21        tuple: Contains city name and availability status ('Available' or 'Unavailable')
22    """
23    destinations = {
24        "Barcelona": "Available",
25        "Tokyo": "Unavailable",
26        "Cape Town": "Available",
27        "Vancouver": "Available",
28        "Dubai": "Unavailable",
29    }
30
31    if city in destinations:
32        return city, destinations[city]
33    else:
34        return city, "City not found"
35
36
37get_vacations = FunctionTool(
38    vacation_destination,
39    description="Search for vacation destinations and if they are availabe or not.",
40)
41
42agent = AssistantAgent(
43    name="assistant",
44    model_client=client,
45    tools=[get_vacations],
46    system_message="You are a travel agent that helps users find vacation destinations.",
47    reflect_on_tool_use=True,
48)
49
50
51async def main():
52    response = await agent.on_messages(
53        [TextMessage(content="I would like to take a trip to Tokyo", source="user")],
54        cancellation_token=CancellationToken(),
55    )
56    print(response.inner_messages)
57    print(response.chat_message)
58
59
60asyncio.run(main())

1[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')]
2source='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'