fix: ollama version string issue

#1800
This commit is contained in:
Timothy J. Baek 2024-04-27 19:19:47 -04:00
parent 85e94cc259
commit 01c4647dfc
1 changed files with 4 additions and 1 deletions

View File

@ -16,6 +16,7 @@ from fastapi.concurrency import run_in_threadpool
from pydantic import BaseModel, ConfigDict
import os
import re
import copy
import random
import requests
@ -216,7 +217,9 @@ async def get_ollama_versions(url_idx: Optional[int] = None):
if len(responses) > 0:
lowest_version = min(
responses,
key=lambda x: tuple(map(int, x["version"].split("-")[0].split("."))),
key=lambda x: tuple(
map(int, re.sub(r"^v|-.*", "", x["version"]).split("."))
),
)
return {"version": lowest_version["version"]}