remove exa from byok

Author: icecrasher321

apps/sim/app/api/tools/search/route.ts

@@ -1,6 +1,5 @@
 import { type NextRequest, NextResponse } from 'next/server'
 import { z } from 'zod'
-import { getBYOKKey } from '@/lib/api-key/byok'
 import { checkHybridAuth } from '@/lib/auth/hybrid'
 import { SEARCH_TOOL_COST } from '@/lib/billing/constants'
 import { env } from '@/lib/core/config/env'
@@ -11,7 +10,6 @@ const logger = createLogger('search')
 
 const SearchRequestSchema = z.object({
   query: z.string().min(1),
-  workspaceId: z.string().optional(),
 })
 
 export const maxDuration = 60
@@ -41,17 +39,7 @@ export async function POST(request: NextRequest) {
     const body = await request.json()
     const validated = SearchRequestSchema.parse(body)
 
-    let exaApiKey = env.EXA_API_KEY
-    let isBYOK = false
-
-    if (validated.workspaceId) {
-      const byokResult = await getBYOKKey(validated.workspaceId, 'exa')
-      if (byokResult) {
-        exaApiKey = byokResult.apiKey
-        isBYOK = true
-        logger.info(`[${requestId}] Using workspace BYOK key for Exa search`)
-      }
-    }
+    const exaApiKey = env.EXA_API_KEY
 
     if (!exaApiKey) {
       logger.error(`[${requestId}] No Exa API key available`)
@@ -64,7 +52,6 @@ export async function POST(request: NextRequest) {
     logger.info(`[${requestId}] Executing search`, {
       userId,
       query: validated.query,
-      isBYOK,
     })
 
     const result = await executeTool('exa_search', {
@@ -100,7 +87,7 @@ export async function POST(request: NextRequest) {
     const cost = {
       input: 0,
       output: 0,
-      total: isBYOK ? 0 : SEARCH_TOOL_COST,
+      total: SEARCH_TOOL_COST,
       tokens: {
         input: 0,
         output: 0,
@@ -119,7 +106,6 @@ export async function POST(request: NextRequest) {
       userId,
       resultCount: results.length,
       cost: cost.total,
-      isBYOK,
     })
 
     return NextResponse.json({

apps/sim/app/api/workspaces/[id]/byok-keys/route.ts

@@ -12,7 +12,7 @@ import { getUserEntityPermissions } from '@/lib/workspaces/permissions/utils'
 
 const logger = createLogger('WorkspaceBYOKKeysAPI')
 
-const VALID_PROVIDERS = ['openai', 'anthropic', 'google', 'mistral', 'exa'] as const
+const VALID_PROVIDERS = ['openai', 'anthropic', 'google', 'mistral'] as const
 
 const UpsertKeySchema = z.object({
   providerId: z.enum(VALID_PROVIDERS),

apps/sim/app/workspace/[workspaceId]/w/[workflowId]/hooks/use-auto-layout.ts

@@ -1,5 +1,4 @@
 import { useCallback } from 'react'
-import { useReactFlow } from 'reactflow'
 import { createLogger } from '@/lib/logs/console/logger'
 import type { AutoLayoutOptions } from '@/app/workspace/[workspaceId]/w/[workflowId]/utils/auto-layout-utils'
 import { applyAutoLayoutAndUpdateStore as applyAutoLayoutStandalone } from '@/app/workspace/[workspaceId]/w/[workflowId]/utils/auto-layout-utils'
@@ -8,36 +7,45 @@ export type { AutoLayoutOptions }
 
 const logger = createLogger('useAutoLayout')
 
+interface UseAutoLayoutOptions {
+  fitView?: (options?: { padding?: number; duration?: number }) => void
+}
+
 /**
- * Hook providing auto-layout functionality for workflows
- * Binds workflowId context and provides memoized callback for React components
- * Includes automatic fitView animation after successful layout
+ * Hook providing auto-layout functionality for workflows.
+ * Binds workflowId context and provides memoized callback for React components.
+ * Optionally accepts a fitView function to animate after successful layout.
+ *
+ * @param workflowId - The workflow ID to apply layout to
+ * @param options - Optional configuration including fitView function from useReactFlow
  */
-export function useAutoLayout(workflowId: string | null) {
-  const { fitView } = useReactFlow()
+export function useAutoLayout(workflowId: string | null, options: UseAutoLayoutOptions = {}) {
+  const { fitView } = options
 
   const applyAutoLayoutAndUpdateStore = useCallback(
-    async (options: AutoLayoutOptions = {}) => {
+    async (layoutOptions: AutoLayoutOptions = {}) => {
       if (!workflowId) {
         return { success: false, error: 'No workflow ID provided' }
       }
-      return applyAutoLayoutStandalone(workflowId, options)
+      return applyAutoLayoutStandalone(workflowId, layoutOptions)
     },
     [workflowId]
   )
 
   /**
-   * Applies auto-layout and animates to fit all blocks in view
+   * Applies auto-layout and optionally animates to fit all blocks in view
    */
   const handleAutoLayout = useCallback(async () => {
     try {
       const result = await applyAutoLayoutAndUpdateStore()
 
       if (result.success) {
         logger.info('Auto layout completed successfully')
-        requestAnimationFrame(() => {
-          fitView({ padding: 0.8, duration: 600 })
-        })
+        if (fitView) {
+          requestAnimationFrame(() => {
+            fitView({ padding: 0.8, duration: 600 })
+          })
+        }
       } else {
         logger.error('Auto layout failed:', result.error)
       }

apps/sim/app/workspace/[workspaceId]/w/[workflowId]/workflow.tsx

@@ -198,7 +198,9 @@ const WorkflowContent = React.memo(() => {
     return resizeLoopNodes(updateNodeDimensions)
   }, [resizeLoopNodes, updateNodeDimensions])
 
-  const { handleAutoLayout: autoLayoutWithFitView } = useAutoLayout(activeWorkflowId || null)
+  const { handleAutoLayout: autoLayoutWithFitView } = useAutoLayout(activeWorkflowId || null, {
+    fitView,
+  })
 
   const isWorkflowEmpty = useMemo(() => Object.keys(blocks).length === 0, [blocks])
 

apps/sim/app/workspace/[workspaceId]/w/components/sidebar/components/settings-modal/components/byok/byok.tsx

@@ -13,7 +13,7 @@ import {
   ModalHeader,
   Trash,
 } from '@/components/emcn'
-import { AnthropicIcon, ExaAIIcon, GeminiIcon, MistralIcon, OpenAIIcon } from '@/components/icons'
+import { AnthropicIcon, GeminiIcon, MistralIcon, OpenAIIcon } from '@/components/icons'
 import { Skeleton } from '@/components/ui'
 import { createLogger } from '@/lib/logs/console/logger'
 import {
@@ -61,13 +61,6 @@ const PROVIDERS: {
     description: 'LLM calls and Knowledge Base OCR',
     placeholder: 'Enter your API key',
   },
-  {
-    id: 'exa',
-    name: 'Exa',
-    icon: ExaAIIcon,
-    description: 'Web Search block',
-    placeholder: 'Enter your API key',
-  },
 ]
 
 function BYOKKeySkeleton() {

apps/sim/hooks/queries/byok-keys.ts

@@ -4,7 +4,7 @@ import { API_ENDPOINTS } from '@/stores/constants'
 
 const logger = createLogger('BYOKKeysQueries')
 
-export type BYOKProviderId = 'openai' | 'anthropic' | 'google' | 'mistral' | 'exa'
+export type BYOKProviderId = 'openai' | 'anthropic' | 'google' | 'mistral'
 
 export interface BYOKKey {
   id: string

apps/sim/tools/search/tool.ts

@@ -25,7 +25,6 @@ export const searchTool: ToolConfig<SearchParams, SearchResponse> = {
     }),
     body: (params) => ({
       query: params.query,
-      workspaceId: params._context?.workspaceId,
     }),
   },
 

apps/sim/tools/search/types.ts

@@ -2,11 +2,6 @@ import type { ToolResponse } from '@/tools/types'
 
 export interface SearchParams {
   query: string
-  _context?: {
-    workflowId?: string
-    workspaceId?: string
-    executionId?: string
-  }
 }
 
 export interface SearchResponse extends ToolResponse {