UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-16 13:40:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

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"]}