Commit 93fda9f
Changed files (1)
lib
net
llm
lib/net/llm/ollama.rb
@@ -35,21 +35,15 @@ module Net
end
def embeddings(input)
- url = build_url("/api/embed")
- payload = { model: model, input: input }
- post_request(url, payload)
+ post_request(build_url("/api/embed"), { model: model, input: input })
end
def tags
- url = build_url("/api/tags")
- response = http.get(url)
- handle_response(response)
+ get_request(build_url("/api/tags"))
end
def show(name)
- url = build_url("/api/show")
- payload = { name: name }
- post_request(url, payload)
+ post_request(build_url("/api/show"), { name: name })
end
private
@@ -59,16 +53,22 @@ module Net
"#{base}#{path}"
end
+ def get_request(url)
+ handle_response(http.get(url))
+ end
+
def post_request(url, payload)
- response = http.post(url, body: payload)
- handle_response(response)
+ handle_response(http.post(url, body: payload))
end
def handle_response(response)
if response.is_a?(Net::HTTPSuccess)
JSON.parse(response.body)
else
- { "code" => response.code, "body" => response.body }
+ {
+ "code" => response.code,
+ "body" => response.body
+ }
end
end