Skip to content

Commit 5ec8635

Browse files
committed
add venice key to image plugin
1 parent 7ad8141 commit 5ec8635

File tree

2 files changed

+10
-3
lines changed

2 files changed

+10
-3
lines changed

packages/plugin-image-generation/src/environment.ts

+7-2
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@ export const imageGenEnvSchema = z
88
HEURIST_API_KEY: z.string().optional(),
99
FAL_API_KEY: z.string().optional(),
1010
OPENAI_API_KEY: z.string().optional(),
11+
VENICE_API_KEY: z.string().optional(),
1112
})
1213
.refine(
1314
(data) => {
@@ -16,12 +17,13 @@ export const imageGenEnvSchema = z
1617
data.TOGETHER_API_KEY ||
1718
data.HEURIST_API_KEY ||
1819
data.FAL_API_KEY ||
19-
data.OPENAI_API_KEY
20+
data.OPENAI_API_KEY ||
21+
data.VENICE_API_KEY
2022
);
2123
},
2224
{
2325
message:
24-
"At least one of ANTHROPIC_API_KEY, TOGETHER_API_KEY, HEURIST_API_KEY, FAL_API_KEY or OPENAI_API_KEY is required",
26+
"At least one of ANTHROPIC_API_KEY, TOGETHER_API_KEY, HEURIST_API_KEY, FAL_API_KEY, OPENAI_API_KEY or VENICE_API_KEY is required",
2527
}
2628
);
2729

@@ -46,6 +48,9 @@ export async function validateImageGenConfig(
4648
OPENAI_API_KEY:
4749
runtime.getSetting("OPENAI_API_KEY") ||
4850
process.env.OPENAI_API_KEY,
51+
VENICE_API_KEY:
52+
runtime.getSetting("VENICE_API_KEY") ||
53+
process.env.VENICE_API_KEY,
4954
};
5055

5156
return imageGenEnvSchema.parse(config);

packages/plugin-image-generation/src/index.ts

+3-1
Original file line numberDiff line numberDiff line change
@@ -84,13 +84,15 @@ const imageGeneration: Action = {
8484
const heuristApiKeyOk = !!runtime.getSetting("HEURIST_API_KEY");
8585
const falApiKeyOk = !!runtime.getSetting("FAL_API_KEY");
8686
const openAiApiKeyOk = !!runtime.getSetting("OPENAI_API_KEY");
87+
const veniceApiKeyOk = !!runtime.getSetting("VENICE_API_KEY");
8788

8889
return (
8990
anthropicApiKeyOk ||
9091
togetherApiKeyOk ||
9192
heuristApiKeyOk ||
9293
falApiKeyOk ||
93-
openAiApiKeyOk
94+
openAiApiKeyOk ||
95+
veniceApiKeyOk
9496
);
9597
},
9698
handler: async (

0 commit comments

Comments
 (0)