Rustとpgvectorを使って、類義語の検索をやってみた

DRANK

PostgreSQLOpenAIembeddingpgvectortech はじめにこの記事は株式会社LabBase テックカレンダー Advent Calendar 2023 24日目の記事です。株式会社LabBaseでエンジニアをしている渡辺創です。今回は pgvector を使って、類義語の提案ができるか試してみたのでそれについて書いていきたいと思います。 背景LabBase就職というサービスを提供しており、研究に取り組んでいる学生が研究概要を登録してくれています。研究を頑張っている学生を採用したい企業の方が研究内容と募集内容のマッチングによって、就職活動・採用活動を支援するサービスとなっています。企業の人事の方が学生の研究内容をキーワードで検索するのですが、ユーザは自分の語彙の中でしか検索することができないので、その語彙を拡張する方法の1つとして、ユーザに類義語を提案しようと考えています。類義語関連だとこちらの記事にも書いたので、ぜひ読んでみてください。 使用する技術などRustpgvectorDockerOpenAI EmbeddingLabBase就職に登録されている技術キーワードの一部 手順データの準備データベースの準備データのインサートキーワードからベクトルを取得し、保存類似キーワードを検索する ディレク…