mirror of
https://github.com/simstudioai/sim.git
synced 2026-09-24 15:45:35 +08:00
fix(cost): restored cost reporting for agent block in console entry (#1253)
This commit is contained in:
@@ -86,7 +86,7 @@ export async function executeProviderRequest(
|
||||
const { prompt: promptTokens = 0, completion: completionTokens = 0 } = response.tokens
|
||||
const useCachedInput = !!request.context && request.context.length > 0
|
||||
|
||||
if (shouldBillModelUsage(response.model, request.apiKey)) {
|
||||
if (shouldBillModelUsage(response.model)) {
|
||||
response.cost = calculateCost(response.model, promptTokens, completionTokens, useCachedInput)
|
||||
} else {
|
||||
response.cost = {
|
||||
|
||||
@@ -548,20 +548,11 @@ export function getHostedModels(): string[] {
|
||||
* Determine if model usage should be billed to the user
|
||||
*
|
||||
* @param model The model name
|
||||
* @param userProvidedApiKey Whether the user provided their own API key
|
||||
* @returns true if the usage should be billed to the user
|
||||
*/
|
||||
export function shouldBillModelUsage(model: string, userProvidedApiKey?: string): boolean {
|
||||
export function shouldBillModelUsage(model: string): boolean {
|
||||
const hostedModels = getHostedModels()
|
||||
if (!hostedModels.includes(model)) {
|
||||
return false
|
||||
}
|
||||
|
||||
if (userProvidedApiKey && userProvidedApiKey.trim() !== '') {
|
||||
return false
|
||||
}
|
||||
|
||||
return true
|
||||
return hostedModels.includes(model)
|
||||
}
|
||||
|
||||
/**
|
||||
|
||||
Reference in New Issue
Block a user