ddi-fw 0.0.122__py3-none-any.whl → 0.0.123__py3-none-any.whl
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.
- ddi_fw/pipeline/pipeline.py +13 -5
- {ddi_fw-0.0.122.dist-info → ddi_fw-0.0.123.dist-info}/METADATA +2 -3
- {ddi_fw-0.0.122.dist-info → ddi_fw-0.0.123.dist-info}/RECORD +5 -5
- {ddi_fw-0.0.122.dist-info → ddi_fw-0.0.123.dist-info}/WHEEL +1 -1
- {ddi_fw-0.0.122.dist-info → ddi_fw-0.0.123.dist-info}/top_level.txt +0 -0
ddi_fw/pipeline/pipeline.py
CHANGED
@@ -67,9 +67,13 @@ class Pipeline:
|
|
67
67
|
if column == None:
|
68
68
|
dictionary = collection.get(
|
69
69
|
include=['embeddings', 'metadatas'])
|
70
|
+
print(
|
71
|
+
f"Embeddings are calculated from {vector_db_collection_name}")
|
70
72
|
else:
|
71
73
|
dictionary = collection.get(include=['embeddings', 'metadatas'], where={
|
72
74
|
"type": {"$eq": f"{column}"}})
|
75
|
+
print(
|
76
|
+
f"Embeddings of {column} are calculated from {vector_db_collection_name}")
|
73
77
|
# Populate the embedding dictionary with embeddings from the vector database
|
74
78
|
for metadata, embedding in zip(dictionary['metadatas'], dictionary['embeddings']):
|
75
79
|
embedding_dict[metadata["type"]
|
@@ -88,9 +92,7 @@ class Pipeline:
|
|
88
92
|
kwargs[k] = v
|
89
93
|
if self.embedding_dict == None:
|
90
94
|
embedding_dict = defaultdict(lambda: defaultdict(list))
|
91
|
-
|
92
|
-
self.__create_or_update_embeddings__(
|
93
|
-
embedding_dict, self.vector_db_persist_directory, self.vector_db_collection_name)
|
95
|
+
# TODO find more effective solution
|
94
96
|
|
95
97
|
if self.column_embedding_configs:
|
96
98
|
for item in self.column_embedding_configs:
|
@@ -99,8 +101,14 @@ class Pipeline:
|
|
99
101
|
col_db_collection = item["vector_db_collection_name"]
|
100
102
|
self.__create_or_update_embeddings__(
|
101
103
|
embedding_dict, col_db_dir, col_db_collection, col)
|
102
|
-
|
103
|
-
|
104
|
+
|
105
|
+
elif self.vector_db_persist_directory:
|
106
|
+
self.__create_or_update_embeddings__(
|
107
|
+
embedding_dict, self.vector_db_persist_directory, self.vector_db_collection_name)
|
108
|
+
|
109
|
+
else:
|
110
|
+
print(
|
111
|
+
f"There is no configuration of Embeddings")
|
104
112
|
|
105
113
|
# if self.embedding_dict == None:
|
106
114
|
# if self.vector_db_persist_directory:
|
@@ -1,6 +1,6 @@
|
|
1
1
|
Metadata-Version: 2.1
|
2
2
|
Name: ddi_fw
|
3
|
-
Version: 0.0.
|
3
|
+
Version: 0.0.123
|
4
4
|
Summary: Do not use :)
|
5
5
|
Author-email: Kıvanç Bayraktar <bayraktarkivanc@gmail.com>
|
6
6
|
Maintainer-email: Kıvanç Bayraktar <bayraktarkivanc@gmail.com>
|
@@ -46,7 +46,6 @@ Requires-Dist: py7zr==0.22.0
|
|
46
46
|
Requires-Dist: openai>=1.52.2
|
47
47
|
Requires-Dist: langchain>=0.3.4
|
48
48
|
Requires-Dist: chromadb>=0.5.15
|
49
|
-
Requires-Dist:
|
49
|
+
Requires-Dist: langchain_community==0.3.3
|
50
50
|
Requires-Dist: datasets==3.0.2
|
51
51
|
Requires-Dist: unstructured==0.16.3
|
52
|
-
|
@@ -72,7 +72,7 @@ ddi_fw/pipeline/__init__.py,sha256=tKDM_rW4vPjlYTeOkNgi9PujDzb4e9O3LK1w5wqnebw,2
|
|
72
72
|
ddi_fw/pipeline/multi_modal_combination_strategy.py,sha256=qIst7vxHaOAhRv4lgozszwa3b1QE4aIrN74t41Xnvr4,1637
|
73
73
|
ddi_fw/pipeline/multi_pipeline.py,sha256=t_Z7d7xRfDnhpQTlqCf7c0isZ5hZlyXavKhC7ePsnJY,5903
|
74
74
|
ddi_fw/pipeline/ner_pipeline.py,sha256=wB7hz4YCOv7UAz6bGE6sSpPXXIdoOflOVK5UCc1fO-o,5586
|
75
|
-
ddi_fw/pipeline/pipeline.py,sha256=
|
75
|
+
ddi_fw/pipeline/pipeline.py,sha256=WQk9vZOLao6YDpqPK7O7NDMqb5pjcPfUaG6_LZL2Qis,8769
|
76
76
|
ddi_fw/test/basic_test.py,sha256=fEOGcZm1ObnsDvMiXNmdmz6YCeUrGc8V0DwlSwGhsq8,376
|
77
77
|
ddi_fw/test/combination_test.py,sha256=TWNE8sf-DSh1Q9-yRaRBc774Sn1kSMGXLwQhd2_Qynk,324
|
78
78
|
ddi_fw/test/compress_json_test.py,sha256=BGny56YqiG-pzhMoDzLKQBQI1E7o3jU0S7VYWtclAx4,1045
|
@@ -91,7 +91,7 @@ ddi_fw/utils/package_helper.py,sha256=erl8_onmhK-41zQoaED2qyDUV9GQxmT9sdoyRp9_q5
|
|
91
91
|
ddi_fw/utils/py7zr_helper.py,sha256=gOqaFIyJvTjUM-btO2x9AQ69jZOS8PoKN0wetYIckJw,4747
|
92
92
|
ddi_fw/utils/utils.py,sha256=szwnxMTDRrZoeNRyDuf3aCbtzriwtaRk4mHSH3asLdA,4301
|
93
93
|
ddi_fw/utils/zip_helper.py,sha256=YRZA4tKZVBJwGQM0_WK6L-y5MoqkKoC-nXuuHK6CU9I,5567
|
94
|
-
ddi_fw-0.0.
|
95
|
-
ddi_fw-0.0.
|
96
|
-
ddi_fw-0.0.
|
97
|
-
ddi_fw-0.0.
|
94
|
+
ddi_fw-0.0.123.dist-info/METADATA,sha256=aklS2YR-_6JHgDWB2vv_Urdb-bUj975olJHX17x6AhA,1965
|
95
|
+
ddi_fw-0.0.123.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
|
96
|
+
ddi_fw-0.0.123.dist-info/top_level.txt,sha256=PMwHICFZTZtcpzQNPV4UQnfNXYIeLR_Ste-Wfc1h810,7
|
97
|
+
ddi_fw-0.0.123.dist-info/RECORD,,
|
File without changes
|