mirror of
https://github.com/stackblitz/bolt.new
synced 2025-06-26 18:17:50 +00:00
Create glhf.chat.ts
add glhf.chat service to providers
This commit is contained in:
parent
608d9338a8
commit
b6ad8971ab
119
app/lib/modules/llm/providers/glhf.chat.ts
Normal file
119
app/lib/modules/llm/providers/glhf.chat.ts
Normal 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;
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user