优化绘图接口

LLM-Red-Team · Apr 25, 2024 · 64e43ff · 64e43ff
1 parent 50083ac
commit 64e43ff
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/api/controllers/chat.ts b/src/api/controllers/chat.ts
@@ -379,8 +379,7 @@ async function generateImages(
   return (async () => {
     logger.info(prompt);
     const messages = [
-
-      { role: "user", content: prompt },
+      { role: "user", content: prompt.indexOf('画') == -1 ? `请画：${prompt}` : prompt },
     ];
     // 请求流
     const token = await acquireToken(refreshToken);