Function fetchOllamaModels

core/llm/fetchModels.ts:64–122

()

}

async function fetchOllamaModels(): Promise<FetchedModel[]> {
  try {
    const response = await fetch("https://ollama.com/library");
    if (!response.ok) {
      throw new Error(`Failed to fetch Ollama library: ${response.status}`);
    }

    const html = await response.text();
    const models: FetchedModel[] = [];
    const items = html.split("x-test-model class=");
    const seen = new Set<string>();

    for (let i = 1; i < items.length; i++) {
      const item = items[i];
      const nameMatch = item.match(/href="\/library\/([^"]+)"/);
      if (!nameMatch) continue;
      const name = nameMatch[1];
      if (seen.has(name)) continue;

      const capabilities: string[] = [];
      const capRegex = /x-test-capability[^>]*>([^<]+)</g;
      let capMatch;
      while ((capMatch = capRegex.exec(item)) !== null) {
        capabilities.push(capMatch[1].trim().toLowerCase());
      }
      if (
        capabilities.some((cap) => OLLAMA_EXCLUDED_CAPABILITIES.includes(cap))
      ) {
        continue;
      }

      const sizes: string[] = [];
      const sizeRegex = /x-test-size[^>]*>([^<]+)</g;
      let sizeMatch;
      while ((sizeMatch = sizeRegex.exec(item)) !== null) {
        sizes.push(sizeMatch[1].trim());
      }

      const descMatch = item.match(/<p class="max-w-lg[^"]*">([^<]+)</);
      const sizeLabel = sizes.length > 0 ? ` (${sizes.join(", ")})` : "";
      const description = descMatch
        ? descMatch[1].trim()
        : `Ollama model: ${name}${sizeLabel}`;

      seen.add(name);
      models.push({
        name,
        description,
        icon: getOllamaIcon(name),
        supportsTools: capabilities.includes("tools"),
      });
    }

    return models;
  } catch (error) {
    console.error("Error fetching Ollama library models:", error);
    return [];
  }