Fail loudly if cost field isn't found

Author: Theodore Li

apps/sim/tools/exa/answer.ts

@@ -1,9 +1,6 @@
-import { createLogger } from '@sim/logger'
 import type { ExaAnswerParams, ExaAnswerResponse } from '@/tools/exa/types'
 import type { ToolConfig } from '@/tools/types'
 
-const logger = createLogger('ExaAnswerTool')
-
 export const answerTool: ToolConfig<ExaAnswerParams, ExaAnswerResponse> = {
   id: 'exa_answer',
   name: 'Exa Answer',
@@ -37,14 +34,11 @@ export const answerTool: ToolConfig<ExaAnswerParams, ExaAnswerResponse> = {
     pricing: {
       type: 'custom',
       getCost: (_params, output) => {
-        // Use __costDollars from Exa API response (internal field, stripped from final output)
         const costDollars = output.__costDollars as { total?: number } | undefined
-        if (costDollars?.total != null) {
-          return { cost: costDollars.total, metadata: { costDollars } }
+        if (costDollars?.total == null) {
+          throw new Error('Exa answer response missing costDollars field')
         }
-        // Fallback: $5/1000 requests
-        logger.warn('Exa answer response missing costDollars, using fallback pricing')
-        return 0.005
+        return { cost: costDollars.total, metadata: { costDollars } }
       },
     },
     rateLimit: {

apps/sim/tools/exa/find_similar_links.ts

@@ -1,9 +1,6 @@
-import { createLogger } from '@sim/logger'
 import type { ExaFindSimilarLinksParams, ExaFindSimilarLinksResponse } from '@/tools/exa/types'
 import type { ToolConfig } from '@/tools/types'
 
-const logger = createLogger('ExaFindSimilarLinksTool')
-
 export const findSimilarLinksTool: ToolConfig<
   ExaFindSimilarLinksParams,
   ExaFindSimilarLinksResponse
@@ -86,16 +83,11 @@ export const findSimilarLinksTool: ToolConfig<
     pricing: {
       type: 'custom',
       getCost: (_params, output) => {
-        // Use __costDollars from Exa API response (internal field, stripped from final output)
         const costDollars = output.__costDollars as { total?: number } | undefined
-        if (costDollars?.total != null) {
-          return { cost: costDollars.total, metadata: { costDollars } }
+        if (costDollars?.total == null) {
+          throw new Error('Exa find_similar_links response missing costDollars field')
         }
-        // Fallback: $5/1000 (1-25 results) or $25/1000 (26-100 results)
-        logger.warn('Exa find_similar_links response missing costDollars, using fallback pricing')
-        const similarLinks = output.similarLinks as unknown[] | undefined
-        const resultCount = similarLinks?.length || 0
-        return resultCount <= 25 ? 0.005 : 0.025
+        return { cost: costDollars.total, metadata: { costDollars } }
       },
     },
     rateLimit: {

apps/sim/tools/exa/get_contents.ts

@@ -1,9 +1,6 @@
-import { createLogger } from '@sim/logger'
 import type { ExaGetContentsParams, ExaGetContentsResponse } from '@/tools/exa/types'
 import type { ToolConfig } from '@/tools/types'
 
-const logger = createLogger('ExaGetContentsTool')
-
 export const getContentsTool: ToolConfig<ExaGetContentsParams, ExaGetContentsResponse> = {
   id: 'exa_get_contents',
   name: 'Exa Get Contents',
@@ -71,15 +68,11 @@ export const getContentsTool: ToolConfig<ExaGetContentsParams, ExaGetContentsRes
     pricing: {
       type: 'custom',
       getCost: (_params, output) => {
-        // Use __costDollars from Exa API response (internal field, stripped from final output)
         const costDollars = output.__costDollars as { total?: number } | undefined
-        if (costDollars?.total != null) {
-          return { cost: costDollars.total, metadata: { costDollars } }
+        if (costDollars?.total == null) {
+          throw new Error('Exa get_contents response missing costDollars field')
         }
-        // Fallback: $1/1000 pages
-        logger.warn('Exa get_contents response missing costDollars, using fallback pricing')
-        const results = output.results as unknown[] | undefined
-        return (results?.length || 0) * 0.001
+        return { cost: costDollars.total, metadata: { costDollars } }
       },
     },
     rateLimit: {

apps/sim/tools/exa/search.ts

@@ -1,9 +1,6 @@
-import { createLogger } from '@sim/logger'
 import type { ExaSearchParams, ExaSearchResponse } from '@/tools/exa/types'
 import type { ToolConfig } from '@/tools/types'
 
-const logger = createLogger('ExaSearchTool')
-
 export const searchTool: ToolConfig<ExaSearchParams, ExaSearchResponse> = {
   id: 'exa_search',
   name: 'Exa Search',
@@ -95,22 +92,12 @@ export const searchTool: ToolConfig<ExaSearchParams, ExaSearchResponse> = {
     byokProviderId: 'exa',
     pricing: {
       type: 'custom',
-      getCost: (params, output) => {
-        // Use __costDollars from Exa API response (internal field, stripped from final output)
+      getCost: (_params, output) => {
         const costDollars = output.__costDollars as { total?: number } | undefined
-        if (costDollars?.total != null) {
-          return { cost: costDollars.total, metadata: { costDollars } }
-        }
-
-        // Fallback: estimate based on search type and result count
-        logger.warn('Exa search response missing costDollars, using fallback pricing')
-        const isDeepSearch = params.type === 'neural'
-        if (isDeepSearch) {
-          return 0.015
+        if (costDollars?.total == null) {
+          throw new Error('Exa search response missing costDollars field')
         }
-        const results = output.results as unknown[] | undefined
-        const resultCount = results?.length || 0
-        return resultCount <= 25 ? 0.005 : 0.025
+        return { cost: costDollars.total, metadata: { costDollars } }
       },
     },
     rateLimit: {