Skip to content

Commit bfd3908

Browse files
committed
remove all but anthropic. remove unnecessary index files.
1 parent a5d0b02 commit bfd3908

File tree

7 files changed

+43
-215
lines changed

7 files changed

+43
-215
lines changed

packages/agent/src/core/llm/examples.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
11
/**
22
* Examples of using the LLM abstraction
33
*/
4+
import { generateText } from './core.js';
5+
import { createProvider } from './provider.js';
46
import { FunctionDefinition, Message } from './types.js';
57

6-
import { createProvider, generateText } from './index.js';
7-
88
/**
99
* Example of using the OpenAI provider
1010
*/

packages/agent/src/core/llm/index.ts

Lines changed: 0 additions & 20 deletions
This file was deleted.

packages/agent/src/core/llm/provider.ts

Lines changed: 38 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,9 @@
11
/**
2-
* LLM Provider interface and factory
2+
* Provider registry and factory implementations
33
*/
4-
import { GenerateOptions, LLMResponse } from './types.js';
4+
5+
import { AnthropicProvider } from './providers/anthropic.js';
6+
import { ProviderOptions, GenerateOptions, LLMResponse } from './types.js';
57

68
/**
79
* Interface for LLM providers
@@ -39,12 +41,39 @@ export interface LLMProvider {
3941
countTokens(text: string): Promise<number>;
4042
}
4143

44+
// Provider factory registry
45+
const providerFactories: Record<
46+
string,
47+
(model: string, options: ProviderOptions) => LLMProvider
48+
> = {
49+
anthropic: (model, options) => new AnthropicProvider(model, options),
50+
};
51+
4252
/**
43-
* Factory function to create a provider
44-
*
45-
* @param providerType Provider type (e.g., 'openai', 'anthropic')
46-
* @param model Model name
47-
* @param options Provider-specific options
48-
* @returns LLM provider instance
53+
* Create a provider instance
4954
*/
50-
export { createProvider, registerProvider } from './providers/index.js';
55+
export function createProvider(
56+
providerType: string,
57+
model: string,
58+
options: ProviderOptions = {},
59+
): LLMProvider {
60+
const factory = providerFactories[providerType.toLowerCase()];
61+
62+
if (!factory) {
63+
throw new Error(
64+
`Provider '${providerType}' not found. Available providers: ${Object.keys(providerFactories).join(', ')}`,
65+
);
66+
}
67+
68+
return factory(model, options);
69+
}
70+
71+
/**
72+
* Register a new provider implementation
73+
*/
74+
export function registerProvider(
75+
providerType: string,
76+
factory: (model: string, options: ProviderOptions) => LLMProvider,
77+
): void {
78+
providerFactories[providerType.toLowerCase()] = factory;
79+
}

packages/agent/src/core/llm/providers/index.ts

Lines changed: 0 additions & 46 deletions
This file was deleted.

packages/agent/src/core/llm/providers/openai.ts

Lines changed: 0 additions & 136 deletions
This file was deleted.

packages/agent/src/core/toolAgent/config.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@ import * as fs from 'fs';
22
import * as os from 'os';
33
import * as path from 'path';
44

5-
import { createProvider, LLMProvider } from '../llm/index.js';
5+
import { createProvider, LLMProvider } from '../llm/provider.js';
66
import { ToolContext } from '../types';
77

88
/**

packages/agent/src/core/toolAgent/toolAgentCore.ts

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import { zodToJsonSchema } from 'zod-to-json-schema';
22

3-
import { Message, ToolUseMessage, generateText } from '../llm/index.js';
3+
import { generateText } from '../llm/core.js';
4+
import { Message, ToolUseMessage } from '../llm/types.js';
45

56
import { DEFAULT_CONFIG } from './config.js';
67
import { logTokenUsage } from './tokenTracking.js';

0 commit comments

Comments
 (0)