pgvector-template 0.1.0__tar.gz → 0.1.1__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (23) hide show
  1. {pgvector_template-0.1.0 → pgvector_template-0.1.1}/PKG-INFO +2 -2
  2. {pgvector_template-0.1.0 → pgvector_template-0.1.1}/pgvector_template/core/manager.py +3 -3
  3. {pgvector_template-0.1.0 → pgvector_template-0.1.1}/pgvector_template.egg-info/PKG-INFO +2 -2
  4. {pgvector_template-0.1.0 → pgvector_template-0.1.1}/pyproject.toml +2 -2
  5. {pgvector_template-0.1.0 → pgvector_template-0.1.1}/LICENSE +0 -0
  6. {pgvector_template-0.1.0 → pgvector_template-0.1.1}/README.md +0 -0
  7. {pgvector_template-0.1.0 → pgvector_template-0.1.1}/pgvector_template/__init__.py +0 -0
  8. {pgvector_template-0.1.0 → pgvector_template-0.1.1}/pgvector_template/core/__init__.py +0 -0
  9. {pgvector_template-0.1.0 → pgvector_template-0.1.1}/pgvector_template/core/document.py +0 -0
  10. {pgvector_template-0.1.0 → pgvector_template-0.1.1}/pgvector_template/core/embedder.py +0 -0
  11. {pgvector_template-0.1.0 → pgvector_template-0.1.1}/pgvector_template/core/retriever.py +0 -0
  12. {pgvector_template-0.1.0 → pgvector_template-0.1.1}/pgvector_template/core/search.py +0 -0
  13. {pgvector_template-0.1.0 → pgvector_template-0.1.1}/pgvector_template/db/__init__.py +0 -0
  14. {pgvector_template-0.1.0 → pgvector_template-0.1.1}/pgvector_template/db/connection.py +0 -0
  15. {pgvector_template-0.1.0 → pgvector_template-0.1.1}/pgvector_template/db/document_db.py +0 -0
  16. {pgvector_template-0.1.0 → pgvector_template-0.1.1}/pgvector_template/search/__init__.py +0 -0
  17. {pgvector_template-0.1.0 → pgvector_template-0.1.1}/pgvector_template/types.py +0 -0
  18. {pgvector_template-0.1.0 → pgvector_template-0.1.1}/pgvector_template/utils/__init__.py +0 -0
  19. {pgvector_template-0.1.0 → pgvector_template-0.1.1}/pgvector_template.egg-info/SOURCES.txt +0 -0
  20. {pgvector_template-0.1.0 → pgvector_template-0.1.1}/pgvector_template.egg-info/dependency_links.txt +0 -0
  21. {pgvector_template-0.1.0 → pgvector_template-0.1.1}/pgvector_template.egg-info/requires.txt +1 -1
  22. {pgvector_template-0.1.0 → pgvector_template-0.1.1}/pgvector_template.egg-info/top_level.txt +0 -0
  23. {pgvector_template-0.1.0 → pgvector_template-0.1.1}/setup.cfg +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: pgvector-template
3
- Version: 0.1.0
3
+ Version: 0.1.1
4
4
  Summary: Template library for flexible PGVector RAG implementations
5
5
  Author-email: DL <v49t9zpqd@mozmail.com>
6
6
  License: MIT
@@ -11,12 +11,12 @@ Classifier: Operating System :: OS Independent
11
11
  Requires-Python: >=3.11
12
12
  Description-Content-Type: text/markdown
13
13
  License-File: LICENSE
14
- Requires-Dist: psycopg[binary]>=3.1.0
15
14
  Requires-Dist: pgvector>=0.2.0
16
15
  Requires-Dist: pydantic<3.0,>=2.11
17
16
  Requires-Dist: sqlalchemy>=2.0.0
18
17
  Requires-Dist: typing-extensions>=4.0.0
19
18
  Provides-Extra: test
19
+ Requires-Dist: psycopg[binary]>=3.1.0; extra == "test"
20
20
  Requires-Dist: pytest>=7.0.0; extra == "test"
21
21
  Requires-Dist: pytest-cov>=4.0.0; extra == "test"
22
22
  Requires-Dist: python-dotenv>=1.0.0; extra == "test"
@@ -19,7 +19,7 @@ class BaseCorpusManagerConfig(BaseModel):
19
19
  schema_name: str
20
20
  document_cls: Type[BaseDocument]
21
21
  embedding_provider: BaseEmbeddingProvider
22
- document_metadata: Type[BaseDocumentMetadata]
22
+ document_metadata_cls: Type[BaseDocumentMetadata]
23
23
 
24
24
  model_config = {"arbitrary_types_allowed": True}
25
25
 
@@ -80,7 +80,7 @@ class BaseCorpusManager(ABC):
80
80
  self,
81
81
  content: str,
82
82
  corpus_metadata: dict[str, Any],
83
- optional_props: BaseDocumentOptionalProps,
83
+ optional_props: BaseDocumentOptionalProps | None = None,
84
84
  ) -> int:
85
85
  """
86
86
  Insert a new `Corpus`, which will be split into 1-or-more `Document`s, depending on its length.
@@ -130,7 +130,7 @@ class BaseCorpusManager(ABC):
130
130
  documents_to_insert = []
131
131
  for i in range(len(document_contents)):
132
132
  chunk_md = self._extract_chunk_metadata(document_contents[i])
133
- base_metadata = self.config.document_metadata(**(corpus_metadata | chunk_md))
133
+ base_metadata = self.config.document_metadata_cls(**(corpus_metadata | chunk_md))
134
134
  documents_to_insert.append(
135
135
  self.config.document_cls.from_props(
136
136
  corpus_id=corpus_id,
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: pgvector-template
3
- Version: 0.1.0
3
+ Version: 0.1.1
4
4
  Summary: Template library for flexible PGVector RAG implementations
5
5
  Author-email: DL <v49t9zpqd@mozmail.com>
6
6
  License: MIT
@@ -11,12 +11,12 @@ Classifier: Operating System :: OS Independent
11
11
  Requires-Python: >=3.11
12
12
  Description-Content-Type: text/markdown
13
13
  License-File: LICENSE
14
- Requires-Dist: psycopg[binary]>=3.1.0
15
14
  Requires-Dist: pgvector>=0.2.0
16
15
  Requires-Dist: pydantic<3.0,>=2.11
17
16
  Requires-Dist: sqlalchemy>=2.0.0
18
17
  Requires-Dist: typing-extensions>=4.0.0
19
18
  Provides-Extra: test
19
+ Requires-Dist: psycopg[binary]>=3.1.0; extra == "test"
20
20
  Requires-Dist: pytest>=7.0.0; extra == "test"
21
21
  Requires-Dist: pytest-cov>=4.0.0; extra == "test"
22
22
  Requires-Dist: python-dotenv>=1.0.0; extra == "test"
@@ -4,12 +4,11 @@ build-backend = "setuptools.build_meta"
4
4
 
5
5
  [project]
6
6
  name = "pgvector-template"
7
- version = "0.1.0"
7
+ version = "0.1.1"
8
8
  description = "Template library for flexible PGVector RAG implementations"
9
9
  authors = [{ name="DL", email="v49t9zpqd@mozmail.com" }]
10
10
  license = { text = "MIT" }
11
11
  dependencies = [
12
- "psycopg[binary]>=3.1.0",
13
12
  "pgvector>=0.2.0",
14
13
  "pydantic>=2.11,<3.0",
15
14
  "sqlalchemy>=2.0.0",
@@ -25,6 +24,7 @@ classifiers = [
25
24
 
26
25
  [project.optional-dependencies]
27
26
  test = [
27
+ "psycopg[binary]>=3.1.0",
28
28
  "pytest>=7.0.0",
29
29
  "pytest-cov>=4.0.0",
30
30
  "python-dotenv>=1.0.0",
@@ -1,10 +1,10 @@
1
- psycopg[binary]>=3.1.0
2
1
  pgvector>=0.2.0
3
2
  pydantic<3.0,>=2.11
4
3
  sqlalchemy>=2.0.0
5
4
  typing-extensions>=4.0.0
6
5
 
7
6
  [test]
7
+ psycopg[binary]>=3.1.0
8
8
  pytest>=7.0.0
9
9
  pytest-cov>=4.0.0
10
10
  python-dotenv>=1.0.0