HexDocs MCP

defmodule HexdocsMcp.Migrations do @moduledoc """ Shared migrations that can be used in regular migrations and tests. This provides a consistent way to create database structure both during regular migrations and in test environments. """ alias HexdocsMcp.Repo @doc """ Creates the embeddings table with 1024 dimensions (for mxbai-embed-large). ## Options * `:prefix` - The prefix to run the migrations in """ def create_embeddings_table(opts \\ []) do prefix = opts[:prefix] create_opts = if prefix, do: ", prefix: #{inspect(prefix)}", else: "" [ """ CREATE TABLE IF NOT EXISTS embeddings( id INTEGER PRIMARY KEY, package TEXT NOT NULL, version TEXT NOT NULL, source_file TEXT NOT NULL, source_type TEXT, start_byte INTEGER, end_byte INTEGER, url TEXT, text_snippet TEXT, text TEXT NOT NULL, content_hash TEXT NOT NULL, embedding FLOAT[1024] NOT NULL, inserted_at TIMESTAMP, updated_at TIMESTAMP, UNIQUE(package, version, source_file, text_snippet) )#{create_opts}; """, "CREATE INDEX IF NOT EXISTS idx_embeddings_package_version ON embeddings(package, version)#{create_opts};", "CREATE INDEX IF NOT EXISTS idx_embeddings_content_hash ON embeddings(package, version, content_hash)#{create_opts};" ] end @doc """ Updates the embeddings table schema to ensure all required columns exist. This should be called on application start to ensure the database schema is up-to-date. """ def update_embeddings_table do %{rows: table_info} = Repo.query!("PRAGMA table_info(embeddings)") has_content_hash? = Enum.any?(table_info, &(Enum.at(&1, 1) == "content_hash")) has_url? = Enum.any?(table_info, &(Enum.at(&1, 1) == "url")) if not has_content_hash? do Repo.query!("ALTER TABLE embeddings ADD COLUMN content_hash TEXT NOT NULL DEFAULT ''") end if not has_url? do Repo.query!("ALTER TABLE embeddings ADD COLUMN url TEXT") end Repo.query!("CREATE INDEX IF NOT EXISTS idx_embeddings_content_hash ON embeddings(package, version, content_hash)") end @doc """ Drops the embeddings table. ## Options * `:prefix` - The prefix to run the migrations in """ def drop_embeddings_table(opts \\ []) do prefix = opts[:prefix] drop_opts = if prefix, do: ", prefix: #{inspect(prefix)}", else: "" ["DROP TABLE IF EXISTS embeddings#{drop_opts};"] end end