File tree 1 file changed +24
-0
lines changed
1 file changed +24
-0
lines changed Original file line number Diff line number Diff line change @@ -2111,6 +2111,30 @@ export const generateImage = async (
2111
2111
console . error ( error ) ;
2112
2112
return { success : false , error : error } ;
2113
2113
}
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 } ;
2114
2138
} else {
2115
2139
let targetSize = `${ data . width } x${ data . height } ` ;
2116
2140
if (
You can’t perform that action at this time.
0 commit comments