From 4fb9d7bdffbe3b27b255c081aff01250597130e9 Mon Sep 17 00:00:00 2001
From: Dustin Healy <dustinhealy1@gmail.com>
Date: Sat, 13 Sep 2025 12:26:39 -0700
Subject: [PATCH] refactor: swap data-service and types to use new cost table

---
 packages/data-provider/src/data-service.ts |  6 ++++--
 packages/data-provider/src/types.ts        | 11 ++++-------
 2 files changed, 8 insertions(+), 9 deletions(-)

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<s.TMessage
   return request.get(endpoints.messages({ conversationId }));
 }
 
-export function getConversationCosts(conversationId: string): Promise<t.TConversationCosts> {
-  return request.get(endpoints.conversationCosts(conversationId));
+export function getModelCosts(
+  modelHistory: Array<{ model: string; endpoint: string }>,
+): Promise<t.TModelCosts> {
+  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<string, { prompt: number; completion: number }>;
 };