Class AgentCostTotalQueryResultMapper

java.lang.Object

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

All Implemented Interfaces:

DashboardComponentQueryResultMapper

public class AgentCostTotalQueryResultMapper
extends Object
implements DashboardComponentQueryResultMapper

Computes the single total cost across all LLM models 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 per-model costs are summed into one value.

Reads the same three groupByNestedFieldFilteredWithSumField aggregations as AgentCostByModelQueryResultMapper (inputTokens, outputTokens, cachedTokens) and applies the same per-1,000,000-tokens cost arithmetic with longest-prefix model matching. Models with no matching price are skipped. Returns a SingleValueDashboardComponentData with the total cost.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
protected static class	AgentCostTotalQueryResultMapper.ModelTokens
protected static interface	AgentCostTotalQueryResultMapper.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
AgentCostTotalQueryResultMapper(AgentModelPricingService agentModelPricingService)

Method Summary

Modifier and Type	Method	Description
protected static BigDecimal	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, AgentCostTotalQueryResultMapper.ModelTokens> tokensByModel, AgentCostTotalQueryResultMapper.TokenAssigner assigner)	Reads the summed token count per model for the given nested-filtered aggregation and applies it to the accumulated AgentCostTotalQueryResultMapper.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

AgentCostTotalQueryResultMapper

public AgentCostTotalQueryResultMapper(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, AgentCostTotalQueryResultMapper.ModelTokens> tokensByModel,
 AgentCostTotalQueryResultMapper.TokenAssigner assigner)

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

computeCost

protected static BigDecimal 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.