@@ -15,43 +15,40 @@ import { VsCodeLmHandler } from "./providers/vscode-lm"
1515import { ApiStream } from "./transform/stream"
1616
1717export interface SingleCompletionHandler {
18- completePrompt ( prompt : string ) : Promise < string >
18+ completePrompt ( prompt : string ) : Promise < string >
1919}
2020
2121export interface ApiHandler {
22- createMessage ( systemPrompt : string , messages : Anthropic . Messages . MessageParam [ ] ) : ApiStream
23- getModel ( ) : { id : string ; info : ModelInfo }
22+ createMessage ( systemPrompt : string , messages : Anthropic . Messages . MessageParam [ ] ) : ApiStream
23+ getModel ( ) : { id : string ; info : ModelInfo }
2424}
2525
2626export function buildApiHandler ( configuration : ApiConfiguration ) : ApiHandler {
27- const { apiProvider, ...options } = configuration
28-
29- switch ( apiProvider ) {
30- case "anthropic" :
31- return new AnthropicHandler ( options )
32- case "glama" :
33- return new GlamaHandler ( options )
34- case "openrouter" :
35- return new OpenRouterHandler ( options )
36- case "bedrock" :
37- return new AwsBedrockHandler ( options )
38- case "vertex" :
39- return new VertexHandler ( options )
40- case "openai" :
41- return new OpenAiHandler ( options )
42- case "ollama" :
43- return new OllamaHandler ( options )
44- case "lmstudio" :
45- return new LmStudioHandler ( options )
46- case "gemini" :
47- return new GeminiHandler ( options )
48- case "openai-native" :
49- return new OpenAiNativeHandler ( options )
50- case "deepseek" :
51- return new DeepSeekHandler ( options )
52- case "vscode-lm" :
53- return new VsCodeLmHandler ( options )
54- default :
55- return new AnthropicHandler ( options )
56- }
27+ const { apiProvider, ...options } = configuration
28+ switch ( apiProvider ) {
29+ case "anthropic" :
30+ return new AnthropicHandler ( options )
31+ case "glama" :
32+ return new GlamaHandler ( options )
33+ case "openrouter" :
34+ return new OpenRouterHandler ( options )
35+ case "bedrock" :
36+ return new AwsBedrockHandler ( options )
37+ case "vertex" :
38+ return new VertexHandler ( options )
39+ case "openai" :
40+ return new OpenAiHandler ( options )
41+ case "ollama" :
42+ return new OllamaHandler ( options )
43+ case "lmstudio" :
44+ return new LmStudioHandler ( options )
45+ case "gemini" :
46+ return new GeminiHandler ( options )
47+ case "openai-native" :
48+ return new OpenAiNativeHandler ( options )
49+ case "deepseek" :
50+ return new DeepSeekHandler ( options )
51+ default :
52+ return new AnthropicHandler ( options )
53+ }
5754}
0 commit comments