@req2rank/core 0.1.0-r7
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/LICENSE +21 -0
- package/dist/adaptive-calibration.d.ts +13 -0
- package/dist/adaptive-calibration.d.ts.map +1 -0
- package/dist/adaptive-calibration.js +39 -0
- package/dist/adaptive-calibration.js.map +1 -0
- package/dist/adaptive-calibration.test.d.ts +2 -0
- package/dist/adaptive-calibration.test.d.ts.map +1 -0
- package/dist/adaptive-calibration.test.js +20 -0
- package/dist/adaptive-calibration.test.js.map +1 -0
- package/dist/checkpoint-key.d.ts +3 -0
- package/dist/checkpoint-key.d.ts.map +1 -0
- package/dist/checkpoint-key.js +29 -0
- package/dist/checkpoint-key.js.map +1 -0
- package/dist/checkpoint-key.test.d.ts +2 -0
- package/dist/checkpoint-key.test.d.ts.map +1 -0
- package/dist/checkpoint-key.test.js +32 -0
- package/dist/checkpoint-key.test.js.map +1 -0
- package/dist/config.d.ts +205 -0
- package/dist/config.d.ts.map +1 -0
- package/dist/config.js +85 -0
- package/dist/config.js.map +1 -0
- package/dist/domain-taxonomy.d.ts +13 -0
- package/dist/domain-taxonomy.d.ts.map +1 -0
- package/dist/domain-taxonomy.js +12 -0
- package/dist/domain-taxonomy.js.map +1 -0
- package/dist/evaluation-panel.d.ts +27 -0
- package/dist/evaluation-panel.d.ts.map +1 -0
- package/dist/evaluation-panel.js +158 -0
- package/dist/evaluation-panel.js.map +1 -0
- package/dist/evaluation-panel.test.d.ts +2 -0
- package/dist/evaluation-panel.test.d.ts.map +1 -0
- package/dist/evaluation-panel.test.js +185 -0
- package/dist/evaluation-panel.test.js.map +1 -0
- package/dist/evidence-chain.d.ts +11 -0
- package/dist/evidence-chain.d.ts.map +1 -0
- package/dist/evidence-chain.js +33 -0
- package/dist/evidence-chain.js.map +1 -0
- package/dist/evidence-chain.test.d.ts +2 -0
- package/dist/evidence-chain.test.d.ts.map +1 -0
- package/dist/evidence-chain.test.js +16 -0
- package/dist/evidence-chain.test.js.map +1 -0
- package/dist/execution-engine.d.ts +29 -0
- package/dist/execution-engine.d.ts.map +1 -0
- package/dist/execution-engine.js +102 -0
- package/dist/execution-engine.js.map +1 -0
- package/dist/execution-engine.test.d.ts +2 -0
- package/dist/execution-engine.test.d.ts.map +1 -0
- package/dist/execution-engine.test.js +86 -0
- package/dist/execution-engine.test.js.map +1 -0
- package/dist/hub-client.d.ts +21 -0
- package/dist/hub-client.d.ts.map +1 -0
- package/dist/hub-client.js +99 -0
- package/dist/hub-client.js.map +1 -0
- package/dist/hub-client.test.d.ts +2 -0
- package/dist/hub-client.test.d.ts.map +1 -0
- package/dist/hub-client.test.js +129 -0
- package/dist/hub-client.test.js.map +1 -0
- package/dist/index.d.ts +19 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +19 -0
- package/dist/index.js.map +1 -0
- package/dist/leaderboard-query.d.ts +29 -0
- package/dist/leaderboard-query.d.ts.map +1 -0
- package/dist/leaderboard-query.js +59 -0
- package/dist/leaderboard-query.js.map +1 -0
- package/dist/leaderboard-query.test.d.ts +2 -0
- package/dist/leaderboard-query.test.d.ts.map +1 -0
- package/dist/leaderboard-query.test.js +34 -0
- package/dist/leaderboard-query.test.js.map +1 -0
- package/dist/local-store.d.ts +18 -0
- package/dist/local-store.d.ts.map +1 -0
- package/dist/local-store.js +181 -0
- package/dist/local-store.js.map +1 -0
- package/dist/local-store.test.d.ts +2 -0
- package/dist/local-store.test.d.ts.map +1 -0
- package/dist/local-store.test.js +153 -0
- package/dist/local-store.test.js.map +1 -0
- package/dist/pipeline-stage-handoff.test.d.ts +2 -0
- package/dist/pipeline-stage-handoff.test.d.ts.map +1 -0
- package/dist/pipeline-stage-handoff.test.js +290 -0
- package/dist/pipeline-stage-handoff.test.js.map +1 -0
- package/dist/pipeline.d.ts +67 -0
- package/dist/pipeline.d.ts.map +1 -0
- package/dist/pipeline.js +493 -0
- package/dist/pipeline.js.map +1 -0
- package/dist/providers/anthropic.d.ts +8 -0
- package/dist/providers/anthropic.d.ts.map +1 -0
- package/dist/providers/anthropic.js +45 -0
- package/dist/providers/anthropic.js.map +1 -0
- package/dist/providers/base.d.ts +36 -0
- package/dist/providers/base.d.ts.map +1 -0
- package/dist/providers/base.js +47 -0
- package/dist/providers/base.js.map +1 -0
- package/dist/providers/custom.d.ts +6 -0
- package/dist/providers/custom.d.ts.map +1 -0
- package/dist/providers/custom.js +6 -0
- package/dist/providers/custom.js.map +1 -0
- package/dist/providers/google.d.ts +8 -0
- package/dist/providers/google.d.ts.map +1 -0
- package/dist/providers/google.js +48 -0
- package/dist/providers/google.js.map +1 -0
- package/dist/providers/index.d.ts +31 -0
- package/dist/providers/index.d.ts.map +1 -0
- package/dist/providers/index.js +63 -0
- package/dist/providers/index.js.map +1 -0
- package/dist/providers/openai.d.ts +18 -0
- package/dist/providers/openai.d.ts.map +1 -0
- package/dist/providers/openai.js +111 -0
- package/dist/providers/openai.js.map +1 -0
- package/dist/providers/providers.test.d.ts +2 -0
- package/dist/providers/providers.test.d.ts.map +1 -0
- package/dist/providers/providers.test.js +71 -0
- package/dist/providers/providers.test.js.map +1 -0
- package/dist/requirement-generator.d.ts +29 -0
- package/dist/requirement-generator.d.ts.map +1 -0
- package/dist/requirement-generator.js +358 -0
- package/dist/requirement-generator.js.map +1 -0
- package/dist/requirement-generator.test.d.ts +2 -0
- package/dist/requirement-generator.test.d.ts.map +1 -0
- package/dist/requirement-generator.test.js +182 -0
- package/dist/requirement-generator.test.js.map +1 -0
- package/dist/sandbox.d.ts +32 -0
- package/dist/sandbox.d.ts.map +1 -0
- package/dist/sandbox.js +124 -0
- package/dist/sandbox.js.map +1 -0
- package/dist/sandbox.test.d.ts +2 -0
- package/dist/sandbox.test.d.ts.map +1 -0
- package/dist/sandbox.test.js +20 -0
- package/dist/sandbox.test.js.map +1 -0
- package/dist/scoring-engine.d.ts +15 -0
- package/dist/scoring-engine.d.ts.map +1 -0
- package/dist/scoring-engine.js +109 -0
- package/dist/scoring-engine.js.map +1 -0
- package/dist/scoring-engine.test.d.ts +2 -0
- package/dist/scoring-engine.test.d.ts.map +1 -0
- package/dist/scoring-engine.test.js +137 -0
- package/dist/scoring-engine.test.js.map +1 -0
- package/dist/submit-payload-builder.d.ts +9 -0
- package/dist/submit-payload-builder.d.ts.map +1 -0
- package/dist/submit-payload-builder.js +23 -0
- package/dist/submit-payload-builder.js.map +1 -0
- package/dist/submit-payload-builder.test.d.ts +2 -0
- package/dist/submit-payload-builder.test.d.ts.map +1 -0
- package/dist/submit-payload-builder.test.js +75 -0
- package/dist/submit-payload-builder.test.js.map +1 -0
- package/dist/submitter-types.d.ts +54 -0
- package/dist/submitter-types.d.ts.map +1 -0
- package/dist/submitter-types.js +2 -0
- package/dist/submitter-types.js.map +1 -0
- package/dist/types.d.ts +40 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +2 -0
- package/dist/types.js.map +1 -0
- package/package.json +36 -0
|
@@ -0,0 +1,45 @@
|
|
|
1
|
+
import { HttpProviderBase } from "./base.js";
|
|
2
|
+
export class AnthropicProvider extends HttpProviderBase {
|
|
3
|
+
id = "anthropic";
|
|
4
|
+
name = "Anthropic";
|
|
5
|
+
constructor(apiKey, baseUrl = "https://api.anthropic.com/v1") {
|
|
6
|
+
super(apiKey, baseUrl);
|
|
7
|
+
}
|
|
8
|
+
async chat(params) {
|
|
9
|
+
this.requireApiKey();
|
|
10
|
+
const startedAt = Date.now();
|
|
11
|
+
const system = params.messages.filter((message) => message.role === "system").map((message) => message.content).join("\n\n");
|
|
12
|
+
const messages = params.messages
|
|
13
|
+
.filter((message) => message.role !== "system")
|
|
14
|
+
.map((message) => ({ role: message.role, content: message.content }));
|
|
15
|
+
const response = await this.fetchWithRetry(`${this.baseUrl}/messages`, {
|
|
16
|
+
method: "POST",
|
|
17
|
+
headers: {
|
|
18
|
+
"Content-Type": "application/json",
|
|
19
|
+
"x-api-key": this.apiKey,
|
|
20
|
+
"anthropic-version": "2023-06-01"
|
|
21
|
+
},
|
|
22
|
+
body: JSON.stringify({
|
|
23
|
+
model: params.model,
|
|
24
|
+
max_tokens: params.maxTokens ?? 2048,
|
|
25
|
+
temperature: params.temperature,
|
|
26
|
+
system: system.length > 0 ? system : undefined,
|
|
27
|
+
messages
|
|
28
|
+
})
|
|
29
|
+
});
|
|
30
|
+
if (!response.ok) {
|
|
31
|
+
throw new Error(`Anthropic request failed: ${response.status} ${response.statusText}`);
|
|
32
|
+
}
|
|
33
|
+
const data = (await response.json());
|
|
34
|
+
const firstTextChunk = data.content?.find((item) => item.type === "text")?.text ?? "";
|
|
35
|
+
return {
|
|
36
|
+
content: firstTextChunk,
|
|
37
|
+
usage: {
|
|
38
|
+
promptTokens: data.usage?.input_tokens ?? 0,
|
|
39
|
+
completionTokens: data.usage?.output_tokens ?? 0
|
|
40
|
+
},
|
|
41
|
+
latencyMs: Date.now() - startedAt
|
|
42
|
+
};
|
|
43
|
+
}
|
|
44
|
+
}
|
|
45
|
+
//# sourceMappingURL=anthropic.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../src/providers/anthropic.ts"],"names":[],"mappings":"AAAA,OAAO,EAA4B,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAUvE,MAAM,OAAO,iBAAkB,SAAQ,gBAAgB;IACrD,EAAE,GAAG,WAAW,CAAC;IACjB,IAAI,GAAG,WAAW,CAAC;IAEnB,YAAY,MAAc,EAAE,OAAO,GAAG,8BAA8B;QAClE,KAAK,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,MAAkB;QAC3B,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,MAAM,MAAM,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC7H,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ;aAC7B,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,QAAQ,CAAC;aAC9C,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAExE,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,GAAG,IAAI,CAAC,OAAO,WAAW,EAAE;YACrE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,WAAW,EAAE,IAAI,CAAC,MAAM;gBACxB,mBAAmB,EAAE,YAAY;aAClC;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,UAAU,EAAE,MAAM,CAAC,SAAS,IAAI,IAAI;gBACpC,WAAW,EAAE,MAAM,CAAC,WAAW;gBAC/B,MAAM,EAAE,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS;gBAC9C,QAAQ;aACT,CAAC;SACH,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,6BAA6B,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;QACzF,CAAC;QAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAA6B,CAAC;QACjE,MAAM,cAAc,GAAG,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC;QAEtF,OAAO;YACL,OAAO,EAAE,cAAc;YACvB,KAAK,EAAE;gBACL,YAAY,EAAE,IAAI,CAAC,KAAK,EAAE,YAAY,IAAI,CAAC;gBAC3C,gBAAgB,EAAE,IAAI,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC;aACjD;YACD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;SAClC,CAAC;IACJ,CAAC;CACF"}
|
|
@@ -0,0 +1,36 @@
|
|
|
1
|
+
export interface Message {
|
|
2
|
+
role: "system" | "user" | "assistant";
|
|
3
|
+
content: string;
|
|
4
|
+
}
|
|
5
|
+
export interface ChatParams {
|
|
6
|
+
model: string;
|
|
7
|
+
messages: Message[];
|
|
8
|
+
temperature?: number;
|
|
9
|
+
maxTokens?: number;
|
|
10
|
+
responseFormat?: "text" | "json";
|
|
11
|
+
}
|
|
12
|
+
export interface ChatResponse {
|
|
13
|
+
content: string;
|
|
14
|
+
usage: {
|
|
15
|
+
promptTokens: number;
|
|
16
|
+
completionTokens: number;
|
|
17
|
+
};
|
|
18
|
+
latencyMs: number;
|
|
19
|
+
}
|
|
20
|
+
export interface LLMProvider {
|
|
21
|
+
id: string;
|
|
22
|
+
name: string;
|
|
23
|
+
chat(params: ChatParams): Promise<ChatResponse>;
|
|
24
|
+
}
|
|
25
|
+
export declare abstract class HttpProviderBase implements LLMProvider {
|
|
26
|
+
abstract id: string;
|
|
27
|
+
abstract name: string;
|
|
28
|
+
protected readonly apiKey: string;
|
|
29
|
+
protected readonly baseUrl: string;
|
|
30
|
+
protected constructor(apiKey: string, baseUrl: string);
|
|
31
|
+
abstract chat(params: ChatParams): Promise<ChatResponse>;
|
|
32
|
+
protected sleep(ms: number): Promise<void>;
|
|
33
|
+
protected fetchWithRetry(url: string, init: RequestInit, timeoutMs?: number): Promise<Response>;
|
|
34
|
+
protected requireApiKey(): void;
|
|
35
|
+
}
|
|
36
|
+
//# sourceMappingURL=base.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../src/providers/base.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,OAAO;IACtB,IAAI,EAAE,QAAQ,GAAG,MAAM,GAAG,WAAW,CAAC;IACtC,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,UAAU;IACzB,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,cAAc,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;CAClC;AAED,MAAM,WAAW,YAAY;IAC3B,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE;QACL,YAAY,EAAE,MAAM,CAAC;QACrB,gBAAgB,EAAE,MAAM,CAAC;KAC1B,CAAC;IACF,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,WAAW;IAC1B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,MAAM,EAAE,UAAU,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC;CACjD;AAED,8BAAsB,gBAAiB,YAAW,WAAW;IAC3D,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IAEtB,SAAS,CAAC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IAClC,SAAS,CAAC,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IAEnC,SAAS,aAAa,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM;IAKrD,QAAQ,CAAC,IAAI,CAAC,MAAM,EAAE,UAAU,GAAG,OAAO,CAAC,YAAY,CAAC;cAExC,KAAK,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;cAIhC,cAAc,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,SAAS,SAAS,GAAG,OAAO,CAAC,QAAQ,CAAC;IAmCrG,SAAS,CAAC,aAAa,IAAI,IAAI;CAKhC"}
|
|
@@ -0,0 +1,47 @@
|
|
|
1
|
+
export class HttpProviderBase {
|
|
2
|
+
apiKey;
|
|
3
|
+
baseUrl;
|
|
4
|
+
constructor(apiKey, baseUrl) {
|
|
5
|
+
this.apiKey = apiKey;
|
|
6
|
+
this.baseUrl = baseUrl;
|
|
7
|
+
}
|
|
8
|
+
async sleep(ms) {
|
|
9
|
+
await new Promise((resolve) => setTimeout(resolve, ms));
|
|
10
|
+
}
|
|
11
|
+
async fetchWithRetry(url, init, timeoutMs = 30_000) {
|
|
12
|
+
let timeoutRetriesLeft = 1;
|
|
13
|
+
for (let attempt = 0; attempt < 4; attempt += 1) {
|
|
14
|
+
const controller = new AbortController();
|
|
15
|
+
const timer = setTimeout(() => controller.abort(), timeoutMs);
|
|
16
|
+
try {
|
|
17
|
+
const response = await fetch(url, {
|
|
18
|
+
...init,
|
|
19
|
+
signal: controller.signal
|
|
20
|
+
});
|
|
21
|
+
if (response.status === 429 && attempt < 3) {
|
|
22
|
+
await this.sleep(2 ** attempt * 1000);
|
|
23
|
+
continue;
|
|
24
|
+
}
|
|
25
|
+
return response;
|
|
26
|
+
}
|
|
27
|
+
catch (error) {
|
|
28
|
+
const timeoutLike = error instanceof Error && (error.name === "AbortError" || error.name === "TimeoutError");
|
|
29
|
+
if (timeoutLike && timeoutRetriesLeft > 0) {
|
|
30
|
+
timeoutRetriesLeft -= 1;
|
|
31
|
+
continue;
|
|
32
|
+
}
|
|
33
|
+
throw error;
|
|
34
|
+
}
|
|
35
|
+
finally {
|
|
36
|
+
clearTimeout(timer);
|
|
37
|
+
}
|
|
38
|
+
}
|
|
39
|
+
throw new Error("Request failed after retry attempts");
|
|
40
|
+
}
|
|
41
|
+
requireApiKey() {
|
|
42
|
+
if (!this.apiKey || this.apiKey.trim().length === 0) {
|
|
43
|
+
throw new Error(`${this.name} API key is required.`);
|
|
44
|
+
}
|
|
45
|
+
}
|
|
46
|
+
}
|
|
47
|
+
//# sourceMappingURL=base.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"base.js","sourceRoot":"","sources":["../../src/providers/base.ts"],"names":[],"mappings":"AA4BA,MAAM,OAAgB,gBAAgB;IAIjB,MAAM,CAAS;IACf,OAAO,CAAS;IAEnC,YAAsB,MAAc,EAAE,OAAe;QACnD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAIS,KAAK,CAAC,KAAK,CAAC,EAAU;QAC9B,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;IAC1D,CAAC;IAES,KAAK,CAAC,cAAc,CAAC,GAAW,EAAE,IAAiB,EAAE,SAAS,GAAG,MAAM;QAC/E,IAAI,kBAAkB,GAAG,CAAC,CAAC;QAE3B,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,CAAC,EAAE,CAAC;YAChD,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;YACzC,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,SAAS,CAAC,CAAC;YAE9D,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;oBAChC,GAAG,IAAI;oBACP,MAAM,EAAE,UAAU,CAAC,MAAM;iBAC1B,CAAC,CAAC;gBAEH,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;oBAC3C,MAAM,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,OAAO,GAAG,IAAI,CAAC,CAAC;oBACtC,SAAS;gBACX,CAAC;gBAED,OAAO,QAAQ,CAAC;YAClB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,WAAW,GAAG,KAAK,YAAY,KAAK,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,YAAY,IAAI,KAAK,CAAC,IAAI,KAAK,cAAc,CAAC,CAAC;gBAC7G,IAAI,WAAW,IAAI,kBAAkB,GAAG,CAAC,EAAE,CAAC;oBAC1C,kBAAkB,IAAI,CAAC,CAAC;oBACxB,SAAS;gBACX,CAAC;gBAED,MAAM,KAAK,CAAC;YACd,CAAC;oBAAS,CAAC;gBACT,YAAY,CAAC,KAAK,CAAC,CAAC;YACtB,CAAC;QACH,CAAC;QAED,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;IACzD,CAAC;IAES,aAAa;QACrB,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACpD,MAAM,IAAI,KAAK,CAAC,GAAG,IAAI,CAAC,IAAI,uBAAuB,CAAC,CAAC;QACvD,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"custom.d.ts","sourceRoot":"","sources":["../../src/providers/custom.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAE7C,qBAAa,8BAA+B,SAAQ,cAAc;IAChE,EAAE,SAAY;IACd,IAAI,SAA8B;CACnC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"custom.js","sourceRoot":"","sources":["../../src/providers/custom.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAE7C,MAAM,OAAO,8BAA+B,SAAQ,cAAc;IAChE,EAAE,GAAG,QAAQ,CAAC;IACd,IAAI,GAAG,0BAA0B,CAAC;CACnC"}
|
|
@@ -0,0 +1,8 @@
|
|
|
1
|
+
import { ChatParams, ChatResponse, HttpProviderBase } from "./base.js";
|
|
2
|
+
export declare class GoogleProvider extends HttpProviderBase {
|
|
3
|
+
id: string;
|
|
4
|
+
name: string;
|
|
5
|
+
constructor(apiKey: string, baseUrl?: string);
|
|
6
|
+
chat(params: ChatParams): Promise<ChatResponse>;
|
|
7
|
+
}
|
|
8
|
+
//# sourceMappingURL=google.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"google.d.ts","sourceRoot":"","sources":["../../src/providers/google.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,gBAAgB,EAAW,MAAM,WAAW,CAAC;AAuBhF,qBAAa,cAAe,SAAQ,gBAAgB;IAClD,EAAE,SAAY;IACd,IAAI,SAAY;gBAEJ,MAAM,EAAE,MAAM,EAAE,OAAO,SAAqD;IAIlF,IAAI,CAAC,MAAM,EAAE,UAAU,GAAG,OAAO,CAAC,YAAY,CAAC;CAmCtD"}
|
|
@@ -0,0 +1,48 @@
|
|
|
1
|
+
import { HttpProviderBase } from "./base.js";
|
|
2
|
+
function toGeminiContents(messages) {
|
|
3
|
+
return messages
|
|
4
|
+
.filter((message) => message.role !== "system")
|
|
5
|
+
.map((message) => ({
|
|
6
|
+
role: message.role === "assistant" ? "model" : "user",
|
|
7
|
+
parts: [{ text: message.content }]
|
|
8
|
+
}));
|
|
9
|
+
}
|
|
10
|
+
export class GoogleProvider extends HttpProviderBase {
|
|
11
|
+
id = "google";
|
|
12
|
+
name = "Google";
|
|
13
|
+
constructor(apiKey, baseUrl = "https://generativelanguage.googleapis.com/v1beta") {
|
|
14
|
+
super(apiKey, baseUrl);
|
|
15
|
+
}
|
|
16
|
+
async chat(params) {
|
|
17
|
+
this.requireApiKey();
|
|
18
|
+
const startedAt = Date.now();
|
|
19
|
+
const response = await this.fetchWithRetry(`${this.baseUrl}/models/${params.model}:generateContent?key=${this.apiKey}`, {
|
|
20
|
+
method: "POST",
|
|
21
|
+
headers: {
|
|
22
|
+
"Content-Type": "application/json"
|
|
23
|
+
},
|
|
24
|
+
body: JSON.stringify({
|
|
25
|
+
contents: toGeminiContents(params.messages),
|
|
26
|
+
generationConfig: {
|
|
27
|
+
temperature: params.temperature,
|
|
28
|
+
maxOutputTokens: params.maxTokens,
|
|
29
|
+
responseMimeType: params.responseFormat === "json" ? "application/json" : "text/plain"
|
|
30
|
+
}
|
|
31
|
+
})
|
|
32
|
+
});
|
|
33
|
+
if (!response.ok) {
|
|
34
|
+
throw new Error(`Google request failed: ${response.status} ${response.statusText}`);
|
|
35
|
+
}
|
|
36
|
+
const data = (await response.json());
|
|
37
|
+
const content = data.candidates?.[0]?.content?.parts?.map((part) => part.text ?? "").join("") ?? "";
|
|
38
|
+
return {
|
|
39
|
+
content,
|
|
40
|
+
usage: {
|
|
41
|
+
promptTokens: data.usageMetadata?.promptTokenCount ?? 0,
|
|
42
|
+
completionTokens: data.usageMetadata?.candidatesTokenCount ?? 0
|
|
43
|
+
},
|
|
44
|
+
latencyMs: Date.now() - startedAt
|
|
45
|
+
};
|
|
46
|
+
}
|
|
47
|
+
}
|
|
48
|
+
//# sourceMappingURL=google.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"google.js","sourceRoot":"","sources":["../../src/providers/google.ts"],"names":[],"mappings":"AAAA,OAAO,EAA4B,gBAAgB,EAAW,MAAM,WAAW,CAAC;AAchF,SAAS,gBAAgB,CAAC,QAAmB;IAC3C,OAAO,QAAQ;SACZ,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,QAAQ,CAAC;SAC9C,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;QACjB,IAAI,EAAE,OAAO,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM;QACrD,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC;KACnC,CAAC,CAAC,CAAC;AACR,CAAC;AAED,MAAM,OAAO,cAAe,SAAQ,gBAAgB;IAClD,EAAE,GAAG,QAAQ,CAAC;IACd,IAAI,GAAG,QAAQ,CAAC;IAEhB,YAAY,MAAc,EAAE,OAAO,GAAG,kDAAkD;QACtF,KAAK,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,MAAkB;QAC3B,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,GAAG,IAAI,CAAC,OAAO,WAAW,MAAM,CAAC,KAAK,wBAAwB,IAAI,CAAC,MAAM,EAAE,EAAE;YACtH,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;aACnC;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,QAAQ,EAAE,gBAAgB,CAAC,MAAM,CAAC,QAAQ,CAAC;gBAC3C,gBAAgB,EAAE;oBAChB,WAAW,EAAE,MAAM,CAAC,WAAW;oBAC/B,eAAe,EAAE,MAAM,CAAC,SAAS;oBACjC,gBAAgB,EAAE,MAAM,CAAC,cAAc,KAAK,MAAM,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,YAAY;iBACvF;aACF,CAAC;SACH,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,0BAA0B,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;QACtF,CAAC;QAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAmB,CAAC;QACvD,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;QAEpG,OAAO;YACL,OAAO;YACP,KAAK,EAAE;gBACL,YAAY,EAAE,IAAI,CAAC,aAAa,EAAE,gBAAgB,IAAI,CAAC;gBACvD,gBAAgB,EAAE,IAAI,CAAC,aAAa,EAAE,oBAAoB,IAAI,CAAC;aAChE;YACD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;SAClC,CAAC;IACJ,CAAC;CACF"}
|
|
@@ -0,0 +1,31 @@
|
|
|
1
|
+
import { z } from "zod";
|
|
2
|
+
import { LLMProvider } from "./base.js";
|
|
3
|
+
export declare const providerConfigSchema: z.ZodEffects<z.ZodObject<{
|
|
4
|
+
provider: z.ZodEnum<["openai", "openai-response", "gemini", "anthropic", "azure-openai", "newapi", "google", "custom"]>;
|
|
5
|
+
apiKey: z.ZodString;
|
|
6
|
+
baseUrl: z.ZodOptional<z.ZodString>;
|
|
7
|
+
}, "strip", z.ZodTypeAny, {
|
|
8
|
+
provider: "custom" | "openai" | "openai-response" | "gemini" | "anthropic" | "azure-openai" | "newapi" | "google";
|
|
9
|
+
apiKey: string;
|
|
10
|
+
baseUrl?: string | undefined;
|
|
11
|
+
}, {
|
|
12
|
+
provider: "custom" | "openai" | "openai-response" | "gemini" | "anthropic" | "azure-openai" | "newapi" | "google";
|
|
13
|
+
apiKey: string;
|
|
14
|
+
baseUrl?: string | undefined;
|
|
15
|
+
}>, {
|
|
16
|
+
provider: "custom" | "openai" | "openai-response" | "gemini" | "anthropic" | "azure-openai" | "newapi" | "google";
|
|
17
|
+
apiKey: string;
|
|
18
|
+
baseUrl?: string | undefined;
|
|
19
|
+
}, {
|
|
20
|
+
provider: "custom" | "openai" | "openai-response" | "gemini" | "anthropic" | "azure-openai" | "newapi" | "google";
|
|
21
|
+
apiKey: string;
|
|
22
|
+
baseUrl?: string | undefined;
|
|
23
|
+
}>;
|
|
24
|
+
export type ProviderConfig = z.infer<typeof providerConfigSchema>;
|
|
25
|
+
export declare function createProvider(config: ProviderConfig): LLMProvider;
|
|
26
|
+
export * from "./base.js";
|
|
27
|
+
export * from "./openai.js";
|
|
28
|
+
export * from "./anthropic.js";
|
|
29
|
+
export * from "./google.js";
|
|
30
|
+
export * from "./custom.js";
|
|
31
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/providers/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,OAAO,EAAE,WAAW,EAAE,MAAM,WAAW,CAAC;AAexC,eAAO,MAAM,oBAAoB;;;;;;;;;;;;;;;;;;;;EAc7B,CAAC;AAEL,MAAM,MAAM,cAAc,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,oBAAoB,CAAC,CAAC;AAElE,wBAAgB,cAAc,CAAC,MAAM,EAAE,cAAc,GAAG,WAAW,CA2BlE;AAED,cAAc,WAAW,CAAC;AAC1B,cAAc,aAAa,CAAC;AAC5B,cAAc,gBAAgB,CAAC;AAC/B,cAAc,aAAa,CAAC;AAC5B,cAAc,aAAa,CAAC"}
|
|
@@ -0,0 +1,63 @@
|
|
|
1
|
+
import { z } from "zod";
|
|
2
|
+
import { AnthropicProvider } from "./anthropic.js";
|
|
3
|
+
import { CustomOpenAICompatibleProvider } from "./custom.js";
|
|
4
|
+
import { GoogleProvider } from "./google.js";
|
|
5
|
+
import { OpenAIProvider } from "./openai.js";
|
|
6
|
+
const providerTypeSchema = z.enum([
|
|
7
|
+
"openai",
|
|
8
|
+
"openai-response",
|
|
9
|
+
"gemini",
|
|
10
|
+
"anthropic",
|
|
11
|
+
"azure-openai",
|
|
12
|
+
"newapi",
|
|
13
|
+
"google",
|
|
14
|
+
"custom"
|
|
15
|
+
]);
|
|
16
|
+
export const providerConfigSchema = z
|
|
17
|
+
.object({
|
|
18
|
+
provider: providerTypeSchema,
|
|
19
|
+
apiKey: z.string().min(1),
|
|
20
|
+
baseUrl: z.string().url().optional()
|
|
21
|
+
})
|
|
22
|
+
.superRefine((value, context) => {
|
|
23
|
+
if ((value.provider === "azure-openai" || value.provider === "newapi") && !value.baseUrl) {
|
|
24
|
+
context.addIssue({
|
|
25
|
+
code: z.ZodIssueCode.custom,
|
|
26
|
+
message: `Required: baseUrl for ${value.provider}`,
|
|
27
|
+
path: ["baseUrl"]
|
|
28
|
+
});
|
|
29
|
+
}
|
|
30
|
+
});
|
|
31
|
+
export function createProvider(config) {
|
|
32
|
+
const parsedConfig = providerConfigSchema.parse(config);
|
|
33
|
+
switch (parsedConfig.provider) {
|
|
34
|
+
case "openai":
|
|
35
|
+
return new OpenAIProvider(parsedConfig.apiKey, parsedConfig.baseUrl);
|
|
36
|
+
case "openai-response":
|
|
37
|
+
return new OpenAIProvider(parsedConfig.apiKey, parsedConfig.baseUrl, { protocol: "responses" });
|
|
38
|
+
case "anthropic":
|
|
39
|
+
return new AnthropicProvider(parsedConfig.apiKey, parsedConfig.baseUrl);
|
|
40
|
+
case "gemini":
|
|
41
|
+
case "google":
|
|
42
|
+
return new GoogleProvider(parsedConfig.apiKey, parsedConfig.baseUrl);
|
|
43
|
+
case "azure-openai":
|
|
44
|
+
return new OpenAIProvider(parsedConfig.apiKey, parsedConfig.baseUrl, {
|
|
45
|
+
authMode: "api-key",
|
|
46
|
+
defaultQuery: {
|
|
47
|
+
"api-version": "2024-10-21"
|
|
48
|
+
}
|
|
49
|
+
});
|
|
50
|
+
case "newapi":
|
|
51
|
+
return new OpenAIProvider(parsedConfig.apiKey, parsedConfig.baseUrl);
|
|
52
|
+
case "custom":
|
|
53
|
+
return new CustomOpenAICompatibleProvider(parsedConfig.apiKey, parsedConfig.baseUrl ?? "http://localhost:11434/v1");
|
|
54
|
+
default:
|
|
55
|
+
throw new Error("Unsupported provider");
|
|
56
|
+
}
|
|
57
|
+
}
|
|
58
|
+
export * from "./base.js";
|
|
59
|
+
export * from "./openai.js";
|
|
60
|
+
export * from "./anthropic.js";
|
|
61
|
+
export * from "./google.js";
|
|
62
|
+
export * from "./custom.js";
|
|
63
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/providers/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,EAAE,8BAA8B,EAAE,MAAM,aAAa,CAAC;AAE7D,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAE7C,MAAM,kBAAkB,GAAG,CAAC,CAAC,IAAI,CAAC;IAChC,QAAQ;IACR,iBAAiB;IACjB,QAAQ;IACR,WAAW;IACX,cAAc;IACd,QAAQ;IACR,QAAQ;IACR,QAAQ;CACT,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,oBAAoB,GAAG,CAAC;KAClC,MAAM,CAAC;IACN,QAAQ,EAAE,kBAAkB;IAC5B,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;IACzB,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE;CACrC,CAAC;KACD,WAAW,CAAC,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE;IAC9B,IAAI,CAAC,KAAK,CAAC,QAAQ,KAAK,cAAc,IAAI,KAAK,CAAC,QAAQ,KAAK,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;QACzF,OAAO,CAAC,QAAQ,CAAC;YACf,IAAI,EAAE,CAAC,CAAC,YAAY,CAAC,MAAM;YAC3B,OAAO,EAAE,yBAAyB,KAAK,CAAC,QAAQ,EAAE;YAClD,IAAI,EAAE,CAAC,SAAS,CAAC;SAClB,CAAC,CAAC;IACL,CAAC;AACH,CAAC,CAAC,CAAC;AAIL,MAAM,UAAU,cAAc,CAAC,MAAsB;IACnD,MAAM,YAAY,GAAG,oBAAoB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IAExD,QAAQ,YAAY,CAAC,QAAQ,EAAE,CAAC;QAC9B,KAAK,QAAQ;YACX,OAAO,IAAI,cAAc,CAAC,YAAY,CAAC,MAAM,EAAE,YAAY,CAAC,OAAO,CAAC,CAAC;QACvE,KAAK,iBAAiB;YACpB,OAAO,IAAI,cAAc,CAAC,YAAY,CAAC,MAAM,EAAE,YAAY,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,CAAC,CAAC;QAClG,KAAK,WAAW;YACd,OAAO,IAAI,iBAAiB,CAAC,YAAY,CAAC,MAAM,EAAE,YAAY,CAAC,OAAO,CAAC,CAAC;QAC1E,KAAK,QAAQ,CAAC;QACd,KAAK,QAAQ;YACX,OAAO,IAAI,cAAc,CAAC,YAAY,CAAC,MAAM,EAAE,YAAY,CAAC,OAAO,CAAC,CAAC;QACvE,KAAK,cAAc;YACjB,OAAO,IAAI,cAAc,CAAC,YAAY,CAAC,MAAM,EAAE,YAAY,CAAC,OAAO,EAAE;gBACnE,QAAQ,EAAE,SAAS;gBACnB,YAAY,EAAE;oBACZ,aAAa,EAAE,YAAY;iBAC5B;aACF,CAAC,CAAC;QACL,KAAK,QAAQ;YACX,OAAO,IAAI,cAAc,CAAC,YAAY,CAAC,MAAM,EAAE,YAAY,CAAC,OAAO,CAAC,CAAC;QACvE,KAAK,QAAQ;YACX,OAAO,IAAI,8BAA8B,CAAC,YAAY,CAAC,MAAM,EAAE,YAAY,CAAC,OAAO,IAAI,2BAA2B,CAAC,CAAC;QACtH;YACE,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;IAC5C,CAAC;AACH,CAAC;AAED,cAAc,WAAW,CAAC;AAC1B,cAAc,aAAa,CAAC;AAC5B,cAAc,gBAAgB,CAAC;AAC/B,cAAc,aAAa,CAAC;AAC5B,cAAc,aAAa,CAAC"}
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
import { ChatParams, ChatResponse, HttpProviderBase } from "./base.js";
|
|
2
|
+
type OpenAIProtocol = "chat-completions" | "responses";
|
|
3
|
+
interface OpenAIProviderOptions {
|
|
4
|
+
protocol?: OpenAIProtocol;
|
|
5
|
+
authMode?: "bearer" | "api-key";
|
|
6
|
+
defaultQuery?: Record<string, string>;
|
|
7
|
+
}
|
|
8
|
+
export declare class OpenAIProvider extends HttpProviderBase {
|
|
9
|
+
id: string;
|
|
10
|
+
name: string;
|
|
11
|
+
private readonly protocol;
|
|
12
|
+
private readonly authMode;
|
|
13
|
+
private readonly defaultQuery;
|
|
14
|
+
constructor(apiKey: string, baseUrl?: string, options?: OpenAIProviderOptions);
|
|
15
|
+
chat(params: ChatParams): Promise<ChatResponse>;
|
|
16
|
+
}
|
|
17
|
+
export {};
|
|
18
|
+
//# sourceMappingURL=openai.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../src/providers/openai.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAyBvE,KAAK,cAAc,GAAG,kBAAkB,GAAG,WAAW,CAAC;AAEvD,UAAU,qBAAqB;IAC7B,QAAQ,CAAC,EAAE,cAAc,CAAC;IAC1B,QAAQ,CAAC,EAAE,QAAQ,GAAG,SAAS,CAAC;IAChC,YAAY,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CACvC;AAqCD,qBAAa,cAAe,SAAQ,gBAAgB;IAClD,EAAE,SAAY;IACd,IAAI,SAAY;IAChB,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAiB;IAC1C,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAuB;IAChD,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAqC;gBAEtD,MAAM,EAAE,MAAM,EAAE,OAAO,SAA8B,EAAE,OAAO,GAAE,qBAA0B;IAOhG,IAAI,CAAC,MAAM,EAAE,UAAU,GAAG,OAAO,CAAC,YAAY,CAAC;CA8EtD"}
|
|
@@ -0,0 +1,111 @@
|
|
|
1
|
+
import { HttpProviderBase } from "./base.js";
|
|
2
|
+
function buildUrl(baseUrl, endpoint, query) {
|
|
3
|
+
const normalizedBase = baseUrl.replace(/\/+$/, "");
|
|
4
|
+
const normalizedEndpoint = endpoint.replace(/^\/+/, "");
|
|
5
|
+
const url = new URL(`${normalizedBase}/${normalizedEndpoint}`);
|
|
6
|
+
if (query) {
|
|
7
|
+
for (const [key, value] of Object.entries(query)) {
|
|
8
|
+
url.searchParams.set(key, value);
|
|
9
|
+
}
|
|
10
|
+
}
|
|
11
|
+
return url.toString();
|
|
12
|
+
}
|
|
13
|
+
function toResponsesInput(messages) {
|
|
14
|
+
return messages.map((message) => ({
|
|
15
|
+
role: message.role,
|
|
16
|
+
content: [{ type: "input_text", text: message.content }]
|
|
17
|
+
}));
|
|
18
|
+
}
|
|
19
|
+
function extractResponsesContent(data) {
|
|
20
|
+
if (typeof data.output_text === "string" && data.output_text.length > 0) {
|
|
21
|
+
return data.output_text;
|
|
22
|
+
}
|
|
23
|
+
return (data.output
|
|
24
|
+
?.flatMap((item) => item.content ?? [])
|
|
25
|
+
.filter((item) => item.type === "output_text" || item.type === "text")
|
|
26
|
+
.map((item) => item.text ?? "")
|
|
27
|
+
.join("") ?? "");
|
|
28
|
+
}
|
|
29
|
+
export class OpenAIProvider extends HttpProviderBase {
|
|
30
|
+
id = "openai";
|
|
31
|
+
name = "OpenAI";
|
|
32
|
+
protocol;
|
|
33
|
+
authMode;
|
|
34
|
+
defaultQuery;
|
|
35
|
+
constructor(apiKey, baseUrl = "https://api.openai.com/v1", options = {}) {
|
|
36
|
+
super(apiKey, baseUrl);
|
|
37
|
+
this.protocol = options.protocol ?? "chat-completions";
|
|
38
|
+
this.authMode = options.authMode ?? "bearer";
|
|
39
|
+
this.defaultQuery = options.defaultQuery;
|
|
40
|
+
}
|
|
41
|
+
async chat(params) {
|
|
42
|
+
this.requireApiKey();
|
|
43
|
+
const startedAt = Date.now();
|
|
44
|
+
const headers = {
|
|
45
|
+
"Content-Type": "application/json"
|
|
46
|
+
};
|
|
47
|
+
if (this.authMode === "api-key") {
|
|
48
|
+
headers["api-key"] = this.apiKey;
|
|
49
|
+
}
|
|
50
|
+
else {
|
|
51
|
+
headers.Authorization = `Bearer ${this.apiKey}`;
|
|
52
|
+
}
|
|
53
|
+
if (this.protocol === "responses") {
|
|
54
|
+
const payload = {
|
|
55
|
+
model: params.model,
|
|
56
|
+
input: toResponsesInput(params.messages),
|
|
57
|
+
temperature: params.temperature,
|
|
58
|
+
max_output_tokens: params.maxTokens,
|
|
59
|
+
text: params.responseFormat === "json"
|
|
60
|
+
? {
|
|
61
|
+
format: {
|
|
62
|
+
type: "json_object"
|
|
63
|
+
}
|
|
64
|
+
}
|
|
65
|
+
: undefined
|
|
66
|
+
};
|
|
67
|
+
const response = await this.fetchWithRetry(buildUrl(this.baseUrl, "responses", this.defaultQuery), {
|
|
68
|
+
method: "POST",
|
|
69
|
+
headers,
|
|
70
|
+
body: JSON.stringify(payload)
|
|
71
|
+
});
|
|
72
|
+
if (!response.ok) {
|
|
73
|
+
throw new Error(`OpenAI request failed: ${response.status} ${response.statusText}`);
|
|
74
|
+
}
|
|
75
|
+
const data = (await response.json());
|
|
76
|
+
return {
|
|
77
|
+
content: extractResponsesContent(data),
|
|
78
|
+
usage: {
|
|
79
|
+
promptTokens: data.usage?.input_tokens ?? 0,
|
|
80
|
+
completionTokens: data.usage?.output_tokens ?? 0
|
|
81
|
+
},
|
|
82
|
+
latencyMs: Date.now() - startedAt
|
|
83
|
+
};
|
|
84
|
+
}
|
|
85
|
+
const payload = {
|
|
86
|
+
model: params.model,
|
|
87
|
+
messages: params.messages,
|
|
88
|
+
temperature: params.temperature,
|
|
89
|
+
max_tokens: params.maxTokens,
|
|
90
|
+
response_format: params.responseFormat === "json" ? { type: "json_object" } : undefined
|
|
91
|
+
};
|
|
92
|
+
const response = await this.fetchWithRetry(buildUrl(this.baseUrl, "chat/completions", this.defaultQuery), {
|
|
93
|
+
method: "POST",
|
|
94
|
+
headers,
|
|
95
|
+
body: JSON.stringify(payload)
|
|
96
|
+
});
|
|
97
|
+
if (!response.ok) {
|
|
98
|
+
throw new Error(`OpenAI request failed: ${response.status} ${response.statusText}`);
|
|
99
|
+
}
|
|
100
|
+
const data = (await response.json());
|
|
101
|
+
return {
|
|
102
|
+
content: data.choices?.[0]?.message?.content ?? "",
|
|
103
|
+
usage: {
|
|
104
|
+
promptTokens: data.usage?.prompt_tokens ?? 0,
|
|
105
|
+
completionTokens: data.usage?.completion_tokens ?? 0
|
|
106
|
+
},
|
|
107
|
+
latencyMs: Date.now() - startedAt
|
|
108
|
+
};
|
|
109
|
+
}
|
|
110
|
+
}
|
|
111
|
+
//# sourceMappingURL=openai.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/providers/openai.ts"],"names":[],"mappings":"AAAA,OAAO,EAA4B,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAiCvE,SAAS,QAAQ,CAAC,OAAe,EAAE,QAAgB,EAAE,KAA8B;IACjF,MAAM,cAAc,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IACnD,MAAM,kBAAkB,GAAG,QAAQ,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IACxD,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,GAAG,cAAc,IAAI,kBAAkB,EAAE,CAAC,CAAC;IAE/D,IAAI,KAAK,EAAE,CAAC;QACV,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACjD,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QACnC,CAAC;IACH,CAAC;IAED,OAAO,GAAG,CAAC,QAAQ,EAAE,CAAC;AACxB,CAAC;AAED,SAAS,gBAAgB,CAAC,QAAgC;IACxD,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;QAChC,IAAI,EAAE,OAAO,CAAC,IAAI;QAClB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC;KACzD,CAAC,CAAC,CAAC;AACN,CAAC;AAED,SAAS,uBAAuB,CAAC,IAAqB;IACpD,IAAI,OAAO,IAAI,CAAC,WAAW,KAAK,QAAQ,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACxE,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAED,OAAO,CACL,IAAI,CAAC,MAAM;QACT,EAAE,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC;SACtC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,aAAa,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC;SACrE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC;SAC9B,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAClB,CAAC;AACJ,CAAC;AAED,MAAM,OAAO,cAAe,SAAQ,gBAAgB;IAClD,EAAE,GAAG,QAAQ,CAAC;IACd,IAAI,GAAG,QAAQ,CAAC;IACC,QAAQ,CAAiB;IACzB,QAAQ,CAAuB;IAC/B,YAAY,CAAqC;IAElE,YAAY,MAAc,EAAE,OAAO,GAAG,2BAA2B,EAAE,UAAiC,EAAE;QACpG,KAAK,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACvB,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,kBAAkB,CAAC;QACvD,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,QAAQ,CAAC;QAC7C,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,MAAkB;QAC3B,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,MAAM,OAAO,GAA2B;YACtC,cAAc,EAAE,kBAAkB;SACnC,CAAC;QACF,IAAI,IAAI,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;YAChC,OAAO,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC;QACnC,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,aAAa,GAAG,UAAU,IAAI,CAAC,MAAM,EAAE,CAAC;QAClD,CAAC;QAED,IAAI,IAAI,CAAC,QAAQ,KAAK,WAAW,EAAE,CAAC;YAClC,MAAM,OAAO,GAAG;gBACd,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,KAAK,EAAE,gBAAgB,CAAC,MAAM,CAAC,QAAQ,CAAC;gBACxC,WAAW,EAAE,MAAM,CAAC,WAAW;gBAC/B,iBAAiB,EAAE,MAAM,CAAC,SAAS;gBACnC,IAAI,EACF,MAAM,CAAC,cAAc,KAAK,MAAM;oBAC9B,CAAC,CAAC;wBACE,MAAM,EAAE;4BACN,IAAI,EAAE,aAAa;yBACpB;qBACF;oBACH,CAAC,CAAC,SAAS;aAChB,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,WAAW,EAAE,IAAI,CAAC,YAAY,CAAC,EAAE;gBACjG,MAAM,EAAE,MAAM;gBACd,OAAO;gBACP,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;aAC9B,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,IAAI,KAAK,CAAC,0BAA0B,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;YACtF,CAAC;YAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAoB,CAAC;YACxD,OAAO;gBACL,OAAO,EAAE,uBAAuB,CAAC,IAAI,CAAC;gBACtC,KAAK,EAAE;oBACL,YAAY,EAAE,IAAI,CAAC,KAAK,EAAE,YAAY,IAAI,CAAC;oBAC3C,gBAAgB,EAAE,IAAI,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC;iBACjD;gBACD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;aAClC,CAAC;QACJ,CAAC;QAED,MAAM,OAAO,GAAG;YACd,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,UAAU,EAAE,MAAM,CAAC,SAAS;YAC5B,eAAe,EAAE,MAAM,CAAC,cAAc,KAAK,MAAM,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC,SAAS;SACxF,CAAC;QAEF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,kBAAkB,EAAE,IAAI,CAAC,YAAY,CAAC,EAAE;YACxG,MAAM,EAAE,MAAM;YACd,OAAO;YACP,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;SAC9B,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,0BAA0B,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;QACtF,CAAC;QAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAyB,CAAC;QAC7D,OAAO;YACL,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE;YAClD,KAAK,EAAE;gBACL,YAAY,EAAE,IAAI,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC;gBAC5C,gBAAgB,EAAE,IAAI,CAAC,KAAK,EAAE,iBAAiB,IAAI,CAAC;aACrD;YACD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;SAClC,CAAC;IACJ,CAAC;CACF"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"providers.test.d.ts","sourceRoot":"","sources":["../../src/providers/providers.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,71 @@
|
|
|
1
|
+
import { afterEach, describe, expect, it, vi } from "vitest";
|
|
2
|
+
import { createProvider } from "./index.js";
|
|
3
|
+
describe("provider adapters", () => {
|
|
4
|
+
afterEach(() => {
|
|
5
|
+
vi.unstubAllGlobals();
|
|
6
|
+
vi.restoreAllMocks();
|
|
7
|
+
});
|
|
8
|
+
it("creates providers for all configured adapter types", () => {
|
|
9
|
+
expect(createProvider({ provider: "openai", apiKey: "k" }).id).toBe("openai");
|
|
10
|
+
expect(createProvider({ provider: "openai-response", apiKey: "k" }).id).toBe("openai");
|
|
11
|
+
expect(createProvider({ provider: "anthropic", apiKey: "k" }).id).toBe("anthropic");
|
|
12
|
+
expect(createProvider({ provider: "gemini", apiKey: "k" }).id).toBe("google");
|
|
13
|
+
expect(createProvider({
|
|
14
|
+
provider: "azure-openai",
|
|
15
|
+
apiKey: "k",
|
|
16
|
+
baseUrl: "https://example.openai.azure.com/openai/deployments/gpt-4o-mini"
|
|
17
|
+
}).id).toBe("openai");
|
|
18
|
+
expect(createProvider({ provider: "newapi", apiKey: "k", baseUrl: "https://newapi.example.com/v1" }).id).toBe("openai");
|
|
19
|
+
});
|
|
20
|
+
it("requires baseUrl for azure-openai and newapi", () => {
|
|
21
|
+
expect(() => createProvider({ provider: "azure-openai", apiKey: "k" })).toThrow("Required");
|
|
22
|
+
expect(() => createProvider({ provider: "newapi", apiKey: "k" })).toThrow("Required");
|
|
23
|
+
});
|
|
24
|
+
it("retries 429 responses with exponential backoff", async () => {
|
|
25
|
+
const provider = createProvider({ provider: "openai", apiKey: "k" });
|
|
26
|
+
const sleeps = [];
|
|
27
|
+
provider.sleep = async (ms) => {
|
|
28
|
+
sleeps.push(ms);
|
|
29
|
+
};
|
|
30
|
+
const fetchMock = vi.fn()
|
|
31
|
+
.mockResolvedValueOnce(new Response("rate-limit", { status: 429, statusText: "Too Many Requests" }))
|
|
32
|
+
.mockResolvedValueOnce(new Response(JSON.stringify({
|
|
33
|
+
choices: [{ message: { content: "ok" } }],
|
|
34
|
+
usage: { prompt_tokens: 10, completion_tokens: 5 }
|
|
35
|
+
}), { status: 200, headers: { "content-type": "application/json" } }));
|
|
36
|
+
vi.stubGlobal("fetch", fetchMock);
|
|
37
|
+
const result = await provider.chat({
|
|
38
|
+
model: "gpt-4o-mini",
|
|
39
|
+
messages: [{ role: "user", content: "hello" }]
|
|
40
|
+
});
|
|
41
|
+
expect(result.content).toBe("ok");
|
|
42
|
+
expect(fetchMock).toHaveBeenCalledTimes(2);
|
|
43
|
+
expect(sleeps).toEqual([1000]);
|
|
44
|
+
});
|
|
45
|
+
it("throws provider specific errors for non-retryable responses", async () => {
|
|
46
|
+
const provider = createProvider({ provider: "gemini", apiKey: "k" });
|
|
47
|
+
vi.stubGlobal("fetch", vi.fn(async () => new Response("bad", { status: 401, statusText: "Unauthorized" })));
|
|
48
|
+
await expect(provider.chat({
|
|
49
|
+
model: "gemini-1.5-pro",
|
|
50
|
+
messages: [{ role: "user", content: "hello" }]
|
|
51
|
+
})).rejects.toThrow("Google request failed: 401 Unauthorized");
|
|
52
|
+
});
|
|
53
|
+
it("supports OpenAI responses API format", async () => {
|
|
54
|
+
const provider = createProvider({ provider: "openai-response", apiKey: "k" });
|
|
55
|
+
const fetchMock = vi.fn().mockResolvedValue(new Response(JSON.stringify({
|
|
56
|
+
output_text: "ok-from-responses",
|
|
57
|
+
usage: { input_tokens: 12, output_tokens: 4 }
|
|
58
|
+
}), { status: 200, headers: { "content-type": "application/json" } }));
|
|
59
|
+
vi.stubGlobal("fetch", fetchMock);
|
|
60
|
+
const result = await provider.chat({
|
|
61
|
+
model: "gpt-4o-mini",
|
|
62
|
+
messages: [{ role: "user", content: "hello" }]
|
|
63
|
+
});
|
|
64
|
+
expect(result.content).toBe("ok-from-responses");
|
|
65
|
+
expect(result.usage.promptTokens).toBe(12);
|
|
66
|
+
expect(result.usage.completionTokens).toBe(4);
|
|
67
|
+
expect(fetchMock).toHaveBeenCalledTimes(1);
|
|
68
|
+
expect(fetchMock.mock.calls[0]?.[0]).toBe("https://api.openai.com/v1/responses");
|
|
69
|
+
});
|
|
70
|
+
});
|
|
71
|
+
//# sourceMappingURL=providers.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"providers.test.js","sourceRoot":"","sources":["../../src/providers/providers.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC7D,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAE5C,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,gBAAgB,EAAE,CAAC;QACtB,EAAE,CAAC,eAAe,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,CAAC,cAAc,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC9E,MAAM,CAAC,cAAc,CAAC,EAAE,QAAQ,EAAE,iBAAiB,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACvF,MAAM,CAAC,cAAc,CAAC,EAAE,QAAQ,EAAE,WAAW,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACpF,MAAM,CAAC,cAAc,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC9E,MAAM,CACJ,cAAc,CAAC;YACb,QAAQ,EAAE,cAAc;YACxB,MAAM,EAAE,GAAG;YACX,OAAO,EAAE,iEAAiE;SAC3E,CAAC,CAAC,EAAE,CACN,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACjB,MAAM,CAAC,cAAc,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,+BAA+B,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC1H,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,CAAC,GAAG,EAAE,CAAC,cAAc,CAAC,EAAE,QAAQ,EAAE,cAAc,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAC5F,MAAM,CAAC,GAAG,EAAE,CAAC,cAAc,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IACxF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC9D,MAAM,QAAQ,GAAG,cAAc,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC;QACrE,MAAM,MAAM,GAAa,EAAE,CAAC;QAC1B,QAAiE,CAAC,KAAK,GAAG,KAAK,EAAE,EAAU,EAAE,EAAE;YAC/F,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAClB,CAAC,CAAC;QAEF,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAgB;aACpC,qBAAqB,CAAC,IAAI,QAAQ,CAAC,YAAY,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,UAAU,EAAE,mBAAmB,EAAE,CAAC,CAAC;aACnG,qBAAqB,CACpB,IAAI,QAAQ,CACV,IAAI,CAAC,SAAS,CAAC;YACb,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,CAAC;YACzC,KAAK,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,iBAAiB,EAAE,CAAC,EAAE;SACnD,CAAC,EACF,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,EAAE,CACjE,CACF,CAAC;QACJ,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;QAElC,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC;YACjC,KAAK,EAAE,aAAa;YACpB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;SAC/C,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC3C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,MAAM,QAAQ,GAAG,cAAc,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC;QACrE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,IAAI,QAAQ,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,UAAU,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC,CAAC;QAE5G,MAAM,MAAM,CACV,QAAQ,CAAC,IAAI,CAAC;YACZ,KAAK,EAAE,gBAAgB;YACvB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;SAC/C,CAAC,CACH,CAAC,OAAO,CAAC,OAAO,CAAC,yCAAyC,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,QAAQ,GAAG,cAAc,CAAC,EAAE,QAAQ,EAAE,iBAAiB,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC;QAC9E,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAgB,CAAC,iBAAiB,CACvD,IAAI,QAAQ,CACV,IAAI,CAAC,SAAS,CAAC;YACb,WAAW,EAAE,mBAAmB;YAChC,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,aAAa,EAAE,CAAC,EAAE;SAC9C,CAAC,EACF,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,EAAE,CACjE,CACF,CAAC;QACF,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;QAElC,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC;YACjC,KAAK,EAAE,aAAa;YACpB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;SAC/C,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QACjD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC3C,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9C,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC3C,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,qCAAqC,CAAC,CAAC;IACnF,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,29 @@
|
|
|
1
|
+
import { Complexity, ProjectRequirement } from "./types.js";
|
|
2
|
+
import { LLMProvider } from "./providers/base.js";
|
|
3
|
+
export interface RequirementSeed {
|
|
4
|
+
id: string;
|
|
5
|
+
template: string;
|
|
6
|
+
skills: string[];
|
|
7
|
+
complexity: Complexity;
|
|
8
|
+
slots: Record<string, string[]>;
|
|
9
|
+
}
|
|
10
|
+
export interface GenerationInput {
|
|
11
|
+
skills: string[];
|
|
12
|
+
complexity: Complexity;
|
|
13
|
+
domain: string;
|
|
14
|
+
scenario: string;
|
|
15
|
+
techStack: string[];
|
|
16
|
+
extraConstraints?: string[];
|
|
17
|
+
seed?: string;
|
|
18
|
+
}
|
|
19
|
+
export interface GenerationModelConfig {
|
|
20
|
+
provider: LLMProvider;
|
|
21
|
+
model: string;
|
|
22
|
+
}
|
|
23
|
+
export declare class RequirementGenerator {
|
|
24
|
+
private readonly seeds;
|
|
25
|
+
constructor(seeds?: RequirementSeed[]);
|
|
26
|
+
generate(input: GenerationInput, modelConfig: GenerationModelConfig): Promise<ProjectRequirement>;
|
|
27
|
+
}
|
|
28
|
+
export declare const defaultSeeds: RequirementSeed[];
|
|
29
|
+
//# sourceMappingURL=requirement-generator.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"requirement-generator.d.ts","sourceRoot":"","sources":["../src/requirement-generator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAC5D,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAGlD,MAAM,WAAW,eAAe;IAC9B,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,UAAU,EAAE,UAAU,CAAC;IACvB,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;CACjC;AAED,MAAM,WAAW,eAAe;IAC9B,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,UAAU,EAAE,UAAU,CAAC;IACvB,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,MAAM,EAAE,CAAC;IACpB,gBAAgB,CAAC,EAAE,MAAM,EAAE,CAAC;IAC5B,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,qBAAqB;IACpC,QAAQ,EAAE,WAAW,CAAC;IACtB,KAAK,EAAE,MAAM,CAAC;CACf;AAwTD,qBAAa,oBAAoB;IAC/B,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAoB;gBAE9B,KAAK,GAAE,eAAe,EAAkB;IAI9C,QAAQ,CAAC,KAAK,EAAE,eAAe,EAAE,WAAW,EAAE,qBAAqB,GAAG,OAAO,CAAC,kBAAkB,CAAC;CA2GxG;AAED,eAAO,MAAM,YAAY,mBAAgB,CAAC"}
|