infio-copilot/src/constants.ts
2025-04-29 11:25:44 +08:00

68 lines
2.0 KiB
TypeScript

import { LLMModel } from './types/llm/model'
// import { ApiProvider } from './utils/api'
export const CHAT_VIEW_TYPE = 'infio-chat-view'
export const APPLY_VIEW_TYPE = 'infio-apply-view'
export const PREVIEW_VIEW_TYPE = 'infio-preview-view'
export const DEFAULT_MODELS: LLMModel[] = []
// export const PROVIDERS: ApiProvider[] = [
// 'Infio',
// 'OpenRouter',
// 'SiliconFlow',
// 'Anthropic',
// 'Deepseek',
// 'OpenAI',
// 'Google',
// 'Groq',
// 'Ollama',
// 'OpenAICompatible',
// ]
export const SUPPORT_EMBEDDING_SIMENTION: number[] = [
384,
512,
768,
1024,
1536
]
export const OPENAI_BASE_URL = 'https://api.openai.com/v1'
export const DEEPSEEK_BASE_URL = 'https://api.deepseek.com'
export const OPENROUTER_BASE_URL = 'https://openrouter.ai/api/v1'
export const GROK_BASE_URL = 'https://api.x.ai/v1'
export const SILICONFLOW_BASE_URL = 'https://api.siliconflow.cn/v1'
export const ALIBABA_QWEN_BASE_URL = 'https://dashscope.aliyuncs.com/compatible-mode/v1'
export const INFIO_BASE_URL = 'https://api.infio.com/api/raw_message'
export const JINA_BASE_URL = 'https://r.jina.ai'
export const SERPER_BASE_URL = 'https://serpapi.com/search'
// Pricing in dollars per million tokens
type ModelPricing = {
input: number
output: number
}
export const OPENAI_PRICES: Record<string, ModelPricing> = {
'gpt-4o': { input: 2.5, output: 10 },
'gpt-4o-mini': { input: 0.15, output: 0.6 },
'deepseek-chat': { input: 0.16, output: 0.32 },
}
export const ANTHROPIC_PRICES: Record<string, ModelPricing> = {
'claude-3-5-sonnet-latest': { input: 3, output: 15 },
'claude-3-5-haiku-latest': { input: 1, output: 5 },
}
// Gemini is currently free for low rate limits
export const GEMINI_PRICES: Record<string, ModelPricing> = {
'gemini-1.5-pro': { input: 0, output: 0 },
'gemini-1.5-flash': { input: 0, output: 0 },
}
export const GROQ_PRICES: Record<string, ModelPricing> = {
'llama-3.1-70b-versatile': { input: 0.59, output: 0.79 },
'llama-3.1-8b-instant': { input: 0.05, output: 0.08 },
}
export const PGLITE_DB_PATH = '.infio_pglite_db.tar.gz'