From 668b17d921f41d689bf49ca845982c8a1db931f1 Mon Sep 17 00:00:00 2001
From: Romain Dauby <romain.dauby@gmail.com>
Date: Sat, 10 Jun 2023 16:02:37 -0400
Subject: [PATCH] Fix imports and requirements

---
 query_data.py    | 2 +-
 requirements.txt | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

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