diff --git a/packages/data-provider/src/data-service.ts b/packages/data-provider/src/data-service.ts index 337a0af894..3c2ee03509 100644 --- a/packages/data-provider/src/data-service.ts +++ b/packages/data-provider/src/data-service.ts @@ -697,8 +697,10 @@ export function getMessagesByConvoId(conversationId: string): Promise { - return request.get(endpoints.conversationCosts(conversationId)); +export function getModelCosts( + modelHistory: Array<{ model: string; endpoint: string }>, +): Promise { + return request.post(endpoints.costs(), { modelHistory }); } export function getPrompt(id: string): Promise<{ prompt: t.TPrompt }> { diff --git a/packages/data-provider/src/types.ts b/packages/data-provider/src/types.ts index caf429a820..a172edde07 100644 --- a/packages/data-provider/src/types.ts +++ b/packages/data-provider/src/types.ts @@ -655,16 +655,13 @@ export type TBalanceResponse = { }; export type TConversationCosts = { - conversationId: string; totals: { prompt: { usd: number; tokenCount: number }; completion: { usd: number; tokenCount: number }; total: { usd: number; tokenCount: number }; }; - perMessage: Array<{ - messageId: string; - tokenType: 'prompt' | 'completion'; - tokenCount: number; - usd: number; - }>; +}; + +export type TModelCosts = { + modelCostTable: Record; };