llama-index-vector-stores-chroma 0.3.0__py3-none-any.whl → 0.4.1__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.
Potentially problematic release.
This version of llama-index-vector-stores-chroma might be problematic. Click here for more details.
- llama_index/vector_stores/chroma/base.py +27 -14
- {llama_index_vector_stores_chroma-0.3.0.dist-info → llama_index_vector_stores_chroma-0.4.1.dist-info}/METADATA +3 -3
- llama_index_vector_stores_chroma-0.4.1.dist-info/RECORD +6 -0
- llama_index_vector_stores_chroma-0.3.0.dist-info/RECORD +0 -6
- {llama_index_vector_stores_chroma-0.3.0.dist-info → llama_index_vector_stores_chroma-0.4.1.dist-info}/WHEEL +0 -0
|
@@ -331,10 +331,10 @@ class ChromaVectorStore(BasePydanticVectorStore):
|
|
|
331
331
|
|
|
332
332
|
if filters:
|
|
333
333
|
where = _to_chroma_filter(filters)
|
|
334
|
-
|
|
335
|
-
where = None
|
|
334
|
+
self._collection.delete(ids=node_ids, where=where)
|
|
336
335
|
|
|
337
|
-
|
|
336
|
+
else:
|
|
337
|
+
self._collection.delete(ids=node_ids)
|
|
338
338
|
|
|
339
339
|
def clear(self) -> None:
|
|
340
340
|
"""Clear the collection."""
|
|
@@ -378,12 +378,19 @@ class ChromaVectorStore(BasePydanticVectorStore):
|
|
|
378
378
|
def _query(
|
|
379
379
|
self, query_embeddings: List["float"], n_results: int, where: dict, **kwargs
|
|
380
380
|
) -> VectorStoreQueryResult:
|
|
381
|
-
|
|
382
|
-
|
|
383
|
-
|
|
384
|
-
|
|
385
|
-
|
|
386
|
-
|
|
381
|
+
if where:
|
|
382
|
+
results = self._collection.query(
|
|
383
|
+
query_embeddings=query_embeddings,
|
|
384
|
+
n_results=n_results,
|
|
385
|
+
where=where,
|
|
386
|
+
**kwargs,
|
|
387
|
+
)
|
|
388
|
+
else:
|
|
389
|
+
results = self._collection.query(
|
|
390
|
+
query_embeddings=query_embeddings,
|
|
391
|
+
n_results=n_results,
|
|
392
|
+
**kwargs,
|
|
393
|
+
)
|
|
387
394
|
|
|
388
395
|
logger.debug(f"> Top {len(results['documents'][0])} nodes:")
|
|
389
396
|
nodes = []
|
|
@@ -429,11 +436,17 @@ class ChromaVectorStore(BasePydanticVectorStore):
|
|
|
429
436
|
def _get(
|
|
430
437
|
self, limit: Optional[int], where: dict, **kwargs
|
|
431
438
|
) -> VectorStoreQueryResult:
|
|
432
|
-
|
|
433
|
-
|
|
434
|
-
|
|
435
|
-
|
|
436
|
-
|
|
439
|
+
if where:
|
|
440
|
+
results = self._collection.get(
|
|
441
|
+
limit=limit,
|
|
442
|
+
where=where,
|
|
443
|
+
**kwargs,
|
|
444
|
+
)
|
|
445
|
+
else:
|
|
446
|
+
results = self._collection.get(
|
|
447
|
+
limit=limit,
|
|
448
|
+
**kwargs,
|
|
449
|
+
)
|
|
437
450
|
|
|
438
451
|
logger.debug(f"> Top {len(results['documents'])} nodes:")
|
|
439
452
|
nodes = []
|
|
@@ -1,11 +1,11 @@
|
|
|
1
1
|
Metadata-Version: 2.1
|
|
2
2
|
Name: llama-index-vector-stores-chroma
|
|
3
|
-
Version: 0.
|
|
3
|
+
Version: 0.4.1
|
|
4
4
|
Summary: llama-index vector_stores chroma integration
|
|
5
5
|
License: MIT
|
|
6
6
|
Author: Your Name
|
|
7
7
|
Author-email: you@example.com
|
|
8
|
-
Requires-Python: >=3.
|
|
8
|
+
Requires-Python: >=3.9,<4.0
|
|
9
9
|
Classifier: License :: OSI Approved :: MIT License
|
|
10
10
|
Classifier: Programming Language :: Python :: 3
|
|
11
11
|
Classifier: Programming Language :: Python :: 3.9
|
|
@@ -13,7 +13,7 @@ Classifier: Programming Language :: Python :: 3.10
|
|
|
13
13
|
Classifier: Programming Language :: Python :: 3.11
|
|
14
14
|
Classifier: Programming Language :: Python :: 3.12
|
|
15
15
|
Requires-Dist: chromadb (>=0.5.17)
|
|
16
|
-
Requires-Dist: llama-index-core (>=0.
|
|
16
|
+
Requires-Dist: llama-index-core (>=0.12.0,<0.13.0)
|
|
17
17
|
Description-Content-Type: text/markdown
|
|
18
18
|
|
|
19
19
|
# LlamaIndex Vector_Stores Integration: Chroma
|
|
@@ -0,0 +1,6 @@
|
|
|
1
|
+
llama_index/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
2
|
+
llama_index/vector_stores/chroma/__init__.py,sha256=QNMK-nHKEt-wmks5mhWfdOKDybpmsqrL4neV-HCA6N4,101
|
|
3
|
+
llama_index/vector_stores/chroma/base.py,sha256=ELnoNImcqpxsEZAcQhYncY8qoRKFR7JwQsO0basA55I,15269
|
|
4
|
+
llama_index_vector_stores_chroma-0.4.1.dist-info/METADATA,sha256=oD3FuNbR4hRlCjoKVT2RW6GoWP5epTO5XeoPTgyA2u4,696
|
|
5
|
+
llama_index_vector_stores_chroma-0.4.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
|
|
6
|
+
llama_index_vector_stores_chroma-0.4.1.dist-info/RECORD,,
|
|
@@ -1,6 +0,0 @@
|
|
|
1
|
-
llama_index/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
2
|
-
llama_index/vector_stores/chroma/__init__.py,sha256=QNMK-nHKEt-wmks5mhWfdOKDybpmsqrL4neV-HCA6N4,101
|
|
3
|
-
llama_index/vector_stores/chroma/base.py,sha256=bMnmCNGwNSPbuCUi_J_DUfvpucNibq8l2U9C4AZOwXY,14845
|
|
4
|
-
llama_index_vector_stores_chroma-0.3.0.dist-info/METADATA,sha256=WAyb4s5iNwdoBqCpS1mq4IQzJkPfKDkaYbv7HKF8kpE,698
|
|
5
|
-
llama_index_vector_stores_chroma-0.3.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
|
|
6
|
-
llama_index_vector_stores_chroma-0.3.0.dist-info/RECORD,,
|
|
File without changes
|