Update app.py
Browse files
app.py
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
import gradio as gr
|
2 |
-
import os, time
|
3 |
|
4 |
from dotenv import load_dotenv, find_dotenv
|
5 |
|
@@ -11,15 +11,18 @@ _ = load_dotenv(find_dotenv())
|
|
11 |
|
12 |
RAG_INGESTION = False # load, split, embed, and store documents
|
13 |
|
|
|
|
|
|
|
|
|
14 |
config = {
|
15 |
"k": 3, # retrieve documents
|
16 |
"model_name": "gpt-4-0314", # llm
|
17 |
"temperature": 0 # llm
|
18 |
}
|
19 |
|
20 |
-
|
21 |
-
|
22 |
-
RAG_LLAMAINDEX = "LlamaIndex"
|
23 |
|
24 |
def invoke(openai_api_key, prompt, rag_option):
|
25 |
if (openai_api_key == ""):
|
|
|
1 |
import gradio as gr
|
2 |
+
import logging, os, sys, time
|
3 |
|
4 |
from dotenv import load_dotenv, find_dotenv
|
5 |
|
|
|
11 |
|
12 |
RAG_INGESTION = False # load, split, embed, and store documents
|
13 |
|
14 |
+
RAG_OFF = "Off"
|
15 |
+
RAG_LANGCHAIN = "LangChain"
|
16 |
+
RAG_LLAMAINDEX = "LlamaIndex"
|
17 |
+
|
18 |
config = {
|
19 |
"k": 3, # retrieve documents
|
20 |
"model_name": "gpt-4-0314", # llm
|
21 |
"temperature": 0 # llm
|
22 |
}
|
23 |
|
24 |
+
logging.basicConfig(stream = sys.stdout, level = logging.INFO)
|
25 |
+
logging.getLogger().addHandler(logging.StreamHandler(stream = sys.stdout))
|
|
|
26 |
|
27 |
def invoke(openai_api_key, prompt, rag_option):
|
28 |
if (openai_api_key == ""):
|