38c13494c2
- Updated constants in app/lib/.server/llm/constants.ts - Modified stream-text functionality in app/lib/.server/llm/stream-text.ts - Updated Anthropic provider in app/lib/modules/llm/providers/anthropic.ts - Modified GitHub provider in app/lib/modules/llm/providers/github.ts - Updated Google provider in app/lib/modules/llm/providers/google.ts - Modified OpenAI provider in app/lib/modules/llm/providers/openai.ts - Updated LLM types in app/lib/modules/llm/types.ts - Modified API route in app/routes/api.llmcall.ts
72 lines
2.2 KiB
TypeScript
72 lines
2.2 KiB
TypeScript
import { BaseProvider } from '~/lib/modules/llm/base-provider';
|
|
import type { ModelInfo } from '~/lib/modules/llm/types';
|
|
import type { IProviderSetting } from '~/types/model';
|
|
import type { LanguageModelV1 } from 'ai';
|
|
import { createOpenAI } from '@ai-sdk/openai';
|
|
|
|
export default class GithubProvider extends BaseProvider {
|
|
name = 'Github';
|
|
getApiKeyLink = 'https://github.com/settings/personal-access-tokens';
|
|
|
|
config = {
|
|
apiTokenKey: 'GITHUB_API_KEY',
|
|
};
|
|
|
|
// find more in https://github.com/marketplace?type=models
|
|
staticModels: ModelInfo[] = [
|
|
{ name: 'gpt-4o', label: 'GPT-4o', provider: 'Github', maxTokenAllowed: 128000, maxCompletionTokens: 16384 },
|
|
{ name: 'o1', label: 'o1-preview', provider: 'Github', maxTokenAllowed: 100000, maxCompletionTokens: 16384 },
|
|
{ name: 'o1-mini', label: 'o1-mini', provider: 'Github', maxTokenAllowed: 65536, maxCompletionTokens: 8192 },
|
|
{
|
|
name: 'gpt-4o-mini',
|
|
label: 'GPT-4o Mini',
|
|
provider: 'Github',
|
|
maxTokenAllowed: 128000,
|
|
maxCompletionTokens: 16384,
|
|
},
|
|
{
|
|
name: 'gpt-4-turbo',
|
|
label: 'GPT-4 Turbo',
|
|
provider: 'Github',
|
|
maxTokenAllowed: 128000,
|
|
maxCompletionTokens: 8192,
|
|
},
|
|
{ name: 'gpt-4', label: 'GPT-4', provider: 'Github', maxTokenAllowed: 8192, maxCompletionTokens: 8192 },
|
|
{
|
|
name: 'gpt-3.5-turbo',
|
|
label: 'GPT-3.5 Turbo',
|
|
provider: 'Github',
|
|
maxTokenAllowed: 16385,
|
|
maxCompletionTokens: 4096,
|
|
},
|
|
];
|
|
|
|
getModelInstance(options: {
|
|
model: string;
|
|
serverEnv: Env;
|
|
apiKeys?: Record<string, string>;
|
|
providerSettings?: Record<string, IProviderSetting>;
|
|
}): LanguageModelV1 {
|
|
const { model, serverEnv, apiKeys, providerSettings } = options;
|
|
|
|
const { apiKey } = this.getProviderBaseUrlAndKey({
|
|
apiKeys,
|
|
providerSettings: providerSettings?.[this.name],
|
|
serverEnv: serverEnv as any,
|
|
defaultBaseUrlKey: '',
|
|
defaultApiTokenKey: 'GITHUB_API_KEY',
|
|
});
|
|
|
|
if (!apiKey) {
|
|
throw new Error(`Missing API key for ${this.name} provider`);
|
|
}
|
|
|
|
const openai = createOpenAI({
|
|
baseURL: 'https://models.inference.ai.azure.com',
|
|
apiKey,
|
|
});
|
|
|
|
return openai(model);
|
|
}
|
|
}
|