adding ast and requests module to utility

BaranziniLab · Dec 4, 2023 · 7012cb4 · 7012cb4
1 parent 8570673
commit 7012cb4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/kg_rag/utility.py b/kg_rag/utility.py
@@ -16,6 +16,8 @@
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM, TextStreamer, GPTQConfig
 from kg_rag.config_loader import *
+import ast
+import requests
 
 memory = Memory("cachegpt", verbose=0)