mirror of
https://github.com/CherryHQ/cherry-studio.git
synced 2025-12-19 14:41:24 +08:00
a5e7aa1342
- Added provider API host formatting utilities to handle differences between Cherry Studio and AI SDK. - Introduced functions for formatting provider API hosts, including support for Azure OpenAI and Vertex AI. - Created a simple API key rotator for managing API key rotation. - Developed shared provider initialization and mapping utilities for resolving provider IDs. - Implemented AI SDK configuration utilities for converting Cherry Studio providers to AI SDK configurations. - Added support for various providers including OpenRouter, Google Vertex AI, and Amazon Bedrock. - Enhanced error handling and logging in the unified messages service for better debugging. - Introduced functions for streaming and generating unified messages using AI SDK.
108 lines
3.1 KiB
TypeScript
108 lines
3.1 KiB
TypeScript
import { type ProviderConfig, registerMultipleProviderConfigs } from '@cherrystudio/ai-core/provider'
|
|
|
|
type ProviderInitializationLogger = {
|
|
warn?: (message: string) => void
|
|
error?: (message: string, error: Error) => void
|
|
}
|
|
|
|
export const SHARED_PROVIDER_CONFIGS: ProviderConfig[] = [
|
|
{
|
|
id: 'openrouter',
|
|
name: 'OpenRouter',
|
|
import: () => import('@openrouter/ai-sdk-provider'),
|
|
creatorFunctionName: 'createOpenRouter',
|
|
supportsImageGeneration: true,
|
|
aliases: ['openrouter']
|
|
},
|
|
{
|
|
id: 'google-vertex',
|
|
name: 'Google Vertex AI',
|
|
import: () => import('@ai-sdk/google-vertex/edge'),
|
|
creatorFunctionName: 'createVertex',
|
|
supportsImageGeneration: true,
|
|
aliases: ['vertexai']
|
|
},
|
|
{
|
|
id: 'google-vertex-anthropic',
|
|
name: 'Google Vertex AI Anthropic',
|
|
import: () => import('@ai-sdk/google-vertex/anthropic/edge'),
|
|
creatorFunctionName: 'createVertexAnthropic',
|
|
supportsImageGeneration: true,
|
|
aliases: ['vertexai-anthropic']
|
|
},
|
|
{
|
|
id: 'azure-anthropic',
|
|
name: 'Azure AI Anthropic',
|
|
import: () => import('@ai-sdk/anthropic'),
|
|
creatorFunctionName: 'createAnthropic',
|
|
supportsImageGeneration: false,
|
|
aliases: ['azure-anthropic']
|
|
},
|
|
{
|
|
id: 'github-copilot-openai-compatible',
|
|
name: 'GitHub Copilot OpenAI Compatible',
|
|
import: () => import('@opeoginni/github-copilot-openai-compatible'),
|
|
creatorFunctionName: 'createGitHubCopilotOpenAICompatible',
|
|
supportsImageGeneration: false,
|
|
aliases: ['copilot', 'github-copilot']
|
|
},
|
|
{
|
|
id: 'bedrock',
|
|
name: 'Amazon Bedrock',
|
|
import: () => import('@ai-sdk/amazon-bedrock'),
|
|
creatorFunctionName: 'createAmazonBedrock',
|
|
supportsImageGeneration: true,
|
|
aliases: ['aws-bedrock']
|
|
},
|
|
{
|
|
id: 'perplexity',
|
|
name: 'Perplexity',
|
|
import: () => import('@ai-sdk/perplexity'),
|
|
creatorFunctionName: 'createPerplexity',
|
|
supportsImageGeneration: false,
|
|
aliases: ['perplexity']
|
|
},
|
|
{
|
|
id: 'mistral',
|
|
name: 'Mistral',
|
|
import: () => import('@ai-sdk/mistral'),
|
|
creatorFunctionName: 'createMistral',
|
|
supportsImageGeneration: false,
|
|
aliases: ['mistral']
|
|
},
|
|
{
|
|
id: 'huggingface',
|
|
name: 'HuggingFace',
|
|
import: () => import('@ai-sdk/huggingface'),
|
|
creatorFunctionName: 'createHuggingFace',
|
|
supportsImageGeneration: true,
|
|
aliases: ['hf', 'hugging-face']
|
|
},
|
|
{
|
|
id: 'ai-gateway',
|
|
name: 'AI Gateway',
|
|
import: () => import('@ai-sdk/gateway'),
|
|
creatorFunctionName: 'createGateway',
|
|
supportsImageGeneration: true,
|
|
aliases: ['gateway']
|
|
},
|
|
{
|
|
id: 'cerebras',
|
|
name: 'Cerebras',
|
|
import: () => import('@ai-sdk/cerebras'),
|
|
creatorFunctionName: 'createCerebras',
|
|
supportsImageGeneration: false
|
|
}
|
|
] as const
|
|
|
|
export function initializeSharedProviders(logger?: ProviderInitializationLogger): void {
|
|
try {
|
|
const successCount = registerMultipleProviderConfigs(SHARED_PROVIDER_CONFIGS)
|
|
if (successCount < SHARED_PROVIDER_CONFIGS.length) {
|
|
logger?.warn?.('Some providers failed to register. Check previous error logs.')
|
|
}
|
|
} catch (error) {
|
|
logger?.error?.('Failed to initialize shared providers', error as Error)
|
|
}
|
|
}
|