Function get_population

contributing/samples/tools/parallel_functions/agent.py:174–211

Get population information for multiple cities. Args: cities: A list of city names. Returns: A dictionary with population data for each city.

(cities: List[str], tool_context: ToolContext)

async def get_population(cities: List[str], tool_context: ToolContext) -> dict:
  """Get population information for multiple cities.

  Args:
    cities: A list of city names.

  Returns:
    A dictionary with population data for each city.
  """
  # Simulate async processing time proportional to number of cities (non-blocking)
  await asyncio.sleep(len(cities) * 0.5)

  # Mock population data
  populations = {
      'New York': 8336817,
      'London': 9648110,
      'Tokyo': 13960000,
      'San Francisco': 873965,
      'Paris': 2161000,
      'Sydney': 5312163,
  }

  results = {}
  for city in cities:
    results[city] = populations.get(city, 1000000)  # default 1M if not found

  # Store in context for testing thread safety
  if 'population_requests' not in tool_context.state:
    tool_context.state['population_requests'] = []
  tool_context.state['population_requests'].append(
      {'cities': cities, 'results': results}
  )

  return {
      'populations': results,
      'total_population': sum(results.values()),
      'cities_count': len(cities),
  }


root_agent = Agent(