hj/pleroma
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

QdrantSearch: Add health checks.

Browse source
This commit is contained in:
Lain Soykaf 2024-05-27 14:21:55 +04:00
commit f4c04e6b2d
3 changed files with 23 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

20
test/pleroma/search/qdrant_search_test.exs
View file

@ -16,15 +16,29 @@ defmodule Pleroma.Search.QdrantSearchTest do
describe "Qdrant search" do
test "returns the correct healthcheck endpoints" do
# No openai healthcheck URL
Config
|> expect(:get, 1, fn
|> expect(:get, 2, fn
[Pleroma.Search.QdrantSearch, key], nil ->
%{qdrant_url: "https://qdrant.url"}[key]
end)
health_endpoints = QdrantSearch.healthcheck_endpoints()
[health_endpoint] = QdrantSearch.healthcheck_endpoints()
assert "https://qdrant.url/healthz" in health_endpoints
assert "https://qdrant.url/healthz" == health_endpoint
# Set openai healthcheck URL
Config
|> expect(:get, 2, fn
[Pleroma.Search.QdrantSearch, key], nil ->
%{qdrant_url: "https://qdrant.url", openai_healthcheck_url: "https://openai.url/health"}[
key
]
end)
[_, health_endpoint] = QdrantSearch.healthcheck_endpoints()
assert "https://openai.url/health" == health_endpoint
end
test "searches for a term by encoding it and sending it to qdrant" do