markdown-ai-rewriter 0.3.3 → 0.4.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +82 -0
- package/dist/cli.js +39 -1
- package/dist/cli.js.map +1 -1
- package/dist/core/image-processor.d.ts +32 -0
- package/dist/core/image-processor.d.ts.map +1 -0
- package/dist/core/image-processor.js +229 -0
- package/dist/core/image-processor.js.map +1 -0
- package/dist/core/image-providers/doubao.d.ts +12 -0
- package/dist/core/image-providers/doubao.d.ts.map +1 -0
- package/dist/core/image-providers/doubao.js +73 -0
- package/dist/core/image-providers/doubao.js.map +1 -0
- package/dist/core/image-providers/fal.d.ts +25 -0
- package/dist/core/image-providers/fal.d.ts.map +1 -0
- package/dist/core/image-providers/fal.js +79 -0
- package/dist/core/image-providers/fal.js.map +1 -0
- package/dist/core/image-providers/gemini.d.ts +11 -0
- package/dist/core/image-providers/gemini.d.ts.map +1 -0
- package/dist/core/image-providers/gemini.js +57 -0
- package/dist/core/image-providers/gemini.js.map +1 -0
- package/dist/core/image-providers/index.d.ts +6 -0
- package/dist/core/image-providers/index.d.ts.map +1 -0
- package/dist/core/image-providers/index.js +20 -0
- package/dist/core/image-providers/index.js.map +1 -0
- package/dist/core/image-providers/minimax.d.ts +20 -0
- package/dist/core/image-providers/minimax.d.ts.map +1 -0
- package/dist/core/image-providers/minimax.js +89 -0
- package/dist/core/image-providers/minimax.js.map +1 -0
- package/dist/core/image-providers/nano-banana.d.ts +36 -0
- package/dist/core/image-providers/nano-banana.d.ts.map +1 -0
- package/dist/core/image-providers/nano-banana.js +90 -0
- package/dist/core/image-providers/nano-banana.js.map +1 -0
- package/dist/core/image-providers/openai.d.ts +12 -0
- package/dist/core/image-providers/openai.d.ts.map +1 -0
- package/dist/core/image-providers/openai.js +82 -0
- package/dist/core/image-providers/openai.js.map +1 -0
- package/dist/core/providers/anthropic.d.ts.map +1 -1
- package/dist/core/providers/anthropic.js +4 -3
- package/dist/core/providers/anthropic.js.map +1 -1
- package/dist/core/providers/azure-openai.d.ts.map +1 -1
- package/dist/core/providers/azure-openai.js +4 -3
- package/dist/core/providers/azure-openai.js.map +1 -1
- package/dist/core/providers/base.d.ts +6 -2
- package/dist/core/providers/base.d.ts.map +1 -1
- package/dist/core/providers/base.js +40 -4
- package/dist/core/providers/base.js.map +1 -1
- package/dist/core/providers/minimax.d.ts.map +1 -1
- package/dist/core/providers/minimax.js +4 -3
- package/dist/core/providers/minimax.js.map +1 -1
- package/dist/core/providers/openai-compatible.d.ts.map +1 -1
- package/dist/core/providers/openai-compatible.js +4 -3
- package/dist/core/providers/openai-compatible.js.map +1 -1
- package/dist/core/providers/openai.d.ts.map +1 -1
- package/dist/core/providers/openai.js +4 -3
- package/dist/core/providers/openai.js.map +1 -1
- package/dist/core/rewriter.d.ts +2 -1
- package/dist/core/rewriter.d.ts.map +1 -1
- package/dist/core/rewriter.js +27 -3
- package/dist/core/rewriter.js.map +1 -1
- package/dist/index.d.ts +6 -0
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +19 -1
- package/dist/index.js.map +1 -1
- package/dist/types/index.d.ts +62 -0
- package/dist/types/index.d.ts.map +1 -1
- package/package.json +2 -2
|
@@ -0,0 +1,79 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.NanoBananaImageProvider = exports.DEFAULT_NANO_BANANA_BASE_URL = void 0;
|
|
4
|
+
exports.DEFAULT_NANO_BANANA_BASE_URL = 'https://generativelanguage.googleapis.com/v1beta/models';
|
|
5
|
+
/**
|
|
6
|
+
* Nano Banana Image Provider (Google Imagen 3)
|
|
7
|
+
*
|
|
8
|
+
* 使用 Google 官方 Generative Language API
|
|
9
|
+
* 支持图生图
|
|
10
|
+
*
|
|
11
|
+
* Models:
|
|
12
|
+
* - imagen-3.0-generate-002 (默认)
|
|
13
|
+
* - imagen-3.0-fast-generate-001
|
|
14
|
+
*/
|
|
15
|
+
class NanoBananaImageProvider {
|
|
16
|
+
constructor(apiKey, model = 'imagen-3.0-generate-002', baseUrl) {
|
|
17
|
+
this.name = 'nano-banana-image';
|
|
18
|
+
this.apiKey = apiKey;
|
|
19
|
+
this.model = model;
|
|
20
|
+
this.baseUrl = baseUrl || exports.DEFAULT_NANO_BANANA_BASE_URL;
|
|
21
|
+
}
|
|
22
|
+
async generateImage(options) {
|
|
23
|
+
const { referenceImage, prompt, aspectRatio = '1:1' } = options;
|
|
24
|
+
try {
|
|
25
|
+
const payload = {
|
|
26
|
+
instances: [
|
|
27
|
+
{
|
|
28
|
+
prompt,
|
|
29
|
+
referenceImages: referenceImage ? [{ referenceImage: { bytesBase64Encoded: this.extractBase64(referenceImage) } }] : undefined,
|
|
30
|
+
},
|
|
31
|
+
],
|
|
32
|
+
parameters: {
|
|
33
|
+
sampleCount: 1,
|
|
34
|
+
aspectRatio: aspectRatio.replace(':', ':'),
|
|
35
|
+
},
|
|
36
|
+
};
|
|
37
|
+
const response = await fetch(`${this.baseUrl}/${this.model}:predict?key=${this.apiKey}`, {
|
|
38
|
+
method: 'POST',
|
|
39
|
+
headers: { 'Content-Type': 'application/json' },
|
|
40
|
+
body: JSON.stringify(payload),
|
|
41
|
+
});
|
|
42
|
+
if (!response.ok) {
|
|
43
|
+
const errorText = await response.text();
|
|
44
|
+
throw new Error(`Google Imagen API error: ${response.status} - ${errorText}`);
|
|
45
|
+
}
|
|
46
|
+
const result = await response.json();
|
|
47
|
+
if (result.error) {
|
|
48
|
+
throw new Error(`Google Imagen API error: ${result.error.message}`);
|
|
49
|
+
}
|
|
50
|
+
const imageData = result.predictions?.[0]?.bytesBase64Encoded;
|
|
51
|
+
if (!imageData) {
|
|
52
|
+
throw new Error('No image data returned');
|
|
53
|
+
}
|
|
54
|
+
return {
|
|
55
|
+
success: true,
|
|
56
|
+
imageData,
|
|
57
|
+
format: 'base64',
|
|
58
|
+
};
|
|
59
|
+
}
|
|
60
|
+
catch (error) {
|
|
61
|
+
return {
|
|
62
|
+
success: false,
|
|
63
|
+
format: 'base64',
|
|
64
|
+
error: error.message,
|
|
65
|
+
};
|
|
66
|
+
}
|
|
67
|
+
}
|
|
68
|
+
/**
|
|
69
|
+
* 从 data URL 或 base64 字符串中提取 base64 数据
|
|
70
|
+
*/
|
|
71
|
+
extractBase64(input) {
|
|
72
|
+
if (input.startsWith('data:')) {
|
|
73
|
+
return input.split(',')[1] || input;
|
|
74
|
+
}
|
|
75
|
+
return input;
|
|
76
|
+
}
|
|
77
|
+
}
|
|
78
|
+
exports.NanoBananaImageProvider = NanoBananaImageProvider;
|
|
79
|
+
//# sourceMappingURL=fal.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"fal.js","sourceRoot":"","sources":["../../../src/core/image-providers/fal.ts"],"names":[],"mappings":";;;AAEa,QAAA,4BAA4B,GAAG,yDAAyD,CAAC;AAEtG;;;;;;;;;GASG;AACH,MAAa,uBAAuB;IAMlC,YACE,MAAc,EACd,QAAgB,yBAAyB,EACzC,OAAgB;QARlB,SAAI,GAAG,mBAAmB,CAAC;QAUzB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,OAAO,GAAG,OAAO,IAAI,oCAA4B,CAAC;IACzD,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,OAA+B;QACjD,MAAM,EAAE,cAAc,EAAE,MAAM,EAAE,WAAW,GAAG,KAAK,EAAE,GAAG,OAAO,CAAC;QAEhE,IAAI,CAAC;YACH,MAAM,OAAO,GAAG;gBACd,SAAS,EAAE;oBACT;wBACE,MAAM;wBACN,eAAe,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,EAAE,kBAAkB,EAAE,IAAI,CAAC,aAAa,CAAC,cAAc,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS;qBAC/H;iBACF;gBACD,UAAU,EAAE;oBACV,WAAW,EAAE,CAAC;oBACd,WAAW,EAAE,WAAW,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC;iBAC3C;aACF,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,KAAK,CAC1B,GAAG,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,KAAK,gBAAgB,IAAI,CAAC,MAAM,EAAE,EAC1D;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;gBAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;aAC9B,CACF,CAAC;YAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACxC,MAAM,IAAI,KAAK,CAAC,4BAA4B,QAAQ,CAAC,MAAM,MAAM,SAAS,EAAE,CAAC,CAAC;YAChF,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,EAGjC,CAAC;YAEF,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;gBACjB,MAAM,IAAI,KAAK,CAAC,4BAA4B,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YACtE,CAAC;YAED,MAAM,SAAS,GAAG,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,kBAAkB,CAAC;YAC9D,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;YAC5C,CAAC;YAED,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,SAAS;gBACT,MAAM,EAAE,QAAQ;aACjB,CAAC;QACJ,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,QAAQ;gBAChB,KAAK,EAAE,KAAK,CAAC,OAAO;aACrB,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,aAAa,CAAC,KAAa;QACjC,IAAI,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;YAC9B,OAAO,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC;QACtC,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;CACF;AApFD,0DAoFC"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import { ImageProvider, ImageGenerationOptions, ImageGenerationResult } from '../../types';
|
|
2
|
+
export declare const DEFAULT_GEMINI_IMAGE_URL = "https://generativelanguage.googleapis.com/v1beta/models";
|
|
3
|
+
export declare class GeminiImageProvider implements ImageProvider {
|
|
4
|
+
name: string;
|
|
5
|
+
private apiKey;
|
|
6
|
+
private model;
|
|
7
|
+
private baseUrl;
|
|
8
|
+
constructor(apiKey: string, model?: string, baseUrl?: string);
|
|
9
|
+
generateImage(options: ImageGenerationOptions): Promise<ImageGenerationResult>;
|
|
10
|
+
}
|
|
11
|
+
//# sourceMappingURL=gemini.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gemini.d.ts","sourceRoot":"","sources":["../../../src/core/image-providers/gemini.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,sBAAsB,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAC;AAE3F,eAAO,MAAM,wBAAwB,4DAA4D,CAAC;AAElG,qBAAa,mBAAoB,YAAW,aAAa;IACvD,IAAI,SAAkB;IACtB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,KAAK,CAAS;IACtB,OAAO,CAAC,OAAO,CAAS;gBAEZ,MAAM,EAAE,MAAM,EAAE,KAAK,GAAE,MAAkC,EAAE,OAAO,CAAC,EAAE,MAAM;IAMjF,aAAa,CAAC,OAAO,EAAE,sBAAsB,GAAG,OAAO,CAAC,qBAAqB,CAAC;CA0DrF"}
|
|
@@ -0,0 +1,57 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.GeminiImageProvider = exports.DEFAULT_GEMINI_IMAGE_URL = void 0;
|
|
4
|
+
exports.DEFAULT_GEMINI_IMAGE_URL = 'https://generativelanguage.googleapis.com/v1beta/models';
|
|
5
|
+
class GeminiImageProvider {
|
|
6
|
+
constructor(apiKey, model = 'imagen-3.0-generate-002', baseUrl) {
|
|
7
|
+
this.name = 'gemini-image';
|
|
8
|
+
this.apiKey = apiKey;
|
|
9
|
+
this.model = model;
|
|
10
|
+
this.baseUrl = baseUrl || exports.DEFAULT_GEMINI_IMAGE_URL;
|
|
11
|
+
}
|
|
12
|
+
async generateImage(options) {
|
|
13
|
+
const { prompt, aspectRatio = '1:1', } = options;
|
|
14
|
+
try {
|
|
15
|
+
const payload = {
|
|
16
|
+
instances: [{ prompt }],
|
|
17
|
+
parameters: {
|
|
18
|
+
sampleCount: 1,
|
|
19
|
+
aspectRatio: aspectRatio.replace(':', ':'),
|
|
20
|
+
},
|
|
21
|
+
};
|
|
22
|
+
const response = await fetch(`${this.baseUrl}/${this.model}:predict?key=${this.apiKey}`, {
|
|
23
|
+
method: 'POST',
|
|
24
|
+
headers: {
|
|
25
|
+
'Content-Type': 'application/json',
|
|
26
|
+
},
|
|
27
|
+
body: JSON.stringify(payload),
|
|
28
|
+
});
|
|
29
|
+
if (!response.ok) {
|
|
30
|
+
const errorText = await response.text();
|
|
31
|
+
throw new Error(`Gemini API error: ${response.status} - ${errorText}`);
|
|
32
|
+
}
|
|
33
|
+
const result = await response.json();
|
|
34
|
+
if (result.error) {
|
|
35
|
+
throw new Error(`Gemini API error: ${result.error.message}`);
|
|
36
|
+
}
|
|
37
|
+
const imageData = result.predictions?.[0]?.bytesBase64Encoded;
|
|
38
|
+
if (!imageData) {
|
|
39
|
+
throw new Error('No image data returned');
|
|
40
|
+
}
|
|
41
|
+
return {
|
|
42
|
+
success: true,
|
|
43
|
+
imageData,
|
|
44
|
+
format: 'base64',
|
|
45
|
+
};
|
|
46
|
+
}
|
|
47
|
+
catch (error) {
|
|
48
|
+
return {
|
|
49
|
+
success: false,
|
|
50
|
+
format: 'base64',
|
|
51
|
+
error: error.message,
|
|
52
|
+
};
|
|
53
|
+
}
|
|
54
|
+
}
|
|
55
|
+
}
|
|
56
|
+
exports.GeminiImageProvider = GeminiImageProvider;
|
|
57
|
+
//# sourceMappingURL=gemini.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gemini.js","sourceRoot":"","sources":["../../../src/core/image-providers/gemini.ts"],"names":[],"mappings":";;;AAEa,QAAA,wBAAwB,GAAG,yDAAyD,CAAC;AAElG,MAAa,mBAAmB;IAM9B,YAAY,MAAc,EAAE,QAAgB,yBAAyB,EAAE,OAAgB;QALvF,SAAI,GAAG,cAAc,CAAC;QAMpB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,OAAO,GAAG,OAAO,IAAI,gCAAwB,CAAC;IACrD,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,OAA+B;QACjD,MAAM,EACJ,MAAM,EACN,WAAW,GAAG,KAAK,GACpB,GAAG,OAAO,CAAC;QAEZ,IAAI,CAAC;YACH,MAAM,OAAO,GAAG;gBACd,SAAS,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC;gBACvB,UAAU,EAAE;oBACV,WAAW,EAAE,CAAC;oBACd,WAAW,EAAE,WAAW,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC;iBAC3C;aACF,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,KAAK,CAC1B,GAAG,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,KAAK,gBAAgB,IAAI,CAAC,MAAM,EAAE,EAC1D;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;iBACnC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;aAC9B,CACF,CAAC;YAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACxC,MAAM,IAAI,KAAK,CAAC,qBAAqB,QAAQ,CAAC,MAAM,MAAM,SAAS,EAAE,CAAC,CAAC;YACzE,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,EAGjC,CAAC;YAEF,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;gBACjB,MAAM,IAAI,KAAK,CAAC,qBAAqB,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YAC/D,CAAC;YAED,MAAM,SAAS,GAAG,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,kBAAkB,CAAC;YAC9D,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;YAC5C,CAAC;YAED,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,SAAS;gBACT,MAAM,EAAE,QAAQ;aACjB,CAAC;QACJ,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,QAAQ;gBAChB,KAAK,EAAE,KAAK,CAAC,OAAO;aACrB,CAAC;QACJ,CAAC;IACH,CAAC;CACF;AAtED,kDAsEC"}
|
|
@@ -0,0 +1,6 @@
|
|
|
1
|
+
export { MiniMaxImageProvider, DEFAULT_MINIMAX_IMAGE_URL } from './minimax';
|
|
2
|
+
export { OpenAIImageProvider, DEFAULT_OPENAI_IMAGE_URL } from './openai';
|
|
3
|
+
export { GeminiImageProvider, DEFAULT_GEMINI_IMAGE_URL } from './gemini';
|
|
4
|
+
export { DoubaoImageProvider, DEFAULT_DOUBAO_IMAGE_URL } from './doubao';
|
|
5
|
+
export { NanoBananaImageProvider, DEFAULT_NANO_BANANA_BASE_URL, IMAGEN_MODELS } from './nano-banana';
|
|
6
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/core/image-providers/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,yBAAyB,EAAE,MAAM,WAAW,CAAC;AAC5E,OAAO,EAAE,mBAAmB,EAAE,wBAAwB,EAAE,MAAM,UAAU,CAAC;AACzE,OAAO,EAAE,mBAAmB,EAAE,wBAAwB,EAAE,MAAM,UAAU,CAAC;AACzE,OAAO,EAAE,mBAAmB,EAAE,wBAAwB,EAAE,MAAM,UAAU,CAAC;AACzE,OAAO,EAAE,uBAAuB,EAAE,4BAA4B,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC"}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.IMAGEN_MODELS = exports.DEFAULT_NANO_BANANA_BASE_URL = exports.NanoBananaImageProvider = exports.DEFAULT_DOUBAO_IMAGE_URL = exports.DoubaoImageProvider = exports.DEFAULT_GEMINI_IMAGE_URL = exports.GeminiImageProvider = exports.DEFAULT_OPENAI_IMAGE_URL = exports.OpenAIImageProvider = exports.DEFAULT_MINIMAX_IMAGE_URL = exports.MiniMaxImageProvider = void 0;
|
|
4
|
+
var minimax_1 = require("./minimax");
|
|
5
|
+
Object.defineProperty(exports, "MiniMaxImageProvider", { enumerable: true, get: function () { return minimax_1.MiniMaxImageProvider; } });
|
|
6
|
+
Object.defineProperty(exports, "DEFAULT_MINIMAX_IMAGE_URL", { enumerable: true, get: function () { return minimax_1.DEFAULT_MINIMAX_IMAGE_URL; } });
|
|
7
|
+
var openai_1 = require("./openai");
|
|
8
|
+
Object.defineProperty(exports, "OpenAIImageProvider", { enumerable: true, get: function () { return openai_1.OpenAIImageProvider; } });
|
|
9
|
+
Object.defineProperty(exports, "DEFAULT_OPENAI_IMAGE_URL", { enumerable: true, get: function () { return openai_1.DEFAULT_OPENAI_IMAGE_URL; } });
|
|
10
|
+
var gemini_1 = require("./gemini");
|
|
11
|
+
Object.defineProperty(exports, "GeminiImageProvider", { enumerable: true, get: function () { return gemini_1.GeminiImageProvider; } });
|
|
12
|
+
Object.defineProperty(exports, "DEFAULT_GEMINI_IMAGE_URL", { enumerable: true, get: function () { return gemini_1.DEFAULT_GEMINI_IMAGE_URL; } });
|
|
13
|
+
var doubao_1 = require("./doubao");
|
|
14
|
+
Object.defineProperty(exports, "DoubaoImageProvider", { enumerable: true, get: function () { return doubao_1.DoubaoImageProvider; } });
|
|
15
|
+
Object.defineProperty(exports, "DEFAULT_DOUBAO_IMAGE_URL", { enumerable: true, get: function () { return doubao_1.DEFAULT_DOUBAO_IMAGE_URL; } });
|
|
16
|
+
var nano_banana_1 = require("./nano-banana");
|
|
17
|
+
Object.defineProperty(exports, "NanoBananaImageProvider", { enumerable: true, get: function () { return nano_banana_1.NanoBananaImageProvider; } });
|
|
18
|
+
Object.defineProperty(exports, "DEFAULT_NANO_BANANA_BASE_URL", { enumerable: true, get: function () { return nano_banana_1.DEFAULT_NANO_BANANA_BASE_URL; } });
|
|
19
|
+
Object.defineProperty(exports, "IMAGEN_MODELS", { enumerable: true, get: function () { return nano_banana_1.IMAGEN_MODELS; } });
|
|
20
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/core/image-providers/index.ts"],"names":[],"mappings":";;;AAAA,qCAA4E;AAAnE,+GAAA,oBAAoB,OAAA;AAAE,oHAAA,yBAAyB,OAAA;AACxD,mCAAyE;AAAhE,6GAAA,mBAAmB,OAAA;AAAE,kHAAA,wBAAwB,OAAA;AACtD,mCAAyE;AAAhE,6GAAA,mBAAmB,OAAA;AAAE,kHAAA,wBAAwB,OAAA;AACtD,mCAAyE;AAAhE,6GAAA,mBAAmB,OAAA;AAAE,kHAAA,wBAAwB,OAAA;AACtD,6CAAqG;AAA5F,sHAAA,uBAAuB,OAAA;AAAE,2HAAA,4BAA4B,OAAA;AAAE,4GAAA,aAAa,OAAA"}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
import { ImageProvider, ImageGenerationOptions, ImageGenerationResult } from '../../types';
|
|
2
|
+
export declare const DEFAULT_MINIMAX_IMAGE_URL = "https://api.minimaxi.com/v1/image_generation";
|
|
3
|
+
/**
|
|
4
|
+
* MiniMax 图片生成 Provider
|
|
5
|
+
*
|
|
6
|
+
* 支持:
|
|
7
|
+
* - 文生图(无参考图)
|
|
8
|
+
* - 图生图(有参考图,使用 subject_reference)
|
|
9
|
+
*
|
|
10
|
+
* Models:
|
|
11
|
+
* - image-01 (默认)
|
|
12
|
+
*/
|
|
13
|
+
export declare class MiniMaxImageProvider implements ImageProvider {
|
|
14
|
+
name: string;
|
|
15
|
+
private apiKey;
|
|
16
|
+
private baseUrl;
|
|
17
|
+
constructor(apiKey: string, baseUrl?: string);
|
|
18
|
+
generateImage(options: ImageGenerationOptions): Promise<ImageGenerationResult>;
|
|
19
|
+
}
|
|
20
|
+
//# sourceMappingURL=minimax.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"minimax.d.ts","sourceRoot":"","sources":["../../../src/core/image-providers/minimax.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,sBAAsB,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAC;AAE3F,eAAO,MAAM,yBAAyB,iDAAiD,CAAC;AAExF;;;;;;;;;GASG;AACH,qBAAa,oBAAqB,YAAW,aAAa;IACxD,IAAI,SAAmB;IACvB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,OAAO,CAAS;gBAEZ,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM;IAKtC,aAAa,CAAC,OAAO,EAAE,sBAAsB,GAAG,OAAO,CAAC,qBAAqB,CAAC;CAgFrF"}
|
|
@@ -0,0 +1,89 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.MiniMaxImageProvider = exports.DEFAULT_MINIMAX_IMAGE_URL = void 0;
|
|
4
|
+
exports.DEFAULT_MINIMAX_IMAGE_URL = 'https://api.minimaxi.com/v1/image_generation';
|
|
5
|
+
/**
|
|
6
|
+
* MiniMax 图片生成 Provider
|
|
7
|
+
*
|
|
8
|
+
* 支持:
|
|
9
|
+
* - 文生图(无参考图)
|
|
10
|
+
* - 图生图(有参考图,使用 subject_reference)
|
|
11
|
+
*
|
|
12
|
+
* Models:
|
|
13
|
+
* - image-01 (默认)
|
|
14
|
+
*/
|
|
15
|
+
class MiniMaxImageProvider {
|
|
16
|
+
constructor(apiKey, baseUrl) {
|
|
17
|
+
this.name = 'minimax-image';
|
|
18
|
+
this.apiKey = apiKey;
|
|
19
|
+
this.baseUrl = baseUrl || process.env.MINIMAX_BASE_URL || exports.DEFAULT_MINIMAX_IMAGE_URL;
|
|
20
|
+
}
|
|
21
|
+
async generateImage(options) {
|
|
22
|
+
const { referenceImage, prompt, aspectRatio = '1:1', responseFormat = 'base64', watermark = false, } = options;
|
|
23
|
+
try {
|
|
24
|
+
const payload = {
|
|
25
|
+
model: 'image-01',
|
|
26
|
+
prompt,
|
|
27
|
+
aspect_ratio: aspectRatio,
|
|
28
|
+
response_format: responseFormat,
|
|
29
|
+
watermark,
|
|
30
|
+
};
|
|
31
|
+
// 如果有参考图,使用图生图模式
|
|
32
|
+
if (referenceImage) {
|
|
33
|
+
payload.subject_reference = [
|
|
34
|
+
{
|
|
35
|
+
type: 'character',
|
|
36
|
+
image_file: referenceImage,
|
|
37
|
+
},
|
|
38
|
+
];
|
|
39
|
+
}
|
|
40
|
+
const response = await fetch(this.baseUrl, {
|
|
41
|
+
method: 'POST',
|
|
42
|
+
headers: {
|
|
43
|
+
'Authorization': `Bearer ${this.apiKey}`,
|
|
44
|
+
'Content-Type': 'application/json',
|
|
45
|
+
},
|
|
46
|
+
body: JSON.stringify(payload),
|
|
47
|
+
});
|
|
48
|
+
if (!response.ok) {
|
|
49
|
+
const errorText = await response.text();
|
|
50
|
+
throw new Error(`MiniMax API error: ${response.status} - ${errorText}`);
|
|
51
|
+
}
|
|
52
|
+
const result = await response.json();
|
|
53
|
+
if (result.base_resp?.status_code !== 0) {
|
|
54
|
+
throw new Error(`MiniMax API error: ${result.base_resp?.status_msg}`);
|
|
55
|
+
}
|
|
56
|
+
if (responseFormat === 'url') {
|
|
57
|
+
const imageUrl = result.data?.image_urls?.[0];
|
|
58
|
+
if (!imageUrl) {
|
|
59
|
+
throw new Error('No image URL returned');
|
|
60
|
+
}
|
|
61
|
+
return {
|
|
62
|
+
success: true,
|
|
63
|
+
imageData: imageUrl,
|
|
64
|
+
format: 'url',
|
|
65
|
+
};
|
|
66
|
+
}
|
|
67
|
+
else {
|
|
68
|
+
const imageBase64 = result.data?.image_base64?.[0];
|
|
69
|
+
if (!imageBase64) {
|
|
70
|
+
throw new Error('No image base64 returned');
|
|
71
|
+
}
|
|
72
|
+
return {
|
|
73
|
+
success: true,
|
|
74
|
+
imageData: imageBase64,
|
|
75
|
+
format: 'base64',
|
|
76
|
+
};
|
|
77
|
+
}
|
|
78
|
+
}
|
|
79
|
+
catch (error) {
|
|
80
|
+
return {
|
|
81
|
+
success: false,
|
|
82
|
+
format: responseFormat,
|
|
83
|
+
error: error.message,
|
|
84
|
+
};
|
|
85
|
+
}
|
|
86
|
+
}
|
|
87
|
+
}
|
|
88
|
+
exports.MiniMaxImageProvider = MiniMaxImageProvider;
|
|
89
|
+
//# sourceMappingURL=minimax.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"minimax.js","sourceRoot":"","sources":["../../../src/core/image-providers/minimax.ts"],"names":[],"mappings":";;;AAEa,QAAA,yBAAyB,GAAG,8CAA8C,CAAC;AAExF;;;;;;;;;GASG;AACH,MAAa,oBAAoB;IAK/B,YAAY,MAAc,EAAE,OAAgB;QAJ5C,SAAI,GAAG,eAAe,CAAC;QAKrB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG,OAAO,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,iCAAyB,CAAC;IACtF,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,OAA+B;QACjD,MAAM,EACJ,cAAc,EACd,MAAM,EACN,WAAW,GAAG,KAAK,EACnB,cAAc,GAAG,QAAQ,EACzB,SAAS,GAAG,KAAK,GAClB,GAAG,OAAO,CAAC;QAEZ,IAAI,CAAC;YACH,MAAM,OAAO,GAAwB;gBACnC,KAAK,EAAE,UAAU;gBACjB,MAAM;gBACN,YAAY,EAAE,WAAW;gBACzB,eAAe,EAAE,cAAc;gBAC/B,SAAS;aACV,CAAC;YAEF,iBAAiB;YACjB,IAAI,cAAc,EAAE,CAAC;gBACnB,OAAO,CAAC,iBAAiB,GAAG;oBAC1B;wBACE,IAAI,EAAE,WAAW;wBACjB,UAAU,EAAE,cAAc;qBAC3B;iBACF,CAAC;YACJ,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE;gBACzC,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;oBACxC,cAAc,EAAE,kBAAkB;iBACnC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;aAC9B,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACxC,MAAM,IAAI,KAAK,CAAC,sBAAsB,QAAQ,CAAC,MAAM,MAAM,SAAS,EAAE,CAAC,CAAC;YAC1E,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,EAGjC,CAAC;YAEF,IAAI,MAAM,CAAC,SAAS,EAAE,WAAW,KAAK,CAAC,EAAE,CAAC;gBACxC,MAAM,IAAI,KAAK,CAAC,sBAAsB,MAAM,CAAC,SAAS,EAAE,UAAU,EAAE,CAAC,CAAC;YACxE,CAAC;YAED,IAAI,cAAc,KAAK,KAAK,EAAE,CAAC;gBAC7B,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;gBAC9C,IAAI,CAAC,QAAQ,EAAE,CAAC;oBACd,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;gBAC3C,CAAC;gBACD,OAAO;oBACL,OAAO,EAAE,IAAI;oBACb,SAAS,EAAE,QAAQ;oBACnB,MAAM,EAAE,KAAK;iBACd,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,CAAC;gBACnD,IAAI,CAAC,WAAW,EAAE,CAAC;oBACjB,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;gBAC9C,CAAC;gBACD,OAAO;oBACL,OAAO,EAAE,IAAI;oBACb,SAAS,EAAE,WAAW;oBACtB,MAAM,EAAE,QAAQ;iBACjB,CAAC;YACJ,CAAC;QACH,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,cAAc;gBACtB,KAAK,EAAE,KAAK,CAAC,OAAO;aACrB,CAAC;QACJ,CAAC;IACH,CAAC;CACF;AA1FD,oDA0FC"}
|
|
@@ -0,0 +1,36 @@
|
|
|
1
|
+
import { ImageProvider, ImageGenerationOptions, ImageGenerationResult } from '../../types';
|
|
2
|
+
export declare const DEFAULT_NANO_BANANA_BASE_URL = "https://generativelanguage.googleapis.com/v1beta/models";
|
|
3
|
+
/**
|
|
4
|
+
* Google Imagen 3 模型
|
|
5
|
+
*/
|
|
6
|
+
export declare const IMAGEN_MODELS: {
|
|
7
|
+
/** 标准版 - 高质量生成 */
|
|
8
|
+
readonly standard: "imagen-3.0-generate-002";
|
|
9
|
+
/** 快速版 - 更快的生成速度 */
|
|
10
|
+
readonly fast: "imagen-3.0-fast-generate-001";
|
|
11
|
+
/** Pro 版 - 支持编辑和 inpainting */
|
|
12
|
+
readonly pro: "imagen-3.0-capability-001";
|
|
13
|
+
};
|
|
14
|
+
/**
|
|
15
|
+
* Nano Banana Image Provider (Google Imagen 3)
|
|
16
|
+
*
|
|
17
|
+
* 使用 Google 官方 Generative Language API
|
|
18
|
+
*
|
|
19
|
+
* Models:
|
|
20
|
+
* - imagen-3.0-generate-002 (标准版,默认)
|
|
21
|
+
* - imagen-3.0-fast-generate-001 (快速版)
|
|
22
|
+
* - imagen-3.0-capability-001 (Pro 版,支持编辑/inpainting)
|
|
23
|
+
*/
|
|
24
|
+
export declare class NanoBananaImageProvider implements ImageProvider {
|
|
25
|
+
name: string;
|
|
26
|
+
private apiKey;
|
|
27
|
+
private model;
|
|
28
|
+
private baseUrl;
|
|
29
|
+
constructor(apiKey: string, model?: string, baseUrl?: string);
|
|
30
|
+
generateImage(options: ImageGenerationOptions): Promise<ImageGenerationResult>;
|
|
31
|
+
/**
|
|
32
|
+
* 从 data URL 或 base64 字符串中提取 base64 数据
|
|
33
|
+
*/
|
|
34
|
+
private extractBase64;
|
|
35
|
+
}
|
|
36
|
+
//# sourceMappingURL=nano-banana.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"nano-banana.d.ts","sourceRoot":"","sources":["../../../src/core/image-providers/nano-banana.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,sBAAsB,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAC;AAE3F,eAAO,MAAM,4BAA4B,4DAA4D,CAAC;AAEtG;;GAEG;AACH,eAAO,MAAM,aAAa;IACxB,kBAAkB;;IAElB,oBAAoB;;IAEpB,+BAA+B;;CAEvB,CAAC;AAEX;;;;;;;;;GASG;AACH,qBAAa,uBAAwB,YAAW,aAAa;IAC3D,IAAI,SAAuB;IAC3B,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,KAAK,CAAS;IACtB,OAAO,CAAC,OAAO,CAAS;gBAGtB,MAAM,EAAE,MAAM,EACd,KAAK,GAAE,MAA+B,EACtC,OAAO,CAAC,EAAE,MAAM;IAOZ,aAAa,CAAC,OAAO,EAAE,sBAAsB,GAAG,OAAO,CAAC,qBAAqB,CAAC;IA2DpF;;OAEG;IACH,OAAO,CAAC,aAAa;CAMtB"}
|
|
@@ -0,0 +1,90 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.NanoBananaImageProvider = exports.IMAGEN_MODELS = exports.DEFAULT_NANO_BANANA_BASE_URL = void 0;
|
|
4
|
+
exports.DEFAULT_NANO_BANANA_BASE_URL = 'https://generativelanguage.googleapis.com/v1beta/models';
|
|
5
|
+
/**
|
|
6
|
+
* Google Imagen 3 模型
|
|
7
|
+
*/
|
|
8
|
+
exports.IMAGEN_MODELS = {
|
|
9
|
+
/** 标准版 - 高质量生成 */
|
|
10
|
+
standard: 'imagen-3.0-generate-002',
|
|
11
|
+
/** 快速版 - 更快的生成速度 */
|
|
12
|
+
fast: 'imagen-3.0-fast-generate-001',
|
|
13
|
+
/** Pro 版 - 支持编辑和 inpainting */
|
|
14
|
+
pro: 'imagen-3.0-capability-001',
|
|
15
|
+
};
|
|
16
|
+
/**
|
|
17
|
+
* Nano Banana Image Provider (Google Imagen 3)
|
|
18
|
+
*
|
|
19
|
+
* 使用 Google 官方 Generative Language API
|
|
20
|
+
*
|
|
21
|
+
* Models:
|
|
22
|
+
* - imagen-3.0-generate-002 (标准版,默认)
|
|
23
|
+
* - imagen-3.0-fast-generate-001 (快速版)
|
|
24
|
+
* - imagen-3.0-capability-001 (Pro 版,支持编辑/inpainting)
|
|
25
|
+
*/
|
|
26
|
+
class NanoBananaImageProvider {
|
|
27
|
+
constructor(apiKey, model = exports.IMAGEN_MODELS.standard, baseUrl) {
|
|
28
|
+
this.name = 'nano-banana-image';
|
|
29
|
+
this.apiKey = apiKey;
|
|
30
|
+
this.model = model;
|
|
31
|
+
this.baseUrl = baseUrl || exports.DEFAULT_NANO_BANANA_BASE_URL;
|
|
32
|
+
}
|
|
33
|
+
async generateImage(options) {
|
|
34
|
+
const { referenceImage, prompt, aspectRatio = '1:1' } = options;
|
|
35
|
+
try {
|
|
36
|
+
const payload = {
|
|
37
|
+
instances: [
|
|
38
|
+
{
|
|
39
|
+
prompt,
|
|
40
|
+
referenceImages: referenceImage ? [{ referenceImage: { bytesBase64Encoded: this.extractBase64(referenceImage) } }] : undefined,
|
|
41
|
+
},
|
|
42
|
+
],
|
|
43
|
+
parameters: {
|
|
44
|
+
sampleCount: 1,
|
|
45
|
+
aspectRatio: aspectRatio.replace(':', ':'),
|
|
46
|
+
},
|
|
47
|
+
};
|
|
48
|
+
const response = await fetch(`${this.baseUrl}/${this.model}:predict?key=${this.apiKey}`, {
|
|
49
|
+
method: 'POST',
|
|
50
|
+
headers: { 'Content-Type': 'application/json' },
|
|
51
|
+
body: JSON.stringify(payload),
|
|
52
|
+
});
|
|
53
|
+
if (!response.ok) {
|
|
54
|
+
const errorText = await response.text();
|
|
55
|
+
throw new Error(`Google Imagen API error: ${response.status} - ${errorText}`);
|
|
56
|
+
}
|
|
57
|
+
const result = await response.json();
|
|
58
|
+
if (result.error) {
|
|
59
|
+
throw new Error(`Google Imagen API error: ${result.error.message}`);
|
|
60
|
+
}
|
|
61
|
+
const imageData = result.predictions?.[0]?.bytesBase64Encoded;
|
|
62
|
+
if (!imageData) {
|
|
63
|
+
throw new Error('No image data returned');
|
|
64
|
+
}
|
|
65
|
+
return {
|
|
66
|
+
success: true,
|
|
67
|
+
imageData,
|
|
68
|
+
format: 'base64',
|
|
69
|
+
};
|
|
70
|
+
}
|
|
71
|
+
catch (error) {
|
|
72
|
+
return {
|
|
73
|
+
success: false,
|
|
74
|
+
format: 'base64',
|
|
75
|
+
error: error.message,
|
|
76
|
+
};
|
|
77
|
+
}
|
|
78
|
+
}
|
|
79
|
+
/**
|
|
80
|
+
* 从 data URL 或 base64 字符串中提取 base64 数据
|
|
81
|
+
*/
|
|
82
|
+
extractBase64(input) {
|
|
83
|
+
if (input.startsWith('data:')) {
|
|
84
|
+
return input.split(',')[1] || input;
|
|
85
|
+
}
|
|
86
|
+
return input;
|
|
87
|
+
}
|
|
88
|
+
}
|
|
89
|
+
exports.NanoBananaImageProvider = NanoBananaImageProvider;
|
|
90
|
+
//# sourceMappingURL=nano-banana.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"nano-banana.js","sourceRoot":"","sources":["../../../src/core/image-providers/nano-banana.ts"],"names":[],"mappings":";;;AAEa,QAAA,4BAA4B,GAAG,yDAAyD,CAAC;AAEtG;;GAEG;AACU,QAAA,aAAa,GAAG;IAC3B,kBAAkB;IAClB,QAAQ,EAAE,yBAAyB;IACnC,oBAAoB;IACpB,IAAI,EAAE,8BAA8B;IACpC,+BAA+B;IAC/B,GAAG,EAAE,2BAA2B;CACxB,CAAC;AAEX;;;;;;;;;GASG;AACH,MAAa,uBAAuB;IAMlC,YACE,MAAc,EACd,QAAgB,qBAAa,CAAC,QAAQ,EACtC,OAAgB;QARlB,SAAI,GAAG,mBAAmB,CAAC;QAUzB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,OAAO,GAAG,OAAO,IAAI,oCAA4B,CAAC;IACzD,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,OAA+B;QACjD,MAAM,EAAE,cAAc,EAAE,MAAM,EAAE,WAAW,GAAG,KAAK,EAAE,GAAG,OAAO,CAAC;QAEhE,IAAI,CAAC;YACH,MAAM,OAAO,GAAG;gBACd,SAAS,EAAE;oBACT;wBACE,MAAM;wBACN,eAAe,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,EAAE,kBAAkB,EAAE,IAAI,CAAC,aAAa,CAAC,cAAc,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS;qBAC/H;iBACF;gBACD,UAAU,EAAE;oBACV,WAAW,EAAE,CAAC;oBACd,WAAW,EAAE,WAAW,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC;iBAC3C;aACF,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,KAAK,CAC1B,GAAG,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,KAAK,gBAAgB,IAAI,CAAC,MAAM,EAAE,EAC1D;gBACE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;gBAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;aAC9B,CACF,CAAC;YAEF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACxC,MAAM,IAAI,KAAK,CAAC,4BAA4B,QAAQ,CAAC,MAAM,MAAM,SAAS,EAAE,CAAC,CAAC;YAChF,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,EAGjC,CAAC;YAEF,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;gBACjB,MAAM,IAAI,KAAK,CAAC,4BAA4B,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YACtE,CAAC;YAED,MAAM,SAAS,GAAG,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,kBAAkB,CAAC;YAC9D,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;YAC5C,CAAC;YAED,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,SAAS;gBACT,MAAM,EAAE,QAAQ;aACjB,CAAC;QACJ,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,QAAQ;gBAChB,KAAK,EAAE,KAAK,CAAC,OAAO;aACrB,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,aAAa,CAAC,KAAa;QACjC,IAAI,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;YAC9B,OAAO,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC;QACtC,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;CACF;AApFD,0DAoFC"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
import { ImageProvider, ImageGenerationOptions, ImageGenerationResult } from '../../types';
|
|
2
|
+
export declare const DEFAULT_OPENAI_IMAGE_URL = "https://api.openai.com/v1/images";
|
|
3
|
+
export declare class OpenAIImageProvider implements ImageProvider {
|
|
4
|
+
name: string;
|
|
5
|
+
private apiKey;
|
|
6
|
+
private model;
|
|
7
|
+
private baseUrl;
|
|
8
|
+
constructor(apiKey: string, model?: string, baseUrl?: string);
|
|
9
|
+
generateImage(options: ImageGenerationOptions): Promise<ImageGenerationResult>;
|
|
10
|
+
private aspectRatioToSize;
|
|
11
|
+
}
|
|
12
|
+
//# sourceMappingURL=openai.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../../src/core/image-providers/openai.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,sBAAsB,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAC;AAE3F,eAAO,MAAM,wBAAwB,qCAAqC,CAAC;AAE3E,qBAAa,mBAAoB,YAAW,aAAa;IACvD,IAAI,SAAkB;IACtB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,KAAK,CAAS;IACtB,OAAO,CAAC,OAAO,CAAS;gBAEZ,MAAM,EAAE,MAAM,EAAE,KAAK,GAAE,MAAmB,EAAE,OAAO,CAAC,EAAE,MAAM;IAMlE,aAAa,CAAC,OAAO,EAAE,sBAAsB,GAAG,OAAO,CAAC,qBAAqB,CAAC;IAuEpF,OAAO,CAAC,iBAAiB;CAY1B"}
|
|
@@ -0,0 +1,82 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.OpenAIImageProvider = exports.DEFAULT_OPENAI_IMAGE_URL = void 0;
|
|
4
|
+
exports.DEFAULT_OPENAI_IMAGE_URL = 'https://api.openai.com/v1/images';
|
|
5
|
+
class OpenAIImageProvider {
|
|
6
|
+
constructor(apiKey, model = 'dall-e-3', baseUrl) {
|
|
7
|
+
this.name = 'openai-image';
|
|
8
|
+
this.apiKey = apiKey;
|
|
9
|
+
this.model = model;
|
|
10
|
+
this.baseUrl = baseUrl || exports.DEFAULT_OPENAI_IMAGE_URL;
|
|
11
|
+
}
|
|
12
|
+
async generateImage(options) {
|
|
13
|
+
const { referenceImage, prompt, aspectRatio = '1:1', responseFormat = 'b64_json', } = options;
|
|
14
|
+
try {
|
|
15
|
+
// OpenAI DALL-E 3 不支持直接图生图,需要用 variations 或 edit
|
|
16
|
+
// 这里使用 generations 接口,通过 prompt 描述参考图
|
|
17
|
+
const size = this.aspectRatioToSize(aspectRatio);
|
|
18
|
+
const payload = {
|
|
19
|
+
model: this.model,
|
|
20
|
+
prompt: `${prompt}. Reference style from the original image.`,
|
|
21
|
+
n: 1,
|
|
22
|
+
size,
|
|
23
|
+
response_format: responseFormat === 'url' ? 'url' : 'b64_json',
|
|
24
|
+
};
|
|
25
|
+
const response = await fetch(`${this.baseUrl}/generations`, {
|
|
26
|
+
method: 'POST',
|
|
27
|
+
headers: {
|
|
28
|
+
'Authorization': `Bearer ${this.apiKey}`,
|
|
29
|
+
'Content-Type': 'application/json',
|
|
30
|
+
},
|
|
31
|
+
body: JSON.stringify(payload),
|
|
32
|
+
});
|
|
33
|
+
if (!response.ok) {
|
|
34
|
+
const errorText = await response.text();
|
|
35
|
+
throw new Error(`OpenAI API error: ${response.status} - ${errorText}`);
|
|
36
|
+
}
|
|
37
|
+
const result = await response.json();
|
|
38
|
+
if (result.error) {
|
|
39
|
+
throw new Error(`OpenAI API error: ${result.error.message}`);
|
|
40
|
+
}
|
|
41
|
+
const imageData = result.data?.[0];
|
|
42
|
+
if (!imageData) {
|
|
43
|
+
throw new Error('No image data returned');
|
|
44
|
+
}
|
|
45
|
+
if (responseFormat === 'url') {
|
|
46
|
+
return {
|
|
47
|
+
success: true,
|
|
48
|
+
imageData: imageData.url,
|
|
49
|
+
format: 'url',
|
|
50
|
+
};
|
|
51
|
+
}
|
|
52
|
+
else {
|
|
53
|
+
return {
|
|
54
|
+
success: true,
|
|
55
|
+
imageData: imageData.b64_json,
|
|
56
|
+
format: 'base64',
|
|
57
|
+
};
|
|
58
|
+
}
|
|
59
|
+
}
|
|
60
|
+
catch (error) {
|
|
61
|
+
return {
|
|
62
|
+
success: false,
|
|
63
|
+
format: responseFormat === 'url' ? 'url' : 'base64',
|
|
64
|
+
error: error.message,
|
|
65
|
+
};
|
|
66
|
+
}
|
|
67
|
+
}
|
|
68
|
+
aspectRatioToSize(aspectRatio) {
|
|
69
|
+
const sizeMap = {
|
|
70
|
+
'1:1': '1024x1024',
|
|
71
|
+
'16:9': '1792x1024',
|
|
72
|
+
'9:16': '1024x1792',
|
|
73
|
+
'4:3': '1024x1024',
|
|
74
|
+
'3:4': '1024x1024',
|
|
75
|
+
'3:2': '1792x1024',
|
|
76
|
+
'2:3': '1024x1792',
|
|
77
|
+
};
|
|
78
|
+
return sizeMap[aspectRatio] || '1024x1024';
|
|
79
|
+
}
|
|
80
|
+
}
|
|
81
|
+
exports.OpenAIImageProvider = OpenAIImageProvider;
|
|
82
|
+
//# sourceMappingURL=openai.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../../src/core/image-providers/openai.ts"],"names":[],"mappings":";;;AAEa,QAAA,wBAAwB,GAAG,kCAAkC,CAAC;AAE3E,MAAa,mBAAmB;IAM9B,YAAY,MAAc,EAAE,QAAgB,UAAU,EAAE,OAAgB;QALxE,SAAI,GAAG,cAAc,CAAC;QAMpB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,OAAO,GAAG,OAAO,IAAI,gCAAwB,CAAC;IACrD,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,OAA+B;QACjD,MAAM,EACJ,cAAc,EACd,MAAM,EACN,WAAW,GAAG,KAAK,EACnB,cAAc,GAAG,UAAU,GAC5B,GAAG,OAAO,CAAC;QAEZ,IAAI,CAAC;YACH,iDAAiD;YACjD,sCAAsC;YACtC,MAAM,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC;YAEjD,MAAM,OAAO,GAAG;gBACd,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,MAAM,EAAE,GAAG,MAAM,4CAA4C;gBAC7D,CAAC,EAAE,CAAC;gBACJ,IAAI;gBACJ,eAAe,EAAE,cAAc,KAAK,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,UAAU;aAC/D,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,cAAc,EAAE;gBAC1D,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;oBACxC,cAAc,EAAE,kBAAkB;iBACnC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;aAC9B,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACxC,MAAM,IAAI,KAAK,CAAC,qBAAqB,QAAQ,CAAC,MAAM,MAAM,SAAS,EAAE,CAAC,CAAC;YACzE,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,EAGjC,CAAC;YAEF,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;gBACjB,MAAM,IAAI,KAAK,CAAC,qBAAqB,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YAC/D,CAAC;YAED,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;YACnC,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;YAC5C,CAAC;YAED,IAAI,cAAc,KAAK,KAAK,EAAE,CAAC;gBAC7B,OAAO;oBACL,OAAO,EAAE,IAAI;oBACb,SAAS,EAAE,SAAS,CAAC,GAAG;oBACxB,MAAM,EAAE,KAAK;iBACd,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,OAAO;oBACL,OAAO,EAAE,IAAI;oBACb,SAAS,EAAE,SAAS,CAAC,QAAQ;oBAC7B,MAAM,EAAE,QAAQ;iBACjB,CAAC;YACJ,CAAC;QACH,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,cAAc,KAAK,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,QAAQ;gBACnD,KAAK,EAAE,KAAK,CAAC,OAAO;aACrB,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,iBAAiB,CAAC,WAAmB;QAC3C,MAAM,OAAO,GAA2B;YACtC,KAAK,EAAE,WAAW;YAClB,MAAM,EAAE,WAAW;YACnB,MAAM,EAAE,WAAW;YACnB,KAAK,EAAE,WAAW;YAClB,KAAK,EAAE,WAAW;YAClB,KAAK,EAAE,WAAW;YAClB,KAAK,EAAE,WAAW;SACnB,CAAC;QACF,OAAO,OAAO,CAAC,WAAW,CAAC,IAAI,WAAW,CAAC;IAC7C,CAAC;CACF;AA/FD,kDA+FC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"anthropic.d.ts","sourceRoot":"","sources":["../../../src/core/providers/anthropic.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAI7C,qBAAa,iBAAkB,SAAQ,YAAY;IACjD,IAAI,SAAe;IACnB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,KAAK,CAAS;IACtB,OAAO,CAAC,MAAM,CAAgC;gBAElC,MAAM,EAAE,MAAM,EAAE,KAAK,GAAE,MAAqC;YAM1D,SAAS;IAQjB,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,cAAc,GAAG,OAAO,CAAC,MAAM,CAAC;
|
|
1
|
+
{"version":3,"file":"anthropic.d.ts","sourceRoot":"","sources":["../../../src/core/providers/anthropic.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAI7C,qBAAa,iBAAkB,SAAQ,YAAY;IACjD,IAAI,SAAe;IACnB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,KAAK,CAAS;IACtB,OAAO,CAAC,MAAM,CAAgC;gBAElC,MAAM,EAAE,MAAM,EAAE,KAAK,GAAE,MAAqC;YAM1D,SAAS;IAQjB,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,cAAc,GAAG,OAAO,CAAC,MAAM,CAAC;CAwBtE"}
|
|
@@ -53,17 +53,18 @@ class AnthropicProvider extends base_1.BaseProvider {
|
|
|
53
53
|
async rewrite(text, options) {
|
|
54
54
|
try {
|
|
55
55
|
const client = await this.getClient();
|
|
56
|
+
const originalLength = text.length;
|
|
56
57
|
const response = await client.messages.create({
|
|
57
58
|
model: this.model,
|
|
58
59
|
max_tokens: options.maxTokens ?? 2000,
|
|
59
60
|
temperature: options.temperature ?? 0.7,
|
|
60
|
-
system: this.getSystemPrompt(options),
|
|
61
|
+
system: this.getSystemPrompt(options, originalLength),
|
|
61
62
|
messages: [
|
|
62
|
-
{ role: 'user', content: this.getUserPrompt(text) },
|
|
63
|
+
{ role: 'user', content: this.getUserPrompt(text, options.preserveLength) },
|
|
63
64
|
],
|
|
64
65
|
});
|
|
65
66
|
const content = response.content[0];
|
|
66
|
-
return content.type === 'text' ? content.text.trim() : text;
|
|
67
|
+
return this.cleanupThinkingContent(content.type === 'text' ? content.text.trim() : text);
|
|
67
68
|
}
|
|
68
69
|
catch (error) {
|
|
69
70
|
if (error.code === 'MODULE_NOT_FOUND') {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../../src/core/providers/anthropic.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,iCAAsC;AAKtC,MAAa,iBAAkB,SAAQ,mBAAY;IAMjD,YAAY,MAAc,EAAE,QAAgB,4BAA4B;QACtE,KAAK,EAAE,CAAC;QANV,SAAI,GAAG,WAAW,CAAC;QAGX,WAAM,GAA2B,IAAI,CAAC;QAI5C,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAEO,KAAK,CAAC,SAAS;QACrB,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,SAAS,GAAG,MAAM,kDAAO,mBAAmB,IAAE,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,CAAC;YAC9E,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAoB,CAAC;QAC1E,CAAC;QACD,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,IAAY,EAAE,OAAuB;QACjD,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;
|
|
1
|
+
{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../../src/core/providers/anthropic.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,iCAAsC;AAKtC,MAAa,iBAAkB,SAAQ,mBAAY;IAMjD,YAAY,MAAc,EAAE,QAAgB,4BAA4B;QACtE,KAAK,EAAE,CAAC;QANV,SAAI,GAAG,WAAW,CAAC;QAGX,WAAM,GAA2B,IAAI,CAAC;QAI5C,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAEO,KAAK,CAAC,SAAS;QACrB,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,SAAS,GAAG,MAAM,kDAAO,mBAAmB,IAAE,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,CAAC;YAC9E,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAoB,CAAC;QAC1E,CAAC;QACD,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,IAAY,EAAE,OAAuB;QACjD,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;YACtC,MAAM,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC;YAEnC,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC;gBAC5C,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,UAAU,EAAE,OAAO,CAAC,SAAS,IAAI,IAAI;gBACrC,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,GAAG;gBACvC,MAAM,EAAE,IAAI,CAAC,eAAe,CAAC,OAAO,EAAE,cAAc,CAAC;gBACrD,QAAQ,EAAE;oBACR,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,OAAO,CAAC,cAAc,CAAC,EAAE;iBAC5E;aACF,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;YACpC,OAAO,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAC3F,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,IAAI,KAAK,CAAC,IAAI,KAAK,kBAAkB,EAAE,CAAC;gBACtC,MAAM,IAAI,KAAK,CAAC,iEAAiE,CAAC,CAAC;YACrF,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,wBAAwB,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;QAC3D,CAAC;IACH,CAAC;CACF;AA5CD,8CA4CC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"azure-openai.d.ts","sourceRoot":"","sources":["../../../src/core/providers/azure-openai.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAI7C,eAAO,MAAM,gCAAgC,eAAe,CAAC;AAE7D,qBAAa,mBAAoB,SAAQ,YAAY;IACnD,IAAI,SAAkB;IACtB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,MAAM,CAAkC;gBAG9C,MAAM,EAAE,MAAM,EACd,UAAU,GAAE,MAAsB,EAClC,QAAQ,CAAC,EAAE,MAAM,EACjB,UAAU,CAAC,EAAE,MAAM;YAeP,SAAS;IAYjB,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,cAAc,GAAG,OAAO,CAAC,MAAM,CAAC;
|
|
1
|
+
{"version":3,"file":"azure-openai.d.ts","sourceRoot":"","sources":["../../../src/core/providers/azure-openai.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAI7C,eAAO,MAAM,gCAAgC,eAAe,CAAC;AAE7D,qBAAa,mBAAoB,SAAQ,YAAY;IACnD,IAAI,SAAkB;IACtB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,MAAM,CAAkC;gBAG9C,MAAM,EAAE,MAAM,EACd,UAAU,GAAE,MAAsB,EAClC,QAAQ,CAAC,EAAE,MAAM,EACjB,UAAU,CAAC,EAAE,MAAM;YAeP,SAAS;IAYjB,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,cAAc,GAAG,OAAO,CAAC,MAAM,CAAC;CAuBtE"}
|
|
@@ -63,16 +63,17 @@ class AzureOpenAIProvider extends base_1.BaseProvider {
|
|
|
63
63
|
async rewrite(text, options) {
|
|
64
64
|
try {
|
|
65
65
|
const client = await this.getClient();
|
|
66
|
+
const originalLength = text.length;
|
|
66
67
|
const response = await client.chat.completions.create({
|
|
67
68
|
model: this.deployment,
|
|
68
69
|
messages: [
|
|
69
|
-
{ role: 'system', content: this.getSystemPrompt(options) },
|
|
70
|
-
{ role: 'user', content: this.getUserPrompt(text) },
|
|
70
|
+
{ role: 'system', content: this.getSystemPrompt(options, originalLength) },
|
|
71
|
+
{ role: 'user', content: this.getUserPrompt(text, options.preserveLength) },
|
|
71
72
|
],
|
|
72
73
|
temperature: options.temperature ?? 0.7,
|
|
73
74
|
max_tokens: options.maxTokens ?? 2000,
|
|
74
75
|
});
|
|
75
|
-
return response.choices[0]?.message?.content?.trim() || text;
|
|
76
|
+
return this.cleanupThinkingContent(response.choices[0]?.message?.content?.trim() || text);
|
|
76
77
|
}
|
|
77
78
|
catch (error) {
|
|
78
79
|
if (error.code === 'MODULE_NOT_FOUND') {
|