clean up privategpt example

This commit is contained in:
Jeffrey Morgan
2023-08-11 00:34:52 -07:00
parent 89a92477ad
commit e863066144
6 changed files with 803 additions and 41 deletions

View File

@@ -1,5 +1,4 @@
#!/usr/bin/env python3
from dotenv import load_dotenv
from langchain.chains import RetrievalQA
from langchain.embeddings import HuggingFaceEmbeddings
from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
@@ -9,8 +8,6 @@ import os
import argparse
import time
load_dotenv()
model = os.environ.get("MODEL", "llama2-uncensored")
embeddings_model_name = os.environ.get("EMBEDDINGS_MODEL_NAME", "all-MiniLM-L6-v2")
persist_directory = os.environ.get("PERSIST_DIRECTORY", "db")