diff --git a/query_data.py b/query_data.py index c0028317f..167146587 100644 --- a/query_data.py +++ b/query_data.py @@ -1,5 +1,5 @@ """Create a ChatVectorDBChain for question/answering.""" -from langchain.callbacks.base import AsyncCallbackManager +from langchain.callbacks.manager import AsyncCallbackManager from langchain.callbacks.tracers import LangChainTracer from langchain.chains import ChatVectorDBChain from langchain.chains.chat_vector_db.prompts import (CONDENSE_QUESTION_PROMPT, diff --git a/requirements.txt b/requirements.txt index 1b7831d96..fdf772eb1 100644 --- a/requirements.txt +++ b/requirements.txt @@ -11,3 +11,4 @@ faiss-cpu bs4 unstructured libmagic +tiktoken