构建协作智能体团队¶

// Stub tool functions — in a real agent these call external services.
func getWeather(_ agent.Context, _ struct{ City string }) (string, error) {
    return "Sunny, 22°C", nil
}

func searchFlights(_ agent.Context, _ struct{ Origin, Destination string }) (string, error) {
    return "3 flights found", nil
}

func bookFlight(_ agent.Context, _ struct{ FlightID string }) (string, error) {
    return "Flight booked", nil
}

// newCollaborativeTeam builds a coordinator agent with two subagents, each
// configured with a different collaboration mode. This is the Go equivalent of:
//
//  weather_agent = Agent(name="weather_checker", mode="single_turn", ...)
//  flight_agent  = Agent(name="flight_booker",   mode="task",        ...)
//  root = Agent(name="travel_planner", sub_agents=[weather_agent, flight_agent])
func newCollaborativeTeam(ctx context.Context) (agent.Agent, error) {
    model, err := gemini.NewModel(ctx, "gemini-flash-latest", &genai.ClientConfig{})
    if err != nil {
        return nil, err
    }

    getWeatherTool, err := functiontool.New(functiontool.Config{
        Name:        "get_weather",
        Description: "Returns the current weather for a city.",
    }, getWeather)
    if err != nil {
        return nil, err
    }

    searchFlightsTool, err := functiontool.New(functiontool.Config{
        Name:        "search_flights",
        Description: "Searches for available flights between two airports.",
    }, searchFlights)
    if err != nil {
        return nil, err
    }

    bookFlightTool, err := functiontool.New(functiontool.Config{
        Name:        "book_flight",
        Description: "Books a specific flight by ID.",
    }, bookFlight)
    if err != nil {
        return nil, err
    }

    // weatherAgent runs in ModeSingleTurn: no user interaction, executes one
    // turn and returns automatically. Equivalent to mode="single_turn" in Python.
    weatherAgent, err := llmagent.New(llmagent.Config{
        Name:        "weather_checker",
        Model:       model,
        Mode:        llmagent.ModeSingleTurn,
        Description: "Checks the current weather for a given city.",
        Instruction: "Use the get_weather tool to look up the current weather.",
        Tools:       []tool.Tool{getWeatherTool},
    })
    if err != nil {
        return nil, err
    }

    // flightAgent runs in ModeTask: may ask the user clarifying questions and
    // automatically returns control to the coordinator when done. Equivalent to
    // mode="task" in Python.
    flightAgent, err := llmagent.New(llmagent.Config{
        Name:        "flight_booker",
        Model:       model,
        Mode:        llmagent.ModeTask,
        Description: "Searches for and books flights.",
        Instruction: "Help the user find and book a flight using the available tools.",
        Tools:       []tool.Tool{searchFlightsTool, bookFlightTool},
    })
    if err != nil {
        return nil, err
    }

    // The coordinator agent declares SubAgents. ADK automatically generates
    // weather_checker and flight_booker delegation tools, named after each
    // subagent, so the coordinator can delegate work to each one.
    return llmagent.New(llmagent.Config{
        Name:        "travel_planner",
        Model:       model,
        Description: "Coordinator agent that delegates to weather and flight subagents.",
        Instruction: "Help the user plan their trip. Use the weather checker and flight booker as needed.",
        SubAgents:   []agent.Agent{weatherAgent, flightAgent},
    })
}