apunkt/rowboat
1
0
Fork 0
mirror of https://github.com/rowboatlabs/rowboat.git synced 2026-05-10 15:52:38 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

make generate_image conditionally show up

Browse source
This commit is contained in:
arkml 2025-09-16 14:59:00 +05:30
parent cdfa7daf98
commit 70cbd8beea
2 changed files with 7 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
apps/rowboat/app/lib/project_templates.ts
View file

@ -3,13 +3,15 @@ import { z } from 'zod';
// Provide a minimal default template to satisfy legacy code paths that
// still reference `templates.default`. Real templates are DB-backed.
const includeGeminiImageTool = !!process.env.GOOGLE_API_KEY;
const defaultTemplate: z.infer<typeof WorkflowTemplate> = {
name: 'Blank Template',
description: 'A blank canvas to build your assistant.',
startAgent: "",
agents: [],
prompts: [],
tools: [
tools: includeGeminiImageTool ? [
{
name: "Generate Image",
description: "Generate an image using Google Gemini given a text prompt. Returns base64-encoded image data and any text parts.",
@ -24,7 +26,7 @@ const defaultTemplate: z.infer<typeof WorkflowTemplate> = {
additionalProperties: true,
},
},
],
] : [],
pipelines: [],
};

5
apps/rowboat/src/application/use-cases/projects/create-project.use-case.ts
View file

@ -95,9 +95,10 @@ export class CreateProjectUseCase implements ICreateProjectUseCase {
}
}
// Ensure the Gemini Image tool is always available by default
// Conditionally include Gemini Image tool by default if GOOGLE_API_KEY is present
const hasGoogleKey = !!process.env.GOOGLE_API_KEY;
const hasImageTool = (workflow.tools || []).some(t => t.isGeminiImage || t.name === 'Generate Image');
if (!hasImageTool) {
if (hasGoogleKey && !hasImageTool) {
const imageTool = {
name: 'Generate Image',
description: 'Generate an image using Google Gemini given a text prompt. Returns base64-encoded image data and any text parts.',