diff --git a/src/backend/src/services/ai/chat/providers/TogetherAiProvider/TogetherAIProvider.ts b/src/backend/src/services/ai/chat/providers/TogetherAiProvider/TogetherAIProvider.ts
index 1ff4e2f7b..f11d97cb0 100644
--- a/src/backend/src/services/ai/chat/providers/TogetherAiProvider/TogetherAIProvider.ts
+++ b/src/backend/src/services/ai/chat/providers/TogetherAiProvider/TogetherAIProvider.ts
@@ -25,8 +25,8 @@ import * as OpenAIUtil from '../../../utils/OpenAIUtil.js';
 import { IChatModel, IChatProvider, ICompleteArguments } from '../types.js';
 
 const TOGETHER_AI_CHAT_COST_MAP = {
-    'input': 'prompt_tokens',
-    'output': 'completion_tokens',
+    prompt_tokens: 'input',
+    completion_tokens: 'output',
 };
 
 export class TogetherAIProvider implements IChatProvider {