GithubProxy/ChatGPT-Next-Web
1
0
Fork 1
mirror of https://github.com/ChatGPTNextWeb/ChatGPT-Next-Web.git synced 2025-05-31 09:50:35 +09:00
Code Issues Packages Projects Releases Wiki Activity

feat: dall-e plugin upgrade

Browse Source
This commit is contained in:
Hk-Gosuto 2023-12-29 18:34:38 +08:00
parent 568ffede5c
commit 4551abdce8
1 changed files with 33 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -41,20 +41,32 @@ export class DallEAPIWrapper extends StructuredTool {
prompt: z prompt: z
.string() .string()
.describe( .describe(
'input must be a english prompt. you can set `quality: "hd"` for enhanced detail.', "A text description of the desired image(s). input must be a english prompt.",
), ),
size: z size: z
.enum(["1024x1024", "1024x1792", "1792x1024"]) .enum(["1024x1024", "1024x1792", "1792x1024"])
.default("1024x1024") .default("1024x1024")
.describe("images size"), .describe("images size"),
quality: z
.enum(["standard", "hd"])
.default("standard")
.describe(
"hd increases image detail and clarity at the cost of doubled consumption, and should not be used unless specified by the user.",
),
style: z
.enum(["vivid", "natural"])
.default("vivid")
.describe(
"vivid leads to the creation of more intense and dramatic images, while Natural results in images that look more realistic and less exaggerated.",
),
}); });
/** @ignore */ /** @ignore */
async _call({ prompt, size }: z.infer<typeof this.schema>) { async _call({ prompt, size, quality, style }: z.infer<typeof this.schema>) {
let imageUrl; let imageUrl;
const apiUrl = `${this.baseURL}/images/generations`; const apiUrl = `${this.baseURL}/images/generations`;
try { try {
const requestOptions = { let requestOptions = {
method: "POST", method: "POST",
headers: { headers: {
"Content-Type": "application/json", "Content-Type": "application/json",
@ -65,8 +77,25 @@ export class DallEAPIWrapper extends StructuredTool {
prompt: prompt, prompt: prompt,
n: this.n, n: this.n,
size: size, size: size,
quality: quality,
style: style,
}), }),
}; };
if (this.model != "dall-e-3") {
requestOptions = {
method: "POST",
headers: {
"Content-Type": "application/json",
Authorization: `Bearer ${this.apiKey}`,
},
body: JSON.stringify({
model: this.model,
prompt: prompt,
n: this.n,
size: size,
}),
};
}
console.log(requestOptions); console.log(requestOptions);
const response = await fetch(apiUrl, requestOptions); const response = await fetch(apiUrl, requestOptions);
const json = await response.json(); const json = await response.json();
@ -98,5 +127,5 @@ export class DallEAPIWrapper extends StructuredTool {
} }
} }
description = `openai's dall-e 3 image generator.`; description = `openai's dall-e image generator.`;
} }