Merge pull request #1340 from joequant/main

allow version to work with -rc tag

Merge pull request #1340 from joequant/main
allow version to work with -rc tag
782cce7c · Timothy Jaeryang Baek · GitHub · fafb8cd2 · 395ca821 · 782cce7c
Unverified Commit 782cce7c authored Apr 06, 2024 by Timothy Jaeryang Baek Committed by GitHub Apr 06, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

backend/apps/ollama/main.py backend/apps/ollama/main.py +2 -1

No files found.
--- a/backend/apps/ollama/main.py
+++ b/backend/apps/ollama/main.py
@@ -215,7 +215,8 @@ 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(".")))
+                responses,
+                key=lambda x: tuple(map(int, x["version"].split("-")[0].split("."))),
            )
            return {"version": lowest_version["version"]}