From 1a72b06a82bfbf72d0d070d2430fa0bc2f5f672c Mon Sep 17 00:00:00 2001 From: Research Assistant Date: Mon, 18 May 2026 17:12:14 +0800 Subject: [PATCH] feat(embedding): create embedding package skeleton --- paperforge/embedding/__init__.py | 31 ++++++++++++++++++++++ paperforge/embedding/providers/__init__.py | 0 2 files changed, 31 insertions(+) create mode 100644 paperforge/embedding/__init__.py create mode 100644 paperforge/embedding/providers/__init__.py diff --git a/paperforge/embedding/__init__.py b/paperforge/embedding/__init__.py new file mode 100644 index 00000000..350c5c13 --- /dev/null +++ b/paperforge/embedding/__init__.py @@ -0,0 +1,31 @@ +from __future__ import annotations + +from paperforge.embedding._chroma import ( + delete_paper_vectors, + get_collection, + get_vector_db_path, +) +from paperforge.embedding.build_state import ( + get_vector_build_state_path, + mark_vector_build_state, + read_vector_build_state, + write_vector_build_state, +) +from paperforge.embedding.builder import embed_paper +from paperforge.embedding.preflight import _preflight_check +from paperforge.embedding.search import retrieve_chunks +from paperforge.embedding.status import get_embed_status + +__all__ = [ + "delete_paper_vectors", + "embed_paper", + "get_collection", + "get_embed_status", + "get_vector_build_state_path", + "get_vector_db_path", + "mark_vector_build_state", + "read_vector_build_state", + "retrieve_chunks", + "write_vector_build_state", + "_preflight_check", +] diff --git a/paperforge/embedding/providers/__init__.py b/paperforge/embedding/providers/__init__.py new file mode 100644 index 00000000..e69de29b