Skip to main content

Embedding extension to omop-graph

Project description

omop-emb

Embedding layer for OMOP CDM.

omop-emb now separates model metadata from embedding storage:

  • model metadata is stored locally in SQLite (metadata.db)
  • embedding vectors are stored by the selected backend (pgvector or faiss)
  • OMOP concept metadata remains in the OMOP CDM database

Installation

omop-emb now exposes backend-specific optional dependencies so installation can match the embedding backend you actually intend to use.

pip install "omop-emb[pgvector]"
pip install "omop-emb[faiss]"
pip install "omop-emb[all]"

Notes:

  • pgvector installs the PostgreSQL/pgvector dependencies.
  • faiss installs the FAISS-based backend dependencies. This currently only includes CPU support
  • all installs both backend stacks for development or mixed environments.
  • A plain pip install omop-emb installs the shared core package only.
  • PostgreSQL-specific embedding dependencies are optional, but omop-emb still requires OMOP CDM database access.
  • Non-PostgreSQL database backends have not yet been tested.

Runtime Configuration

Common environment variables:

  • OMOP_EMB_BACKEND: backend name (pgvector or faiss) used by the backend factory.
  • OMOP_EMB_BASE_STORAGE_DIR: local base directory for omop-emb artifacts, including local metadata (metadata.db) and FAISS files. If unset, omop-emb defaults to ./.omop_emb in the current working directory.
  • OMOP_DATABASE_URL: SQLAlchemy URL for the OMOP CDM database.

Extended documentation can be found here.

Project Roadmap

  • Interface for PostgreSQL storage of vectors
  • Interface for FAISS storage of embeddings
  • Extensive unit testing
    • Backend testing
    • Corruption and restoration of DB testing
  • Support non-Flat indices for each backend
  • faiss GPU support
  • pgvectorscale support
  • Vector-quantisation for more efficient storage

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

omop_emb-0.3.1.tar.gz (204.7 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

omop_emb-0.3.1-py3-none-any.whl (42.4 kB view details)

Uploaded Python 3

File details

Details for the file omop_emb-0.3.1.tar.gz.

File metadata

  • Download URL: omop_emb-0.3.1.tar.gz
  • Upload date:
  • Size: 204.7 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: uv/0.11.6 {"installer":{"name":"uv","version":"0.11.6","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}

File hashes

Hashes for omop_emb-0.3.1.tar.gz
Algorithm Hash digest
SHA256 ebdf116041335f49ea84080420a646def0e9033abfbf160cac5a9b49b0fc216c
MD5 2f56fe95ac754fa8cc4d68caa89584de
BLAKE2b-256 cce180c3c27f3b98fd14208bf616b903d74c53970a2b1f3657a16b3c74a96c63

See more details on using hashes here.

File details

Details for the file omop_emb-0.3.1-py3-none-any.whl.

File metadata

  • Download URL: omop_emb-0.3.1-py3-none-any.whl
  • Upload date:
  • Size: 42.4 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: uv/0.11.6 {"installer":{"name":"uv","version":"0.11.6","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}

File hashes

Hashes for omop_emb-0.3.1-py3-none-any.whl
Algorithm Hash digest
SHA256 2dec7998e31612fd011d3dd48959754196273afc936415deb32e1faf91496577
MD5 40ee18c38c068d6cbe8edb613c42014d
BLAKE2b-256 4545448c1013e4440197e244c7270425a5a9b7d918baa4dc41294bfdcbaee3a3

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page