Skip to content

Commit

Permalink
updated chromadb python package to latest (0.4.12) to mitigate error …
Browse files Browse the repository at this point in the history
…while adding golden records (#175)

Co-authored-by: Paul Markus <pmarkus@rockhampton-mgt.com>
  • Loading branch information
ppmarkus and pmrockhampton authored Sep 21, 2023
1 parent fdaea65 commit 10fcf3e
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 6 deletions.
6 changes: 1 addition & 5 deletions dataherald/vector_store/chroma.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
from typing import Any, List

import chromadb
from chromadb.config import Settings
from overrides import override

from dataherald.config import System
Expand All @@ -12,13 +11,10 @@ class Chroma(VectorStore):
def __init__(
self,
system: System,
chroma_db_impl: str = "duckdb+parquet",
persist_directory: str = "/app/chroma",
):
super().__init__(system)
self.chroma_client = chromadb.Client(
Settings(chroma_db_impl=chroma_db_impl, persist_directory=persist_directory)
)
self.chroma_client = chromadb.PersistentClient(path=persist_directory)

@override
def query(
Expand Down
2 changes: 1 addition & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ snowflake-sqlalchemy==1.4.7
databricks-sql-connector==2.7.0
sqlalchemy-databricks==0.2.0
sqlalchemy-bigquery==1.6.1
chromadb==0.3.26
chromadb==0.4.12
pytest-dotenv==0.5.2
pinecone-client==2.2.2
cryptography==40.0.2
Expand Down

0 comments on commit 10fcf3e

Please sign in to comment.