gencode-ai 0.1.0
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/.env.example +11 -0
- package/CLAUDE.md +70 -0
- package/LICENSE +21 -0
- package/README.md +117 -0
- package/dist/agent/agent.d.ts +84 -0
- package/dist/agent/agent.d.ts.map +1 -0
- package/dist/agent/agent.js +233 -0
- package/dist/agent/agent.js.map +1 -0
- package/dist/agent/index.d.ts +6 -0
- package/dist/agent/index.d.ts.map +1 -0
- package/dist/agent/index.js +6 -0
- package/dist/agent/index.js.map +1 -0
- package/dist/agent/types.d.ts +47 -0
- package/dist/agent/types.d.ts.map +1 -0
- package/dist/agent/types.js +5 -0
- package/dist/agent/types.js.map +1 -0
- package/dist/cli/components/App.d.ts +14 -0
- package/dist/cli/components/App.d.ts.map +1 -0
- package/dist/cli/components/App.js +395 -0
- package/dist/cli/components/App.js.map +1 -0
- package/dist/cli/components/CommandSuggestions.d.ts +13 -0
- package/dist/cli/components/CommandSuggestions.d.ts.map +1 -0
- package/dist/cli/components/CommandSuggestions.js +32 -0
- package/dist/cli/components/CommandSuggestions.js.map +1 -0
- package/dist/cli/components/Header.d.ts +9 -0
- package/dist/cli/components/Header.d.ts.map +1 -0
- package/dist/cli/components/Header.js +13 -0
- package/dist/cli/components/Header.js.map +1 -0
- package/dist/cli/components/Input.d.ts +13 -0
- package/dist/cli/components/Input.d.ts.map +1 -0
- package/dist/cli/components/Input.js +27 -0
- package/dist/cli/components/Input.js.map +1 -0
- package/dist/cli/components/Logo.d.ts +2 -0
- package/dist/cli/components/Logo.d.ts.map +1 -0
- package/dist/cli/components/Logo.js +8 -0
- package/dist/cli/components/Logo.js.map +1 -0
- package/dist/cli/components/Messages.d.ts +37 -0
- package/dist/cli/components/Messages.d.ts.map +1 -0
- package/dist/cli/components/Messages.js +106 -0
- package/dist/cli/components/Messages.js.map +1 -0
- package/dist/cli/components/ModelSelector.d.ts +13 -0
- package/dist/cli/components/ModelSelector.d.ts.map +1 -0
- package/dist/cli/components/ModelSelector.js +72 -0
- package/dist/cli/components/ModelSelector.js.map +1 -0
- package/dist/cli/components/Spinner.d.ts +12 -0
- package/dist/cli/components/Spinner.d.ts.map +1 -0
- package/dist/cli/components/Spinner.js +45 -0
- package/dist/cli/components/Spinner.js.map +1 -0
- package/dist/cli/components/index.d.ts +12 -0
- package/dist/cli/components/index.d.ts.map +1 -0
- package/dist/cli/components/index.js +12 -0
- package/dist/cli/components/index.js.map +1 -0
- package/dist/cli/components/theme.d.ts +31 -0
- package/dist/cli/components/theme.d.ts.map +1 -0
- package/dist/cli/components/theme.js +36 -0
- package/dist/cli/components/theme.js.map +1 -0
- package/dist/cli/index-legacy.d.ts +7 -0
- package/dist/cli/index-legacy.d.ts.map +1 -0
- package/dist/cli/index-legacy.js +431 -0
- package/dist/cli/index-legacy.js.map +1 -0
- package/dist/cli/index.d.ts +7 -0
- package/dist/cli/index.d.ts.map +1 -0
- package/dist/cli/index.js +116 -0
- package/dist/cli/index.js.map +1 -0
- package/dist/cli/ink-cli.d.ts +7 -0
- package/dist/cli/ink-cli.d.ts.map +1 -0
- package/dist/cli/ink-cli.js +105 -0
- package/dist/cli/ink-cli.js.map +1 -0
- package/dist/cli/session-picker.d.ts +16 -0
- package/dist/cli/session-picker.d.ts.map +1 -0
- package/dist/cli/session-picker.js +280 -0
- package/dist/cli/session-picker.js.map +1 -0
- package/dist/cli/ui.d.ts +61 -0
- package/dist/cli/ui.d.ts.map +1 -0
- package/dist/cli/ui.js +364 -0
- package/dist/cli/ui.js.map +1 -0
- package/dist/config/index.d.ts +7 -0
- package/dist/config/index.d.ts.map +1 -0
- package/dist/config/index.js +6 -0
- package/dist/config/index.js.map +1 -0
- package/dist/config/manager.d.ts +31 -0
- package/dist/config/manager.d.ts.map +1 -0
- package/dist/config/manager.js +65 -0
- package/dist/config/manager.js.map +1 -0
- package/dist/config/types.d.ts +22 -0
- package/dist/config/types.d.ts.map +1 -0
- package/dist/config/types.js +6 -0
- package/dist/config/types.js.map +1 -0
- package/dist/index.d.ts +12 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +21 -0
- package/dist/index.js.map +1 -0
- package/dist/memory/index.d.ts +10 -0
- package/dist/memory/index.d.ts.map +1 -0
- package/dist/memory/index.js +9 -0
- package/dist/memory/index.js.map +1 -0
- package/dist/memory/init.d.ts +20 -0
- package/dist/memory/init.d.ts.map +1 -0
- package/dist/memory/init.js +332 -0
- package/dist/memory/init.js.map +1 -0
- package/dist/memory/manager.d.ts +85 -0
- package/dist/memory/manager.d.ts.map +1 -0
- package/dist/memory/manager.js +234 -0
- package/dist/memory/manager.js.map +1 -0
- package/dist/memory/types.d.ts +74 -0
- package/dist/memory/types.d.ts.map +1 -0
- package/dist/memory/types.js +6 -0
- package/dist/memory/types.js.map +1 -0
- package/dist/permissions/index.d.ts +7 -0
- package/dist/permissions/index.d.ts.map +1 -0
- package/dist/permissions/index.js +6 -0
- package/dist/permissions/index.js.map +1 -0
- package/dist/permissions/manager.d.ts +32 -0
- package/dist/permissions/manager.d.ts.map +1 -0
- package/dist/permissions/manager.js +79 -0
- package/dist/permissions/manager.js.map +1 -0
- package/dist/permissions/types.d.ts +14 -0
- package/dist/permissions/types.d.ts.map +1 -0
- package/dist/permissions/types.js +17 -0
- package/dist/permissions/types.js.map +1 -0
- package/dist/providers/anthropic.d.ts +20 -0
- package/dist/providers/anthropic.d.ts.map +1 -0
- package/dist/providers/anthropic.js +185 -0
- package/dist/providers/anthropic.js.map +1 -0
- package/dist/providers/gemini.d.ts +21 -0
- package/dist/providers/gemini.d.ts.map +1 -0
- package/dist/providers/gemini.js +241 -0
- package/dist/providers/gemini.js.map +1 -0
- package/dist/providers/index.d.ts +34 -0
- package/dist/providers/index.d.ts.map +1 -0
- package/dist/providers/index.js +72 -0
- package/dist/providers/index.js.map +1 -0
- package/dist/providers/openai.d.ts +19 -0
- package/dist/providers/openai.d.ts.map +1 -0
- package/dist/providers/openai.js +221 -0
- package/dist/providers/openai.js.map +1 -0
- package/dist/providers/types.d.ts +125 -0
- package/dist/providers/types.d.ts.map +1 -0
- package/dist/providers/types.js +6 -0
- package/dist/providers/types.js.map +1 -0
- package/dist/session/index.d.ts +6 -0
- package/dist/session/index.d.ts.map +1 -0
- package/dist/session/index.js +6 -0
- package/dist/session/index.js.map +1 -0
- package/dist/session/manager.d.ts +101 -0
- package/dist/session/manager.d.ts.map +1 -0
- package/dist/session/manager.js +295 -0
- package/dist/session/manager.js.map +1 -0
- package/dist/session/types.d.ts +39 -0
- package/dist/session/types.d.ts.map +1 -0
- package/dist/session/types.js +10 -0
- package/dist/session/types.js.map +1 -0
- package/dist/tools/builtin/bash.d.ts +7 -0
- package/dist/tools/builtin/bash.d.ts.map +1 -0
- package/dist/tools/builtin/bash.js +80 -0
- package/dist/tools/builtin/bash.js.map +1 -0
- package/dist/tools/builtin/edit.d.ts +7 -0
- package/dist/tools/builtin/edit.d.ts.map +1 -0
- package/dist/tools/builtin/edit.js +32 -0
- package/dist/tools/builtin/edit.js.map +1 -0
- package/dist/tools/builtin/glob.d.ts +7 -0
- package/dist/tools/builtin/glob.d.ts.map +1 -0
- package/dist/tools/builtin/glob.js +36 -0
- package/dist/tools/builtin/glob.js.map +1 -0
- package/dist/tools/builtin/grep.d.ts +7 -0
- package/dist/tools/builtin/grep.d.ts.map +1 -0
- package/dist/tools/builtin/grep.js +59 -0
- package/dist/tools/builtin/grep.js.map +1 -0
- package/dist/tools/builtin/read.d.ts +7 -0
- package/dist/tools/builtin/read.d.ts.map +1 -0
- package/dist/tools/builtin/read.js +29 -0
- package/dist/tools/builtin/read.js.map +1 -0
- package/dist/tools/builtin/write.d.ts +7 -0
- package/dist/tools/builtin/write.d.ts.map +1 -0
- package/dist/tools/builtin/write.js +24 -0
- package/dist/tools/builtin/write.js.map +1 -0
- package/dist/tools/index.d.ts +38 -0
- package/dist/tools/index.d.ts.map +1 -0
- package/dist/tools/index.js +32 -0
- package/dist/tools/index.js.map +1 -0
- package/dist/tools/registry.d.ts +22 -0
- package/dist/tools/registry.d.ts.map +1 -0
- package/dist/tools/registry.js +71 -0
- package/dist/tools/registry.js.map +1 -0
- package/dist/tools/types.d.ts +62 -0
- package/dist/tools/types.d.ts.map +1 -0
- package/dist/tools/types.js +126 -0
- package/dist/tools/types.js.map +1 -0
- package/docs/README.md +16 -0
- package/docs/proposals/0001-web-fetch-tool.md +293 -0
- package/docs/proposals/0002-web-search-tool.md +306 -0
- package/docs/proposals/0003-task-subagents.md +333 -0
- package/docs/proposals/0004-plan-mode.md +338 -0
- package/docs/proposals/0005-todo-system.md +299 -0
- package/docs/proposals/0006-memory-system.md +539 -0
- package/docs/proposals/0007-context-management.md +429 -0
- package/docs/proposals/0008-checkpointing.md +327 -0
- package/docs/proposals/0009-hooks-system.md +343 -0
- package/docs/proposals/0010-mcp-integration.md +382 -0
- package/docs/proposals/0011-custom-commands.md +374 -0
- package/docs/proposals/0012-ask-user-question.md +317 -0
- package/docs/proposals/0013-multi-edit-tool.md +345 -0
- package/docs/proposals/0014-lsp-tool.md +478 -0
- package/docs/proposals/0015-ls-tool.md +407 -0
- package/docs/proposals/0016-kill-shell-tool.md +455 -0
- package/docs/proposals/0017-background-tasks.md +489 -0
- package/docs/proposals/0018-parallel-tool-execution.md +415 -0
- package/docs/proposals/0019-session-enhancements.md +462 -0
- package/docs/proposals/0020-session-summarization.md +447 -0
- package/docs/proposals/0021-skills-system.md +409 -0
- package/docs/proposals/0022-plugin-system.md +467 -0
- package/docs/proposals/0023-permission-enhancements.md +470 -0
- package/docs/proposals/0024-keyboard-shortcuts.md +443 -0
- package/docs/proposals/0025-cost-tracking.md +447 -0
- package/docs/proposals/0026-git-integration.md +475 -0
- package/docs/proposals/0027-enhanced-read-tool.md +514 -0
- package/docs/proposals/0028-enhanced-bash-tool.md +511 -0
- package/docs/proposals/0029-notebook-edit-tool.md +413 -0
- package/docs/proposals/0030-plugin-marketplace.md +360 -0
- package/docs/proposals/0031-command-suggestions.md +295 -0
- package/docs/proposals/0032-ide-integrations.md +328 -0
- package/docs/proposals/0033-enterprise-deployment.md +221 -0
- package/docs/proposals/0034-sandboxing.md +273 -0
- package/docs/proposals/0035-auto-updater.md +311 -0
- package/docs/proposals/0036-enhanced-glob-tool.md +267 -0
- package/docs/proposals/0037-enhanced-grep-tool.md +360 -0
- package/docs/proposals/0038-interactive-cli-ui.md +373 -0
- package/docs/proposals/0039-streaming-enhancements.md +359 -0
- package/docs/proposals/0040-multi-provider-enhancements.md +369 -0
- package/docs/proposals/README.md +84 -0
- package/docs/proposals/TEMPLATE.md +57 -0
- package/docs/proposals/research/claude-code-research.md +307 -0
- package/examples/agent-demo.ts +115 -0
- package/examples/basic.ts +166 -0
- package/package.json +50 -0
- package/src/agent/agent.ts +276 -0
- package/src/agent/index.ts +6 -0
- package/src/agent/types.ts +62 -0
- package/src/cli/components/App.tsx +565 -0
- package/src/cli/components/CommandSuggestions.tsx +58 -0
- package/src/cli/components/Header.tsx +36 -0
- package/src/cli/components/Input.tsx +60 -0
- package/src/cli/components/Logo.tsx +16 -0
- package/src/cli/components/Messages.tsx +210 -0
- package/src/cli/components/ModelSelector.tsx +135 -0
- package/src/cli/components/Spinner.tsx +72 -0
- package/src/cli/components/index.ts +21 -0
- package/src/cli/components/theme.ts +36 -0
- package/src/cli/index.tsx +136 -0
- package/src/config/index.ts +7 -0
- package/src/config/manager.ts +77 -0
- package/src/config/types.ts +25 -0
- package/src/index.ts +86 -0
- package/src/permissions/index.ts +7 -0
- package/src/permissions/manager.ts +97 -0
- package/src/permissions/types.ts +29 -0
- package/src/providers/anthropic.ts +224 -0
- package/src/providers/gemini.ts +295 -0
- package/src/providers/index.ts +97 -0
- package/src/providers/openai.ts +261 -0
- package/src/providers/types.ts +181 -0
- package/src/session/index.ts +6 -0
- package/src/session/manager.ts +354 -0
- package/src/session/types.ts +49 -0
- package/src/tools/builtin/bash.ts +92 -0
- package/src/tools/builtin/edit.ts +37 -0
- package/src/tools/builtin/glob.ts +42 -0
- package/src/tools/builtin/grep.ts +67 -0
- package/src/tools/builtin/read.ts +34 -0
- package/src/tools/builtin/write.ts +27 -0
- package/src/tools/index.ts +36 -0
- package/src/tools/registry.ts +83 -0
- package/src/tools/types.ts +172 -0
- package/tsconfig.json +21 -0
|
@@ -0,0 +1,185 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Anthropic Provider Implementation
|
|
3
|
+
* Supports Claude 3.5 Sonnet, Claude 3 Opus, Claude 3 Haiku, etc.
|
|
4
|
+
*/
|
|
5
|
+
import Anthropic from '@anthropic-ai/sdk';
|
|
6
|
+
export class AnthropicProvider {
|
|
7
|
+
name = 'anthropic';
|
|
8
|
+
client;
|
|
9
|
+
constructor(config = {}) {
|
|
10
|
+
this.client = new Anthropic({
|
|
11
|
+
apiKey: config.apiKey ?? process.env.ANTHROPIC_API_KEY,
|
|
12
|
+
baseURL: config.baseURL,
|
|
13
|
+
});
|
|
14
|
+
}
|
|
15
|
+
async complete(options) {
|
|
16
|
+
const messages = this.convertMessages(options.messages);
|
|
17
|
+
const tools = options.tools ? this.convertTools(options.tools) : undefined;
|
|
18
|
+
const response = await this.client.messages.create({
|
|
19
|
+
model: options.model,
|
|
20
|
+
messages,
|
|
21
|
+
tools,
|
|
22
|
+
system: options.systemPrompt,
|
|
23
|
+
max_tokens: options.maxTokens ?? 4096,
|
|
24
|
+
temperature: options.temperature,
|
|
25
|
+
});
|
|
26
|
+
return this.convertResponse(response);
|
|
27
|
+
}
|
|
28
|
+
async *stream(options) {
|
|
29
|
+
const messages = this.convertMessages(options.messages);
|
|
30
|
+
const tools = options.tools ? this.convertTools(options.tools) : undefined;
|
|
31
|
+
const stream = this.client.messages.stream({
|
|
32
|
+
model: options.model,
|
|
33
|
+
messages,
|
|
34
|
+
tools,
|
|
35
|
+
system: options.systemPrompt,
|
|
36
|
+
max_tokens: options.maxTokens ?? 4096,
|
|
37
|
+
temperature: options.temperature,
|
|
38
|
+
});
|
|
39
|
+
const toolInputBuffers = new Map();
|
|
40
|
+
for await (const event of stream) {
|
|
41
|
+
if (event.type === 'content_block_start') {
|
|
42
|
+
const block = event.content_block;
|
|
43
|
+
if (block.type === 'tool_use') {
|
|
44
|
+
toolInputBuffers.set(event.index, {
|
|
45
|
+
id: block.id,
|
|
46
|
+
name: block.name,
|
|
47
|
+
input: '',
|
|
48
|
+
});
|
|
49
|
+
yield { type: 'tool_start', id: block.id, name: block.name };
|
|
50
|
+
}
|
|
51
|
+
}
|
|
52
|
+
else if (event.type === 'content_block_delta') {
|
|
53
|
+
const delta = event.delta;
|
|
54
|
+
if (delta.type === 'text_delta') {
|
|
55
|
+
yield { type: 'text', text: delta.text };
|
|
56
|
+
}
|
|
57
|
+
else if (delta.type === 'input_json_delta') {
|
|
58
|
+
const buffer = toolInputBuffers.get(event.index);
|
|
59
|
+
if (buffer) {
|
|
60
|
+
buffer.input += delta.partial_json;
|
|
61
|
+
yield { type: 'tool_input', id: buffer.id, input: delta.partial_json };
|
|
62
|
+
}
|
|
63
|
+
}
|
|
64
|
+
}
|
|
65
|
+
}
|
|
66
|
+
// Get final message
|
|
67
|
+
const finalMessage = await stream.finalMessage();
|
|
68
|
+
const content = this.convertContent(finalMessage.content);
|
|
69
|
+
yield {
|
|
70
|
+
type: 'done',
|
|
71
|
+
response: {
|
|
72
|
+
content,
|
|
73
|
+
stopReason: this.convertStopReason(finalMessage.stop_reason),
|
|
74
|
+
usage: {
|
|
75
|
+
inputTokens: finalMessage.usage.input_tokens,
|
|
76
|
+
outputTokens: finalMessage.usage.output_tokens,
|
|
77
|
+
},
|
|
78
|
+
},
|
|
79
|
+
};
|
|
80
|
+
}
|
|
81
|
+
convertMessages(messages) {
|
|
82
|
+
const result = [];
|
|
83
|
+
for (const msg of messages) {
|
|
84
|
+
// Skip system messages - handled separately
|
|
85
|
+
if (msg.role === 'system') {
|
|
86
|
+
continue;
|
|
87
|
+
}
|
|
88
|
+
if (msg.role === 'user') {
|
|
89
|
+
const content = this.convertToAnthropicContent(msg.content, 'user');
|
|
90
|
+
result.push({ role: 'user', content });
|
|
91
|
+
}
|
|
92
|
+
else if (msg.role === 'assistant') {
|
|
93
|
+
const content = this.convertToAnthropicContent(msg.content, 'assistant');
|
|
94
|
+
result.push({ role: 'assistant', content });
|
|
95
|
+
}
|
|
96
|
+
}
|
|
97
|
+
return result;
|
|
98
|
+
}
|
|
99
|
+
convertToAnthropicContent(content, role) {
|
|
100
|
+
if (typeof content === 'string') {
|
|
101
|
+
return content;
|
|
102
|
+
}
|
|
103
|
+
const result = [];
|
|
104
|
+
for (const item of content) {
|
|
105
|
+
if (item.type === 'text') {
|
|
106
|
+
result.push({ type: 'text', text: item.text });
|
|
107
|
+
}
|
|
108
|
+
else if (item.type === 'tool_use' && role === 'assistant') {
|
|
109
|
+
result.push({
|
|
110
|
+
type: 'tool_use',
|
|
111
|
+
id: item.id,
|
|
112
|
+
name: item.name,
|
|
113
|
+
input: item.input,
|
|
114
|
+
});
|
|
115
|
+
}
|
|
116
|
+
else if (item.type === 'tool_result' && role === 'user') {
|
|
117
|
+
result.push({
|
|
118
|
+
type: 'tool_result',
|
|
119
|
+
tool_use_id: item.toolUseId,
|
|
120
|
+
content: item.content,
|
|
121
|
+
is_error: item.isError,
|
|
122
|
+
});
|
|
123
|
+
}
|
|
124
|
+
}
|
|
125
|
+
return result.length > 0 ? result : '';
|
|
126
|
+
}
|
|
127
|
+
convertTools(tools) {
|
|
128
|
+
return tools.map((tool) => ({
|
|
129
|
+
name: tool.name,
|
|
130
|
+
description: tool.description,
|
|
131
|
+
input_schema: tool.parameters,
|
|
132
|
+
}));
|
|
133
|
+
}
|
|
134
|
+
convertResponse(response) {
|
|
135
|
+
return {
|
|
136
|
+
content: this.convertContent(response.content),
|
|
137
|
+
stopReason: this.convertStopReason(response.stop_reason),
|
|
138
|
+
usage: {
|
|
139
|
+
inputTokens: response.usage.input_tokens,
|
|
140
|
+
outputTokens: response.usage.output_tokens,
|
|
141
|
+
},
|
|
142
|
+
};
|
|
143
|
+
}
|
|
144
|
+
convertContent(content) {
|
|
145
|
+
return content.map((block) => {
|
|
146
|
+
if (block.type === 'text') {
|
|
147
|
+
return { type: 'text', text: block.text };
|
|
148
|
+
}
|
|
149
|
+
else if (block.type === 'tool_use') {
|
|
150
|
+
return {
|
|
151
|
+
type: 'tool_use',
|
|
152
|
+
id: block.id,
|
|
153
|
+
name: block.name,
|
|
154
|
+
input: block.input,
|
|
155
|
+
};
|
|
156
|
+
}
|
|
157
|
+
// Fallback for unknown types
|
|
158
|
+
return { type: 'text', text: '' };
|
|
159
|
+
});
|
|
160
|
+
}
|
|
161
|
+
convertStopReason(reason) {
|
|
162
|
+
switch (reason) {
|
|
163
|
+
case 'tool_use':
|
|
164
|
+
return 'tool_use';
|
|
165
|
+
case 'max_tokens':
|
|
166
|
+
return 'max_tokens';
|
|
167
|
+
case 'stop_sequence':
|
|
168
|
+
return 'stop_sequence';
|
|
169
|
+
case 'end_turn':
|
|
170
|
+
default:
|
|
171
|
+
return 'end_turn';
|
|
172
|
+
}
|
|
173
|
+
}
|
|
174
|
+
async listModels() {
|
|
175
|
+
const models = [];
|
|
176
|
+
for await (const model of this.client.models.list()) {
|
|
177
|
+
models.push({
|
|
178
|
+
id: model.id,
|
|
179
|
+
name: model.display_name || model.id,
|
|
180
|
+
});
|
|
181
|
+
}
|
|
182
|
+
return models;
|
|
183
|
+
}
|
|
184
|
+
}
|
|
185
|
+
//# sourceMappingURL=anthropic.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../src/providers/anthropic.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,SAAS,MAAM,mBAAmB,CAAC;AAkB1C,MAAM,OAAO,iBAAiB;IACnB,IAAI,GAAG,WAAW,CAAC;IACpB,MAAM,CAAY;IAE1B,YAAY,SAA0B,EAAE;QACtC,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC;YAC1B,MAAM,EAAE,MAAM,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,iBAAiB;YACtD,OAAO,EAAE,MAAM,CAAC,OAAO;SACxB,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,OAA0B;QACvC,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACxD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAE3E,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC;YACjD,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,QAAQ;YACR,KAAK;YACL,MAAM,EAAE,OAAO,CAAC,YAAY;YAC5B,UAAU,EAAE,OAAO,CAAC,SAAS,IAAI,IAAI;YACrC,WAAW,EAAE,OAAO,CAAC,WAAW;SACjC,CAAC,CAAC;QAEH,OAAO,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;IACxC,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CAAC,OAA0B;QACtC,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACxD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAE3E,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC;YACzC,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,QAAQ;YACR,KAAK;YACL,MAAM,EAAE,OAAO,CAAC,YAAY;YAC5B,UAAU,EAAE,OAAO,CAAC,SAAS,IAAI,IAAI;YACrC,WAAW,EAAE,OAAO,CAAC,WAAW;SACjC,CAAC,CAAC;QAEH,MAAM,gBAAgB,GAA6D,IAAI,GAAG,EAAE,CAAC;QAE7F,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YACjC,IAAI,KAAK,CAAC,IAAI,KAAK,qBAAqB,EAAE,CAAC;gBACzC,MAAM,KAAK,GAAG,KAAK,CAAC,aAAa,CAAC;gBAElC,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBAC9B,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,EAAE;wBAChC,EAAE,EAAE,KAAK,CAAC,EAAE;wBACZ,IAAI,EAAE,KAAK,CAAC,IAAI;wBAChB,KAAK,EAAE,EAAE;qBACV,CAAC,CAAC;oBACH,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,EAAE,EAAE,KAAK,CAAC,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC;gBAC/D,CAAC;YACH,CAAC;iBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,qBAAqB,EAAE,CAAC;gBAChD,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;gBAE1B,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;oBAChC,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC;gBAC3C,CAAC;qBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,kBAAkB,EAAE,CAAC;oBAC7C,MAAM,MAAM,GAAG,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;oBACjD,IAAI,MAAM,EAAE,CAAC;wBACX,MAAM,CAAC,KAAK,IAAI,KAAK,CAAC,YAAY,CAAC;wBACnC,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,EAAE,EAAE,MAAM,CAAC,EAAE,EAAE,KAAK,EAAE,KAAK,CAAC,YAAY,EAAE,CAAC;oBACzE,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,oBAAoB;QACpB,MAAM,YAAY,GAAG,MAAM,MAAM,CAAC,YAAY,EAAE,CAAC;QACjD,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAE1D,MAAM;YACJ,IAAI,EAAE,MAAM;YACZ,QAAQ,EAAE;gBACR,OAAO;gBACP,UAAU,EAAE,IAAI,CAAC,iBAAiB,CAAC,YAAY,CAAC,WAAW,CAAC;gBAC5D,KAAK,EAAE;oBACL,WAAW,EAAE,YAAY,CAAC,KAAK,CAAC,YAAY;oBAC5C,YAAY,EAAE,YAAY,CAAC,KAAK,CAAC,aAAa;iBAC/C;aACF;SACF,CAAC;IACJ,CAAC;IAEO,eAAe,CAAC,QAAmB;QACzC,MAAM,MAAM,GAAuB,EAAE,CAAC;QAEtC,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;YAC3B,4CAA4C;YAC5C,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC1B,SAAS;YACX,CAAC;YAED,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACxB,MAAM,OAAO,GAAG,IAAI,CAAC,yBAAyB,CAAC,GAAG,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;gBACpE,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;YACzC,CAAC;iBAAM,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gBACpC,MAAM,OAAO,GAAG,IAAI,CAAC,yBAAyB,CAAC,GAAG,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;gBACzE,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,CAAC;YAC9C,CAAC;QACH,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,yBAAyB,CAC/B,OAAkC,EAClC,IAA0B;QAE1B,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;YAChC,OAAO,OAAO,CAAC;QACjB,CAAC;QAED,MAAM,MAAM,GAAuB,EAAE,CAAC;QAEtC,KAAK,MAAM,IAAI,IAAI,OAAO,EAAE,CAAC;YAC3B,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACzB,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;YACjD,CAAC;iBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,IAAI,IAAI,KAAK,WAAW,EAAE,CAAC;gBAC5D,MAAM,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,UAAU;oBAChB,EAAE,EAAE,IAAI,CAAC,EAAE;oBACX,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,KAAK,EAAE,IAAI,CAAC,KAAK;iBAClB,CAAC,CAAC;YACL,CAAC;iBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,aAAa,IAAI,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC1D,MAAM,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,aAAa;oBACnB,WAAW,EAAE,IAAI,CAAC,SAAS;oBAC3B,OAAO,EAAE,IAAI,CAAC,OAAO;oBACrB,QAAQ,EAAE,IAAI,CAAC,OAAO;iBACvB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,OAAO,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC;IACzC,CAAC;IAEO,YAAY,CAAC,KAAuB;QAC1C,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC1B,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,YAAY,EAAE,IAAI,CAAC,UAAwC;SAC5D,CAAC,CAAC,CAAC;IACN,CAAC;IAEO,eAAe,CAAC,QAA2B;QACjD,OAAO;YACL,OAAO,EAAE,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,OAAO,CAAC;YAC9C,UAAU,EAAE,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,WAAW,CAAC;YACxD,KAAK,EAAE;gBACL,WAAW,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY;gBACxC,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,aAAa;aAC3C;SACF,CAAC;IACJ,CAAC;IAEO,cAAc,CAAC,OAAiC;QACtD,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;YAC3B,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC1B,OAAO,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC;YACrD,CAAC;iBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBACrC,OAAO;oBACL,IAAI,EAAE,UAAmB;oBACzB,EAAE,EAAE,KAAK,CAAC,EAAE;oBACZ,IAAI,EAAE,KAAK,CAAC,IAAI;oBAChB,KAAK,EAAE,KAAK,CAAC,KAAgC;iBAC9C,CAAC;YACJ,CAAC;YACD,6BAA6B;YAC7B,OAAO,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;QAC7C,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,iBAAiB,CAAC,MAAwC;QAChE,QAAQ,MAAM,EAAE,CAAC;YACf,KAAK,UAAU;gBACb,OAAO,UAAU,CAAC;YACpB,KAAK,YAAY;gBACf,OAAO,YAAY,CAAC;YACtB,KAAK,eAAe;gBAClB,OAAO,eAAe,CAAC;YACzB,KAAK,UAAU,CAAC;YAChB;gBACE,OAAO,UAAU,CAAC;QACtB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU;QACd,MAAM,MAAM,GAAgB,EAAE,CAAC;QAC/B,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;YACpD,MAAM,CAAC,IAAI,CAAC;gBACV,EAAE,EAAE,KAAK,CAAC,EAAE;gBACZ,IAAI,EAAE,KAAK,CAAC,YAAY,IAAI,KAAK,CAAC,EAAE;aACrC,CAAC,CAAC;QACL,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;CACF"}
|
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Google Gemini Provider Implementation
|
|
3
|
+
* Supports Gemini 1.5 Pro, Gemini 1.5 Flash, Gemini 2.0, etc.
|
|
4
|
+
*/
|
|
5
|
+
import type { LLMProvider, CompletionOptions, CompletionResponse, StreamChunk, GeminiConfig, ModelInfo } from './types.js';
|
|
6
|
+
export declare class GeminiProvider implements LLMProvider {
|
|
7
|
+
readonly name = "gemini";
|
|
8
|
+
private client;
|
|
9
|
+
private apiKey;
|
|
10
|
+
constructor(config?: GeminiConfig);
|
|
11
|
+
complete(options: CompletionOptions): Promise<CompletionResponse>;
|
|
12
|
+
stream(options: CompletionOptions): AsyncGenerator<StreamChunk, void, unknown>;
|
|
13
|
+
private convertMessages;
|
|
14
|
+
private convertToParts;
|
|
15
|
+
private convertTools;
|
|
16
|
+
private convertSchema;
|
|
17
|
+
private convertResponse;
|
|
18
|
+
private getStopReason;
|
|
19
|
+
listModels(): Promise<ModelInfo[]>;
|
|
20
|
+
}
|
|
21
|
+
//# sourceMappingURL=gemini.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gemini.d.ts","sourceRoot":"","sources":["../../src/providers/gemini.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAIH,OAAO,KAAK,EACV,WAAW,EACX,iBAAiB,EACjB,kBAAkB,EAClB,WAAW,EAKX,YAAY,EAEZ,SAAS,EACV,MAAM,YAAY,CAAC;AAEpB,qBAAa,cAAe,YAAW,WAAW;IAChD,QAAQ,CAAC,IAAI,YAAY;IACzB,OAAO,CAAC,MAAM,CAAqB;IACnC,OAAO,CAAC,MAAM,CAAS;gBAEX,MAAM,GAAE,YAAiB;IAS/B,QAAQ,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAiBhE,MAAM,CAAC,OAAO,EAAE,iBAAiB,GAAG,cAAc,CAAC,WAAW,EAAE,IAAI,EAAE,OAAO,CAAC;IAuErF,OAAO,CAAC,eAAe;IAoBvB,OAAO,CAAC,cAAc;IAgCtB,OAAO,CAAC,YAAY;IAapB,OAAO,CAAC,aAAa;IAsCrB,OAAO,CAAC,eAAe;IAiCvB,OAAO,CAAC,aAAa;IAef,UAAU,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;CAoBzC"}
|
|
@@ -0,0 +1,241 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Google Gemini Provider Implementation
|
|
3
|
+
* Supports Gemini 1.5 Pro, Gemini 1.5 Flash, Gemini 2.0, etc.
|
|
4
|
+
*/
|
|
5
|
+
import { GoogleGenerativeAI, SchemaType } from '@google/generative-ai';
|
|
6
|
+
export class GeminiProvider {
|
|
7
|
+
name = 'gemini';
|
|
8
|
+
client;
|
|
9
|
+
apiKey;
|
|
10
|
+
constructor(config = {}) {
|
|
11
|
+
const apiKey = config.apiKey ?? process.env.GOOGLE_API_KEY ?? process.env.GEMINI_API_KEY;
|
|
12
|
+
if (!apiKey) {
|
|
13
|
+
throw new Error('Gemini API key is required. Set GOOGLE_API_KEY or GEMINI_API_KEY.');
|
|
14
|
+
}
|
|
15
|
+
this.apiKey = apiKey;
|
|
16
|
+
this.client = new GoogleGenerativeAI(apiKey);
|
|
17
|
+
}
|
|
18
|
+
async complete(options) {
|
|
19
|
+
const model = this.client.getGenerativeModel({
|
|
20
|
+
model: options.model,
|
|
21
|
+
systemInstruction: options.systemPrompt,
|
|
22
|
+
generationConfig: {
|
|
23
|
+
maxOutputTokens: options.maxTokens,
|
|
24
|
+
temperature: options.temperature,
|
|
25
|
+
},
|
|
26
|
+
tools: options.tools ? this.convertTools(options.tools) : undefined,
|
|
27
|
+
});
|
|
28
|
+
const contents = this.convertMessages(options.messages);
|
|
29
|
+
const result = await model.generateContent({ contents });
|
|
30
|
+
return this.convertResponse(result);
|
|
31
|
+
}
|
|
32
|
+
async *stream(options) {
|
|
33
|
+
const model = this.client.getGenerativeModel({
|
|
34
|
+
model: options.model,
|
|
35
|
+
systemInstruction: options.systemPrompt,
|
|
36
|
+
generationConfig: {
|
|
37
|
+
maxOutputTokens: options.maxTokens,
|
|
38
|
+
temperature: options.temperature,
|
|
39
|
+
},
|
|
40
|
+
tools: options.tools ? this.convertTools(options.tools) : undefined,
|
|
41
|
+
});
|
|
42
|
+
const contents = this.convertMessages(options.messages);
|
|
43
|
+
const result = await model.generateContentStream({ contents });
|
|
44
|
+
let textContent = '';
|
|
45
|
+
const functionCalls = [];
|
|
46
|
+
let callIndex = 0;
|
|
47
|
+
for await (const chunk of result.stream) {
|
|
48
|
+
const parts = chunk.candidates?.[0]?.content?.parts ?? [];
|
|
49
|
+
for (const part of parts) {
|
|
50
|
+
if ('text' in part && part.text) {
|
|
51
|
+
textContent += part.text;
|
|
52
|
+
yield { type: 'text', text: part.text };
|
|
53
|
+
}
|
|
54
|
+
else if ('functionCall' in part && part.functionCall) {
|
|
55
|
+
const fc = part.functionCall;
|
|
56
|
+
const id = `call_${callIndex++}`;
|
|
57
|
+
functionCalls.push({
|
|
58
|
+
id,
|
|
59
|
+
name: fc.name,
|
|
60
|
+
args: fc.args ?? {},
|
|
61
|
+
});
|
|
62
|
+
yield { type: 'tool_start', id, name: fc.name };
|
|
63
|
+
yield { type: 'tool_input', id, input: JSON.stringify(fc.args) };
|
|
64
|
+
}
|
|
65
|
+
}
|
|
66
|
+
}
|
|
67
|
+
// Build final response
|
|
68
|
+
const content = [];
|
|
69
|
+
if (textContent) {
|
|
70
|
+
content.push({ type: 'text', text: textContent });
|
|
71
|
+
}
|
|
72
|
+
for (const fc of functionCalls) {
|
|
73
|
+
content.push({
|
|
74
|
+
type: 'tool_use',
|
|
75
|
+
id: fc.id,
|
|
76
|
+
name: fc.name,
|
|
77
|
+
input: fc.args,
|
|
78
|
+
});
|
|
79
|
+
}
|
|
80
|
+
const finalResponse = await result.response;
|
|
81
|
+
const stopReason = this.getStopReason(finalResponse, functionCalls.length > 0);
|
|
82
|
+
yield {
|
|
83
|
+
type: 'done',
|
|
84
|
+
response: {
|
|
85
|
+
content,
|
|
86
|
+
stopReason,
|
|
87
|
+
usage: finalResponse.usageMetadata
|
|
88
|
+
? {
|
|
89
|
+
inputTokens: finalResponse.usageMetadata.promptTokenCount ?? 0,
|
|
90
|
+
outputTokens: finalResponse.usageMetadata.candidatesTokenCount ?? 0,
|
|
91
|
+
}
|
|
92
|
+
: undefined,
|
|
93
|
+
},
|
|
94
|
+
};
|
|
95
|
+
}
|
|
96
|
+
convertMessages(messages) {
|
|
97
|
+
const contents = [];
|
|
98
|
+
for (const msg of messages) {
|
|
99
|
+
// Skip system messages - handled via systemInstruction
|
|
100
|
+
if (msg.role === 'system') {
|
|
101
|
+
continue;
|
|
102
|
+
}
|
|
103
|
+
const role = msg.role === 'assistant' ? 'model' : 'user';
|
|
104
|
+
const parts = this.convertToParts(msg.content, role);
|
|
105
|
+
if (parts.length > 0) {
|
|
106
|
+
contents.push({ role, parts });
|
|
107
|
+
}
|
|
108
|
+
}
|
|
109
|
+
return contents;
|
|
110
|
+
}
|
|
111
|
+
convertToParts(content, role) {
|
|
112
|
+
if (typeof content === 'string') {
|
|
113
|
+
return [{ text: content }];
|
|
114
|
+
}
|
|
115
|
+
const parts = [];
|
|
116
|
+
for (const item of content) {
|
|
117
|
+
if (item.type === 'text') {
|
|
118
|
+
parts.push({ text: item.text });
|
|
119
|
+
}
|
|
120
|
+
else if (item.type === 'tool_use' && role === 'model') {
|
|
121
|
+
// Function call from model
|
|
122
|
+
parts.push({
|
|
123
|
+
functionCall: {
|
|
124
|
+
name: item.name,
|
|
125
|
+
args: item.input,
|
|
126
|
+
},
|
|
127
|
+
});
|
|
128
|
+
}
|
|
129
|
+
else if (item.type === 'tool_result' && role === 'user') {
|
|
130
|
+
// Function response
|
|
131
|
+
parts.push({
|
|
132
|
+
functionResponse: {
|
|
133
|
+
name: item.toolUseId, // Use toolUseId as name reference
|
|
134
|
+
response: { result: item.content },
|
|
135
|
+
},
|
|
136
|
+
});
|
|
137
|
+
}
|
|
138
|
+
}
|
|
139
|
+
return parts;
|
|
140
|
+
}
|
|
141
|
+
convertTools(tools) {
|
|
142
|
+
return [
|
|
143
|
+
{
|
|
144
|
+
functionDeclarations: tools.map((tool) => ({
|
|
145
|
+
name: tool.name,
|
|
146
|
+
description: tool.description,
|
|
147
|
+
parameters: this.convertSchema(tool.parameters),
|
|
148
|
+
})),
|
|
149
|
+
},
|
|
150
|
+
];
|
|
151
|
+
}
|
|
152
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
153
|
+
convertSchema(schema) {
|
|
154
|
+
const convertType = (type) => {
|
|
155
|
+
switch (type) {
|
|
156
|
+
case 'string':
|
|
157
|
+
return SchemaType.STRING;
|
|
158
|
+
case 'number':
|
|
159
|
+
case 'integer':
|
|
160
|
+
return SchemaType.NUMBER;
|
|
161
|
+
case 'boolean':
|
|
162
|
+
return SchemaType.BOOLEAN;
|
|
163
|
+
case 'array':
|
|
164
|
+
return SchemaType.ARRAY;
|
|
165
|
+
case 'object':
|
|
166
|
+
default:
|
|
167
|
+
return SchemaType.OBJECT;
|
|
168
|
+
}
|
|
169
|
+
};
|
|
170
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
171
|
+
const result = {
|
|
172
|
+
type: convertType(schema.type),
|
|
173
|
+
description: schema.description,
|
|
174
|
+
};
|
|
175
|
+
if (schema.properties) {
|
|
176
|
+
result.properties = {};
|
|
177
|
+
for (const [key, value] of Object.entries(schema.properties)) {
|
|
178
|
+
result.properties[key] = this.convertSchema(value);
|
|
179
|
+
}
|
|
180
|
+
}
|
|
181
|
+
if (schema.required) {
|
|
182
|
+
result.required = schema.required;
|
|
183
|
+
}
|
|
184
|
+
return result;
|
|
185
|
+
}
|
|
186
|
+
convertResponse(result) {
|
|
187
|
+
const response = result.response;
|
|
188
|
+
const parts = response.candidates?.[0]?.content?.parts ?? [];
|
|
189
|
+
const content = [];
|
|
190
|
+
let callIndex = 0;
|
|
191
|
+
for (const part of parts) {
|
|
192
|
+
if ('text' in part && part.text) {
|
|
193
|
+
content.push({ type: 'text', text: part.text });
|
|
194
|
+
}
|
|
195
|
+
else if ('functionCall' in part && part.functionCall) {
|
|
196
|
+
content.push({
|
|
197
|
+
type: 'tool_use',
|
|
198
|
+
id: `call_${callIndex++}`,
|
|
199
|
+
name: part.functionCall.name,
|
|
200
|
+
input: part.functionCall.args ?? {},
|
|
201
|
+
});
|
|
202
|
+
}
|
|
203
|
+
}
|
|
204
|
+
const hasFunctionCalls = parts.some((p) => 'functionCall' in p);
|
|
205
|
+
return {
|
|
206
|
+
content,
|
|
207
|
+
stopReason: this.getStopReason(response, hasFunctionCalls),
|
|
208
|
+
usage: response.usageMetadata
|
|
209
|
+
? {
|
|
210
|
+
inputTokens: response.usageMetadata.promptTokenCount ?? 0,
|
|
211
|
+
outputTokens: response.usageMetadata.candidatesTokenCount ?? 0,
|
|
212
|
+
}
|
|
213
|
+
: undefined,
|
|
214
|
+
};
|
|
215
|
+
}
|
|
216
|
+
getStopReason(response, hasFunctionCalls) {
|
|
217
|
+
if (hasFunctionCalls) {
|
|
218
|
+
return 'tool_use';
|
|
219
|
+
}
|
|
220
|
+
const finishReason = response.candidates?.[0]?.finishReason;
|
|
221
|
+
switch (finishReason) {
|
|
222
|
+
case 'MAX_TOKENS':
|
|
223
|
+
return 'max_tokens';
|
|
224
|
+
case 'STOP':
|
|
225
|
+
default:
|
|
226
|
+
return 'end_turn';
|
|
227
|
+
}
|
|
228
|
+
}
|
|
229
|
+
async listModels() {
|
|
230
|
+
const response = await fetch(`https://generativelanguage.googleapis.com/v1beta/models?key=${this.apiKey}`);
|
|
231
|
+
const data = (await response.json());
|
|
232
|
+
return (data.models || [])
|
|
233
|
+
.filter((m) => m.supportedGenerationMethods?.includes('generateContent'))
|
|
234
|
+
.map((m) => ({
|
|
235
|
+
id: m.name.replace('models/', ''),
|
|
236
|
+
name: m.displayName,
|
|
237
|
+
description: m.description,
|
|
238
|
+
}));
|
|
239
|
+
}
|
|
240
|
+
}
|
|
241
|
+
//# sourceMappingURL=gemini.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gemini.js","sourceRoot":"","sources":["../../src/providers/gemini.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,kBAAkB,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAgBvE,MAAM,OAAO,cAAc;IAChB,IAAI,GAAG,QAAQ,CAAC;IACjB,MAAM,CAAqB;IAC3B,MAAM,CAAS;IAEvB,YAAY,SAAuB,EAAE;QACnC,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;QACzF,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,mEAAmE,CAAC,CAAC;QACvF,CAAC;QACD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,IAAI,kBAAkB,CAAC,MAAM,CAAC,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,OAA0B;QACvC,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC;YAC3C,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,iBAAiB,EAAE,OAAO,CAAC,YAAY;YACvC,gBAAgB,EAAE;gBAChB,eAAe,EAAE,OAAO,CAAC,SAAS;gBAClC,WAAW,EAAE,OAAO,CAAC,WAAW;aACjC;YACD,KAAK,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS;SACpE,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACxD,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,eAAe,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;QAEzD,OAAO,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;IACtC,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CAAC,OAA0B;QACtC,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC;YAC3C,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,iBAAiB,EAAE,OAAO,CAAC,YAAY;YACvC,gBAAgB,EAAE;gBAChB,eAAe,EAAE,OAAO,CAAC,SAAS;gBAClC,WAAW,EAAE,OAAO,CAAC,WAAW;aACjC;YACD,KAAK,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS;SACpE,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACxD,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,qBAAqB,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;QAE/D,IAAI,WAAW,GAAG,EAAE,CAAC;QACrB,MAAM,aAAa,GAAuE,EAAE,CAAC;QAC7F,IAAI,SAAS,GAAG,CAAC,CAAC;QAElB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YACxC,MAAM,KAAK,GAAG,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC;YAE1D,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,IAAI,MAAM,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;oBAChC,WAAW,IAAI,IAAI,CAAC,IAAI,CAAC;oBACzB,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;gBAC1C,CAAC;qBAAM,IAAI,cAAc,IAAI,IAAI,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;oBACvD,MAAM,EAAE,GAAG,IAAI,CAAC,YAAY,CAAC;oBAC7B,MAAM,EAAE,GAAG,QAAQ,SAAS,EAAE,EAAE,CAAC;oBACjC,aAAa,CAAC,IAAI,CAAC;wBACjB,EAAE;wBACF,IAAI,EAAE,EAAE,CAAC,IAAI;wBACb,IAAI,EAAG,EAAE,CAAC,IAAgC,IAAI,EAAE;qBACjD,CAAC,CAAC;oBACH,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC;oBAChD,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;gBACnE,CAAC;YACH,CAAC;QACH,CAAC;QAED,uBAAuB;QACvB,MAAM,OAAO,GAAqB,EAAE,CAAC;QACrC,IAAI,WAAW,EAAE,CAAC;YAChB,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC;QACpD,CAAC;QACD,KAAK,MAAM,EAAE,IAAI,aAAa,EAAE,CAAC;YAC/B,OAAO,CAAC,IAAI,CAAC;gBACX,IAAI,EAAE,UAAU;gBAChB,EAAE,EAAE,EAAE,CAAC,EAAE;gBACT,IAAI,EAAE,EAAE,CAAC,IAAI;gBACb,KAAK,EAAE,EAAE,CAAC,IAAI;aACf,CAAC,CAAC;QACL,CAAC;QAED,MAAM,aAAa,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC;QAC5C,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC,aAAa,EAAE,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAE/E,MAAM;YACJ,IAAI,EAAE,MAAM;YACZ,QAAQ,EAAE;gBACR,OAAO;gBACP,UAAU;gBACV,KAAK,EAAE,aAAa,CAAC,aAAa;oBAChC,CAAC,CAAC;wBACE,WAAW,EAAE,aAAa,CAAC,aAAa,CAAC,gBAAgB,IAAI,CAAC;wBAC9D,YAAY,EAAE,aAAa,CAAC,aAAa,CAAC,oBAAoB,IAAI,CAAC;qBACpE;oBACH,CAAC,CAAC,SAAS;aACd;SACF,CAAC;IACJ,CAAC;IAEO,eAAe,CAAC,QAAmB;QACzC,MAAM,QAAQ,GAAc,EAAE,CAAC;QAE/B,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;YAC3B,uDAAuD;YACvD,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC1B,SAAS;YACX,CAAC;YAED,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC;YACzD,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;YAErD,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACrB,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;YACjC,CAAC;QACH,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAEO,cAAc,CAAC,OAAkC,EAAE,IAAY;QACrE,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;YAChC,OAAO,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;QAC7B,CAAC;QAED,MAAM,KAAK,GAAW,EAAE,CAAC;QAEzB,KAAK,MAAM,IAAI,IAAI,OAAO,EAAE,CAAC;YAC3B,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACzB,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;YAClC,CAAC;iBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,IAAI,IAAI,KAAK,OAAO,EAAE,CAAC;gBACxD,2BAA2B;gBAC3B,KAAK,CAAC,IAAI,CAAC;oBACT,YAAY,EAAE;wBACZ,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,IAAI,EAAE,IAAI,CAAC,KAAK;qBACjB;iBACF,CAAC,CAAC;YACL,CAAC;iBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,aAAa,IAAI,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC1D,oBAAoB;gBACpB,KAAK,CAAC,IAAI,CAAC;oBACT,gBAAgB,EAAE;wBAChB,IAAI,EAAE,IAAI,CAAC,SAAS,EAAE,kCAAkC;wBACxD,QAAQ,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,OAAO,EAAE;qBACnC;iBACF,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,YAAY,CAAC,KAAuB;QAC1C,OAAO;YACL;gBACE,oBAAoB,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;oBACzC,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,WAAW,EAAE,IAAI,CAAC,WAAW;oBAC7B,UAAU,EAAE,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,UAAU,CAAC;iBAChD,CAAC,CAAC;aACJ;SACF,CAAC;IACJ,CAAC;IAED,8DAA8D;IACtD,aAAa,CAAC,MAAkB;QACtC,MAAM,WAAW,GAAG,CAAC,IAAY,EAAc,EAAE;YAC/C,QAAQ,IAAI,EAAE,CAAC;gBACb,KAAK,QAAQ;oBACX,OAAO,UAAU,CAAC,MAAM,CAAC;gBAC3B,KAAK,QAAQ,CAAC;gBACd,KAAK,SAAS;oBACZ,OAAO,UAAU,CAAC,MAAM,CAAC;gBAC3B,KAAK,SAAS;oBACZ,OAAO,UAAU,CAAC,OAAO,CAAC;gBAC5B,KAAK,OAAO;oBACV,OAAO,UAAU,CAAC,KAAK,CAAC;gBAC1B,KAAK,QAAQ,CAAC;gBACd;oBACE,OAAO,UAAU,CAAC,MAAM,CAAC;YAC7B,CAAC;QACH,CAAC,CAAC;QAEF,8DAA8D;QAC9D,MAAM,MAAM,GAAQ;YAClB,IAAI,EAAE,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC;YAC9B,WAAW,EAAE,MAAM,CAAC,WAAW;SAChC,CAAC;QAEF,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;YACtB,MAAM,CAAC,UAAU,GAAG,EAAE,CAAC;YACvB,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC7D,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;YACrD,CAAC;QACH,CAAC;QAED,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;YACpB,MAAM,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QACpC,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,eAAe,CAAC,MAA6B;QACnD,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QACjC,MAAM,KAAK,GAAG,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC;QAC7D,MAAM,OAAO,GAAqB,EAAE,CAAC;QACrC,IAAI,SAAS,GAAG,CAAC,CAAC;QAElB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,MAAM,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;gBAChC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;YAClD,CAAC;iBAAM,IAAI,cAAc,IAAI,IAAI,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;gBACvD,OAAO,CAAC,IAAI,CAAC;oBACX,IAAI,EAAE,UAAU;oBAChB,EAAE,EAAE,QAAQ,SAAS,EAAE,EAAE;oBACzB,IAAI,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI;oBAC5B,KAAK,EAAG,IAAI,CAAC,YAAY,CAAC,IAAgC,IAAI,EAAE;iBACjE,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,MAAM,gBAAgB,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,cAAc,IAAI,CAAC,CAAC,CAAC;QAEhE,OAAO;YACL,OAAO;YACP,UAAU,EAAE,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,gBAAgB,CAAC;YAC1D,KAAK,EAAE,QAAQ,CAAC,aAAa;gBAC3B,CAAC,CAAC;oBACE,WAAW,EAAE,QAAQ,CAAC,aAAa,CAAC,gBAAgB,IAAI,CAAC;oBACzD,YAAY,EAAE,QAAQ,CAAC,aAAa,CAAC,oBAAoB,IAAI,CAAC;iBAC/D;gBACH,CAAC,CAAC,SAAS;SACd,CAAC;IACJ,CAAC;IAEO,aAAa,CAAC,QAA2C,EAAE,gBAAyB;QAC1F,IAAI,gBAAgB,EAAE,CAAC;YACrB,OAAO,UAAU,CAAC;QACpB,CAAC;QAED,MAAM,YAAY,GAAG,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,CAAC;QAC5D,QAAQ,YAAY,EAAE,CAAC;YACrB,KAAK,YAAY;gBACf,OAAO,YAAY,CAAC;YACtB,KAAK,MAAM,CAAC;YACZ;gBACE,OAAO,UAAU,CAAC;QACtB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU;QACd,MAAM,QAAQ,GAAG,MAAM,KAAK,CAC1B,+DAA+D,IAAI,CAAC,MAAM,EAAE,CAC7E,CAAC;QACF,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAOlC,CAAC;QACF,OAAO,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC;aACvB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,0BAA0B,EAAE,QAAQ,CAAC,iBAAiB,CAAC,CAAC;aACxE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACX,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC;YACjC,IAAI,EAAE,CAAC,CAAC,WAAW;YACnB,WAAW,EAAE,CAAC,CAAC,WAAW;SAC3B,CAAC,CAAC,CAAC;IACR,CAAC;CACF"}
|
|
@@ -0,0 +1,34 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* LLM Providers - Unified interface for OpenAI, Anthropic, and Gemini
|
|
3
|
+
*/
|
|
4
|
+
export * from './types.js';
|
|
5
|
+
export { OpenAIProvider } from './openai.js';
|
|
6
|
+
export { AnthropicProvider } from './anthropic.js';
|
|
7
|
+
export { GeminiProvider } from './gemini.js';
|
|
8
|
+
import type { LLMProvider, OpenAIConfig, AnthropicConfig, GeminiConfig } from './types.js';
|
|
9
|
+
export type ProviderName = 'openai' | 'anthropic' | 'gemini';
|
|
10
|
+
export type ProviderConfigMap = {
|
|
11
|
+
openai: OpenAIConfig;
|
|
12
|
+
anthropic: AnthropicConfig;
|
|
13
|
+
gemini: GeminiConfig;
|
|
14
|
+
};
|
|
15
|
+
export interface CreateProviderOptions<T extends ProviderName = ProviderName> {
|
|
16
|
+
provider: T;
|
|
17
|
+
config?: ProviderConfigMap[T];
|
|
18
|
+
}
|
|
19
|
+
/**
|
|
20
|
+
* Create a provider instance by name
|
|
21
|
+
*/
|
|
22
|
+
export declare function createProvider(options: CreateProviderOptions): LLMProvider;
|
|
23
|
+
/**
|
|
24
|
+
* Infer provider from model name
|
|
25
|
+
*/
|
|
26
|
+
export declare function inferProvider(model: string): ProviderName;
|
|
27
|
+
/**
|
|
28
|
+
* Common model aliases
|
|
29
|
+
*/
|
|
30
|
+
export declare const ModelAliases: Record<string, {
|
|
31
|
+
provider: ProviderName;
|
|
32
|
+
model: string;
|
|
33
|
+
}>;
|
|
34
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/providers/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,cAAc,YAAY,CAAC;AAC3B,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAE7C,OAAO,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAK3F,MAAM,MAAM,YAAY,GAAG,QAAQ,GAAG,WAAW,GAAG,QAAQ,CAAC;AAC7D,MAAM,MAAM,iBAAiB,GAAG;IAC9B,MAAM,EAAE,YAAY,CAAC;IACrB,SAAS,EAAE,eAAe,CAAC;IAC3B,MAAM,EAAE,YAAY,CAAC;CACtB,CAAC;AAEF,MAAM,WAAW,qBAAqB,CAAC,CAAC,SAAS,YAAY,GAAG,YAAY;IAC1E,QAAQ,EAAE,CAAC,CAAC;IACZ,MAAM,CAAC,EAAE,iBAAiB,CAAC,CAAC,CAAC,CAAC;CAC/B;AAED;;GAEG;AACH,wBAAgB,cAAc,CAAC,OAAO,EAAE,qBAAqB,GAAG,WAAW,CAW1E;AAED;;GAEG;AACH,wBAAgB,aAAa,CAAC,KAAK,EAAE,MAAM,GAAG,YAAY,CA2BzD;AAED;;GAEG;AACH,eAAO,MAAM,YAAY,EAAE,MAAM,CAAC,MAAM,EAAE;IAAE,QAAQ,EAAE,YAAY,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,CAmBlF,CAAC"}
|
|
@@ -0,0 +1,72 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* LLM Providers - Unified interface for OpenAI, Anthropic, and Gemini
|
|
3
|
+
*/
|
|
4
|
+
export * from './types.js';
|
|
5
|
+
export { OpenAIProvider } from './openai.js';
|
|
6
|
+
export { AnthropicProvider } from './anthropic.js';
|
|
7
|
+
export { GeminiProvider } from './gemini.js';
|
|
8
|
+
import { OpenAIProvider } from './openai.js';
|
|
9
|
+
import { AnthropicProvider } from './anthropic.js';
|
|
10
|
+
import { GeminiProvider } from './gemini.js';
|
|
11
|
+
/**
|
|
12
|
+
* Create a provider instance by name
|
|
13
|
+
*/
|
|
14
|
+
export function createProvider(options) {
|
|
15
|
+
switch (options.provider) {
|
|
16
|
+
case 'openai':
|
|
17
|
+
return new OpenAIProvider(options.config);
|
|
18
|
+
case 'anthropic':
|
|
19
|
+
return new AnthropicProvider(options.config);
|
|
20
|
+
case 'gemini':
|
|
21
|
+
return new GeminiProvider(options.config);
|
|
22
|
+
default:
|
|
23
|
+
throw new Error(`Unknown provider: ${options.provider}`);
|
|
24
|
+
}
|
|
25
|
+
}
|
|
26
|
+
/**
|
|
27
|
+
* Infer provider from model name
|
|
28
|
+
*/
|
|
29
|
+
export function inferProvider(model) {
|
|
30
|
+
const modelLower = model.toLowerCase();
|
|
31
|
+
// OpenAI models
|
|
32
|
+
if (modelLower.includes('gpt') ||
|
|
33
|
+
modelLower.includes('o1') ||
|
|
34
|
+
modelLower.includes('o3') ||
|
|
35
|
+
modelLower.startsWith('text-') ||
|
|
36
|
+
modelLower.startsWith('davinci') ||
|
|
37
|
+
modelLower.startsWith('curie')) {
|
|
38
|
+
return 'openai';
|
|
39
|
+
}
|
|
40
|
+
// Anthropic models
|
|
41
|
+
if (modelLower.includes('claude')) {
|
|
42
|
+
return 'anthropic';
|
|
43
|
+
}
|
|
44
|
+
// Gemini models
|
|
45
|
+
if (modelLower.includes('gemini') || modelLower.includes('palm')) {
|
|
46
|
+
return 'gemini';
|
|
47
|
+
}
|
|
48
|
+
// Default to OpenAI (most common)
|
|
49
|
+
return 'openai';
|
|
50
|
+
}
|
|
51
|
+
/**
|
|
52
|
+
* Common model aliases
|
|
53
|
+
*/
|
|
54
|
+
export const ModelAliases = {
|
|
55
|
+
// OpenAI
|
|
56
|
+
'gpt-4o': { provider: 'openai', model: 'gpt-4o' },
|
|
57
|
+
'gpt-4o-mini': { provider: 'openai', model: 'gpt-4o-mini' },
|
|
58
|
+
'gpt-4-turbo': { provider: 'openai', model: 'gpt-4-turbo' },
|
|
59
|
+
'o1': { provider: 'openai', model: 'o1' },
|
|
60
|
+
'o1-mini': { provider: 'openai', model: 'o1-mini' },
|
|
61
|
+
'o3-mini': { provider: 'openai', model: 'o3-mini' },
|
|
62
|
+
// Anthropic
|
|
63
|
+
'claude-opus': { provider: 'anthropic', model: 'claude-opus-4-5-20251101' },
|
|
64
|
+
'claude-sonnet': { provider: 'anthropic', model: 'claude-sonnet-4-20250514' },
|
|
65
|
+
'claude-haiku': { provider: 'anthropic', model: 'claude-haiku-4-20250514' },
|
|
66
|
+
'claude-3.5-sonnet': { provider: 'anthropic', model: 'claude-3-5-sonnet-20241022' },
|
|
67
|
+
// Gemini
|
|
68
|
+
'gemini-2.0-flash': { provider: 'gemini', model: 'gemini-2.0-flash' },
|
|
69
|
+
'gemini-1.5-pro': { provider: 'gemini', model: 'gemini-1.5-pro' },
|
|
70
|
+
'gemini-1.5-flash': { provider: 'gemini', model: 'gemini-1.5-flash' },
|
|
71
|
+
};
|
|
72
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/providers/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,cAAc,YAAY,CAAC;AAC3B,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAG7C,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAc7C;;GAEG;AACH,MAAM,UAAU,cAAc,CAAC,OAA8B;IAC3D,QAAQ,OAAO,CAAC,QAAQ,EAAE,CAAC;QACzB,KAAK,QAAQ;YACX,OAAO,IAAI,cAAc,CAAC,OAAO,CAAC,MAAsB,CAAC,CAAC;QAC5D,KAAK,WAAW;YACd,OAAO,IAAI,iBAAiB,CAAC,OAAO,CAAC,MAAyB,CAAC,CAAC;QAClE,KAAK,QAAQ;YACX,OAAO,IAAI,cAAc,CAAC,OAAO,CAAC,MAAsB,CAAC,CAAC;QAC5D;YACE,MAAM,IAAI,KAAK,CAAC,qBAAqB,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC7D,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,aAAa,CAAC,KAAa;IACzC,MAAM,UAAU,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;IAEvC,gBAAgB;IAChB,IACE,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC;QAC1B,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC;QACzB,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC;QACzB,UAAU,CAAC,UAAU,CAAC,OAAO,CAAC;QAC9B,UAAU,CAAC,UAAU,CAAC,SAAS,CAAC;QAChC,UAAU,CAAC,UAAU,CAAC,OAAO,CAAC,EAC9B,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,mBAAmB;IACnB,IAAI,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;QAClC,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,gBAAgB;IAChB,IAAI,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,UAAU,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;QACjE,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,kCAAkC;IAClC,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,MAAM,YAAY,GAA8D;IACrF,SAAS;IACT,QAAQ,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE;IACjD,aAAa,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,aAAa,EAAE;IAC3D,aAAa,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,aAAa,EAAE;IAC3D,IAAI,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE;IACzC,SAAS,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,SAAS,EAAE;IACnD,SAAS,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,SAAS,EAAE;IAEnD,YAAY;IACZ,aAAa,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,KAAK,EAAE,0BAA0B,EAAE;IAC3E,eAAe,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,KAAK,EAAE,0BAA0B,EAAE;IAC7E,cAAc,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,KAAK,EAAE,yBAAyB,EAAE;IAC3E,mBAAmB,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,KAAK,EAAE,4BAA4B,EAAE;IAEnF,SAAS;IACT,kBAAkB,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,kBAAkB,EAAE;IACrE,gBAAgB,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,gBAAgB,EAAE;IACjE,kBAAkB,EAAE,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,kBAAkB,EAAE;CACtE,CAAC"}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* OpenAI Provider Implementation
|
|
3
|
+
* Supports GPT-4, GPT-4o, GPT-3.5-turbo, and other OpenAI models
|
|
4
|
+
*/
|
|
5
|
+
import type { LLMProvider, CompletionOptions, CompletionResponse, StreamChunk, OpenAIConfig, ModelInfo } from './types.js';
|
|
6
|
+
export declare class OpenAIProvider implements LLMProvider {
|
|
7
|
+
readonly name = "openai";
|
|
8
|
+
private client;
|
|
9
|
+
constructor(config?: OpenAIConfig);
|
|
10
|
+
complete(options: CompletionOptions): Promise<CompletionResponse>;
|
|
11
|
+
stream(options: CompletionOptions): AsyncGenerator<StreamChunk, void, unknown>;
|
|
12
|
+
private convertMessages;
|
|
13
|
+
private convertTools;
|
|
14
|
+
private convertResponse;
|
|
15
|
+
private convertStopReason;
|
|
16
|
+
private getTextContent;
|
|
17
|
+
listModels(): Promise<ModelInfo[]>;
|
|
18
|
+
}
|
|
19
|
+
//# sourceMappingURL=openai.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../src/providers/openai.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAGH,OAAO,KAAK,EACV,WAAW,EACX,iBAAiB,EACjB,kBAAkB,EAClB,WAAW,EAKX,YAAY,EACZ,SAAS,EACV,MAAM,YAAY,CAAC;AAKpB,qBAAa,cAAe,YAAW,WAAW;IAChD,QAAQ,CAAC,IAAI,YAAY;IACzB,OAAO,CAAC,MAAM,CAAS;gBAEX,MAAM,GAAE,YAAiB;IAQ/B,QAAQ,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAehE,MAAM,CAAC,OAAO,EAAE,iBAAiB,GAAG,cAAc,CAAC,WAAW,EAAE,IAAI,EAAE,OAAO,CAAC;IAwErF,OAAO,CAAC,eAAe;IAiEvB,OAAO,CAAC,YAAY;IAWpB,OAAO,CAAC,eAAe;IAiCvB,OAAO,CAAC,iBAAiB;IAYzB,OAAO,CAAC,cAAc;IAUhB,UAAU,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;CAQzC"}
|