コンテンツにスキップ

AIチャットの設定

Info

AIチャットにはGramps Web APIバージョン2.5.0以上が必要です。バージョン3.6.0では、よりインテリジェントなインタラクションのためのツール呼び出し機能が導入されました。

Gramps Web APIは、リトリーバル拡張生成(RAG)とツール呼び出しを組み合わせた技術を介して、大規模言語モデル(LLM)を使用して系譜データベースに関する質問をすることをサポートしています。

仕組み

AIアシスタントは、2つの補完的なアプローチを使用します:

リトリーバル拡張生成(RAG)ベクトル埋め込みモデルが、Grampsデータベース内のすべてのオブジェクトのインデックスを、オブジェクトの意味をエンコードした数値ベクトルの形式で作成します。ユーザーが質問をすると、その質問もベクトルに変換され、データベース内のオブジェクトと比較されます。このセマンティック検索は、質問に最も意味的に類似したオブジェクトを返します。

ツール呼び出し(v3.6.0以上):AIアシスタントは、今や特定のツールを使用して系譜データを直接クエリすることができます。これらのツールにより、アシスタントはデータベースを検索し、特定の基準で人々/イベント/家族/場所をフィルタリングし、個人間の関係を計算し、詳細なオブジェクト情報を取得することができます。これにより、アシスタントは複雑な系譜の質問に正確に答える能力が大幅に向上します。

Gramps Web APIでチャットエンドポイントを有効にするには、3つのステップが必要です:

  1. 必要な依存関係のインストール、
  2. セマンティック検索の有効化、
  3. LLMプロバイダーの設定。

これらの3つのステップについては、以下に順を追って説明します。最後に、オーナーまたは管理者はどのユーザーがチャット機能にアクセスできるかを設定する必要があります。

必要な依存関係のインストール

AIチャットには、Sentence TransformersおよびPyTorchライブラリをインストールする必要があります。

Gramps Webの標準Dockerイメージには、amd64(例:64ビットデスクトップPC)およびarm64(例:64ビットRaspberry Pi)アーキテクチャ用にこれらが事前にインストールされています。残念ながら、AIチャットはPyTorchのサポートがないため、armv7(例:32ビットRaspberry Pi)アーキテクチャではサポートされていません。

pipを介してGramps Web APIをインストールする際(Dockerイメージを使用する場合は必要ありません)、必要な依存関係は以下のコマンドでインストールされます。

pip install gramps_webapi[ai]

セマンティック検索の有効化

必要な依存関係がインストールされている場合、セマンティック検索を有効にするのは、VECTOR_EMBEDDING_MODEL設定オプションを設定するだけで済む場合があります(例:GRAMPSWEB_VECTOR_EMBEDDING_MODEL環境変数を設定することで)。詳細はサーバー設定を参照してください。これは、Sentence Transformersライブラリによってサポートされているモデルの任意の文字列である必要があります。このプロジェクトのドキュメントで詳細と利用可能なモデルを確認してください。

Warning

デフォルトのDockerイメージには、GPUサポートのあるPyTorchバージョンが含まれていないことに注意してください。GPUにアクセスできる場合(これはセマンティックインデックスを大幅に高速化します)、GPU対応のPyTorchバージョンをインストールしてください。

モデルを選択する際には、いくつかの考慮事項があります。

  • モデルを変更すると、ツリー(またはマルチツリーセットアップのすべてのツリー)のセマンティック検索インデックスを手動で再作成する必要があります。そうしないと、エラーや無意味な結果が発生します。Gramps Webは、構成された埋め込みモデルが既存のインデックスと一致しなくなったときに検出し、管理者に管理設定から完全な再インデックスをトリガーするように促す持続的な通知を表示します。
  • モデルは、正確性/一般性と計算時間/ストレージスペースの間のトレードオフです。強力なGPUにアクセスできないシステムでGramps Web APIを実行している場合、大きなモデルは通常、実際には遅すぎます。
  • データベース全体が英語であり、すべてのユーザーが英語でチャット質問をすることが期待される場合を除き、多言語埋め込みモデルが必要です。これらは純粋な英語モデルよりもはるかに少ないです。

モデルがローカルキャッシュに存在しない場合、新しい構成でGramps Web APIが初めて起動されるときにダウンロードされます。モデルsentence-transformers/distiluse-base-multilingual-cased-v2は、標準のDockerイメージを使用する際にすでにローカルで利用可能です。このモデルは良い出発点であり、多言語入力をサポートしています。

異なるモデルについての学びをコミュニティと共有してください!

Info

Sentence Transformersライブラリは大量のメモリを消費するため、ワーカープロセスが終了する可能性があります。一般的な目安として、セマンティック検索が有効になっている場合、各Gunicornワーカーは約200MBのメモリを消費し、各Celeryワーカーはアイドル時でも約500MBのメモリを消費し、埋め込みを計算しているときは最大1GBに達します。CPUとメモリの使用制限を参照して、メモリ使用量を制限する設定を確認してください。また、一時的なメモリ使用量のスパイクによるOOMエラーを防ぐために、十分に大きなスワップパーティションを用意することをお勧めします。

リモート埋め込みAPIの使用

ローカルのSentence Transformersモデルを実行する代わりに、セマンティック検索用のリモートOpenAI互換埋め込みAPIを使用することができます。これは、埋め込み計算を別のサービス(例:Ollama)にオフロードしたい場合や、クラウド埋め込みプロバイダー(例:OpenAI)を使用したい場合、またはSentence TransformersおよびPyTorchライブラリをメモリにロードすることを避けたい場合に便利です。

リモートAPIは、OpenAI埋め込みエンドポイント/v1/embeddings)と互換性がある必要があります。

リモート埋め込みAPIを使用するには、次の設定オプションを設定します(サーバー設定を参照):

キー 説明
VECTOR_EMBEDDING_MODEL リモートプロバイダーに渡すモデル名
VECTOR_EMBEDDING_BASE_URL リモートAPIのベースURL
VECTOR_EMBEDDING_API_KEY APIキー(プロバイダーが認証を必要とする場合のみ必要)

Ollamaを使用した埋め込み

Docker ComposeでGramps Webをデプロイする際、Ollamaサービスを追加し、埋め込みと(オプションで)LLMの両方に使用できます:

services:
  grampsweb: &grampsweb
    # ... 既存の設定 ...
    environment:
      GRAMPSWEB_VECTOR_EMBEDDING_MODEL: nomic-embed-text
      GRAMPSWEB_VECTOR_EMBEDDING_BASE_URL: http://ollama:11434

  grampsweb_celery: &grampsweb_celery
    # ... 既存の設定 ...
    environment:
      GRAMPSWEB_VECTOR_EMBEDDING_MODEL: nomic-embed-text
      GRAMPSWEB_VECTOR_EMBEDDING_BASE_URL: http://ollama:11434

  ollama:
    image: ollama/ollama
    container_name: ollama
    ports:
      - "11434:11434"
    volumes:
      - ollama_data:/root/.ollama

volumes:
  ollama_data:

サービスを起動した後、Ollamaに埋め込みモデルをプルします:

docker compose exec ollama ollama pull nomic-embed-text

Info

Ollamaを埋め込みに使用する場合、Sentence TransformersおよびPyTorchライブラリは必要なく、Gramps Web APIワーカーのメモリ使用量が大幅に削減されます。

OpenAIを使用した埋め込み

OpenAI埋め込みAPIを使用するには、ベースURLをOpenAI APIに設定し、APIキーを提供します:

environment:
  GRAMPSWEB_VECTOR_EMBEDDING_MODEL: text-embedding-3-small
  GRAMPSWEB_VECTOR_EMBEDDING_BASE_URL: https://api.openai.com
  GRAMPSWEB_VECTOR_EMBEDDING_API_KEY: sk-...

Warning

埋め込みモデルを変更すると、ツリー(またはマルチツリーセットアップのすべてのツリー)のすべてのレコードを再インデックスする必要があります。異なるモデルは異なる次元のベクトルを生成するためです。

LLMプロバイダーの設定

LLMとの通信にはPydantic AIフレームワークを使用し、OpenAI互換APIをサポートしています。これにより、Ollamaを介してローカルにデプロイされたLLM(OllamaのOpenAI互換性を参照)や、OpenAI、Anthropic、Hugging Face TGI(テキスト生成推論)などのホストAPIを使用できます。LLMは、設定パラメータLLM_MODELおよびLLM_BASE_URLを介して構成されます。

OpenAI APIを介してホストされたLLMの使用

OpenAI APIを使用する場合、LLM_BASE_URLは設定しなくてもよく、LLM_MODELはOpenAIモデルの1つ(例:gpt-4o-mini)に設定する必要があります。LLMはRAGとツール呼び出しの両方を使用して質問に答えます:セマンティック検索結果から関連情報を選択し、特化したツールを使用してデータベースに直接クエリを実行できます。深い系譜や歴史的知識は必要ありません。したがって、小型または安価なモデルが十分かどうかを試すことができます。

また、アカウントにサインアップし、APIキーを取得してOPENAI_API_KEY環境変数に保存する必要があります。

Info

LLM_MODELは設定パラメータです。環境変数を介して設定したい場合は、GRAMPSWEB_LLM_MODELを使用してください(設定を参照)。OPENAI_API_KEYは設定パラメータではなく、Pydantic AIライブラリによって直接使用される環境変数であるため、プレフィックスを付けないでください。

Mistral AIの使用

Mistral AIのホストモデルを使用するには、LLM_MODELを設定する際にモデル名の先頭にmistral:を付けます。

Mistral AIアカウントにサインアップし、APIキーを取得してMISTRAL_API_KEY環境変数に保存する必要があります。Pydantic AIは自動的に正しいMistral APIエンドポイントを使用するため、LLM_BASE_URLを設定する必要はありません。

Docker Composeを使用して環境変数で構成する例:

environment:
  GRAMPSWEB_LLM_MODEL: mistral:mistral-large-latest
  MISTRAL_API_KEY: your-mistral-api-key-here
  GRAMPSWEB_VECTOR_EMBEDDING_MODEL: sentence-transformers/distiluse-base-multilingual-cased-v2

Ollamaを介してローカルLLMの使用

Ollamaは、LLMをローカルで実行する便利な方法です。詳細についてはOllamaのドキュメントを参照してください。LLMはかなりの計算リソースを必要とし、最小のモデルを除いてはGPUサポートなしでは遅すぎる可能性があります。tinyllamaがニーズに合うかどうかを試してみてください。合わない場合は、より大きなモデルを試してください。コミュニティと経験を共有してください!

Docker ComposeでGramps Webをデプロイする際、Ollamaサービスを追加できます。

services:
  ollama:
    image: ollama/ollama
    container_name: ollama
    ports:
      - "11434:11434"
    volumes:
      - ollama_data:/root/.ollama

volumes:
    ollama_data:

次に、LLM_BASE_URL設定パラメータをhttp://ollama:11434/v1に設定します。LLM_MODELをOllamaがサポートするモデルに設定し、ollama pull <model>でコンテナ内にプルします。最後に、OPENAI_API_KEYollamaに設定します。

Ollamaに関する問題をトラブルシューティングするには、Ollamaサービス環境で環境変数OLLAMA_DEBUG=1を設定してデバッグログを有効にできます。

Info

Gramps Web AIチャットにOllamaを使用している場合は、コミュニティをサポートするために、欠けている詳細をこのドキュメントに追加してください。

他のプロバイダーの使用

他のプロバイダーのドキュメントを提出し、コミュニティと経験を共有してください!