1 line
5.8 KiB
JSON
1 line
5.8 KiB
JSON
{"repo": "astitvapg/open-source-ai-artifacts", "files": {"lib/models.ts": {"69eb353b98b6548238f67d0f44eba7af9f59cd85": {"t": 1785745569.2950037, "n": 2717, "h": "803b86c948425bc2", "c": "import { createAnthropic } from '@ai-sdk/anthropic'\nimport { createGoogleGenerativeAI } from '@ai-sdk/google'\nimport { createVertex } from '@ai-sdk/google-vertex'\nimport { createMistral } from '@ai-sdk/mistral'\nimport { createOpenAI } from '@ai-sdk/openai'\nimport { createOllama } from 'ollama-ai-provider'\n\nexport type LLMModel = {\n id: string\n name: string\n provider: string\n providerId: string\n}\n\nexport type LLMModelConfig = {\n model?: string\n apiKey?: string\n baseURL?: string\n temperature?: number\n topP?: number\n topK?: number\n frequencyPenalty?: number\n presencePenalty?: number\n maxTokens?: number\n}\n\nexport function getModelClient(model: LLMModel, config: LLMModelConfig) {\n const { id: modelNameString, providerId } = model\n const { apiKey, baseURL } = config\n\n const providerConfigs = {\n anthropic: () => createAnthropic({ apiKey, baseURL })(modelNameString),\n openai: () => createOpenAI({ apiKey, baseURL })(modelNameString),\n google: () =>\n createGoogleGenerativeAI({ apiKey, baseURL })(modelNameString),\n mistral: () => createMistral({ apiKey, baseURL })(modelNameString),\n groq: () =>\n createOpenAI({\n apiKey: apiKey || process.env.GROQ_API_KEY,\n baseURL: baseURL || 'https://api.groq.com/openai/v1',\n })(modelNameString),\n togetherai: () =>\n createOpenAI({\n apiKey: apiKey || process.env.TOGETHER_API_KEY,\n baseURL: baseURL || 'https://api.together.xyz/v1',\n })(modelNameString),\n ollama: () => createOllama({ baseURL })(modelNameString),\n fireworks: () =>\n createOpenAI({\n apiKey: apiKey || process.env.FIREWORKS_API_KEY,\n baseURL: baseURL || 'https://api.fireworks.ai/inference/v1',\n })(modelNameString),\n vertex: () =>\n createVertex({\n googleAuthOptions: {\n credentials: JSON.parse(\n process.env.GOOGLE_VERTEX_CREDENTIALS || '{}',\n ),\n },\n })(modelNameString),\n xai: () =>\n createOpenAI({\n apiKey: apiKey || process.env.XAI_API_KEY,\n baseURL: baseURL || 'https://api.x.ai/v1',\n })(modelNameString),\n deepseek: () =>\n createOpenAI({\n apiKey: apiKey || process.env.DEEPSEEK_API_KEY,\n baseURL: baseURL || 'https://api.deepseek.com/v1',\n })(modelNameString),\n }\n\n const createClient =\n providerConfigs[providerId as keyof typeof providerConfigs]\n\n if (!createClient) {\n throw new Error(`Unsupported provider: ${providerId}`)\n }\n\n return createClient()\n}\n\nexport function getDefaultMode(model: LLMModel) {\n const { id: modelNameString, providerId } = model\n\n // monkey patch fireworks\n if (providerId === 'fireworks') {\n return 'json'\n }\n\n return 'auto'\n}\n"}, "0da951cea98d3e5aaedde4453aa47a2802c379ed": {"t": 1785754333.7867703, "n": 2717, "h": "803b86c948425bc2", "c": "import { createAnthropic } from '@ai-sdk/anthropic'\nimport { createGoogleGenerativeAI } from '@ai-sdk/google'\nimport { createVertex } from '@ai-sdk/google-vertex'\nimport { createMistral } from '@ai-sdk/mistral'\nimport { createOpenAI } from '@ai-sdk/openai'\nimport { createOllama } from 'ollama-ai-provider'\n\nexport type LLMModel = {\n id: string\n name: string\n provider: string\n providerId: string\n}\n\nexport type LLMModelConfig = {\n model?: string\n apiKey?: string\n baseURL?: string\n temperature?: number\n topP?: number\n topK?: number\n frequencyPenalty?: number\n presencePenalty?: number\n maxTokens?: number\n}\n\nexport function getModelClient(model: LLMModel, config: LLMModelConfig) {\n const { id: modelNameString, providerId } = model\n const { apiKey, baseURL } = config\n\n const providerConfigs = {\n anthropic: () => createAnthropic({ apiKey, baseURL })(modelNameString),\n openai: () => createOpenAI({ apiKey, baseURL })(modelNameString),\n google: () =>\n createGoogleGenerativeAI({ apiKey, baseURL })(modelNameString),\n mistral: () => createMistral({ apiKey, baseURL })(modelNameString),\n groq: () =>\n createOpenAI({\n apiKey: apiKey || process.env.GROQ_API_KEY,\n baseURL: baseURL || 'https://api.groq.com/openai/v1',\n })(modelNameString),\n togetherai: () =>\n createOpenAI({\n apiKey: apiKey || process.env.TOGETHER_API_KEY,\n baseURL: baseURL || 'https://api.together.xyz/v1',\n })(modelNameString),\n ollama: () => createOllama({ baseURL })(modelNameString),\n fireworks: () =>\n createOpenAI({\n apiKey: apiKey || process.env.FIREWORKS_API_KEY,\n baseURL: baseURL || 'https://api.fireworks.ai/inference/v1',\n })(modelNameString),\n vertex: () =>\n createVertex({\n googleAuthOptions: {\n credentials: JSON.parse(\n process.env.GOOGLE_VERTEX_CREDENTIALS || '{}',\n ),\n },\n })(modelNameString),\n xai: () =>\n createOpenAI({\n apiKey: apiKey || process.env.XAI_API_KEY,\n baseURL: baseURL || 'https://api.x.ai/v1',\n })(modelNameString),\n deepseek: () =>\n createOpenAI({\n apiKey: apiKey || process.env.DEEPSEEK_API_KEY,\n baseURL: baseURL || 'https://api.deepseek.com/v1',\n })(modelNameString),\n }\n\n const createClient =\n providerConfigs[providerId as keyof typeof providerConfigs]\n\n if (!createClient) {\n throw new Error(`Unsupported provider: ${providerId}`)\n }\n\n return createClient()\n}\n\nexport function getDefaultMode(model: LLMModel) {\n const { id: modelNameString, providerId } = model\n\n // monkey patch fireworks\n if (providerId === 'fireworks') {\n return 'json'\n }\n\n return 'auto'\n}\n"}}}} |