get version

monofuel · Jan 13, 2024 · c7815f7 · c7815f7
1 parent 50722e8
commit c7815f7
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/llama_leap.nim b/src/llama_leap.nim
@@ -350,3 +350,11 @@ proc generateEmbeddings*(
 # TODO: POST /api/copy
 # TODO: DELETE /api/delete
 # TODO: POST /api/push
+
+proc getVersion*(api: OllamaAPI): string =
+  let url = api.baseUrl / "version"
+  let resp = api.curlPool.get(url, timeout = api.curlTimeout)
+  if resp.code != 200:
+    raise newException(CatchableError, &"ollama version failed: {resp.code} {resp.body}")
+  let json = fromJson(resp.body)
+  result = json["version"].getStr
diff --git a/tests/test_llama_leap.nim b/tests/test_llama_leap.nim
@@ -15,6 +15,10 @@ suite "llama_leap":
   teardown:
     ollama.close()
 
+  suite "version":
+    test "version":
+      echo "> " & ollama.getVersion()
+
   suite "pull":
     test "pull model":
       ollama.pullModel(TestModel)