iod/page-assist
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: Improve ollama2 model fetching

Browse Source 
This commit introduces a more efficient approach to fetching ollama2 models, ensuring proper filtering and handling of providers. This enhances the robustness and reliability of the model loading process, streamlining the overall user experience.
This commit is contained in:
n4ze3m 2024-11-10 17:30:33 +05:30
parent 2409ebc75d
commit 55f3838b6d
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/db/models.ts
View File

@ -407,6 +407,10 @@ export const ollamaFormatAllCustomModels = async (
(provider) => provider.provider === "llamafile" (provider) => provider.provider === "llamafile"
) )
const ollamaProviders = allProviders.filter(
(provider) => provider.provider === "ollama2"
)
const lmModelsPromises = lmstudioProviders.map((provider) => const lmModelsPromises = lmstudioProviders.map((provider) =>
dynamicFetchLMStudio({ dynamicFetchLMStudio({
baseUrl: provider.baseUrl, baseUrl: provider.baseUrl,
@ -421,12 +425,11 @@ export const ollamaFormatAllCustomModels = async (
}) })
) )
const ollamaModelsPromises = allProviders.map((provider) => ( const ollamaModelsPromises = ollamaProviders.map((provider) =>
dynamicFetchOllama2({ dynamicFetchOllama2({
baseUrl: provider.baseUrl, baseUrl: provider.baseUrl,
providerId: provider.id providerId: provider.id
}) }))
))
const lmModelsFetch = await Promise.all(lmModelsPromises) const lmModelsFetch = await Promise.all(lmModelsPromises)