fix chromadb ut

This commit is contained in:
betterwang 2024-03-07 13:56:30 +08:00
parent 2ffa9e12b9
commit 91c58d04cf
3 changed files with 3 additions and 2 deletions

1
.gitignore vendored
View file

@ -27,6 +27,7 @@ share/python-wheels/
.installed.cfg
*.egg
MANIFEST
metagpt/tools/schemas/
# PyInstaller
# Usually these files are written by a python scripts from a template

View file

@ -11,7 +11,7 @@ import chromadb
class ChromaStore:
"""If inherited from BaseStore, or importing other modules from metagpt, a Python exception occurs, which is strange."""
def __init__(self, name, get_or_create: bool = True):
def __init__(self, name: str, get_or_create: bool = False):
client = chromadb.Client()
collection = client.create_collection(name, get_or_create=get_or_create)
self.client = client

View file

@ -12,7 +12,7 @@ from metagpt.document_store.chromadb_store import ChromaStore
def test_chroma_store():
"""FIXMEchroma使用感觉很诡异一用Python就挂测试用例里也是"""
# 创建 ChromaStore 实例,使用 'sample_collection' 集合
document_store = ChromaStore("sample_collection_1")
document_store = ChromaStore("sample_collection_1", get_or_create=True)
# 使用 write 方法添加多个文档
document_store.write(