Class AgentCostByModelQueryResultMapper

java.lang.Object

com.flowable.platform.service.dashboard.resultmapper.extractor.AgentCostByModelQueryResultMapper

All Implemented Interfaces:

DashboardComponentQueryResultMapper

public class AgentCostByModelQueryResultMapper
extends Object
implements DashboardComponentQueryResultMapper

Computes the total cost per LLM model by combining the input/output/cached token sums from the _flowable_AgentInvocationTokensByModelByTypeQuery query result with the model prices stored in the configuration store for the tenant.

The query produces three groupByNestedFieldFilteredWithSumField aggregations (inputTokens, outputTokens, cachedTokens). Each is structured as /aggregations/{aggName}/{aggName}_filtered/{aggName}_inner/buckets, where the bucket key is the model name and the summed token count lives at {aggName}_sum/value.

Cost is tokens / 1_000_000 * price per token type, using the longest model-prefix match from the effective price list. Models with no matching price are skipped.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
protected static class	AgentCostByModelQueryResultMapper.ModelTokens
protected static interface	AgentCostByModelQueryResultMapper.TokenAssigner

Field Summary

Fields

Modifier and Type	Field	Description
protected final AgentModelPricingService	agentModelPricingService
protected static final String	CACHED_TOKENS_AGGREGATION
protected static final String	INPUT_TOKENS_AGGREGATION
protected static final String	OUTPUT_TOKENS_AGGREGATION
protected static final BigDecimal	TOKENS_PER_PRICE_UNIT
static final String	TYPE

Constructor Summary

Constructors

Constructor	Description
AgentCostByModelQueryResultMapper(AgentModelPricingService agentModelPricingService)

Method Summary

Modifier and Type	Method	Description
protected static double	computeCost(double inputTokens, double outputTokens, double cachedTokens, AgentModelPrice price)	Computes the cost for a single model.
protected static BigDecimal	costFor(double tokens, BigDecimal pricePerUnit)
String	getType()
DashboardComponentData	map(DashboardComponentDefinitionModel dashboardComponentDefinitionModel, Map<String, tools.jackson.databind.JsonNode> queryPayload, tools.jackson.databind.JsonNode queryResult)
DashboardComponentData	map(DashboardComponentDefinitionModel dashboardComponentDefinitionModel, Map<String, tools.jackson.databind.JsonNode> queryPayload, tools.jackson.databind.JsonNode queryResult, String tenantId)
protected static AgentModelPrice	matchPrice(String model, List<AgentModelPrice> prices)	Returns the price whose modelPrefix is the longest prefix of the given model name, or null when no configured prefix matches.
protected void	readSummedTokens(tools.jackson.databind.JsonNode queryResult, String aggregationName, Map<String, AgentCostByModelQueryResultMapper.ModelTokens> tokensByModel, AgentCostByModelQueryResultMapper.TokenAssigner assigner)	Reads the summed token count per model for the given nested-filtered aggregation and applies it to the accumulated AgentCostByModelQueryResultMapper.ModelTokens for that model.

Methods inherited from class Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Details

TYPE

public static final String TYPE

See Also:

• Constant Field Values

TOKENS_PER_PRICE_UNIT

protected static final BigDecimal TOKENS_PER_PRICE_UNIT

INPUT_TOKENS_AGGREGATION

protected static final String INPUT_TOKENS_AGGREGATION

See Also:

• Constant Field Values

OUTPUT_TOKENS_AGGREGATION

protected static final String OUTPUT_TOKENS_AGGREGATION

See Also:

• Constant Field Values

CACHED_TOKENS_AGGREGATION

protected static final String CACHED_TOKENS_AGGREGATION

See Also:

• Constant Field Values

agentModelPricingService

protected final AgentModelPricingService agentModelPricingService

Constructor Details

AgentCostByModelQueryResultMapper

public AgentCostByModelQueryResultMapper(AgentModelPricingService agentModelPricingService)

Method Details

getType

public String getType()

Specified by:

getType in interface DashboardComponentQueryResultMapper

map

public DashboardComponentData map(DashboardComponentDefinitionModel dashboardComponentDefinitionModel,
 Map<String, tools.jackson.databind.JsonNode> queryPayload,
 tools.jackson.databind.JsonNode queryResult)

Specified by:

map in interface DashboardComponentQueryResultMapper

map

public DashboardComponentData map(DashboardComponentDefinitionModel dashboardComponentDefinitionModel,
 Map<String, tools.jackson.databind.JsonNode> queryPayload,
 tools.jackson.databind.JsonNode queryResult,
 String tenantId)

Specified by:

map in interface DashboardComponentQueryResultMapper

readSummedTokens

protected void readSummedTokens(tools.jackson.databind.JsonNode queryResult,
 String aggregationName,
 Map<String, AgentCostByModelQueryResultMapper.ModelTokens> tokensByModel,
 AgentCostByModelQueryResultMapper.TokenAssigner assigner)

Reads the summed token count per model for the given nested-filtered aggregation and applies it to the accumulated AgentCostByModelQueryResultMapper.ModelTokens for that model. Mirrors groupByNestedFieldFilteredWithSumField's result structure.

computeCost

protected static double computeCost(double inputTokens,
 double outputTokens,
 double cachedTokens,
 AgentModelPrice price)

Computes the cost for a single model. Each token type is priced per 1,000,000 tokens; a null price for a token type contributes zero.

costFor

protected static BigDecimal costFor(double tokens,
 BigDecimal pricePerUnit)

matchPrice

protected static AgentModelPrice matchPrice(String model,
 List<AgentModelPrice> prices)

Returns the price whose modelPrefix is the longest prefix of the given model name, or null when no configured prefix matches.