bstraehle commited on
Commit
06d3ac4
·
1 Parent(s): cd9c510

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -38,7 +38,6 @@ MODEL_NAME = "gpt-4"
38
  def invoke(openai_api_key, use_rag, prompt):
39
  print(os.listdir("../"))
40
  print(os.listdir("../app/"))
41
- #print(os.listdir("../data/"))
42
  llm = ChatOpenAI(model_name = MODEL_NAME,
43
  openai_api_key = openai_api_key,
44
  temperature = 0)
@@ -60,6 +59,7 @@ def invoke(openai_api_key, use_rag, prompt):
60
  chain_type_kwargs = {"prompt": RAG_CHAIN_PROMPT},
61
  retriever = vector_db.as_retriever(search_kwargs = {"k": 3}),
62
  return_source_documents = True)
 
63
  result = rag_chain({"query": prompt})
64
  result = result["result"]
65
  else:
 
38
  def invoke(openai_api_key, use_rag, prompt):
39
  print(os.listdir("../"))
40
  print(os.listdir("../app/"))
 
41
  llm = ChatOpenAI(model_name = MODEL_NAME,
42
  openai_api_key = openai_api_key,
43
  temperature = 0)
 
59
  chain_type_kwargs = {"prompt": RAG_CHAIN_PROMPT},
60
  retriever = vector_db.as_retriever(search_kwargs = {"k": 3}),
61
  return_source_documents = True)
62
+ print(os.listdir("/data/chroma/"))
63
  result = rag_chain({"query": prompt})
64
  result = result["result"]
65
  else: