llmjs2 0.0.2 → 1.0.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/dist/agent.js ADDED
@@ -0,0 +1,189 @@
1
+ /**
2
+ * Agent - Stateful conversation manager with tool support
3
+ */
4
+ import { completion, streamCompletion, LLMError, } from './index.js';
5
+ /**
6
+ * Stateful agent for conversations and tool use
7
+ */
8
+ export class Agent {
9
+ constructor(config) {
10
+ this.conversationHistory = [];
11
+ if (!config.model) {
12
+ throw new LLMError('Model is required in agent config', 'MISSING_MODEL');
13
+ }
14
+ this.config = config;
15
+ // Initialize conversation with system instruction
16
+ if (config.instruction) {
17
+ this.conversationHistory.push({
18
+ role: 'system',
19
+ content: config.instruction,
20
+ });
21
+ }
22
+ }
23
+ /**
24
+ * Generate a response for a user message
25
+ */
26
+ async generate(request) {
27
+ if (!request.userPrompt) {
28
+ throw new LLMError('userPrompt is required', 'MISSING_USER_PROMPT');
29
+ }
30
+ // Build user message with context
31
+ let userMessage = request.userPrompt;
32
+ if (request.images && request.images.length > 0) {
33
+ userMessage += `\n\n[Images: ${request.images.length} image(s) attached]`;
34
+ }
35
+ if (request.references && request.references.length > 0) {
36
+ userMessage += '\n\nReferences:\n' + request.references.join('\n---\n');
37
+ }
38
+ if (request.context && Object.keys(request.context).length > 0) {
39
+ userMessage +=
40
+ '\n\nContext: ' + JSON.stringify(request.context, null, 2);
41
+ }
42
+ // Add user message to history
43
+ this.conversationHistory.push({
44
+ role: 'user',
45
+ content: userMessage,
46
+ });
47
+ // Get initial completion
48
+ let completion_result = await this.getCompletion();
49
+ let toolCalls = completion_result.toolCalls;
50
+ let maxIterations = 10;
51
+ let iterations = 0;
52
+ // Handle tool calls in a loop
53
+ while (toolCalls && toolCalls.length > 0 && iterations < maxIterations) {
54
+ iterations++;
55
+ // Execute all tool calls
56
+ const toolResults = [];
57
+ for (const toolCall of toolCalls) {
58
+ try {
59
+ let result;
60
+ if (this.config.toolExecutor) {
61
+ // Use provided tool executor
62
+ result = this.config.toolExecutor(toolCall.name, toolCall.arguments);
63
+ }
64
+ else {
65
+ // Fallback: return error
66
+ result = `Tool '${toolCall.name}' not configured`;
67
+ }
68
+ toolResults.push(`Tool: ${toolCall.name}\nResult: ${result}`);
69
+ }
70
+ catch (error) {
71
+ toolResults.push(`Tool: ${toolCall.name}\nError: ${error instanceof Error ? error.message : String(error)}`);
72
+ }
73
+ }
74
+ // Add tool results to conversation
75
+ this.conversationHistory.push({
76
+ role: 'user',
77
+ content: 'Tool Results:\n' + toolResults.join('\n\n'),
78
+ });
79
+ // Get next completion
80
+ completion_result = await this.getCompletion();
81
+ toolCalls = completion_result.toolCalls;
82
+ }
83
+ // Add final assistant response to history
84
+ this.conversationHistory.push({
85
+ role: 'assistant',
86
+ content: completion_result.content,
87
+ });
88
+ return {
89
+ response: completion_result.content,
90
+ completion: completion_result,
91
+ };
92
+ }
93
+ /**
94
+ * Get completion from the model
95
+ */
96
+ async getCompletion() {
97
+ const completionRequest = {
98
+ model: this.config.model,
99
+ apiKey: this.config.apiKey,
100
+ baseUrl: this.config.baseUrl,
101
+ messages: this.conversationHistory,
102
+ tools: this.config.tools,
103
+ };
104
+ return completion(completionRequest);
105
+ }
106
+ /**
107
+ * Stream a response for a user message
108
+ */
109
+ async *generateStream(request) {
110
+ if (!request.userPrompt) {
111
+ throw new LLMError('userPrompt is required', 'MISSING_USER_PROMPT');
112
+ }
113
+ // Build user message with context
114
+ let userMessage = request.userPrompt;
115
+ if (request.images && request.images.length > 0) {
116
+ userMessage += `\n\n[Images: ${request.images.length} image(s) attached]`;
117
+ }
118
+ if (request.references && request.references.length > 0) {
119
+ userMessage += '\n\nReferences:\n' + request.references.join('\n---\n');
120
+ }
121
+ if (request.context && Object.keys(request.context).length > 0) {
122
+ userMessage +=
123
+ '\n\nContext: ' + JSON.stringify(request.context, null, 2);
124
+ }
125
+ // Add user message to history
126
+ this.conversationHistory.push({
127
+ role: 'user',
128
+ content: userMessage,
129
+ });
130
+ // Create completion request
131
+ const completionRequest = {
132
+ model: this.config.model,
133
+ apiKey: this.config.apiKey,
134
+ baseUrl: this.config.baseUrl,
135
+ messages: this.conversationHistory,
136
+ tools: this.config.tools,
137
+ };
138
+ // Stream and collect response
139
+ let fullResponse = '';
140
+ const stream = streamCompletion(completionRequest);
141
+ for await (const chunk of stream) {
142
+ fullResponse += chunk.delta;
143
+ yield chunk;
144
+ }
145
+ // Add assistant response to history
146
+ this.conversationHistory.push({
147
+ role: 'assistant',
148
+ content: fullResponse,
149
+ });
150
+ }
151
+ /**
152
+ * Get conversation history
153
+ */
154
+ getHistory() {
155
+ return [...this.conversationHistory];
156
+ }
157
+ /**
158
+ * Clear conversation history (keeps system instruction if set)
159
+ */
160
+ clearHistory() {
161
+ if (this.config.instruction) {
162
+ this.conversationHistory = [
163
+ {
164
+ role: 'system',
165
+ content: this.config.instruction,
166
+ },
167
+ ];
168
+ }
169
+ else {
170
+ this.conversationHistory = [];
171
+ }
172
+ }
173
+ /**
174
+ * Add a message to history
175
+ */
176
+ addMessage(role, content) {
177
+ this.conversationHistory.push({
178
+ role,
179
+ content,
180
+ });
181
+ }
182
+ /**
183
+ * Get the current configuration
184
+ */
185
+ getConfig() {
186
+ return { ...this.config };
187
+ }
188
+ }
189
+ //# sourceMappingURL=agent.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"agent.js","sourceRoot":"","sources":["../src/agent.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EACL,UAAU,EACV,gBAAgB,EAMhB,QAAQ,GACT,MAAM,YAAY,CAAC;AAqDpB;;GAEG;AACH,MAAM,OAAO,KAAK;IAIhB,YAAY,MAAmB;QAFvB,wBAAmB,GAAc,EAAE,CAAC;QAG1C,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;YAClB,MAAM,IAAI,QAAQ,CAAC,mCAAmC,EAAE,eAAe,CAAC,CAAC;QAC3E,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QAErB,kDAAkD;QAClD,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC;YACvB,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC;gBAC5B,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,MAAM,CAAC,WAAW;aAC5B,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ,CAAC,OAA6B;QAC1C,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;YACxB,MAAM,IAAI,QAAQ,CAAC,wBAAwB,EAAE,qBAAqB,CAAC,CAAC;QACtE,CAAC;QAED,kCAAkC;QAClC,IAAI,WAAW,GAAG,OAAO,CAAC,UAAU,CAAC;QAErC,IAAI,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChD,WAAW,IAAI,gBAAgB,OAAO,CAAC,MAAM,CAAC,MAAM,qBAAqB,CAAC;QAC5E,CAAC;QAED,IAAI,OAAO,CAAC,UAAU,IAAI,OAAO,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxD,WAAW,IAAI,mBAAmB,GAAG,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC1E,CAAC;QAED,IAAI,OAAO,CAAC,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC/D,WAAW;gBACT,eAAe,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QAC/D,CAAC;QAED,8BAA8B;QAC9B,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC;YAC5B,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,WAAW;SACrB,CAAC,CAAC;QAEH,yBAAyB;QACzB,IAAI,iBAAiB,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QACnD,IAAI,SAAS,GAAG,iBAAiB,CAAC,SAAS,CAAC;QAC5C,IAAI,aAAa,GAAG,EAAE,CAAC;QACvB,IAAI,UAAU,GAAG,CAAC,CAAC;QAEnB,8BAA8B;QAC9B,OAAO,SAAS,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,UAAU,GAAG,aAAa,EAAE,CAAC;YACvE,UAAU,EAAE,CAAC;YAEb,yBAAyB;YACzB,MAAM,WAAW,GAAa,EAAE,CAAC;YAEjC,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;gBACjC,IAAI,CAAC;oBACH,IAAI,MAAc,CAAC;oBAEnB,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;wBAC7B,6BAA6B;wBAC7B,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC,SAAS,CAAC,CAAC;oBACvE,CAAC;yBAAM,CAAC;wBACN,yBAAyB;wBACzB,MAAM,GAAG,SAAS,QAAQ,CAAC,IAAI,kBAAkB,CAAC;oBACpD,CAAC;oBAED,WAAW,CAAC,IAAI,CAAC,SAAS,QAAQ,CAAC,IAAI,aAAa,MAAM,EAAE,CAAC,CAAC;gBAChE,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,WAAW,CAAC,IAAI,CAAC,SAAS,QAAQ,CAAC,IAAI,YAAY,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;gBAC/G,CAAC;YACH,CAAC;YAED,mCAAmC;YACnC,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC;gBAC5B,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,iBAAiB,GAAG,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC;aACtD,CAAC,CAAC;YAEH,sBAAsB;YACtB,iBAAiB,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;YAC/C,SAAS,GAAG,iBAAiB,CAAC,SAAS,CAAC;QAC1C,CAAC;QAED,0CAA0C;QAC1C,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC;YAC5B,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,iBAAiB,CAAC,OAAO;SACnC,CAAC,CAAC;QAEH,OAAO;YACL,QAAQ,EAAE,iBAAiB,CAAC,OAAO;YACnC,UAAU,EAAE,iBAAiB;SAC9B,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,aAAa;QACzB,MAAM,iBAAiB,GAAsB;YAC3C,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;YACxB,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM;YAC1B,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO;YAC5B,QAAQ,EAAE,IAAI,CAAC,mBAAmB;YAClC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;SACzB,CAAC;QAEF,OAAO,UAAU,CAAC,iBAAiB,CAAC,CAAC;IACvC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,CAAC,cAAc,CACnB,OAA6B;QAE7B,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;YACxB,MAAM,IAAI,QAAQ,CAAC,wBAAwB,EAAE,qBAAqB,CAAC,CAAC;QACtE,CAAC;QAED,kCAAkC;QAClC,IAAI,WAAW,GAAG,OAAO,CAAC,UAAU,CAAC;QAErC,IAAI,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChD,WAAW,IAAI,gBAAgB,OAAO,CAAC,MAAM,CAAC,MAAM,qBAAqB,CAAC;QAC5E,CAAC;QAED,IAAI,OAAO,CAAC,UAAU,IAAI,OAAO,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxD,WAAW,IAAI,mBAAmB,GAAG,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC1E,CAAC;QAED,IAAI,OAAO,CAAC,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC/D,WAAW;gBACT,eAAe,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QAC/D,CAAC;QAED,8BAA8B;QAC9B,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC;YAC5B,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,WAAW;SACrB,CAAC,CAAC;QAEH,4BAA4B;QAC5B,MAAM,iBAAiB,GAAsB;YAC3C,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;YACxB,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM;YAC1B,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO;YAC5B,QAAQ,EAAE,IAAI,CAAC,mBAAmB;YAClC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;SACzB,CAAC;QAEF,8BAA8B;QAC9B,IAAI,YAAY,GAAG,EAAE,CAAC;QAEtB,MAAM,MAAM,GAAG,gBAAgB,CAAC,iBAAiB,CAAC,CAAC;QACnD,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YACjC,YAAY,IAAI,KAAK,CAAC,KAAK,CAAC;YAC5B,MAAM,KAAK,CAAC;QACd,CAAC;QAED,oCAAoC;QACpC,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC;YAC5B,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,YAAY;SACtB,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,UAAU;QACR,OAAO,CAAC,GAAG,IAAI,CAAC,mBAAmB,CAAC,CAAC;IACvC,CAAC;IAED;;OAEG;IACH,YAAY;QACV,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;YAC5B,IAAI,CAAC,mBAAmB,GAAG;gBACzB;oBACE,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;iBACjC;aACF,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,mBAAmB,GAAG,EAAE,CAAC;QAChC,CAAC;IACH,CAAC;IAED;;OAEG;IACH,UAAU,CAAC,IAAqC,EAAE,OAAe;QAC/D,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC;YAC5B,IAAI;YACJ,OAAO;SACR,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,SAAS;QACP,OAAO,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;IAC5B,CAAC;CACF"}
@@ -0,0 +1,74 @@
1
+ /**
2
+ * llmjs2 - Enterprise-grade LLM abstraction layer
3
+ * Unified API for OpenAI and Ollama
4
+ */
5
+ import { CompletionRequest, CompletionResponse, CompletionChunk, CompletionOptions } from './types.js';
6
+ import { LLMError } from './providers/base.js';
7
+ /**
8
+ * Configure global settings
9
+ */
10
+ export declare function configure(options: CompletionOptions): void;
11
+ /**
12
+ * Create a completion request
13
+ *
14
+ * @example
15
+ * import { completion } from 'llmjs2';
16
+ *
17
+ * const result = await completion({
18
+ * model: 'openai/gpt-4',
19
+ * apiKey: 'sk-...',
20
+ * messages: [
21
+ * { role: 'user', content: 'Hello!' }
22
+ * ]
23
+ * });
24
+ *
25
+ * console.log(result.content);
26
+ */
27
+ export declare function completion(request: CompletionRequest): Promise<CompletionResponse>;
28
+ /**
29
+ * Stream a completion request
30
+ *
31
+ * @example
32
+ * import { streamCompletion } from 'llmjs2';
33
+ *
34
+ * const stream = streamCompletion({
35
+ * model: 'openai/gpt-4',
36
+ * apiKey: 'sk-...',
37
+ * messages: [
38
+ * { role: 'user', content: 'Write a poem' }
39
+ * ]
40
+ * });
41
+ *
42
+ * for await (const chunk of stream) {
43
+ * process.stdout.write(chunk.delta);
44
+ * }
45
+ */
46
+ export declare function streamCompletion(request: CompletionRequest): AsyncIterable<CompletionChunk>;
47
+ /**
48
+ * Validate provider connectivity and configuration
49
+ */
50
+ export declare function validateProvider(model: string, apiKey?: string, baseUrl?: string): Promise<void>;
51
+ /**
52
+ * Clear provider cache
53
+ */
54
+ export declare function clearProviderCache(): void;
55
+ /**
56
+ * Export types for consumers
57
+ */
58
+ export type { CompletionRequest, CompletionResponse, CompletionChunk, CompletionOptions, Message, MessageRole, Tool, ProviderType, ProviderConfig, ProviderError, } from './types.js';
59
+ /**
60
+ * Export error class
61
+ */
62
+ export { LLMError };
63
+ /**
64
+ * Export provider classes for advanced use cases
65
+ */
66
+ export { OpenAIProvider } from './providers/openai.js';
67
+ export { OllamaProvider } from './providers/ollama.js';
68
+ export { BaseProvider } from './providers/base.js';
69
+ /**
70
+ * Export Agent for stateful conversations
71
+ */
72
+ export { Agent } from './agent.js';
73
+ export type { AgentConfig, AgentGenerateRequest, AgentGenerateResponse } from './agent.js';
74
+ //# sourceMappingURL=index.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EACL,iBAAiB,EACjB,kBAAkB,EAClB,eAAe,EACf,iBAAiB,EAGlB,MAAM,YAAY,CAAC;AAGpB,OAAO,EAAE,QAAQ,EAAE,MAAM,qBAAqB,CAAC;AAY/C;;GAEG;AACH,wBAAgB,SAAS,CAAC,OAAO,EAAE,iBAAiB,GAAG,IAAI,CAE1D;AAgFD;;;;;;;;;;;;;;;GAeG;AACH,wBAAsB,UAAU,CAC9B,OAAO,EAAE,iBAAiB,GACzB,OAAO,CAAC,kBAAkB,CAAC,CAqB7B;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,wBAAuB,gBAAgB,CACrC,OAAO,EAAE,iBAAiB,GACzB,aAAa,CAAC,eAAe,CAAC,CAqBhC;AAED;;GAEG;AACH,wBAAsB,gBAAgB,CACpC,KAAK,EAAE,MAAM,EACb,MAAM,CAAC,EAAE,MAAM,EACf,OAAO,CAAC,EAAE,MAAM,GACf,OAAO,CAAC,IAAI,CAAC,CAuBf;AAED;;GAEG;AACH,wBAAgB,kBAAkB,IAAI,IAAI,CAEzC;AAED;;GAEG;AACH,YAAY,EACV,iBAAiB,EACjB,kBAAkB,EAClB,eAAe,EACf,iBAAiB,EACjB,OAAO,EACP,WAAW,EACX,IAAI,EACJ,YAAY,EACZ,cAAc,EACd,aAAa,GACd,MAAM,YAAY,CAAC;AAEpB;;GAEG;AACH,OAAO,EAAE,QAAQ,EAAE,CAAC;AAEpB;;GAEG;AACH,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAEnD;;GAEG;AACH,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AACnC,YAAY,EAAE,WAAW,EAAE,oBAAoB,EAAE,qBAAqB,EAAE,MAAM,YAAY,CAAC"}
package/dist/index.js ADDED
@@ -0,0 +1,191 @@
1
+ /**
2
+ * llmjs2 - Enterprise-grade LLM abstraction layer
3
+ * Unified API for OpenAI and Ollama
4
+ */
5
+ import { OpenAIProvider } from './providers/openai.js';
6
+ import { OllamaProvider } from './providers/ollama.js';
7
+ import { LLMError } from './providers/base.js';
8
+ /**
9
+ * Global configuration for completion
10
+ */
11
+ let globalOptions = {};
12
+ /**
13
+ * Provider cache
14
+ */
15
+ const providerCache = new Map();
16
+ /**
17
+ * Configure global settings
18
+ */
19
+ export function configure(options) {
20
+ globalOptions = { ...globalOptions, ...options };
21
+ }
22
+ /**
23
+ * Get provider type from model string
24
+ */
25
+ function getProviderType(model) {
26
+ if (model.startsWith('openai/')) {
27
+ return 'openai';
28
+ }
29
+ if (model.startsWith('ollama/')) {
30
+ return 'ollama';
31
+ }
32
+ // Default to openai if no prefix specified
33
+ return 'openai';
34
+ }
35
+ /**
36
+ * Create appropriate provider instance
37
+ */
38
+ function createProvider(providerType, request) {
39
+ const cacheKey = `${providerType}:${request.apiKey || 'default'}:${request.baseUrl || 'default'}`;
40
+ // Return cached provider if available
41
+ if (providerCache.has(cacheKey)) {
42
+ return providerCache.get(cacheKey);
43
+ }
44
+ let provider;
45
+ switch (providerType) {
46
+ case 'openai': {
47
+ const apiKey = request.apiKey || process.env.OPENAI_API_KEY;
48
+ if (!apiKey) {
49
+ throw new LLMError('OpenAI API key is required. Pass it as apiKey in the request or set OPENAI_API_KEY environment variable.', 'MISSING_API_KEY');
50
+ }
51
+ provider = new OpenAIProvider({
52
+ type: 'openai',
53
+ apiKey: apiKey,
54
+ baseUrl: request.baseUrl,
55
+ timeout: request.timeout ?? globalOptions.globalTimeout,
56
+ retry: request.retry ?? globalOptions.globalRetry,
57
+ });
58
+ break;
59
+ }
60
+ case 'ollama': {
61
+ provider = new OllamaProvider({
62
+ type: 'ollama',
63
+ baseUrl: request.baseUrl,
64
+ timeout: request.timeout ?? globalOptions.globalTimeout,
65
+ retry: request.retry ?? globalOptions.globalRetry,
66
+ });
67
+ break;
68
+ }
69
+ default:
70
+ throw new LLMError(`Unsupported provider: ${providerType}`, 'UNKNOWN_PROVIDER');
71
+ }
72
+ // Apply global settings
73
+ if (globalOptions.debug) {
74
+ provider.setDebug(true);
75
+ }
76
+ if (globalOptions.logger) {
77
+ provider.setLogger(globalOptions.logger);
78
+ }
79
+ // Cache the provider
80
+ providerCache.set(cacheKey, provider);
81
+ return provider;
82
+ }
83
+ /**
84
+ * Create a completion request
85
+ *
86
+ * @example
87
+ * import { completion } from 'llmjs2';
88
+ *
89
+ * const result = await completion({
90
+ * model: 'openai/gpt-4',
91
+ * apiKey: 'sk-...',
92
+ * messages: [
93
+ * { role: 'user', content: 'Hello!' }
94
+ * ]
95
+ * });
96
+ *
97
+ * console.log(result.content);
98
+ */
99
+ export async function completion(request) {
100
+ if (!request.model) {
101
+ throw new LLMError('Model is required', 'MISSING_MODEL');
102
+ }
103
+ const providerType = getProviderType(request.model);
104
+ const provider = createProvider(providerType, request);
105
+ try {
106
+ return await provider.complete(request);
107
+ }
108
+ catch (error) {
109
+ if (error instanceof LLMError) {
110
+ throw error;
111
+ }
112
+ throw new LLMError(`Completion failed: ${error instanceof Error ? error.message : String(error)}`, 'COMPLETION_FAILED', undefined, error);
113
+ }
114
+ }
115
+ /**
116
+ * Stream a completion request
117
+ *
118
+ * @example
119
+ * import { streamCompletion } from 'llmjs2';
120
+ *
121
+ * const stream = streamCompletion({
122
+ * model: 'openai/gpt-4',
123
+ * apiKey: 'sk-...',
124
+ * messages: [
125
+ * { role: 'user', content: 'Write a poem' }
126
+ * ]
127
+ * });
128
+ *
129
+ * for await (const chunk of stream) {
130
+ * process.stdout.write(chunk.delta);
131
+ * }
132
+ */
133
+ export async function* streamCompletion(request) {
134
+ if (!request.model) {
135
+ throw new LLMError('Model is required', 'MISSING_MODEL');
136
+ }
137
+ const providerType = getProviderType(request.model);
138
+ const provider = createProvider(providerType, request);
139
+ try {
140
+ yield* provider.completeStream(request);
141
+ }
142
+ catch (error) {
143
+ if (error instanceof LLMError) {
144
+ throw error;
145
+ }
146
+ throw new LLMError(`Stream failed: ${error instanceof Error ? error.message : String(error)}`, 'STREAM_FAILED', undefined, error);
147
+ }
148
+ }
149
+ /**
150
+ * Validate provider connectivity and configuration
151
+ */
152
+ export async function validateProvider(model, apiKey, baseUrl) {
153
+ const providerType = getProviderType(model);
154
+ const request = {
155
+ model,
156
+ apiKey,
157
+ baseUrl,
158
+ messages: [{ role: 'user', content: 'test' }],
159
+ };
160
+ const provider = createProvider(providerType, request);
161
+ try {
162
+ await provider.validate();
163
+ }
164
+ catch (error) {
165
+ if (error instanceof LLMError) {
166
+ throw error;
167
+ }
168
+ throw new LLMError(`Validation failed: ${error instanceof Error ? error.message : String(error)}`, 'VALIDATION_FAILED');
169
+ }
170
+ }
171
+ /**
172
+ * Clear provider cache
173
+ */
174
+ export function clearProviderCache() {
175
+ providerCache.clear();
176
+ }
177
+ /**
178
+ * Export error class
179
+ */
180
+ export { LLMError };
181
+ /**
182
+ * Export provider classes for advanced use cases
183
+ */
184
+ export { OpenAIProvider } from './providers/openai.js';
185
+ export { OllamaProvider } from './providers/ollama.js';
186
+ export { BaseProvider } from './providers/base.js';
187
+ /**
188
+ * Export Agent for stateful conversations
189
+ */
190
+ export { Agent } from './agent.js';
191
+ //# sourceMappingURL=index.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAUH,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EAAE,QAAQ,EAAE,MAAM,qBAAqB,CAAC;AAE/C;;GAEG;AACH,IAAI,aAAa,GAAsB,EAAE,CAAC;AAE1C;;GAEG;AACH,MAAM,aAAa,GAA2B,IAAI,GAAG,EAAE,CAAC;AAExD;;GAEG;AACH,MAAM,UAAU,SAAS,CAAC,OAA0B;IAClD,aAAa,GAAG,EAAE,GAAG,aAAa,EAAE,GAAG,OAAO,EAAE,CAAC;AACnD,CAAC;AAED;;GAEG;AACH,SAAS,eAAe,CAAC,KAAa;IACpC,IAAI,KAAK,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QAChC,OAAO,QAAQ,CAAC;IAClB,CAAC;IACD,IAAI,KAAK,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QAChC,OAAO,QAAQ,CAAC;IAClB,CAAC;IACD,2CAA2C;IAC3C,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED;;GAEG;AACH,SAAS,cAAc,CACrB,YAA0B,EAC1B,OAA0B;IAE1B,MAAM,QAAQ,GAAG,GAAG,YAAY,IAAI,OAAO,CAAC,MAAM,IAAI,SAAS,IAAI,OAAO,CAAC,OAAO,IAAI,SAAS,EAAE,CAAC;IAElG,sCAAsC;IACtC,IAAI,aAAa,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;QAChC,OAAO,aAAa,CAAC,GAAG,CAAC,QAAQ,CAAE,CAAC;IACtC,CAAC;IAED,IAAI,QAAmB,CAAC;IAExB,QAAQ,YAAY,EAAE,CAAC;QACrB,KAAK,QAAQ,CAAC,CAAC,CAAC;YACd,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;YAC5D,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,MAAM,IAAI,QAAQ,CAChB,0GAA0G,EAC1G,iBAAiB,CAClB,CAAC;YACJ,CAAC;YACD,QAAQ,GAAG,IAAI,cAAc,CAAC;gBAC5B,IAAI,EAAE,QAAQ;gBACd,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,aAAa,CAAC,aAAa;gBACvD,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,aAAa,CAAC,WAAW;aAClD,CAAC,CAAC;YACH,MAAM;QACR,CAAC;QAED,KAAK,QAAQ,CAAC,CAAC,CAAC;YACd,QAAQ,GAAG,IAAI,cAAc,CAAC;gBAC5B,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,aAAa,CAAC,aAAa;gBACvD,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,aAAa,CAAC,WAAW;aAClD,CAAC,CAAC;YACH,MAAM;QACR,CAAC;QAED;YACE,MAAM,IAAI,QAAQ,CAAC,yBAAyB,YAAY,EAAE,EAAE,kBAAkB,CAAC,CAAC;IACpF,CAAC;IAED,wBAAwB;IACxB,IAAI,aAAa,CAAC,KAAK,EAAE,CAAC;QACxB,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IAED,IAAI,aAAa,CAAC,MAAM,EAAE,CAAC;QACzB,QAAQ,CAAC,SAAS,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;IAC3C,CAAC;IAED,qBAAqB;IACrB,aAAa,CAAC,GAAG,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;IAEtC,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED;;;;;;;;;;;;;;;GAeG;AACH,MAAM,CAAC,KAAK,UAAU,UAAU,CAC9B,OAA0B;IAE1B,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACnB,MAAM,IAAI,QAAQ,CAAC,mBAAmB,EAAE,eAAe,CAAC,CAAC;IAC3D,CAAC;IAED,MAAM,YAAY,GAAG,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IACpD,MAAM,QAAQ,GAAG,cAAc,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;IAEvD,IAAI,CAAC;QACH,OAAO,MAAM,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;IAC1C,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,KAAK,YAAY,QAAQ,EAAE,CAAC;YAC9B,MAAM,KAAK,CAAC;QACd,CAAC;QACD,MAAM,IAAI,QAAQ,CAChB,sBAAsB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,EAC9E,mBAAmB,EACnB,SAAS,EACT,KAAK,CACN,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,MAAM,CAAC,KAAK,SAAS,CAAC,CAAC,gBAAgB,CACrC,OAA0B;IAE1B,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACnB,MAAM,IAAI,QAAQ,CAAC,mBAAmB,EAAE,eAAe,CAAC,CAAC;IAC3D,CAAC;IAED,MAAM,YAAY,GAAG,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IACpD,MAAM,QAAQ,GAAG,cAAc,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;IAEvD,IAAI,CAAC;QACH,KAAK,CAAC,CAAC,QAAQ,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;IAC1C,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,KAAK,YAAY,QAAQ,EAAE,CAAC;YAC9B,MAAM,KAAK,CAAC;QACd,CAAC;QACD,MAAM,IAAI,QAAQ,CAChB,kBAAkB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,EAC1E,eAAe,EACf,SAAS,EACT,KAAK,CACN,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,KAAa,EACb,MAAe,EACf,OAAgB;IAEhB,MAAM,YAAY,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;IAE5C,MAAM,OAAO,GAAsB;QACjC,KAAK;QACL,MAAM;QACN,OAAO;QACP,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;KAC9C,CAAC;IAEF,MAAM,QAAQ,GAAG,cAAc,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;IAEvD,IAAI,CAAC;QACH,MAAM,QAAQ,CAAC,QAAQ,EAAE,CAAC;IAC5B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,KAAK,YAAY,QAAQ,EAAE,CAAC;YAC9B,MAAM,KAAK,CAAC;QACd,CAAC;QACD,MAAM,IAAI,QAAQ,CAChB,sBAAsB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,EAC9E,mBAAmB,CACpB,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB;IAChC,aAAa,CAAC,KAAK,EAAE,CAAC;AACxB,CAAC;AAkBD;;GAEG;AACH,OAAO,EAAE,QAAQ,EAAE,CAAC;AAEpB;;GAEG;AACH,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAEnD;;GAEG;AACH,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC"}
@@ -0,0 +1,58 @@
1
+ /**
2
+ * Base provider class with common functionality
3
+ */
4
+ import { IProvider, CompletionRequest, CompletionResponse, CompletionChunk, ProviderConfig, ProviderError } from '../types.js';
5
+ /**
6
+ * Validation helper for requests
7
+ */
8
+ export declare function validateCompletionRequest(request: CompletionRequest): void;
9
+ /**
10
+ * Custom error class for provider errors
11
+ */
12
+ export declare class LLMError extends Error implements ProviderError {
13
+ code?: string;
14
+ statusCode?: number;
15
+ details?: unknown;
16
+ retryable?: boolean;
17
+ constructor(message: string, code?: string, statusCode?: number, details?: unknown, retryable?: boolean);
18
+ }
19
+ /**
20
+ * Retry helper with exponential backoff
21
+ */
22
+ export declare function withRetry<T>(fn: () => Promise<T>, options?: {
23
+ maxRetries?: number;
24
+ backoffMultiplier?: number;
25
+ initialDelayMs?: number;
26
+ }): Promise<T>;
27
+ /**
28
+ * Base provider class
29
+ */
30
+ export declare abstract class BaseProvider implements IProvider {
31
+ protected config: ProviderConfig;
32
+ protected debug: boolean;
33
+ protected logger: (level: string, message: string, data?: unknown) => void;
34
+ constructor(config: ProviderConfig);
35
+ setDebug(debug: boolean): void;
36
+ setLogger(logger: (level: string, message: string, data?: unknown) => void): void;
37
+ abstract complete(request: CompletionRequest): Promise<CompletionResponse>;
38
+ abstract completeStream(request: CompletionRequest): AsyncIterable<CompletionChunk>;
39
+ abstract validate(): Promise<void>;
40
+ abstract parseModel(model: string): string;
41
+ /**
42
+ * Get the timeout for a request
43
+ */
44
+ protected getTimeout(request?: CompletionRequest): number;
45
+ /**
46
+ * Get retry config
47
+ */
48
+ protected getRetryConfig(request?: CompletionRequest): {
49
+ maxRetries: number;
50
+ backoffMultiplier: number;
51
+ initialDelayMs: number;
52
+ };
53
+ /**
54
+ * Get headers including custom headers
55
+ */
56
+ protected getHeaders(request?: CompletionRequest): Record<string, string>;
57
+ }
58
+ //# sourceMappingURL=base.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../src/providers/base.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EACL,SAAS,EACT,iBAAiB,EACjB,kBAAkB,EAClB,eAAe,EACf,cAAc,EACd,aAAa,EACd,MAAM,aAAa,CAAC;AAErB;;GAEG;AACH,wBAAgB,yBAAyB,CAAC,OAAO,EAAE,iBAAiB,GAAG,IAAI,CAoD1E;AAED;;GAEG;AACH,qBAAa,QAAS,SAAQ,KAAM,YAAW,aAAa;IAC1D,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,SAAS,CAAC,EAAE,OAAO,CAAC;gBAGlB,OAAO,EAAE,MAAM,EACf,IAAI,CAAC,EAAE,MAAM,EACb,UAAU,CAAC,EAAE,MAAM,EACnB,OAAO,CAAC,EAAE,OAAO,EACjB,SAAS,CAAC,EAAE,OAAO;CAStB;AAED;;GAEG;AACH,wBAAsB,SAAS,CAAC,CAAC,EAC/B,EAAE,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,EACpB,OAAO,GAAE;IACP,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,cAAc,CAAC,EAAE,MAAM,CAAC;CACpB,GACL,OAAO,CAAC,CAAC,CAAC,CAiCZ;AAED;;GAEG;AACH,8BAAsB,YAAa,YAAW,SAAS;IACrD,SAAS,CAAC,MAAM,EAAE,cAAc,CAAC;IACjC,SAAS,CAAC,KAAK,EAAE,OAAO,CAAC;IACzB,SAAS,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,OAAO,KAAK,IAAI,CAAC;gBAE/D,MAAM,EAAE,cAAc;IAUlC,QAAQ,CAAC,KAAK,EAAE,OAAO,GAAG,IAAI;IAI9B,SAAS,CACP,MAAM,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,OAAO,KAAK,IAAI,GAC/D,IAAI;IAIP,QAAQ,CAAC,QAAQ,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAE1E,QAAQ,CAAC,cAAc,CACrB,OAAO,EAAE,iBAAiB,GACzB,aAAa,CAAC,eAAe,CAAC;IAEjC,QAAQ,CAAC,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IAElC,QAAQ,CAAC,UAAU,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM;IAE1C;;OAEG;IACH,SAAS,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,iBAAiB,GAAG,MAAM;IAQzD;;OAEG;IACH,SAAS,CAAC,cAAc,CAAC,OAAO,CAAC,EAAE,iBAAiB;;;;;IAcpD;;OAEG;IACH,SAAS,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,iBAAiB,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;CAM1E"}