From 0531844265d0e67f0d71f04a9c1ab853c3e6782c Mon Sep 17 00:00:00 2001 From: Pietro Martino Lugato Date: Thu, 24 Oct 2024 15:49:08 -0400 Subject: [PATCH] chroma issues --- a2rchi/chains/chain.py | 2 +- a2rchi/utils/data_manager.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/a2rchi/chains/chain.py b/a2rchi/chains/chain.py index 6103de8..9c2ba94 100644 --- a/a2rchi/chains/chain.py +++ b/a2rchi/chains/chain.py @@ -1,7 +1,7 @@ from a2rchi.chains.base import BaseSubMITChain as BaseChain from chromadb.config import Settings -from langchain_chroma import Chroma +from langchain_chroma.vectorstores import Chroma import chromadb import time diff --git a/a2rchi/utils/data_manager.py b/a2rchi/utils/data_manager.py index 535e415..a2680e4 100644 --- a/a2rchi/utils/data_manager.py +++ b/a2rchi/utils/data_manager.py @@ -5,7 +5,7 @@ from langchain_community.document_loaders import PyPDFLoader from langchain_community.document_loaders import BSHTMLLoader from langchain_openai import OpenAIEmbeddings -from langchain_chroma import Chroma +from langchain_chroma.vectorstores import Chroma from langchain_text_splitters.character import CharacterTextSplitter import chromadb