chore: linting (#597)

This commit is contained in:
cachho
2023-09-12 18:04:38 +02:00
committed by GitHub
parent 0f9a10c598
commit 03146946fa
11 changed files with 25 additions and 35 deletions

View File

@@ -47,4 +47,4 @@ class BaseVectorDB(JSONSerializable):
raise NotImplementedError
def set_collection_name(self, name: str):
raise NotImplementedError
raise NotImplementedError

View File

@@ -1,5 +1,5 @@
import logging
from typing import Dict, List, Optional, Any
from typing import Any, Dict, List, Optional
from chromadb import Collection, QueryResult
from langchain.docstore.document import Document
@@ -105,9 +105,7 @@ class ChromaDB(BaseVectorDB):
args["where"] = where
if limit:
args["limit"] = limit
return self.collection.get(
**args
)
return self.collection.get(**args)
def get_advanced(self, where):
return self.collection.get(where=where, limit=1)