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

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