Skip to content

Commit a6da6c1

Browse files
fix: NEAR AI image generation
1 parent 2c06f56 commit a6da6c1

File tree

1 file changed

+24
-0
lines changed

1 file changed

+24
-0
lines changed

packages/core/src/generation.ts

+24
Original file line numberDiff line numberDiff line change
@@ -2111,6 +2111,30 @@ export const generateImage = async (
21112111
console.error(error);
21122112
return { success: false, error: error };
21132113
}
2114+
} else if (runtime.imageModelProvider === ModelProviderName.NEARAI) {
2115+
let targetSize = `${data.width}x${data.height}`;
2116+
if (
2117+
targetSize !== "1024x1024" &&
2118+
targetSize !== "1792x1024" &&
2119+
targetSize !== "1024x1792"
2120+
) {
2121+
targetSize = "1024x1024";
2122+
}
2123+
const openai = new OpenAI({
2124+
baseURL: settings.NEARAI_API_URL || "https://api.near.ai/v1",
2125+
apiKey,
2126+
});
2127+
const response = await openai.images.generate({
2128+
model,
2129+
prompt: data.prompt,
2130+
size: targetSize as "1024x1024" | "1792x1024" | "1024x1792",
2131+
n: data.count,
2132+
response_format: "b64_json",
2133+
});
2134+
const base64s = response.data.map(
2135+
(image) => `data:image/png;base64,${image.b64_json}`
2136+
);
2137+
return { success: true, data: base64s };
21142138
} else {
21152139
let targetSize = `${data.width}x${data.height}`;
21162140
if (

0 commit comments

Comments
 (0)