atabey 0.0.12 → 0.0.13
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/ATABEY.md +45 -499
- package/PRIVACY.md +161 -0
- package/README.md +353 -88
- package/bin/validate-al-registry.js +4 -4
- package/dist/framework-mcp/src/tools/definitions.js +23 -1
- package/dist/framework-mcp/src/tools/definitions.js.map +1 -1
- package/dist/framework-mcp/src/tools/gateway/llm_gateway_query.d.ts +19 -0
- package/dist/framework-mcp/src/tools/gateway/llm_gateway_query.js +138 -0
- package/dist/framework-mcp/src/tools/gateway/llm_gateway_query.js.map +1 -0
- package/dist/framework-mcp/src/tools/gateway/llm_gateway_status.d.ts +8 -0
- package/dist/framework-mcp/src/tools/gateway/llm_gateway_status.js +91 -0
- package/dist/framework-mcp/src/tools/gateway/llm_gateway_status.js.map +1 -0
- package/dist/framework-mcp/src/tools/index.d.ts +1 -1
- package/dist/framework-mcp/src/tools/index.js +35 -20
- package/dist/framework-mcp/src/tools/index.js.map +1 -1
- package/dist/framework-mcp/src/tools/quality/analyze_code_quality.d.ts +11 -0
- package/dist/framework-mcp/src/tools/quality/analyze_code_quality.js +166 -0
- package/dist/framework-mcp/src/tools/quality/analyze_code_quality.js.map +1 -0
- package/dist/framework-mcp/src/tools/quality/check_architecture_compliance.d.ts +11 -0
- package/dist/framework-mcp/src/tools/quality/check_architecture_compliance.js +135 -0
- package/dist/framework-mcp/src/tools/quality/check_architecture_compliance.js.map +1 -0
- package/dist/framework-mcp/src/tools/schemas.d.ts +88 -8
- package/dist/framework-mcp/src/tools/schemas.js +29 -0
- package/dist/framework-mcp/src/tools/schemas.js.map +1 -1
- package/dist/framework-mcp/src/tools/shell/run_command.js +2 -2
- package/dist/framework-mcp/src/tools/shell/run_command.js.map +1 -1
- package/dist/src/cli/commands/check.js +2 -2
- package/dist/src/cli/commands/check.js.map +1 -1
- package/dist/src/cli/commands/dashboard.d.ts +1 -1
- package/dist/src/cli/commands/dashboard.js +264 -41
- package/dist/src/cli/commands/dashboard.js.map +1 -1
- package/dist/src/cli/commands/gateway.d.ts +1 -0
- package/dist/src/cli/commands/gateway.js +172 -0
- package/dist/src/cli/commands/gateway.js.map +1 -0
- package/dist/src/cli/commands/init/create-agent.js +3 -3
- package/dist/src/cli/commands/init/create-agent.js.map +1 -1
- package/dist/src/cli/commands/init/scaffold-core.d.ts +1 -1
- package/dist/src/cli/commands/init/scaffold-core.js +1 -1
- package/dist/src/cli/commands/init/scaffold-core.js.map +1 -1
- package/dist/src/cli/commands/init.d.ts +2 -0
- package/dist/src/cli/commands/init.js +105 -23
- package/dist/src/cli/commands/init.js.map +1 -1
- package/dist/src/cli/commands/mcp.d.ts +7 -0
- package/dist/src/cli/commands/mcp.js +45 -22
- package/dist/src/cli/commands/mcp.js.map +1 -1
- package/dist/src/cli/commands/orchestrate.d.ts +8 -7
- package/dist/src/cli/commands/orchestrate.js +2 -1
- package/dist/src/cli/commands/orchestrate.js.map +1 -1
- package/dist/src/cli/commands/quickstart.d.ts +5 -5
- package/dist/src/cli/commands/quickstart.js +36 -36
- package/dist/src/cli/commands/quickstart.js.map +1 -1
- package/dist/src/cli/index.js +230 -320
- package/dist/src/cli/index.js.map +1 -1
- package/dist/src/cli/platforms/core.d.ts +12 -0
- package/dist/src/cli/platforms/core.js +73 -0
- package/dist/src/cli/platforms/core.js.map +1 -0
- package/dist/src/cli/platforms/index.d.ts +5 -0
- package/dist/src/cli/platforms/index.js +6 -0
- package/dist/src/cli/platforms/index.js.map +1 -0
- package/dist/src/cli/platforms/paths.d.ts +15 -0
- package/dist/src/cli/platforms/paths.js +102 -0
- package/dist/src/cli/platforms/paths.js.map +1 -0
- package/dist/src/cli/platforms/scaffold.d.ts +2 -0
- package/dist/src/cli/platforms/scaffold.js +73 -0
- package/dist/src/cli/platforms/scaffold.js.map +1 -0
- package/dist/src/cli/platforms/utils.d.ts +4 -0
- package/dist/src/cli/platforms/utils.js +76 -0
- package/dist/src/cli/platforms/utils.js.map +1 -0
- package/dist/src/cli/shims.js +4 -4
- package/dist/src/cli/utils/fs.d.ts +1 -1
- package/dist/src/cli/utils/fs.js +1 -1
- package/dist/src/cli/utils/fs.js.map +1 -1
- package/dist/src/modules/agents/definitions.d.ts +2 -2
- package/dist/src/modules/agents/definitions.js +59 -23
- package/dist/src/modules/agents/definitions.js.map +1 -1
- package/dist/src/modules/contracts/api.d.ts +178 -0
- package/dist/src/modules/contracts/api.js +56 -0
- package/dist/src/modules/contracts/api.js.map +1 -0
- package/dist/src/modules/contracts/index.d.ts +5 -0
- package/dist/src/modules/contracts/index.js +6 -0
- package/dist/src/modules/contracts/index.js.map +1 -0
- package/dist/src/modules/engines/agent-executor.d.ts +27 -28
- package/dist/src/modules/engines/agent-executor.js +97 -69
- package/dist/src/modules/engines/agent-executor.js.map +1 -1
- package/dist/src/modules/engines/quality-gate.d.ts +7 -7
- package/dist/src/modules/engines/quality-gate.js +7 -7
- package/dist/src/modules/engines/routing-engine.d.ts +18 -27
- package/dist/src/modules/engines/routing-engine.js +70 -44
- package/dist/src/modules/engines/routing-engine.js.map +1 -1
- package/dist/src/modules/gateway/circuit-breaker.d.ts +40 -0
- package/dist/src/modules/gateway/circuit-breaker.js +163 -0
- package/dist/src/modules/gateway/circuit-breaker.js.map +1 -0
- package/dist/src/modules/gateway/cost-tracker.d.ts +55 -0
- package/dist/src/modules/gateway/cost-tracker.js +127 -0
- package/dist/src/modules/gateway/cost-tracker.js.map +1 -0
- package/dist/src/modules/gateway/gateway.d.ts +68 -0
- package/dist/src/modules/gateway/gateway.js +235 -0
- package/dist/src/modules/gateway/gateway.js.map +1 -0
- package/dist/src/modules/gateway/index.d.ts +19 -0
- package/dist/src/modules/gateway/index.js +19 -0
- package/dist/src/modules/gateway/index.js.map +1 -0
- package/dist/src/modules/gateway/load-balancer.d.ts +86 -0
- package/dist/src/modules/gateway/load-balancer.js +230 -0
- package/dist/src/modules/gateway/load-balancer.js.map +1 -0
- package/dist/src/modules/gateway/providers/anthropic.d.ts +14 -0
- package/dist/src/modules/gateway/providers/anthropic.js +169 -0
- package/dist/src/modules/gateway/providers/anthropic.js.map +1 -0
- package/dist/src/modules/gateway/providers/base.d.ts +38 -0
- package/dist/src/modules/gateway/providers/base.js +42 -0
- package/dist/src/modules/gateway/providers/base.js.map +1 -0
- package/dist/src/modules/gateway/providers/google.d.ts +18 -0
- package/dist/src/modules/gateway/providers/google.js +176 -0
- package/dist/src/modules/gateway/providers/google.js.map +1 -0
- package/dist/src/modules/gateway/providers/grok.d.ts +14 -0
- package/dist/src/modules/gateway/providers/grok.js +166 -0
- package/dist/src/modules/gateway/providers/grok.js.map +1 -0
- package/dist/src/modules/gateway/providers/index.d.ts +21 -0
- package/dist/src/modules/gateway/providers/index.js +38 -0
- package/dist/src/modules/gateway/providers/index.js.map +1 -0
- package/dist/src/modules/gateway/providers/ollama.d.ts +23 -0
- package/dist/src/modules/gateway/providers/ollama.js +187 -0
- package/dist/src/modules/gateway/providers/ollama.js.map +1 -0
- package/dist/src/modules/gateway/providers/openai.d.ts +14 -0
- package/dist/src/modules/gateway/providers/openai.js +170 -0
- package/dist/src/modules/gateway/providers/openai.js.map +1 -0
- package/dist/src/modules/gateway/rate-limiter.d.ts +41 -0
- package/dist/src/modules/gateway/rate-limiter.js +119 -0
- package/dist/src/modules/gateway/rate-limiter.js.map +1 -0
- package/dist/src/modules/gateway/types.d.ts +114 -0
- package/dist/src/modules/gateway/types.js +8 -0
- package/dist/src/modules/gateway/types.js.map +1 -0
- package/dist/src/modules/memory/core.d.ts +4 -12
- package/dist/src/modules/memory/core.js +10 -17
- package/dist/src/modules/memory/core.js.map +1 -1
- package/dist/src/modules/memory/embedding.d.ts +28 -0
- package/dist/src/modules/memory/embedding.js +131 -0
- package/dist/src/modules/memory/embedding.js.map +1 -0
- package/dist/src/modules/providers/definitions.d.ts +9 -0
- package/dist/src/modules/providers/definitions.js +141 -0
- package/dist/src/modules/providers/definitions.js.map +1 -0
- package/dist/src/modules/providers/registry.d.ts +12 -0
- package/dist/src/modules/providers/registry.js +19 -0
- package/dist/src/modules/providers/registry.js.map +1 -0
- package/dist/src/modules/providers/shared.d.ts +1 -0
- package/dist/src/modules/providers/shared.js +105 -0
- package/dist/src/modules/providers/shared.js.map +1 -0
- package/dist/src/modules/providers/types.d.ts +14 -0
- package/dist/src/modules/providers/types.js +2 -0
- package/dist/src/modules/providers/types.js.map +1 -0
- package/dist/src/modules/skills/adapter-skills.d.ts +68 -0
- package/dist/src/modules/skills/adapter-skills.js +450 -0
- package/dist/src/modules/skills/adapter-skills.js.map +1 -0
- package/dist/src/shared/audit.d.ts +105 -0
- package/dist/src/shared/audit.js +220 -0
- package/dist/src/shared/audit.js.map +1 -0
- package/dist/src/shared/fs.js +4 -2
- package/dist/src/shared/fs.js.map +1 -1
- package/dist/src/shared/lock.d.ts +82 -0
- package/dist/src/shared/lock.js +160 -0
- package/dist/src/shared/lock.js.map +1 -0
- package/dist/src/shared/logger.d.ts +2 -0
- package/dist/src/shared/logger.js +15 -5
- package/dist/src/shared/logger.js.map +1 -1
- package/dist/src/shared/pii.d.ts +37 -0
- package/dist/src/shared/pii.js +122 -0
- package/dist/src/shared/pii.js.map +1 -0
- package/dist/src/shared/retention.d.ts +63 -0
- package/dist/src/shared/retention.js +171 -0
- package/dist/src/shared/retention.js.map +1 -0
- package/dist/src/shared/storage.d.ts +3 -1
- package/dist/src/shared/storage.js +3 -1
- package/dist/src/shared/storage.js.map +1 -1
- package/dist/tests/adapter.test.js +1 -1
- package/dist/tests/adapter.test.js.map +1 -1
- package/dist/tests/adapters/paths.test.js +4 -4
- package/dist/tests/adapters/paths.test.js.map +1 -1
- package/dist/tests/cli/commands/dashboard-coverage.test.d.ts +1 -0
- package/dist/tests/cli/commands/dashboard-coverage.test.js +88 -0
- package/dist/tests/cli/commands/dashboard-coverage.test.js.map +1 -0
- package/dist/tests/cli/init/adapter_init.test.js +4 -5
- package/dist/tests/cli/init/adapter_init.test.js.map +1 -1
- package/dist/tests/cli/init/unified_init.test.js +5 -7
- package/dist/tests/cli/init/unified_init.test.js.map +1 -1
- package/dist/tests/integration/agent_flow.test.js +14 -8
- package/dist/tests/integration/agent_flow.test.js.map +1 -1
- package/dist/tests/modules/adapters/adapters.test.js +2 -2
- package/dist/tests/modules/adapters/adapters.test.js.map +1 -1
- package/dist/tests/modules/contracts/contracts.test.d.ts +1 -0
- package/dist/tests/modules/contracts/contracts.test.js +73 -0
- package/dist/tests/modules/contracts/contracts.test.js.map +1 -0
- package/dist/tests/modules/engines/agent-executor.test.js +30 -23
- package/dist/tests/modules/engines/agent-executor.test.js.map +1 -1
- package/dist/tests/modules/gateway/circuit-breaker.test.d.ts +1 -0
- package/dist/tests/modules/gateway/circuit-breaker.test.js +124 -0
- package/dist/tests/modules/gateway/circuit-breaker.test.js.map +1 -0
- package/dist/tests/modules/gateway/cost-tracker.test.d.ts +1 -0
- package/dist/tests/modules/gateway/cost-tracker.test.js +74 -0
- package/dist/tests/modules/gateway/cost-tracker.test.js.map +1 -0
- package/dist/tests/modules/gateway/load-balancer.test.d.ts +1 -0
- package/dist/tests/modules/gateway/load-balancer.test.js +70 -0
- package/dist/tests/modules/gateway/load-balancer.test.js.map +1 -0
- package/dist/tests/modules/gateway/rate-limiter.test.d.ts +1 -0
- package/dist/tests/modules/gateway/rate-limiter.test.js +62 -0
- package/dist/tests/modules/gateway/rate-limiter.test.js.map +1 -0
- package/dist/tests/orchestrator-dependencies.test.js +23 -6
- package/dist/tests/orchestrator-dependencies.test.js.map +1 -1
- package/dist/tests/shared/audit.test.d.ts +1 -0
- package/dist/tests/shared/audit.test.js +53 -0
- package/dist/tests/shared/audit.test.js.map +1 -0
- package/dist/tests/storage-operations.test.d.ts +1 -0
- package/dist/tests/storage-operations.test.js +116 -0
- package/dist/tests/storage-operations.test.js.map +1 -0
- package/framework-mcp/README.md +324 -77
- package/framework-mcp/dist/dashboard/assets/index-BAPGLiWE.js +267 -0
- package/framework-mcp/dist/dashboard/index.html +15 -0
- package/framework-mcp/dist/framework-mcp/src/tools/definitions.js +23 -1
- package/framework-mcp/dist/framework-mcp/src/tools/gateway/llm_gateway_query.js +137 -0
- package/framework-mcp/dist/framework-mcp/src/tools/gateway/llm_gateway_status.js +90 -0
- package/framework-mcp/dist/framework-mcp/src/tools/index.js +35 -20
- package/framework-mcp/dist/framework-mcp/src/tools/quality/analyze_code_quality.js +165 -0
- package/framework-mcp/dist/framework-mcp/src/tools/quality/check_architecture_compliance.js +134 -0
- package/framework-mcp/dist/framework-mcp/src/tools/schemas.js +29 -0
- package/framework-mcp/dist/framework-mcp/src/tools/shell/run_command.js +2 -2
- package/framework-mcp/dist/src/modules/memory/core.js +10 -17
- package/framework-mcp/dist/src/modules/memory/embedding.js +130 -0
- package/framework-mcp/dist/src/shared/fs.js +4 -2
- package/framework-mcp/dist/src/shared/logger.js +15 -5
- package/framework-mcp/dist/src/shared/pii.js +121 -0
- package/framework-mcp/dist/src/shared/storage.js +3 -1
- package/framework-mcp/package.json +1 -1
- package/package.json +10 -4
- package/templates/full/ATABEY_FULL.md +515 -0
|
@@ -0,0 +1,169 @@
|
|
|
1
|
+
import { BaseProvider } from "./base.js";
|
|
2
|
+
export class AnthropicProvider extends BaseProvider {
|
|
3
|
+
constructor(config) {
|
|
4
|
+
super({
|
|
5
|
+
id: "anthropic",
|
|
6
|
+
name: "Anthropic",
|
|
7
|
+
baseUrl: "https://api.anthropic.com/v1",
|
|
8
|
+
apiKey: process.env.ANTHROPIC_API_KEY || "",
|
|
9
|
+
models: [
|
|
10
|
+
"claude-3-5-sonnet-latest", "claude-3-5-haiku-latest",
|
|
11
|
+
"claude-3-opus-latest", "claude-3-sonnet-20240229",
|
|
12
|
+
"claude-3-haiku-20240307",
|
|
13
|
+
],
|
|
14
|
+
defaultModel: "claude-3-5-sonnet-latest",
|
|
15
|
+
maxRetries: 3,
|
|
16
|
+
timeoutMs: 60000,
|
|
17
|
+
rpmLimit: 200,
|
|
18
|
+
tpmLimit: 100000,
|
|
19
|
+
costPer1kInputTokens: 0.003,
|
|
20
|
+
costPer1kOutputTokens: 0.015,
|
|
21
|
+
enabled: true,
|
|
22
|
+
priority: 2,
|
|
23
|
+
weight: 4,
|
|
24
|
+
...config,
|
|
25
|
+
});
|
|
26
|
+
}
|
|
27
|
+
get id() { return "anthropic"; }
|
|
28
|
+
get name() { return "Anthropic Claude"; }
|
|
29
|
+
async chat(request) {
|
|
30
|
+
const startTime = Date.now();
|
|
31
|
+
const model = request.model || this.config.defaultModel;
|
|
32
|
+
const traceId = request.traceId;
|
|
33
|
+
if (!this.config.apiKey) {
|
|
34
|
+
return this.errorResponse("Anthropic API key not configured. Set ANTHROPIC_API_KEY env var.", model, traceId);
|
|
35
|
+
}
|
|
36
|
+
try {
|
|
37
|
+
const controller = new AbortController();
|
|
38
|
+
const timeout = setTimeout(() => controller.abort(), this.config.timeoutMs);
|
|
39
|
+
const systemMessage = request.system;
|
|
40
|
+
const messages = request.messages.filter(m => m.role !== "system");
|
|
41
|
+
const response = await fetch(`${this.config.baseUrl}/messages`, {
|
|
42
|
+
method: "POST",
|
|
43
|
+
headers: {
|
|
44
|
+
"Content-Type": "application/json",
|
|
45
|
+
"x-api-key": this.config.apiKey,
|
|
46
|
+
"anthropic-version": "2023-06-01",
|
|
47
|
+
},
|
|
48
|
+
body: JSON.stringify({
|
|
49
|
+
model,
|
|
50
|
+
system: systemMessage,
|
|
51
|
+
messages: messages.map(m => ({
|
|
52
|
+
role: m.role === "assistant" ? "assistant" : "user",
|
|
53
|
+
content: m.content,
|
|
54
|
+
})),
|
|
55
|
+
temperature: request.temperature ?? 0.7,
|
|
56
|
+
max_tokens: request.maxTokens ?? 4096,
|
|
57
|
+
stream: false,
|
|
58
|
+
}),
|
|
59
|
+
signal: controller.signal,
|
|
60
|
+
});
|
|
61
|
+
clearTimeout(timeout);
|
|
62
|
+
if (!response.ok) {
|
|
63
|
+
const errorText = await response.text();
|
|
64
|
+
return this.errorResponse(`Anthropic API error (${response.status}): ${errorText}`, model, traceId);
|
|
65
|
+
}
|
|
66
|
+
const data = await response.json();
|
|
67
|
+
const latencyMs = Date.now() - startTime;
|
|
68
|
+
const content = data.content?.map(c => c.text).join("") || "";
|
|
69
|
+
const inputTokens = data.usage?.input_tokens || this.estimateTokens(request.messages.map(m => m.content).join(" "));
|
|
70
|
+
const outputTokens = data.usage?.output_tokens || this.estimateTokens(content);
|
|
71
|
+
const cost = (inputTokens / 1000) * this.config.costPer1kInputTokens +
|
|
72
|
+
(outputTokens / 1000) * this.config.costPer1kOutputTokens;
|
|
73
|
+
return {
|
|
74
|
+
success: true,
|
|
75
|
+
content,
|
|
76
|
+
model: data.model || model,
|
|
77
|
+
provider: "anthropic",
|
|
78
|
+
usage: { inputTokens, outputTokens, totalTokens: inputTokens + outputTokens },
|
|
79
|
+
latencyMs,
|
|
80
|
+
cost,
|
|
81
|
+
cached: false,
|
|
82
|
+
traceId,
|
|
83
|
+
};
|
|
84
|
+
}
|
|
85
|
+
catch (error) {
|
|
86
|
+
if (error instanceof Error && error.name === "AbortError") {
|
|
87
|
+
return this.timeoutError(model, this.config.timeoutMs, traceId);
|
|
88
|
+
}
|
|
89
|
+
return this.errorResponse(`Anthropic request failed: ${error.message}`, model, traceId);
|
|
90
|
+
}
|
|
91
|
+
}
|
|
92
|
+
async *chatStream(request) {
|
|
93
|
+
const model = request.model || this.config.defaultModel;
|
|
94
|
+
if (!this.config.apiKey) {
|
|
95
|
+
yield { content: "Anthropic API key not configured.", done: true, provider: "anthropic", model };
|
|
96
|
+
return;
|
|
97
|
+
}
|
|
98
|
+
try {
|
|
99
|
+
const systemMessage = request.system;
|
|
100
|
+
const messages = request.messages.filter(m => m.role !== "system");
|
|
101
|
+
const response = await fetch(`${this.config.baseUrl}/messages`, {
|
|
102
|
+
method: "POST",
|
|
103
|
+
headers: {
|
|
104
|
+
"Content-Type": "application/json",
|
|
105
|
+
"x-api-key": this.config.apiKey,
|
|
106
|
+
"anthropic-version": "2023-06-01",
|
|
107
|
+
},
|
|
108
|
+
body: JSON.stringify({
|
|
109
|
+
model,
|
|
110
|
+
system: systemMessage,
|
|
111
|
+
messages: messages.map(m => ({
|
|
112
|
+
role: m.role === "assistant" ? "assistant" : "user",
|
|
113
|
+
content: m.content,
|
|
114
|
+
})),
|
|
115
|
+
temperature: request.temperature ?? 0.7,
|
|
116
|
+
max_tokens: request.maxTokens ?? 4096,
|
|
117
|
+
stream: true,
|
|
118
|
+
}),
|
|
119
|
+
});
|
|
120
|
+
if (!response.ok) {
|
|
121
|
+
yield { content: `Anthropic stream error: ${response.status}`, done: true, provider: "anthropic", model };
|
|
122
|
+
return;
|
|
123
|
+
}
|
|
124
|
+
const reader = response.body?.getReader();
|
|
125
|
+
if (!reader) {
|
|
126
|
+
yield { content: "No response body", done: true, provider: "anthropic", model };
|
|
127
|
+
return;
|
|
128
|
+
}
|
|
129
|
+
const decoder = new TextDecoder();
|
|
130
|
+
let buffer = "";
|
|
131
|
+
while (true) {
|
|
132
|
+
const { done, value } = await reader.read();
|
|
133
|
+
if (done)
|
|
134
|
+
break;
|
|
135
|
+
buffer += decoder.decode(value, { stream: true });
|
|
136
|
+
const lines = buffer.split("\n");
|
|
137
|
+
buffer = lines.pop() || "";
|
|
138
|
+
for (const line of lines) {
|
|
139
|
+
if (line.startsWith("data: ")) {
|
|
140
|
+
const data = line.slice(6).trim();
|
|
141
|
+
try {
|
|
142
|
+
const parsed = JSON.parse(data);
|
|
143
|
+
if (parsed.type === "content_block_delta" && parsed.delta?.text) {
|
|
144
|
+
yield { content: parsed.delta.text, done: false, provider: "anthropic", model };
|
|
145
|
+
}
|
|
146
|
+
if (parsed.type === "message_stop") {
|
|
147
|
+
yield { content: "", done: true, provider: "anthropic", model };
|
|
148
|
+
return;
|
|
149
|
+
}
|
|
150
|
+
}
|
|
151
|
+
catch {
|
|
152
|
+
// Skip malformed
|
|
153
|
+
}
|
|
154
|
+
}
|
|
155
|
+
}
|
|
156
|
+
}
|
|
157
|
+
yield { content: "", done: true, provider: "anthropic", model };
|
|
158
|
+
}
|
|
159
|
+
catch (error) {
|
|
160
|
+
yield {
|
|
161
|
+
content: `Anthropic stream failed: ${error.message}`,
|
|
162
|
+
done: true,
|
|
163
|
+
provider: "anthropic",
|
|
164
|
+
model,
|
|
165
|
+
};
|
|
166
|
+
}
|
|
167
|
+
}
|
|
168
|
+
}
|
|
169
|
+
//# sourceMappingURL=anthropic.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../../../../src/modules/gateway/providers/anthropic.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AAEzC,MAAM,OAAO,iBAAkB,SAAQ,YAAY;IAC/C,YAAY,MAAgC;QACxC,KAAK,CAAC;YACF,EAAE,EAAE,WAAW;YACf,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,8BAA8B;YACvC,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,iBAAiB,IAAI,EAAE;YAC3C,MAAM,EAAE;gBACJ,0BAA0B,EAAE,yBAAyB;gBACrD,sBAAsB,EAAE,0BAA0B;gBAClD,yBAAyB;aAC5B;YACD,YAAY,EAAE,0BAA0B;YACxC,UAAU,EAAE,CAAC;YACb,SAAS,EAAE,KAAK;YAChB,QAAQ,EAAE,GAAG;YACb,QAAQ,EAAE,MAAM;YAChB,oBAAoB,EAAE,KAAK;YAC3B,qBAAqB,EAAE,KAAK;YAC5B,OAAO,EAAE,IAAI;YACb,QAAQ,EAAE,CAAC;YACX,MAAM,EAAE,CAAC;YACT,GAAG,MAAM;SACZ,CAAC,CAAC;IACP,CAAC;IAED,IAAI,EAAE,KAAa,OAAO,WAAW,CAAC,CAAC,CAAC;IACxC,IAAI,IAAI,KAAa,OAAO,kBAAkB,CAAC,CAAC,CAAC;IAEjD,KAAK,CAAC,IAAI,CAAC,OAAuB;QAC9B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC;QACxD,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;QAEhC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YACtB,OAAO,IAAI,CAAC,aAAa,CAAC,kEAAkE,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;QAClH,CAAC;QAED,IAAI,CAAC;YACD,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;YACzC,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YAE5E,MAAM,aAAa,GAAG,OAAO,CAAC,MAAM,CAAC;YACrC,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;YAEnE,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,WAAW,EAAE;gBAC5D,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACL,cAAc,EAAE,kBAAkB;oBAClC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM;oBAC/B,mBAAmB,EAAE,YAAY;iBACpC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACjB,KAAK;oBACL,MAAM,EAAE,aAAa;oBACrB,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;wBACzB,IAAI,EAAE,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM;wBACnD,OAAO,EAAE,CAAC,CAAC,OAAO;qBACrB,CAAC,CAAC;oBACH,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,GAAG;oBACvC,UAAU,EAAE,OAAO,CAAC,SAAS,IAAI,IAAI;oBACrC,MAAM,EAAE,KAAK;iBAChB,CAAC;gBACF,MAAM,EAAE,UAAU,CAAC,MAAM;aAC5B,CAAC,CAAC;YAEH,YAAY,CAAC,OAAO,CAAC,CAAC;YAEtB,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACf,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACxC,OAAO,IAAI,CAAC,aAAa,CACrB,wBAAwB,QAAQ,CAAC,MAAM,MAAM,SAAS,EAAE,EACxD,KAAK,EACL,OAAO,CACV,CAAC;YACN,CAAC;YAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAIrC,CAAC;YAEI,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YACzC,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;YAC9D,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,EAAE,YAAY,IAAI,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;YACpH,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,EAAE,aAAa,IAAI,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;YAC/E,MAAM,IAAI,GAAG,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,oBAAoB;gBAC7D,CAAC,YAAY,GAAG,IAAI,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,qBAAqB,CAAC;YAEjE,OAAO;gBACH,OAAO,EAAE,IAAI;gBACb,OAAO;gBACP,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,KAAK;gBAC1B,QAAQ,EAAE,WAAW;gBACrB,KAAK,EAAE,EAAE,WAAW,EAAE,YAAY,EAAE,WAAW,EAAE,WAAW,GAAG,YAAY,EAAE;gBAC7E,SAAS;gBACT,IAAI;gBACJ,MAAM,EAAE,KAAK;gBACb,OAAO;aACV,CAAC;QACN,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACtB,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBACxD,OAAO,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;YACpE,CAAC;YACD,OAAO,IAAI,CAAC,aAAa,CACrB,6BAA8B,KAAe,CAAC,OAAO,EAAE,EACvD,KAAK,EACL,OAAO,CACV,CAAC;QACN,CAAC;IACL,CAAC;IAED,KAAK,CAAC,CAAC,UAAU,CAAC,OAAuB;QACrC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC;QAExD,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YACtB,MAAM,EAAE,OAAO,EAAE,mCAAmC,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;YACjG,OAAO;QACX,CAAC;QAED,IAAI,CAAC;YACD,MAAM,aAAa,GAAG,OAAO,CAAC,MAAM,CAAC;YACrC,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;YAEnE,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,WAAW,EAAE;gBAC5D,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACL,cAAc,EAAE,kBAAkB;oBAClC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM;oBAC/B,mBAAmB,EAAE,YAAY;iBACpC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACjB,KAAK;oBACL,MAAM,EAAE,aAAa;oBACrB,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;wBACzB,IAAI,EAAE,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM;wBACnD,OAAO,EAAE,CAAC,CAAC,OAAO;qBACrB,CAAC,CAAC;oBACH,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,GAAG;oBACvC,UAAU,EAAE,OAAO,CAAC,SAAS,IAAI,IAAI;oBACrC,MAAM,EAAE,IAAI;iBACf,CAAC;aACL,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACf,MAAM,EAAE,OAAO,EAAE,2BAA2B,QAAQ,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;gBAC1G,OAAO;YACX,CAAC;YAED,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;YAC1C,IAAI,CAAC,MAAM,EAAE,CAAC;gBACV,MAAM,EAAE,OAAO,EAAE,kBAAkB,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;gBAChF,OAAO;YACX,CAAC;YAED,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;YAClC,IAAI,MAAM,GAAG,EAAE,CAAC;YAEhB,OAAO,IAAI,EAAE,CAAC;gBACV,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC5C,IAAI,IAAI;oBAAE,MAAM;gBAEhB,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;gBAClD,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBACjC,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;gBAE3B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;oBACvB,IAAI,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;wBAC5B,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;wBAClC,IAAI,CAAC;4BACD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAI3C,CAAC;4BACY,IAAI,MAAM,CAAC,IAAI,KAAK,qBAAqB,IAAI,MAAM,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC;gCAC9D,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;4BACpF,CAAC;4BACD,IAAI,MAAM,CAAC,IAAI,KAAK,cAAc,EAAE,CAAC;gCACjC,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;gCAChE,OAAO;4BACX,CAAC;wBACL,CAAC;wBAAC,MAAM,CAAC;4BACL,iBAAiB;wBACrB,CAAC;oBACL,CAAC;gBACL,CAAC;YACL,CAAC;YAED,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;QACpE,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACtB,MAAM;gBACF,OAAO,EAAE,4BAA6B,KAAe,CAAC,OAAO,EAAE;gBAC/D,IAAI,EAAE,IAAI;gBACV,QAAQ,EAAE,WAAW;gBACrB,KAAK;aACR,CAAC;QACN,CAAC;IACL,CAAC;CACJ"}
|
|
@@ -0,0 +1,38 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* ATABEY LLM Gateway — Base Provider Interface
|
|
3
|
+
*
|
|
4
|
+
* Abstract class that all LLM providers must implement.
|
|
5
|
+
*/
|
|
6
|
+
import type { GatewayRequest, GatewayResponse, ProviderConfig, StreamChunk } from "../types.js";
|
|
7
|
+
export declare abstract class BaseProvider {
|
|
8
|
+
readonly config: ProviderConfig;
|
|
9
|
+
constructor(config: ProviderConfig);
|
|
10
|
+
abstract get id(): string;
|
|
11
|
+
abstract get name(): string;
|
|
12
|
+
/**
|
|
13
|
+
* Send a chat completion request to the provider.
|
|
14
|
+
* Must be implemented by each provider.
|
|
15
|
+
*/
|
|
16
|
+
abstract chat(request: GatewayRequest): Promise<GatewayResponse>;
|
|
17
|
+
/**
|
|
18
|
+
* Stream a chat completion response.
|
|
19
|
+
* Returns async generator yielding StreamChunks.
|
|
20
|
+
*/
|
|
21
|
+
abstract chatStream(request: GatewayRequest): AsyncGenerator<StreamChunk>;
|
|
22
|
+
/**
|
|
23
|
+
* Check if this provider supports the given model.
|
|
24
|
+
*/
|
|
25
|
+
supportsModel(model: string): boolean;
|
|
26
|
+
/**
|
|
27
|
+
* Estimate token count for a string (rough approximation).
|
|
28
|
+
*/
|
|
29
|
+
protected estimateTokens(text: string): number;
|
|
30
|
+
/**
|
|
31
|
+
* Build standard error response.
|
|
32
|
+
*/
|
|
33
|
+
protected errorResponse(message: string, model: string, traceId?: string): GatewayResponse;
|
|
34
|
+
/**
|
|
35
|
+
* Build timeout error response.
|
|
36
|
+
*/
|
|
37
|
+
protected timeoutError(model: string, timeoutMs: number, traceId?: string): GatewayResponse;
|
|
38
|
+
}
|
|
@@ -0,0 +1,42 @@
|
|
|
1
|
+
export class BaseProvider {
|
|
2
|
+
config;
|
|
3
|
+
constructor(config) {
|
|
4
|
+
this.config = config;
|
|
5
|
+
}
|
|
6
|
+
/**
|
|
7
|
+
* Check if this provider supports the given model.
|
|
8
|
+
*/
|
|
9
|
+
supportsModel(model) {
|
|
10
|
+
return this.config.models.some(m => m.toLowerCase() === model.toLowerCase());
|
|
11
|
+
}
|
|
12
|
+
/**
|
|
13
|
+
* Estimate token count for a string (rough approximation).
|
|
14
|
+
*/
|
|
15
|
+
estimateTokens(text) {
|
|
16
|
+
// Rough estimate: ~4 chars per token for English
|
|
17
|
+
return Math.ceil(text.length / 4);
|
|
18
|
+
}
|
|
19
|
+
/**
|
|
20
|
+
* Build standard error response.
|
|
21
|
+
*/
|
|
22
|
+
errorResponse(message, model, traceId) {
|
|
23
|
+
return {
|
|
24
|
+
success: false,
|
|
25
|
+
content: message,
|
|
26
|
+
model,
|
|
27
|
+
provider: this.config.id,
|
|
28
|
+
usage: { inputTokens: 0, outputTokens: 0, totalTokens: 0 },
|
|
29
|
+
latencyMs: 0,
|
|
30
|
+
cost: 0,
|
|
31
|
+
cached: false,
|
|
32
|
+
traceId,
|
|
33
|
+
};
|
|
34
|
+
}
|
|
35
|
+
/**
|
|
36
|
+
* Build timeout error response.
|
|
37
|
+
*/
|
|
38
|
+
timeoutError(model, timeoutMs, traceId) {
|
|
39
|
+
return this.errorResponse(`Provider ${this.config.id} timed out after ${timeoutMs}ms`, model, traceId);
|
|
40
|
+
}
|
|
41
|
+
}
|
|
42
|
+
//# sourceMappingURL=base.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"base.js","sourceRoot":"","sources":["../../../../../src/modules/gateway/providers/base.ts"],"names":[],"mappings":"AAOA,MAAM,OAAgB,YAAY;IACd,MAAM,CAAiB;IAEvC,YAAY,MAAsB;QAC9B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACzB,CAAC;IAiBH;;OAEG;IACH,aAAa,CAAC,KAAa;QACvB,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAC/B,CAAC,CAAC,WAAW,EAAE,KAAK,KAAK,CAAC,WAAW,EAAE,CAC1C,CAAC;IACN,CAAC;IAED;;OAEG;IACO,cAAc,CAAC,IAAY;QACjC,iDAAiD;QACjD,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IACtC,CAAC;IAED;;OAEG;IACO,aAAa,CACnB,OAAe,EACf,KAAa,EACb,OAAgB;QAEhB,OAAO;YACH,OAAO,EAAE,KAAK;YACd,OAAO,EAAE,OAAO;YAChB,KAAK;YACL,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE;YACxB,KAAK,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE;YAC1D,SAAS,EAAE,CAAC;YACZ,IAAI,EAAE,CAAC;YACP,MAAM,EAAE,KAAK;YACb,OAAO;SACV,CAAC;IACN,CAAC;IAED;;OAEG;IACO,YAAY,CAAC,KAAa,EAAE,SAAiB,EAAE,OAAgB;QACrE,OAAO,IAAI,CAAC,aAAa,CACrB,YAAY,IAAI,CAAC,MAAM,CAAC,EAAE,oBAAoB,SAAS,IAAI,EAC3D,KAAK,EACL,OAAO,CACV,CAAC;IACN,CAAC;CACF"}
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* ATABEY LLM Gateway — Google (Gemini) Provider
|
|
3
|
+
*
|
|
4
|
+
* Supports Gemini 2.0 Flash, Gemini 2.0 Pro, Gemini 1.5 Pro models via Google AI API.
|
|
5
|
+
*/
|
|
6
|
+
import { GatewayRequest, GatewayResponse, ProviderConfig, StreamChunk } from "../types.js";
|
|
7
|
+
import { BaseProvider } from "./base.js";
|
|
8
|
+
export declare class GoogleProvider extends BaseProvider {
|
|
9
|
+
constructor(config?: Partial<ProviderConfig>);
|
|
10
|
+
get id(): string;
|
|
11
|
+
get name(): string;
|
|
12
|
+
chat(request: GatewayRequest): Promise<GatewayResponse>;
|
|
13
|
+
chatStream(request: GatewayRequest): AsyncGenerator<StreamChunk>;
|
|
14
|
+
/**
|
|
15
|
+
* Convert our standard messages to Gemini's content format.
|
|
16
|
+
*/
|
|
17
|
+
private toGeminiContents;
|
|
18
|
+
}
|
|
@@ -0,0 +1,176 @@
|
|
|
1
|
+
import { BaseProvider } from "./base.js";
|
|
2
|
+
export class GoogleProvider extends BaseProvider {
|
|
3
|
+
constructor(config) {
|
|
4
|
+
super({
|
|
5
|
+
id: "google",
|
|
6
|
+
name: "Google Gemini",
|
|
7
|
+
baseUrl: "https://generativelanguage.googleapis.com/v1beta",
|
|
8
|
+
apiKey: process.env.GOOGLE_API_KEY || "",
|
|
9
|
+
models: [
|
|
10
|
+
"gemini-2.0-flash", "gemini-2.0-flash-lite",
|
|
11
|
+
"gemini-2.0-pro", "gemini-1.5-pro",
|
|
12
|
+
"gemini-1.5-flash", "gemini-1.5-flash-8b",
|
|
13
|
+
],
|
|
14
|
+
defaultModel: "gemini-2.0-flash",
|
|
15
|
+
maxRetries: 3,
|
|
16
|
+
timeoutMs: 30000,
|
|
17
|
+
rpmLimit: 360,
|
|
18
|
+
tpmLimit: 360000,
|
|
19
|
+
costPer1kInputTokens: 0.0001,
|
|
20
|
+
costPer1kOutputTokens: 0.0004,
|
|
21
|
+
enabled: true,
|
|
22
|
+
priority: 3,
|
|
23
|
+
weight: 3,
|
|
24
|
+
...config,
|
|
25
|
+
});
|
|
26
|
+
}
|
|
27
|
+
get id() { return "google"; }
|
|
28
|
+
get name() { return "Google Gemini"; }
|
|
29
|
+
async chat(request) {
|
|
30
|
+
const startTime = Date.now();
|
|
31
|
+
const model = request.model || this.config.defaultModel;
|
|
32
|
+
const traceId = request.traceId;
|
|
33
|
+
if (!this.config.apiKey) {
|
|
34
|
+
return this.errorResponse("Google API key not configured. Set GOOGLE_API_KEY env var.", model, traceId);
|
|
35
|
+
}
|
|
36
|
+
try {
|
|
37
|
+
const controller = new AbortController();
|
|
38
|
+
const timeout = setTimeout(() => controller.abort(), this.config.timeoutMs);
|
|
39
|
+
// Convert messages to Gemini format
|
|
40
|
+
const contents = this.toGeminiContents(request);
|
|
41
|
+
const url = `${this.config.baseUrl}/models/${model}:generateContent?key=${this.config.apiKey}`;
|
|
42
|
+
const response = await fetch(url, {
|
|
43
|
+
method: "POST",
|
|
44
|
+
headers: { "Content-Type": "application/json" },
|
|
45
|
+
body: JSON.stringify({
|
|
46
|
+
contents,
|
|
47
|
+
systemInstruction: request.system ? { parts: [{ text: request.system }] } : undefined,
|
|
48
|
+
generationConfig: {
|
|
49
|
+
temperature: request.temperature ?? 0.7,
|
|
50
|
+
maxOutputTokens: request.maxTokens ?? 4096,
|
|
51
|
+
},
|
|
52
|
+
}),
|
|
53
|
+
signal: controller.signal,
|
|
54
|
+
});
|
|
55
|
+
clearTimeout(timeout);
|
|
56
|
+
if (!response.ok) {
|
|
57
|
+
const errorText = await response.text();
|
|
58
|
+
return this.errorResponse(`Gemini API error (${response.status}): ${errorText}`, model, traceId);
|
|
59
|
+
}
|
|
60
|
+
const data = await response.json();
|
|
61
|
+
const candidates = data.candidates;
|
|
62
|
+
const usageMetadata = data.usageMetadata;
|
|
63
|
+
const content = candidates?.[0]?.content?.parts?.map(p => p.text).join("") || "";
|
|
64
|
+
const latencyMs = Date.now() - startTime;
|
|
65
|
+
const inputTokens = usageMetadata?.promptTokenCount || this.estimateTokens(request.messages.map(m => m.content).join(" "));
|
|
66
|
+
const outputTokens = usageMetadata?.candidatesTokenCount || this.estimateTokens(content);
|
|
67
|
+
const cost = (inputTokens / 1000) * this.config.costPer1kInputTokens +
|
|
68
|
+
(outputTokens / 1000) * this.config.costPer1kOutputTokens;
|
|
69
|
+
return {
|
|
70
|
+
success: true,
|
|
71
|
+
content,
|
|
72
|
+
model: model,
|
|
73
|
+
provider: "google",
|
|
74
|
+
usage: { inputTokens, outputTokens, totalTokens: inputTokens + outputTokens },
|
|
75
|
+
latencyMs,
|
|
76
|
+
cost,
|
|
77
|
+
cached: false,
|
|
78
|
+
traceId,
|
|
79
|
+
};
|
|
80
|
+
}
|
|
81
|
+
catch (error) {
|
|
82
|
+
if (error instanceof Error && error.name === "AbortError") {
|
|
83
|
+
return this.timeoutError(model, this.config.timeoutMs, traceId);
|
|
84
|
+
}
|
|
85
|
+
return this.errorResponse(`Gemini request failed: ${error.message}`, model, traceId);
|
|
86
|
+
}
|
|
87
|
+
}
|
|
88
|
+
async *chatStream(request) {
|
|
89
|
+
const model = request.model || this.config.defaultModel;
|
|
90
|
+
if (!this.config.apiKey) {
|
|
91
|
+
yield { content: "Google API key not configured.", done: true, provider: "google", model };
|
|
92
|
+
return;
|
|
93
|
+
}
|
|
94
|
+
try {
|
|
95
|
+
const contents = this.toGeminiContents(request);
|
|
96
|
+
const url = `${this.config.baseUrl}/models/${model}:streamGenerateContent?key=${this.config.apiKey}&alt=sse`;
|
|
97
|
+
const response = await fetch(url, {
|
|
98
|
+
method: "POST",
|
|
99
|
+
headers: { "Content-Type": "application/json" },
|
|
100
|
+
body: JSON.stringify({
|
|
101
|
+
contents,
|
|
102
|
+
systemInstruction: request.system ? { parts: [{ text: request.system }] } : undefined,
|
|
103
|
+
generationConfig: {
|
|
104
|
+
temperature: request.temperature ?? 0.7,
|
|
105
|
+
maxOutputTokens: request.maxTokens ?? 4096,
|
|
106
|
+
},
|
|
107
|
+
}),
|
|
108
|
+
});
|
|
109
|
+
if (!response.ok) {
|
|
110
|
+
yield { content: `Gemini stream error: ${response.status}`, done: true, provider: "google", model };
|
|
111
|
+
return;
|
|
112
|
+
}
|
|
113
|
+
const reader = response.body?.getReader();
|
|
114
|
+
if (!reader) {
|
|
115
|
+
yield { content: "No response body", done: true, provider: "google", model };
|
|
116
|
+
return;
|
|
117
|
+
}
|
|
118
|
+
const decoder = new TextDecoder();
|
|
119
|
+
let buffer = "";
|
|
120
|
+
while (true) {
|
|
121
|
+
const { done, value } = await reader.read();
|
|
122
|
+
if (done)
|
|
123
|
+
break;
|
|
124
|
+
buffer += decoder.decode(value, { stream: true });
|
|
125
|
+
const lines = buffer.split("\n");
|
|
126
|
+
buffer = lines.pop() || "";
|
|
127
|
+
for (const line of lines) {
|
|
128
|
+
if (line.startsWith("data: ")) {
|
|
129
|
+
const data = line.slice(6).trim();
|
|
130
|
+
try {
|
|
131
|
+
const parsed = JSON.parse(data);
|
|
132
|
+
const text = parsed.candidates?.[0]?.content?.parts?.[0]?.text || "";
|
|
133
|
+
if (text) {
|
|
134
|
+
yield { content: text, done: false, provider: "google", model };
|
|
135
|
+
}
|
|
136
|
+
}
|
|
137
|
+
catch {
|
|
138
|
+
// Skip
|
|
139
|
+
}
|
|
140
|
+
}
|
|
141
|
+
}
|
|
142
|
+
}
|
|
143
|
+
yield { content: "", done: true, provider: "google", model };
|
|
144
|
+
}
|
|
145
|
+
catch (error) {
|
|
146
|
+
yield {
|
|
147
|
+
content: `Gemini stream failed: ${error.message}`,
|
|
148
|
+
done: true,
|
|
149
|
+
provider: "google",
|
|
150
|
+
model,
|
|
151
|
+
};
|
|
152
|
+
}
|
|
153
|
+
}
|
|
154
|
+
/**
|
|
155
|
+
* Convert our standard messages to Gemini's content format.
|
|
156
|
+
*/
|
|
157
|
+
toGeminiContents(request) {
|
|
158
|
+
const result = [];
|
|
159
|
+
let lastRole = "";
|
|
160
|
+
for (const msg of request.messages) {
|
|
161
|
+
if (msg.role === "system")
|
|
162
|
+
continue; // Systems handled via systemInstruction
|
|
163
|
+
const role = msg.role === "assistant" ? "model" : "user";
|
|
164
|
+
// Merge consecutive same-role messages
|
|
165
|
+
if (role === lastRole && result.length > 0) {
|
|
166
|
+
result[result.length - 1].parts.push({ text: msg.content });
|
|
167
|
+
}
|
|
168
|
+
else {
|
|
169
|
+
result.push({ role, parts: [{ text: msg.content }] });
|
|
170
|
+
lastRole = role;
|
|
171
|
+
}
|
|
172
|
+
}
|
|
173
|
+
return result;
|
|
174
|
+
}
|
|
175
|
+
}
|
|
176
|
+
//# sourceMappingURL=google.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"google.js","sourceRoot":"","sources":["../../../../../src/modules/gateway/providers/google.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AAEzC,MAAM,OAAO,cAAe,SAAQ,YAAY;IAC5C,YAAY,MAAgC;QACxC,KAAK,CAAC;YACF,EAAE,EAAE,QAAQ;YACZ,IAAI,EAAE,eAAe;YACrB,OAAO,EAAE,kDAAkD;YAC3D,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,EAAE;YACxC,MAAM,EAAE;gBACJ,kBAAkB,EAAE,uBAAuB;gBAC3C,gBAAgB,EAAE,gBAAgB;gBAClC,kBAAkB,EAAE,qBAAqB;aAC5C;YACD,YAAY,EAAE,kBAAkB;YAChC,UAAU,EAAE,CAAC;YACb,SAAS,EAAE,KAAK;YAChB,QAAQ,EAAE,GAAG;YACb,QAAQ,EAAE,MAAM;YAChB,oBAAoB,EAAE,MAAM;YAC5B,qBAAqB,EAAE,MAAM;YAC7B,OAAO,EAAE,IAAI;YACb,QAAQ,EAAE,CAAC;YACX,MAAM,EAAE,CAAC;YACT,GAAG,MAAM;SACZ,CAAC,CAAC;IACP,CAAC;IAED,IAAI,EAAE,KAAa,OAAO,QAAQ,CAAC,CAAC,CAAC;IACrC,IAAI,IAAI,KAAa,OAAO,eAAe,CAAC,CAAC,CAAC;IAE9C,KAAK,CAAC,IAAI,CAAC,OAAuB;QAC9B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC;QACxD,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;QAEhC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YACtB,OAAO,IAAI,CAAC,aAAa,CAAC,4DAA4D,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;QAC5G,CAAC;QAED,IAAI,CAAC;YACD,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;YACzC,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YAE5E,oCAAoC;YACpC,MAAM,QAAQ,GAAG,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;YAEhD,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,WAAW,KAAK,wBAAwB,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YAE/F,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;gBAC9B,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;gBAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACjB,QAAQ;oBACR,iBAAiB,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS;oBACrF,gBAAgB,EAAE;wBACd,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,GAAG;wBACvC,eAAe,EAAE,OAAO,CAAC,SAAS,IAAI,IAAI;qBAC7C;iBACJ,CAAC;gBACF,MAAM,EAAE,UAAU,CAAC,MAAM;aAC5B,CAAC,CAAC;YAEH,YAAY,CAAC,OAAO,CAAC,CAAC;YAEtB,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACf,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACxC,OAAO,IAAI,CAAC,aAAa,CACrB,qBAAqB,QAAQ,CAAC,MAAM,MAAM,SAAS,EAAE,EACrD,KAAK,EACL,OAAO,CACV,CAAC;YACN,CAAC;YAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAA6B,CAAC;YAC9D,MAAM,UAAU,GAAG,IAAI,CAAC,UAEhB,CAAC;YACT,MAAM,aAAa,GAAG,IAAI,CAAC,aAIpB,CAAC;YAER,MAAM,OAAO,GAAG,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;YAEjF,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YACzC,MAAM,WAAW,GAAG,aAAa,EAAE,gBAAgB,IAAI,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;YAC3H,MAAM,YAAY,GAAG,aAAa,EAAE,oBAAoB,IAAI,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;YACzF,MAAM,IAAI,GAAG,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,oBAAoB;gBAC7D,CAAC,YAAY,GAAG,IAAI,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,qBAAqB,CAAC;YAEjE,OAAO;gBACH,OAAO,EAAE,IAAI;gBACb,OAAO;gBACP,KAAK,EAAE,KAAK;gBACZ,QAAQ,EAAE,QAAQ;gBAClB,KAAK,EAAE,EAAE,WAAW,EAAE,YAAY,EAAE,WAAW,EAAE,WAAW,GAAG,YAAY,EAAE;gBAC7E,SAAS;gBACT,IAAI;gBACJ,MAAM,EAAE,KAAK;gBACb,OAAO;aACV,CAAC;QACN,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACtB,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBACxD,OAAO,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;YACpE,CAAC;YACD,OAAO,IAAI,CAAC,aAAa,CACrB,0BAA2B,KAAe,CAAC,OAAO,EAAE,EACpD,KAAK,EACL,OAAO,CACV,CAAC;QACN,CAAC;IACL,CAAC;IAED,KAAK,CAAC,CAAC,UAAU,CAAC,OAAuB;QACrC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC;QAExD,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YACtB,MAAM,EAAE,OAAO,EAAE,gCAAgC,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;YAC3F,OAAO;QACX,CAAC;QAED,IAAI,CAAC;YACD,MAAM,QAAQ,GAAG,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;YAChD,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,WAAW,KAAK,8BAA8B,IAAI,CAAC,MAAM,CAAC,MAAM,UAAU,CAAC;YAE7G,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;gBAC9B,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;gBAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACjB,QAAQ;oBACR,iBAAiB,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS;oBACrF,gBAAgB,EAAE;wBACd,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,GAAG;wBACvC,eAAe,EAAE,OAAO,CAAC,SAAS,IAAI,IAAI;qBAC7C;iBACJ,CAAC;aACL,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACf,MAAM,EAAE,OAAO,EAAE,wBAAwB,QAAQ,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;gBACpG,OAAO;YACX,CAAC;YAED,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;YAC1C,IAAI,CAAC,MAAM,EAAE,CAAC;gBACV,MAAM,EAAE,OAAO,EAAE,kBAAkB,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;gBAC7E,OAAO;YACX,CAAC;YAED,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;YAClC,IAAI,MAAM,GAAG,EAAE,CAAC;YAEhB,OAAO,IAAI,EAAE,CAAC;gBACV,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC5C,IAAI,IAAI;oBAAE,MAAM;gBAEhB,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;gBAClD,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBACjC,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;gBAE3B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;oBACvB,IAAI,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;wBAC5B,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;wBAClC,IAAI,CAAC;4BACD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAI3C,CAAC;4BACY,MAAM,IAAI,GAAG,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC;4BACrE,IAAI,IAAI,EAAE,CAAC;gCACP,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;4BACpE,CAAC;wBACL,CAAC;wBAAC,MAAM,CAAC;4BACL,OAAO;wBACX,CAAC;oBACL,CAAC;gBACL,CAAC;YACL,CAAC;YAED,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;QACjE,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACtB,MAAM;gBACF,OAAO,EAAE,yBAA0B,KAAe,CAAC,OAAO,EAAE;gBAC5D,IAAI,EAAE,IAAI;gBACV,QAAQ,EAAE,QAAQ;gBAClB,KAAK;aACR,CAAC;QACN,CAAC;IACL,CAAC;IAED;;KAEC;IACO,gBAAgB,CAAC,OAAuB;QAI5C,MAAM,MAAM,GAA4D,EAAE,CAAC;QAC3E,IAAI,QAAQ,GAAG,EAAE,CAAC;QAElB,KAAK,MAAM,GAAG,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;YACjC,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ;gBAAE,SAAS,CAAC,wCAAwC;YAE7E,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC;YAEzD,uCAAuC;YACvC,IAAI,IAAI,KAAK,QAAQ,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACzC,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;YAChE,CAAC;iBAAM,CAAC;gBACJ,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;gBACtD,QAAQ,GAAG,IAAI,CAAC;YACpB,CAAC;QACL,CAAC;QAED,OAAO,MAAM,CAAC;IAClB,CAAC;CACJ"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* ATABEY LLM Gateway — Grok (xAI) Provider
|
|
3
|
+
*
|
|
4
|
+
* Supports Grok-2, Grok-2-mini, Grok-beta models via xAI API.
|
|
5
|
+
*/
|
|
6
|
+
import { GatewayRequest, GatewayResponse, ProviderConfig, StreamChunk } from "../types.js";
|
|
7
|
+
import { BaseProvider } from "./base.js";
|
|
8
|
+
export declare class GrokProvider extends BaseProvider {
|
|
9
|
+
constructor(config?: Partial<ProviderConfig>);
|
|
10
|
+
get id(): string;
|
|
11
|
+
get name(): string;
|
|
12
|
+
chat(request: GatewayRequest): Promise<GatewayResponse>;
|
|
13
|
+
chatStream(request: GatewayRequest): AsyncGenerator<StreamChunk>;
|
|
14
|
+
}
|
|
@@ -0,0 +1,166 @@
|
|
|
1
|
+
import { BaseProvider } from "./base.js";
|
|
2
|
+
export class GrokProvider extends BaseProvider {
|
|
3
|
+
constructor(config) {
|
|
4
|
+
super({
|
|
5
|
+
id: "grok",
|
|
6
|
+
name: "xAI Grok",
|
|
7
|
+
baseUrl: "https://api.x.ai/v1",
|
|
8
|
+
apiKey: process.env.XAI_API_KEY || "",
|
|
9
|
+
models: [
|
|
10
|
+
"grok-2", "grok-2-mini", "grok-beta",
|
|
11
|
+
],
|
|
12
|
+
defaultModel: "grok-2",
|
|
13
|
+
maxRetries: 3,
|
|
14
|
+
timeoutMs: 30000,
|
|
15
|
+
rpmLimit: 100,
|
|
16
|
+
tpmLimit: 50000,
|
|
17
|
+
costPer1kInputTokens: 0.005,
|
|
18
|
+
costPer1kOutputTokens: 0.015,
|
|
19
|
+
enabled: true,
|
|
20
|
+
priority: 4,
|
|
21
|
+
weight: 2,
|
|
22
|
+
...config,
|
|
23
|
+
});
|
|
24
|
+
}
|
|
25
|
+
get id() { return "grok"; }
|
|
26
|
+
get name() { return "xAI Grok"; }
|
|
27
|
+
async chat(request) {
|
|
28
|
+
const startTime = Date.now();
|
|
29
|
+
const model = request.model || this.config.defaultModel;
|
|
30
|
+
const traceId = request.traceId;
|
|
31
|
+
if (!this.config.apiKey) {
|
|
32
|
+
return this.errorResponse("xAI API key not configured. Set XAI_API_KEY env var.", model, traceId);
|
|
33
|
+
}
|
|
34
|
+
try {
|
|
35
|
+
const controller = new AbortController();
|
|
36
|
+
const timeout = setTimeout(() => controller.abort(), this.config.timeoutMs);
|
|
37
|
+
const response = await fetch(`${this.config.baseUrl}/chat/completions`, {
|
|
38
|
+
method: "POST",
|
|
39
|
+
headers: {
|
|
40
|
+
"Content-Type": "application/json",
|
|
41
|
+
Authorization: `Bearer ${this.config.apiKey}`,
|
|
42
|
+
},
|
|
43
|
+
body: JSON.stringify({
|
|
44
|
+
model,
|
|
45
|
+
messages: [
|
|
46
|
+
...(request.system ? [{ role: "system", content: request.system }] : []),
|
|
47
|
+
...request.messages.map(m => ({
|
|
48
|
+
role: m.role,
|
|
49
|
+
content: m.content,
|
|
50
|
+
})),
|
|
51
|
+
],
|
|
52
|
+
temperature: request.temperature ?? 0.7,
|
|
53
|
+
max_tokens: request.maxTokens ?? 4096,
|
|
54
|
+
stream: false,
|
|
55
|
+
}),
|
|
56
|
+
signal: controller.signal,
|
|
57
|
+
});
|
|
58
|
+
clearTimeout(timeout);
|
|
59
|
+
if (!response.ok) {
|
|
60
|
+
const errorText = await response.text();
|
|
61
|
+
return this.errorResponse(`xAI API error (${response.status}): ${errorText}`, model, traceId);
|
|
62
|
+
}
|
|
63
|
+
const data = await response.json();
|
|
64
|
+
const latencyMs = Date.now() - startTime;
|
|
65
|
+
const content = data.choices?.[0]?.message?.content || "";
|
|
66
|
+
const inputTokens = data.usage?.prompt_tokens || this.estimateTokens(request.messages.map(m => m.content).join(" "));
|
|
67
|
+
const outputTokens = data.usage?.completion_tokens || this.estimateTokens(content);
|
|
68
|
+
const cost = (inputTokens / 1000) * this.config.costPer1kInputTokens +
|
|
69
|
+
(outputTokens / 1000) * this.config.costPer1kOutputTokens;
|
|
70
|
+
return {
|
|
71
|
+
success: true,
|
|
72
|
+
content,
|
|
73
|
+
model: data.model || model,
|
|
74
|
+
provider: "grok",
|
|
75
|
+
usage: { inputTokens, outputTokens, totalTokens: inputTokens + outputTokens },
|
|
76
|
+
latencyMs,
|
|
77
|
+
cost,
|
|
78
|
+
cached: false,
|
|
79
|
+
traceId,
|
|
80
|
+
};
|
|
81
|
+
}
|
|
82
|
+
catch (error) {
|
|
83
|
+
if (error instanceof Error && error.name === "AbortError") {
|
|
84
|
+
return this.timeoutError(model, this.config.timeoutMs, traceId);
|
|
85
|
+
}
|
|
86
|
+
return this.errorResponse(`xAI request failed: ${error.message}`, model, traceId);
|
|
87
|
+
}
|
|
88
|
+
}
|
|
89
|
+
async *chatStream(request) {
|
|
90
|
+
const model = request.model || this.config.defaultModel;
|
|
91
|
+
if (!this.config.apiKey) {
|
|
92
|
+
yield { content: "xAI API key not configured.", done: true, provider: "grok", model };
|
|
93
|
+
return;
|
|
94
|
+
}
|
|
95
|
+
try {
|
|
96
|
+
const response = await fetch(`${this.config.baseUrl}/chat/completions`, {
|
|
97
|
+
method: "POST",
|
|
98
|
+
headers: {
|
|
99
|
+
"Content-Type": "application/json",
|
|
100
|
+
Authorization: `Bearer ${this.config.apiKey}`,
|
|
101
|
+
},
|
|
102
|
+
body: JSON.stringify({
|
|
103
|
+
model,
|
|
104
|
+
messages: [
|
|
105
|
+
...(request.system ? [{ role: "system", content: request.system }] : []),
|
|
106
|
+
...request.messages.map(m => ({
|
|
107
|
+
role: m.role,
|
|
108
|
+
content: m.content,
|
|
109
|
+
})),
|
|
110
|
+
],
|
|
111
|
+
temperature: request.temperature ?? 0.7,
|
|
112
|
+
max_tokens: request.maxTokens ?? 4096,
|
|
113
|
+
stream: true,
|
|
114
|
+
}),
|
|
115
|
+
});
|
|
116
|
+
if (!response.ok) {
|
|
117
|
+
yield { content: `xAI stream error: ${response.status}`, done: true, provider: "grok", model };
|
|
118
|
+
return;
|
|
119
|
+
}
|
|
120
|
+
const reader = response.body?.getReader();
|
|
121
|
+
if (!reader) {
|
|
122
|
+
yield { content: "No response body", done: true, provider: "grok", model };
|
|
123
|
+
return;
|
|
124
|
+
}
|
|
125
|
+
const decoder = new TextDecoder();
|
|
126
|
+
let buffer = "";
|
|
127
|
+
while (true) {
|
|
128
|
+
const { done, value } = await reader.read();
|
|
129
|
+
if (done)
|
|
130
|
+
break;
|
|
131
|
+
buffer += decoder.decode(value, { stream: true });
|
|
132
|
+
const lines = buffer.split("\n");
|
|
133
|
+
buffer = lines.pop() || "";
|
|
134
|
+
for (const line of lines) {
|
|
135
|
+
if (line.startsWith("data: ")) {
|
|
136
|
+
const data = line.slice(6).trim();
|
|
137
|
+
if (data === "[DONE]") {
|
|
138
|
+
yield { content: "", done: true, provider: "grok", model };
|
|
139
|
+
return;
|
|
140
|
+
}
|
|
141
|
+
try {
|
|
142
|
+
const parsed = JSON.parse(data);
|
|
143
|
+
const content = parsed.choices?.[0]?.delta?.content || "";
|
|
144
|
+
if (content) {
|
|
145
|
+
yield { content, done: false, provider: "grok", model };
|
|
146
|
+
}
|
|
147
|
+
}
|
|
148
|
+
catch {
|
|
149
|
+
// Skip malformed
|
|
150
|
+
}
|
|
151
|
+
}
|
|
152
|
+
}
|
|
153
|
+
}
|
|
154
|
+
yield { content: "", done: true, provider: "grok", model };
|
|
155
|
+
}
|
|
156
|
+
catch (error) {
|
|
157
|
+
yield {
|
|
158
|
+
content: `xAI stream failed: ${error.message}`,
|
|
159
|
+
done: true,
|
|
160
|
+
provider: "grok",
|
|
161
|
+
model,
|
|
162
|
+
};
|
|
163
|
+
}
|
|
164
|
+
}
|
|
165
|
+
}
|
|
166
|
+
//# sourceMappingURL=grok.js.map
|