xinyin025/ChatGPT-Next-Web
1
0
Fork 0
mirror of https://github.com/ChatGPTNextWeb/ChatGPT-Next-Web.git synced 2025-11-13 20:53:45 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

feat: support custom dalle model

Browse Source
This commit is contained in:
Hk-Gosuto
2023-12-21 12:36:57 +08:00
parent 666e62d75a
commit 4044891f85
2 changed files with 13 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
app/api/langchain-tools/dalle_image_generator.ts
View File

@@ -7,6 +7,7 @@ export class DallEAPIWrapper extends StructuredTool {
n = 1;
apiKey: string;
baseURL?: string;
model: string;
noStorage: boolean;
@@ -26,6 +27,7 @@ export class DallEAPIWrapper extends StructuredTool {
this.callback = callback;
this.noStorage = !!process.env.DALLE_NO_IMAGE_STORAGE;
this.model = process.env.DALLE_MODEL ?? "dall-e-3";
}
async saveImageFromUrl(url: string) {
@@ -59,18 +61,25 @@ export class DallEAPIWrapper extends StructuredTool {
Authorization: `Bearer ${this.apiKey}`,
},
body: JSON.stringify({
model: "dall-e-3",
model: this.model,
prompt: prompt,
n: this.n,
size: size,
}),
};
console.log(requestOptions);
const response = await fetch(apiUrl, requestOptions);
const json = await response.json();
console.log("[DALL-E]", json);
imageUrl = json.data[0].url;
try {
console.log("[DALL-E]", json);
imageUrl = json.data[0].url;
} catch (e) {
if (this.callback != null) await this.callback(JSON.stringify(json));
throw e;
}
} catch (e) {
console.error("[DALL-E]", e);
return (e as Error).message;
}
if (!imageUrl) return "No image was generated";
try {