From 7012cb4354aaa7a4de36d430210a579ecdb22075 Mon Sep 17 00:00:00 2001 From: Karthik Soman Date: Sun, 3 Dec 2023 23:40:21 -0800 Subject: [PATCH] adding ast and requests module to utility --- kg_rag/utility.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/kg_rag/utility.py b/kg_rag/utility.py index cf1928d..823f658 100644 --- 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)