diff --git a/application/app.py b/application/app.py index 70bd189..c620caa 100644 --- a/application/app.py +++ b/application/app.py @@ -196,7 +196,7 @@ def api_answer(): else: qa_chain = load_qa_chain(llm=llm, chain_type="map_reduce", combine_prompt=c_prompt, question_prompt=q_prompt) - chain = VectorDBQA(combine_documents_chain=qa_chain, vectorstore=docsearch, k=4) + chain = VectorDBQA(combine_documents_chain=qa_chain, vectorstore=docsearch, k=3) result = chain({"query": question}) print(result) diff --git a/docker-compose.yaml b/docker-compose.yaml index 172b6ba..6a4b4cf 100644 --- a/docker-compose.yaml +++ b/docker-compose.yaml @@ -21,7 +21,9 @@ services: ports: - "5001:5001" volumes: - - app_data_container:/app + - ./application/indexes:/app/indexes + - ./application/inputs:/app/inputs + - ./application/vectors:/app/vectors depends_on: - redis - mongo @@ -55,5 +57,4 @@ services: volumes: - mongodb_data_container: - app_data_container: \ No newline at end of file + mongodb_data_container: \ No newline at end of file