apunkt/rowboat
1
0
Fork 0
mirror of https://github.com/rowboatlabs/rowboat.git synced 2026-06-12 19:55:19 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

should the image generation tool only if google api key is present

Browse source
This commit is contained in:
arkml 2025-09-16 13:35:01 +05:30
parent cbcc1aa8a6
commit a246d2550e
1 changed files with 22 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1,30 +1,35 @@
import { WorkflowTemplate } from "./types/workflow_types";
import { WorkflowTemplate, WorkflowTool } from "./types/workflow_types";
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 tools: z.infer<typeof WorkflowTool>[] = [];
// Only include the image generation tool if GOOGLE_API_KEY is present
if (process.env.GOOGLE_API_KEY) {
tools.push({
name: "Generate Image",
description: "Generate an image using Google Gemini given a text prompt. Returns base64-encoded image data and any text parts.",
isGeminiImage: true,
parameters: {
type: 'object',
properties: {
prompt: { type: 'string', description: 'Text prompt describing the image to generate' },
modelName: { type: 'string', description: 'Optional Gemini model override' },
},
required: ['prompt'],
additionalProperties: true,
},
});
}
const defaultTemplate: z.infer<typeof WorkflowTemplate> = {
name: 'Blank Template',
description: 'A blank canvas to build your assistant.',
startAgent: "",
agents: [],
prompts: [],
tools: [
{
name: "Generate Image",
description: "Generate an image using Google Gemini given a text prompt. Returns base64-encoded image data and any text parts.",
isGeminiImage: true,
parameters: {
type: 'object',
properties: {
prompt: { type: 'string', description: 'Text prompt describing the image to generate' },
modelName: { type: 'string', description: 'Optional Gemini model override' },
},
required: ['prompt'],
additionalProperties: true,
},
},
],
tools,
pipelines: [],
};