Skip to content

Commit fc475d4

Browse files
committed
chore: better place for calculateInputTokens utility function
1 parent 4c44946 commit fc475d4

File tree

3 files changed

+10
-8
lines changed

3 files changed

+10
-8
lines changed

refact-agent/gui/src/components/ChatContent/UsageCounter/UsageCounter.tsx

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,12 @@
11
import React from "react";
22
import { Card, Flex, HoverCard, Text } from "@radix-ui/themes";
33
import { ArrowDownIcon, ArrowUpIcon } from "@radix-ui/react-icons";
4+
45
import { ScrollArea } from "../../ScrollArea";
5-
import { Usage } from "../../../services/refact";
6+
import { calculateUsageInputTokens } from "../../../utils/calculateUsageInputTokens";
7+
import type { Usage } from "../../../services/refact";
8+
69
import styles from "./UsageCounter.module.css";
7-
import { calculateInputTokens } from "./UsageCounter.utils";
810

911
type UsageCounterProps = {
1012
usage: Usage;
@@ -31,12 +33,12 @@ const TokenDisplay: React.FC<{ label: string; value: number }> = ({
3133
);
3234

3335
export const UsageCounter: React.FC<UsageCounterProps> = ({ usage }) => {
34-
const inputTokens = calculateInputTokens(usage, [
36+
const inputTokens = calculateUsageInputTokens(usage, [
3537
"prompt_tokens",
3638
"cache_creation_input_tokens",
3739
"cache_read_input_tokens",
3840
]);
39-
const outputTokens = calculateInputTokens(usage, ["completion_tokens"]);
41+
const outputTokens = calculateUsageInputTokens(usage, ["completion_tokens"]);
4042

4143
return (
4244
<HoverCard.Root>

refact-agent/gui/src/features/Chat/Thread/reducer.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ import {
4343
isToolCallMessage,
4444
validateToolCall,
4545
} from "../../../services/refact";
46-
import { calculateInputTokens } from "../../../components/ChatContent/UsageCounter/UsageCounter.utils";
46+
import { calculateUsageInputTokens } from "../../../utils/calculateUsageInputTokens";
4747

4848
const RECOMMENDED_MAXIMUM_PROMPT_TOKENS_AMOUNT = 30000;
4949

@@ -228,7 +228,7 @@ export const chatReducer = createReducer(initialState, (builder) => {
228228
const { usage } = action.payload;
229229
state.thread.usage = usage;
230230

231-
const inputTokensAmount = calculateInputTokens(usage, [
231+
const inputTokensAmount = calculateUsageInputTokens(usage, [
232232
"prompt_tokens",
233233
"cache_creation_input_tokens",
234234
"cache_read_input_tokens",

refact-agent/gui/src/components/ChatContent/UsageCounter/UsageCounter.utils.ts renamed to refact-agent/gui/src/utils/calculateUsageInputTokens.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
1-
import { Usage } from "../../../services/refact";
1+
import { Usage } from "../services/refact";
22

3-
export const calculateInputTokens = (
3+
export const calculateUsageInputTokens = (
44
usage: Usage,
55
keys: (keyof Usage)[],
66
): number =>

0 commit comments

Comments
 (0)