diff --git a/libraries/nestjs-libraries/src/agent/agent.graph.service.ts b/libraries/nestjs-libraries/src/agent/agent.graph.service.ts index e6c72f79..fb5195ff 100644 --- a/libraries/nestjs-libraries/src/agent/agent.graph.service.ts +++ b/libraries/nestjs-libraries/src/agent/agent.graph.service.ts @@ -15,9 +15,8 @@ import { z } from 'zod'; import { MediaService } from '@gitroom/nestjs-libraries/database/prisma/media/media.service'; import { UploadFactory } from '@gitroom/nestjs-libraries/upload/upload.factory'; import { GeneratorDto } from '@gitroom/nestjs-libraries/dtos/generator/generator.dto'; -import { IntegrationService } from '@gitroom/nestjs-libraries/database/prisma/integrations/integration.service'; -const tools = [new TavilySearchResults({ maxResults: 3 })]; +const tools = !process.env.TAVILY_API_KEY ? [] : [new TavilySearchResults({ maxResults: 3 })]; const toolNode = new ToolNode(tools); const model = new ChatOpenAI({