mirror of
https://github.com/stackblitz-labs/bolt.diy
synced 2025-06-26 18:26:38 +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