Skip to content
Merged
Original file line number Diff line number Diff line change
Expand Up @@ -83,7 +83,7 @@ def process_data_for_chroma(data):
elif isinstance(value, list):
# Store lists as JSON strings with special prefix
processed_data[f"{key}__list"] = json.dumps(value)
elif isinstance(value, (str, int, float, bool)) or value is None:
elif isinstance(value, (str, int, float, bool)):
processed_data[key] = value
else:
processed_data[key] = str(value)
Expand Down Expand Up @@ -553,8 +553,4 @@ async def get_collection_names(self):
Returns a list of collection names.
"""
client = await self.get_connection()
collections = await client.list_collections()
return [
collection.name if hasattr(collection, "name") else collection["name"]
for collection in collections
]
return await client.list_collections()
4 changes: 2 additions & 2 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -102,8 +102,8 @@ posthog = ["posthog>=3.5.0,<4"]
falkordb = ["falkordb>=1.0.9,<2.0.0"]
groq = ["groq>=0.8.0,<1.0.0"]
chromadb = [
"chromadb>=0.3.0,<0.7",
"pypika==0.48.8",
"chromadb>=0.6,<0.7",
"pypika==0.48.9",
]
docs = ["unstructured[csv, doc, docx, epub, md, odt, org, ppt, pptx, rst, rtf, tsv, xlsx]>=0.18.1,<19"]
codegraph = [
Expand Down
Loading
Loading