Create glhf.chat.ts

add glhf.chat service to providers
This commit is contained in:
Bruno Braga 2025-01-01 20:54:53 -03:00 committed by GitHub
parent 608d9338a8
commit b6ad8971ab
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -0,0 +1,119 @@
import { BaseProvider, getOpenAILikeModel } 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';
export default class GLHFProvider extends BaseProvider {
name = 'glhf.chat';
getApiKeyLink = 'https://glhf.chat/users/settings/api';
config = {
baseUrlKey: 'GLHF_API_BASE_URL',
apiTokenKey: 'GLHF_API_KEY',
};
defaultBaseUrl = 'https://glhf.chat/api/openai/v1';
get staticModels(): ModelInfo[] {
return [
{
name: 'hf:mistralai/Mistral-7B-Instruct-v0.3',
label: 'Mistral-7B-Instruct',
provider: this.name,
maxTokenAllowed: 8000,
},
{
name: 'hf:qwen/Qwen2.5-Coder-32B-Instruct',
label: 'Qwen2.5-Coder-32B',
provider: this.name,
maxTokenAllowed: 8000,
},
{
name: 'hf:deepseek-ai/DeepSeek-Coder-V2-Lite-Instruct',
label: 'DeepSeek-Coder-V2-Lite',
provider: this.name,
maxTokenAllowed: 8000,
},
{
name: 'hf:nvidia/Llama-3.1-Nemotron-70B-Instruct-HF',
label: 'Llama-3.1-Nemotron-70B',
provider: this.name,
maxTokenAllowed: 8000,
},
{
name: 'hf:google/codegemma-7b-it',
label: 'CodeGemma-7B',
provider: this.name,
maxTokenAllowed: 8000,
}
];
}
async getDynamicModels(
apiKeys?: Record<string, string>,
settings?: IProviderSetting,
serverEnv: Record<string, string> = {},
): Promise<ModelInfo[]> {
// Retornamos apenas os modelos estáticos, evitando duplicação
return this.staticModels;
}
getModelInstance(options: {
model?: string;
serverEnv: Record<string, string>;
apiKeys?: Record<string, string>;
providerSettings?: Record<string, IProviderSetting>;
}): LanguageModelV1 {
const { model, serverEnv, apiKeys, providerSettings } = options;
const { baseUrl, apiKey } = this.getProviderBaseUrlAndKey({
apiKeys,
providerSettings: providerSettings?.[this.name],
serverEnv,
defaultBaseUrlKey: this.config.baseUrlKey,
defaultApiTokenKey: this.config.apiTokenKey,
});
const effectiveBaseUrl = baseUrl || this.defaultBaseUrl;
if (!apiKey) {
throw new Error(`Missing API key for ${this.name} provider`);
}
// Usa o primeiro modelo como padrão se nenhum for especificado
const modelToUse = model || this.staticModels[0].name;
return getOpenAILikeModel(effectiveBaseUrl, apiKey, modelToUse);
}
async testApiConnection(): Promise<void> {
const { baseUrl, apiKey } = this.getProviderBaseUrlAndKey({
serverEnv: process.env as any,
defaultBaseUrlKey: this.config.baseUrlKey,
defaultApiTokenKey: this.config.apiTokenKey,
});
const effectiveBaseUrl = baseUrl || this.defaultBaseUrl;
if (!apiKey) {
throw new Error('Missing API key for GLHF provider during connection test.');
}
try {
const response = await fetch(`${effectiveBaseUrl}/models`, {
headers: {
'Authorization': `Bearer ${apiKey}`,
'Content-Type': 'application/json'
},
});
if (!response.ok) {
throw new Error(`GLHF API connection failed: ${response.status} ${response.statusText}`);
}
console.log('GLHF API connection successful.');
} catch (error) {
console.error('Error during GLHF API connection test:', error);
throw error;
}
}
}