diff --git a/src/services/ai/tools.ts b/src/services/ai/tools.ts index d772816..f4de906 100644 --- a/src/services/ai/tools.ts +++ b/src/services/ai/tools.ts @@ -121,6 +121,20 @@ const tools: FunctionDeclaration[] = [ required: [], }, }, + { + name: "generateImage", + description: "Generates an image or a picture based on the prompt", + parametersJsonSchema: { + type: "object", + properties: { + prompt: { + type: "string", + description: "a very detailed prompt to generate an image", + }, + }, + required: ["prompt"], + }, + }, ]; const toolFunctions: AIToolFunction[] = [ @@ -276,6 +290,16 @@ const toolFunctions: AIToolFunction[] = [ }; }, }, + { + name: "generateImage", + function: (matrix, args) => { + matrix.client.sendTextMessage(matrix.roomId, `!img ${args.prompt}`); + + return { + message: "trying to generate...", + }; + }, + }, ]; export { tools, toolFunctions };