Skip to content

Commit c8308d2

Browse files
committed
cleanup
1 parent 52336ca commit c8308d2

File tree

1 file changed

+1
-24
lines changed

1 file changed

+1
-24
lines changed

cli/src/hooks/use-gravity-ad.ts

Lines changed: 1 addition & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -44,25 +44,6 @@ const extractTextFromMessageContent = (content: Message['content']): string => {
4444
.join('\n')
4545
}
4646

47-
/**
48-
* Convert LLM message history to ad API format.
49-
* Includes all message types (user, assistant, tool, system) for analytics.
50-
*/
51-
const convertToAdMessages = (messages: Message[]): AdMessage[] => {
52-
const adMessages: AdMessage[] = []
53-
54-
for (const message of messages.slice(-MAX_MESSAGES_FOR_AD)) {
55-
const textContent = extractTextFromMessageContent(message.content)
56-
if (textContent) {
57-
// Map all roles to user/assistant as required by the API
58-
const role = message.role === 'user' ? 'user' : 'assistant'
59-
adMessages.push({ role, content: textContent })
60-
}
61-
}
62-
63-
return adMessages
64-
}
65-
6647
export type GravityAdState = {
6748
ad: AdResponse | null
6849
isLoading: boolean
@@ -99,9 +80,6 @@ export const useGravityAd = (): GravityAdState => {
9980
// Has the first ad been fetched?
10081
const isStartedRef = useRef<boolean>(false)
10182

102-
// Get runState from chat store
103-
const runState = useChatStore((state) => state.runState)
104-
10583
// Fire impression via web API when ad changes (grants credits)
10684
useEffect(() => {
10785
if (ad?.impUrl && !impressionFiredRef.current.has(ad.impUrl)) {
@@ -176,10 +154,9 @@ export const useGravityAd = (): GravityAdState => {
176154
const currentRunState = useChatStore.getState().runState
177155
const messageHistory =
178156
currentRunState?.sessionState?.mainAgentState?.messageHistory ?? []
179-
const adMessages = convertToAdMessages(messageHistory)
180157

181158
logger.info(
182-
{ messageCount: adMessages.length },
159+
{ messageCount: messageHistory.length },
183160
'[gravity] Fetching ad from web API',
184161
)
185162

0 commit comments

Comments
 (0)