feat: dalle support model list

TBXark · Jan 16, 2025 · 0129727 · 0129727
1 parent 3b75afd
commit 0129727
Show file tree

Hide file tree

Showing 4 changed files with 41 additions and 24 deletions.
diff --git a/dist/buildinfo.json b/dist/buildinfo.json
diff --git a/dist/index.js b/dist/index.js
diff --git a/packages/lib/core/src/agent/openai.ts b/packages/lib/core/src/agent/openai.ts
@@ -13,7 +13,7 @@ import type {
 } from './types';
 import { ImageSupportFormat, loadOpenAIModelList, renderOpenAIMessages } from '#/agent/openai_compatibility';
 import { requestChatCompletions } from './request';
-import { bearerHeader, convertStringToResponseMessages, getAgentUserConfigFieldName } from './utils';
+import {bearerHeader, convertStringToResponseMessages, getAgentUserConfigFieldName, loadModelsList} from './utils';
 
 function openAIApiKey(context: AgentUserConfig): string {
     const length = context.OPENAI_API_KEY.length;
@@ -48,7 +48,7 @@ export class Dalle implements ImageAgent {
 
     readonly enable: AgentEnable = ctx => ctx.OPENAI_API_KEY.length > 0;
     readonly model: AgentModel = ctx => ctx.DALL_E_MODEL;
-    readonly modelList: AgentModelList = ctx => Promise.resolve([ctx.DALL_E_MODEL]);
+    readonly modelList: AgentModelList = ctx => loadModelsList(ctx.DALL_E_MODEL);
 
     readonly request: ImageAgentRequest = async (prompt: string, context: AgentUserConfig): Promise<string | Blob> => {
         const url = `${context.OPENAI_API_BASE}/images/generations`;

diff --git a/packages/lib/core/src/config/version.ts b/packages/lib/core/src/config/version.ts
@@ -1,2 +1,2 @@
-export const BUILD_TIMESTAMP = 1735091598;
-export const BUILD_VERSION = '2b81827';
+export const BUILD_TIMESTAMP = 1736996817;
+export const BUILD_VERSION = '3b75afd';