Update app.py

app.py

@@ -41,11 +41,11 @@ def invoke(openai_api_key, use_rag, prompt):
             splits = text_splitter.split_documents(docs)
             vector_db = Chroma.from_documents(documents = splits, embedding = OpenAIEmbeddings(), persist_directory = CHROMA_DIR)
         rag_chain = RetrievalQA.from_chain_type(llm, retriever = vector_db.as_retriever(search_kwargs = {"k": 3}), return_source_documents = True, chain_type_kwargs = {"prompt": CHAIN_PROMPT})
-        result = rag_chain({"question": prompt})
+        result = rag_chain({"query": prompt})
         result = result["result"]
     else:
         chain = LLMChain(llm = llm, prompt = CHAIN_PROMPT)
-        result = chain.run({"context": "", "question": prompt})
+        result = chain.run({"context": "", "query": prompt})
     #print(result)
     return result