This commit is contained in:
Lain Soykaf 2024-05-18 15:02:22 +04:00
parent 7923ede8ba
commit 3345ddd2d4
2 changed files with 8 additions and 7 deletions

View File

@ -1,16 +1,17 @@
defmodule Pleroma.Search.QdrantSearch do defmodule Pleroma.Search.QdrantSearch do
@behaviour Pleroma.Search.SearchBackend @behaviour Pleroma.Search.SearchBackend
import Ecto.Query import Ecto.Query
alias Pleroma.Activity
alias __MODULE__.QdrantClient alias Pleroma.Activity
alias __MODULE__.OllamaClient
alias Pleroma.Config.Getting, as: Config alias Pleroma.Config.Getting, as: Config
alias __MODULE__.OllamaClient
alias __MODULE__.QdrantClient
import Pleroma.Search.Meilisearch, only: [object_to_search_data: 1] import Pleroma.Search.Meilisearch, only: [object_to_search_data: 1]
@impl true @impl true
def create_index() do def create_index do
payload = Config.get([Pleroma.Search.QdrantSearch, :qdrant_index_configuration]) payload = Config.get([Pleroma.Search.QdrantSearch, :qdrant_index_configuration])
with {:ok, %{status: 200}} <- QdrantClient.put("/collections/posts", payload) do with {:ok, %{status: 200}} <- QdrantClient.put("/collections/posts", payload) do
@ -21,7 +22,7 @@ defmodule Pleroma.Search.QdrantSearch do
end end
@impl true @impl true
def drop_index() do def drop_index do
with {:ok, %{status: 200}} <- QdrantClient.delete("/collections/posts") do with {:ok, %{status: 200}} <- QdrantClient.delete("/collections/posts") do
:ok :ok
else else

View File

@ -9,9 +9,9 @@ defmodule Pleroma.Search.QdrantSearchTest do
import Pleroma.Factory import Pleroma.Factory
import Mox import Mox
alias Pleroma.Web.CommonAPI
alias Pleroma.UnstubbedConfigMock, as: Config
alias Pleroma.Search.QdrantSearch alias Pleroma.Search.QdrantSearch
alias Pleroma.UnstubbedConfigMock, as: Config
alias Pleroma.Web.CommonAPI
alias Pleroma.Workers.SearchIndexingWorker alias Pleroma.Workers.SearchIndexingWorker
describe "Qdrant search" do describe "Qdrant search" do