tdpeuter/open-webui
tdpeuter/open-webui
1
Fork 0
forked from open-webui/open-webui
Code Issues Pull requests 1 Releases 2 Packages Activity Actions

fix: ollama version string issue

Browse source 
#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
Show stats Download patch file Download diff file Expand all files Collapse all files

5
backend/apps/ollama/main.py
View file

@ -16,6 +16,7 @@ from fastapi.concurrency import run_in_threadpool
from pydantic import BaseModel, ConfigDict from pydantic import BaseModel, ConfigDict
import os import os
import re
import copy import copy
import random import random
import requests import requests
@ -216,7 +217,9 @@ async def get_ollama_versions(url_idx: Optional[int] = None):
if len(responses) > 0: if len(responses) > 0:
lowest_version = min( lowest_version = min(
responses, 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"]} return {"version": lowest_version["version"]}